From 3ab66346610d681cf764f24bbdd8d5416d7ef1b6 Mon Sep 17 00:00:00 2001
From: Nathan Fisher <jeang3nie@hitchhiker-linux.org>
Date: Wed, 22 Jan 2025 12:16:36 -0500
Subject: [PATCH] Initial commit

---
 .gitignore    |   1 +
 Cargo.lock    |   7 ++
 Cargo.toml    |   6 ++
 src/encode.rs | 176 ++++++++++++++++++++++++++++++++++++++++++++++++++
 src/lib.rs    |   4 ++
 5 files changed, 194 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 Cargo.lock
 create mode 100644 Cargo.toml
 create mode 100644 src/encode.rs
 create mode 100644 src/lib.rs

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..ea8c4bf
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/Cargo.lock b/Cargo.lock
new file mode 100644
index 0000000..1f73b8c
--- /dev/null
+++ b/Cargo.lock
@@ -0,0 +1,7 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "b2"
+version = "0.1.0"
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..c9ffd86
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,6 @@
+[package]
+name = "b2"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
diff --git a/src/encode.rs b/src/encode.rs
new file mode 100644
index 0000000..133b0e4
--- /dev/null
+++ b/src/encode.rs
@@ -0,0 +1,176 @@
+use std::io::{self, ErrorKind, Read, Write};
+
+#[derive(Default, Clone, Copy)]
+pub enum Style {
+    #[default]
+    Plain,
+    Spaces,
+    Prefixed,
+    Table,
+}
+
+pub struct Encoder<R: Read, W: Write> {
+    reader: R,
+    writer: W,
+    style: Style,
+    wrap: Option<usize>,
+}
+
+impl<R: Read, W: Write> Encoder<R, W> {
+    pub fn new(reader: R, writer: W, style: Option<Style>, mut wrap: Option<usize>) -> Self {
+        let style = style.unwrap_or_default();
+        if let Style::Table = style {
+            wrap = match wrap {
+                Some(u) => Some(u),
+                None => Some(5),
+            };
+        }
+        Self {
+            reader,
+            writer,
+            style,
+            wrap,
+        }
+    }
+
+    pub fn encode(mut self) -> Result<W, io::Error> {
+        let mut bytes = self.reader.bytes();
+        let mut total = 0;
+        loop {
+            match bytes.next() {
+                Some(Ok(b)) => match self.style {
+                    Style::Plain => {
+                        if let Some(w) = self.wrap {
+                            if total > 0 && total % w == 0 {
+                                writeln!(self.writer)?;
+                            }
+                        }
+                        write!(self.writer, "{b:08b}")?;
+                    }
+                    Style::Spaces => {
+                        if let Some(w) = self.wrap {
+                            if total > 0 && total % w == 0 {
+                                writeln!(self.writer)?;
+                            } else {
+                                write!(self.writer, " ")?;
+                            }
+                        } else if total > 0 {
+                            write!(self.writer, " ")?;
+                        }
+                        write!(self.writer, "{b:08b}")?;
+                    }
+                    Style::Prefixed => {
+                        if let Some(w) = self.wrap {
+                            if total > 0 && total % w == 0 {
+                                writeln!(self.writer)?;
+                            } else if total > 0 {
+                                write!(self.writer, " ")?;
+                            }
+                        } else if total > 0 {
+                            write!(self.writer, " ")?;
+                        }
+                        write!(self.writer, "{b:#010b}")?;
+                    }
+                    Style::Table => {
+                        if let Some(w) = self.wrap {
+                            if total % w == 0 {
+                                if total > 0 {
+                                    write!(self.writer, "\n{total:>5}\t{b:#010b}")?;
+                                } else {
+                                    write!(self.writer, "{total:>5}\t{b:#010b}")?;
+                                }
+                            } else {
+                                write!(self.writer, " {b:#010b}")?;
+                            }
+                        } else {
+                            unreachable!();
+                        }
+                    }
+                },
+                Some(Err(e)) if e.kind() == ErrorKind::UnexpectedEof => break,
+                Some(Err(e)) if e.kind() == ErrorKind::Interrupted => continue,
+                Some(Err(e)) => return Err(e),
+                None => break,
+            }
+            total += 1;
+        }
+        if let Some(w) = self.wrap {
+            if total % w != 0 {
+                writeln!(self.writer)?;
+            }
+        } else {
+            writeln!(self.writer)?;
+        }
+        Ok(self.writer)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn hello_plain() {
+        let encoder = Encoder::new("Hello, World!".as_bytes(), vec![], None, None);
+        assert_eq!(
+            encoder.encode().unwrap(),
+            b"01001000011001010110110001101100011011110010110000100000010101110110111101110010011011000110010000100001\n"
+        );
+    }
+
+    #[test]
+    fn hello_spaced() {
+        let encoder = Encoder::new(
+            "Hello, World!".as_bytes(),
+            vec![],
+            Some(Style::Spaces),
+            None,
+        );
+        assert_eq!(
+            encoder.encode().unwrap(),
+            b"01001000 01100101 01101100 01101100 01101111 00101100 00100000 01010111 01101111 01110010 01101100 01100100 00100001\n"
+        );
+    }
+
+    #[test]
+    fn hello_prefixed() {
+        let encoder = Encoder::new(
+            "Hello, World!".as_bytes(),
+            vec![],
+            Some(Style::Prefixed),
+            None,
+        );
+        assert_eq!(
+            encoder.encode().unwrap(),
+            b"0b01001000 0b01100101 0b01101100 0b01101100 0b01101111 0b00101100 0b00100000 0b01010111 0b01101111 0b01110010 0b01101100 0b01100100 0b00100001\n"
+        );
+    }
+
+    #[test]
+    fn hello_prefixed_wrapped() {
+        let encoder = Encoder::new(
+            "Hello, World!".as_bytes(),
+            vec![],
+            Some(Style::Prefixed),
+            Some(4),
+        );
+        assert_eq!(
+            encoder.encode().unwrap(),
+            b"0b01001000 0b01100101 0b01101100 0b01101100\n0b01101111 0b00101100 0b00100000 0b01010111\n0b01101111 0b01110010 0b01101100 0b01100100\n0b00100001\n"
+        );
+    }
+
+    #[test]
+    fn hello_table() {
+        let encoder = Encoder::new(
+            "Hello, World!".as_bytes(),
+            vec![],
+            Some(Style::Table),
+            Some(4),
+        );
+        assert_eq!(
+            encoder.encode().unwrap(),
+            b"    0\t0b01001000 0b01100101 0b01101100 0b01101100\n    4\t0b01101111 0b00101100 0b00100000 0b01010111\n    8\t0b01101111 0b01110010 0b01101100 0b01100100\n   12\t0b00100001\n"
+        );
+    }
+}
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..033777e
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,4 @@
+#![warn(clippy::all, clippy::pedantic)]
+
+mod encode;
+pub use encode::{Encoder, Style};