From 6c6da1ba3ea3e6a77a10ecc320007cbd06f1ccfc Mon Sep 17 00:00:00 2001
From: Ryan Butler <thebutlah@gmail.com>
Date: Sat, 18 May 2024 18:58:00 -0400
Subject: [PATCH] implement encode_varint

---
 crates/did-simple/src/varint.rs | 69 +++++++++++++++++++++++++++------
 1 file changed, 58 insertions(+), 11 deletions(-)

diff --git a/crates/did-simple/src/varint.rs b/crates/did-simple/src/varint.rs
index 4ce2049..a565b39 100644
--- a/crates/did-simple/src/varint.rs
+++ b/crates/did-simple/src/varint.rs
@@ -1,4 +1,61 @@
-pub const fn decode_varint(encoded: &[u8]) -> Result<u16, DecodeError> {
+/// bitmask for 7 least significant bits
+const LSB_7: u8 = u8::MAX / 2;
+/// bitmask for most significant bit
+const MSB: u8 = !LSB_7;
+
+#[inline]
+const fn msb_is_1(val: u8) -> bool {
+	val & MSB == MSB
+}
+
+/// Encodes a value as a varint.
+pub(crate) const fn encode_varint(value: u16) -> (usize, [u8; 3]) {
+	let mut out_buf = [0; 3];
+	// ilog2 can be used to get the (0-indexed from LSB position) of the MSB.
+	// We then add one to it, since we are trying to indicate length, not position.
+	let in_bit_length: u16 = if let Some(x) = value.checked_ilog2() {
+		(x + 1) as u16
+	} else {
+		return (0, out_buf); // empty buffer
+	};
+
+	let mut out = 0u32;
+	let mut in_bit_pos = 0u16;
+	// Each time we need to write a carry bit into the MSB, we increment this.
+	let mut carry_counter = 0;
+	// Have to use macro because consts can't use closures or &mut in functions :(
+	macro_rules! copy_chunk {
+		() => {
+			let retrieved_bits = value & ((LSB_7 as u16) << in_bit_pos);
+			// if the carry bits weren't a thing, we could keep the position intact
+			// and just directly copy the chunk. But we need to shift left to account
+			// for prior carry bits
+			out |= (retrieved_bits as u32) << carry_counter;
+			in_bit_pos += 7;
+		};
+	}
+	copy_chunk!();
+	while in_bit_pos < in_bit_length {
+		carry_counter += 1;
+		out |= 1 << (carry_counter * 8 - 1); // turns on MSB to indicate carry.
+		copy_chunk!();
+	}
+
+	// the number of bytes to write is equivalent to the carry counter + 1.
+	let num_output_bytes = carry_counter + 1;
+
+	// No for loops or copy_from_slice in const fn :(
+	let mut idx = 0;
+	let output_bytes = out.to_le_bytes();
+	while idx < num_output_bytes {
+		out_buf[idx] = output_bytes[idx];
+		idx += 1;
+	}
+
+	(num_output_bytes, out_buf)
+}
+
+pub(crate) const fn decode_varint(encoded: &[u8]) -> Result<u16, DecodeError> {
 	// TODO: Technically, some three byte encodings could fit into a u16, we
 	// should support those in the future.
 	// Luckily none of them are used for did:key afaik.
@@ -9,16 +66,6 @@ pub const fn decode_varint(encoded: &[u8]) -> Result<u16, DecodeError> {
 		return Err(DecodeError::MissingBytes);
 	}
 
-	/// bitmask for 7 least significant bits
-	const LSB_7: u8 = u8::MAX / 2;
-	/// bitmask for most significant bit
-	const MSB: u8 = !LSB_7;
-
-	#[inline]
-	const fn msb_is_1(val: u8) -> bool {
-		val & MSB == MSB
-	}
-
 	let a = encoded[0];
 	let mut result: u16 = (a & LSB_7) as u16;
 	if msb_is_1(a) {