From ae99dac7e0f5cd7c192e23edd2ac1844f8f89691 Mon Sep 17 00:00:00 2001
From: Kevaundray Wedderburn <kevtheappdev@gmail.com>
Date: Sun, 11 May 2025 22:20:39 +0100
Subject: [PATCH] move Input to separate module

---
 crates/zkvm-interface/src/input.rs | 96 ++++++++++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 crates/zkvm-interface/src/input.rs

diff --git a/crates/zkvm-interface/src/input.rs b/crates/zkvm-interface/src/input.rs
new file mode 100644
index 0000000..c0f63ba
--- /dev/null
+++ b/crates/zkvm-interface/src/input.rs
@@ -0,0 +1,96 @@
+use serde::Serialize;
+
+/// Represents a builder for input data to be passed to a ZKVM guest program.
+/// Values are serialized sequentially into an internal byte buffer.
+#[derive(Debug, Default)]
+pub struct Input {
+    buf: Vec<u8>,
+    ranges: Vec<(usize, usize)>,
+}
+
+impl Input {
+    /// Create an empty input buffer.
+    pub fn new() -> Self {
+        Self::default()
+    }
+
+    /// Append a value, serializing it with `bincode`.
+    pub fn write<T: Serialize>(&mut self, value: &T) -> Result<(), bincode::Error> {
+        let start = self.buf.len();
+        bincode::serialize_into(&mut self.buf, value)?;
+        let end = self.buf.len();
+        self.ranges.push((start, end - start));
+        Ok(())
+    }
+
+    /// Number of elements written.
+    pub fn len(&self) -> usize {
+        self.ranges.len()
+    }
+
+    pub fn is_empty(&self) -> bool {
+        self.ranges.is_empty()
+    }
+
+    /* ---------------- Read‑side helpers ---------------- */
+
+    /// Entire concatenated payload as one slice.
+    pub fn bytes(&self) -> &[u8] {
+        &self.buf
+    }
+
+    /// Iterator over individual chunks (the originally written objects).
+    pub fn chunked_iter(&self) -> impl ExactSizeIterator<Item = &[u8]> + '_ {
+        self.ranges.iter().map(|&(s, len)| &self.buf[s..s + len])
+    }
+
+    /// Byte‑wise iterator (rarely needed).
+    pub fn iter(&self) -> std::slice::Iter<'_, u8> {
+        self.buf.iter()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn input_empty() {
+        let input = Input::new();
+        assert!(input.is_empty());
+        assert_eq!(input.len(), 0);
+        assert!(input.bytes().is_empty());
+        assert_eq!(input.chunked_iter().count(), 0);
+    }
+
+    #[test]
+    fn input_write_and_read() {
+        let mut input = Input::new();
+        let a: u32 = 42;
+        let b: &str = "hello";
+
+        input.write(&a).unwrap();
+        input.write(&b).unwrap();
+
+        // length bookkeeping
+        assert_eq!(input.len(), 2);
+        assert!(!input.is_empty());
+
+        // chunk iteration and deserialization
+        let chunks: Vec<&[u8]> = input.chunked_iter().collect();
+        assert_eq!(chunks.len(), 2);
+        let a_back: u32 = bincode::deserialize(chunks[0]).unwrap();
+        assert_eq!(a_back, a);
+        let b_back: String = bincode::deserialize(chunks[1]).unwrap();
+        assert_eq!(b_back, b);
+
+        // contiguous bytes match manual serialization
+        let mut expected = Vec::<u8>::new();
+        bincode::serialize_into(&mut expected, &a).unwrap();
+        bincode::serialize_into(&mut expected, &b).unwrap();
+        assert_eq!(input.bytes(), expected.as_slice());
+
+        // iter() covers same length
+        assert_eq!(input.iter().count(), expected.len());
+    }
+}