add pretty print

Cargo.lock

@@ -27,11 +27,33 @@ dependencies = [
  "winapi",
 ]
 
+[[package]]
+name = "debug2"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5e70062ea3df6318129246273f4b1748900752293d47e7610b5e8a84bc14c44c"
+dependencies = [
+ "debug2-derive",
+]
+
+[[package]]
+name = "debug2-derive"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "77b37e8941dd632993cf5b4ea33a630faa132fdc42481a8c2621dede631e871a"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+ "synstructure",
+]
+
 [[package]]
 name = "dilaria"
 version = "0.1.0"
 dependencies = [
  "bumpalo",
+ "debug2",
  "insta",
  "rustc-hash",
 ]
@@ -187,6 +209,18 @@ dependencies = [
  "unicode-xid",
 ]
 
+[[package]]
+name = "synstructure"
+version = "0.12.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f36bdaa60a83aca3921b5259d5400cbf5e90fc51931376a9bd4a0eb79aa7210f"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+ "unicode-xid",
+]
+
 [[package]]
 name = "terminal_size"
 version = "0.1.17"

Cargo.toml

@@ -7,11 +7,16 @@ edition = "2021"
 
 [dependencies]
 bumpalo = { version = "3.8.0", features = ["collections", "boxed"] }
+debug2 = { version = "0.1.0", optional = true }
 rustc-hash = { version = "1.1.0", optional = true }
 
 
 [features]
 fxhash = ["rustc-hash"]
+pretty = ["debug2"]
+
+# todo: we don't actually want this as a default feature
+default = ["pretty"]
 
 [dev-dependencies]
 insta = "1.9.0"

src/bytecode.rs

@@ -18,6 +18,7 @@
 use crate::errors::Span;
 use crate::vm::Value;
 use bumpalo::collections::Vec;
+use debug2::Formatter;
 
 /// This struct contains all data for a function.
 #[derive(Debug)]
@@ -34,8 +35,21 @@ pub struct FnBlock<'bc> {
     pub arity: u8,
 }
 
+#[cfg(feature = "pretty")]
+impl debug2::Debug for FnBlock<'_> {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        f.debug_struct("FnBlock")
+            .field("code", &self.code.as_slice())
+            .field("stack_sizes", &self.stack_sizes.as_slice())
+            .field("spans", &self.spans.as_slice())
+            .field("arity", &self.arity)
+            .finish()
+    }
+}
+
 /// A bytecode instruction. For more details on the structure of the bytecode, read the module level docs [`bytecode`](`self`)
 #[derive(Debug, Clone, Copy)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 pub enum Instr {
     /// An operation that does nothing.
     Nop,

src/compile.rs

@@ -62,7 +62,7 @@ pub fn compile<'ast, 'bc, 'gc>(
     ast: &'ast Program,
     bytecode_bump: &'bc Bump,
     rt: &'gc mut RtAlloc,
-) -> Result<Vec<'bc, FnBlock<'bc>>, CompilerError> {
+) -> Result<&'bc [FnBlock<'bc>], CompilerError> {
     let mut compiler = Compiler {
         blocks: Vec::new_in(bytecode_bump),
         current_block: 0,
@@ -73,7 +73,7 @@ pub fn compile<'ast, 'bc, 'gc>(
 
     compiler.compile(ast)?;
 
-    Ok(compiler.blocks)
+    Ok(compiler.blocks.into_bump_slice())
 }
 
 impl<'bc, 'gc> Compiler<'bc, 'gc> {

src/errors.rs

@@ -13,6 +13,7 @@ pub use span::Span;
 mod span {
 
     #[derive(Debug, Default, Copy, Clone, PartialOrd, PartialEq, Ord, Eq, Hash)]
+    #[cfg_attr(feature = "pretty", derive(debug2::Debug))]
     pub struct Span {
         pub start: usize,
         pub end: usize,

src/gc.rs

@@ -30,6 +30,21 @@ impl<T: ?Sized> Deref for Gc<T> {
     }
 }
 
+#[cfg(feature = "pretty")]
+impl<T: debug2::Debug> debug2::Debug for Gc<T> {
+    fn fmt(&self, f: &mut debug2::Formatter<'_>) -> std::fmt::Result {
+        T::fmt(&*self, f)
+    }
+}
+
+#[cfg(feature = "pretty")]
+impl debug2::Debug for Gc<str> {
+    fn fmt(&self, f: &mut debug2::Formatter<'_>) -> std::fmt::Result {
+        let str = self.deref();
+        debug2::Debug::fmt(&str, f)
+    }
+}
+
 impl<T: Debug + ?Sized> Debug for Gc<T> {
     fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
         T::fmt(self, f)
@@ -46,6 +61,7 @@ impl<T: ?Sized> Copy for Gc<T> {}
 
 /// An reference to an interned String. Hashing and Equality are O(1) and just look at the pointer address
 #[derive(Clone, Copy)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 pub struct Symbol {
     gc: Gc<str>,
 }
@@ -58,17 +74,20 @@ type ObjectMap = HashMap<Symbol, Value>;
 /// ```
 /// This is inside the local x now.
 #[derive(Clone, Copy)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 pub struct Object {
     gc: Gc<HeapObject>,
 }
 
 #[derive(Debug)]
 #[repr(C)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 struct HeapObject {
     kind: HeapObjectKind,
 }
 
 #[derive(Debug)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 enum HeapObjectKind {
     String(Gc<str>),
     Object(ObjectMap),

src/lib.rs

@@ -63,8 +63,15 @@ fn process_ast(program: &str, ast: &Program, mut runtime: RtAlloc, cfg: &mut Con
     match bytecode {
         Ok(code) => {
             if cfg.debug {
+                #[cfg(feature = "pretty")]
+                {
+                    println!("Bytecode:\n{}\n", debug2::pprint(code));
+                }
+                #[cfg(not(feature = "pretty"))]
+                {
                     println!("Bytecode:\n{:#?}\n", code);
                 }
+            }
 
             let result = vm::execute(&code, runtime, cfg.stdout);
             if let Err(result) = result {

src/vm.rs

@@ -24,6 +24,7 @@ pub fn execute<'bc>(
 }
 
 #[derive(Debug, Clone, Copy)]
+#[cfg_attr(feature = "pretty", derive(debug2::Debug))]
 pub enum Value {
     Null,
     Bool(bool),
@@ -63,6 +64,8 @@ impl<'bc> Vm<'bc, '_> {
                 None => return Ok(()),
             }
             self.pc += 1;
+            // debug stack size assertion
+            todo!()
         }
     }
 

test.dil

@@ -1,7 +1,6 @@
 let i = 0;
 
-while i < 100 {
+while i < 100000 {
-    print i;
     i = i + 1;
 }
 

tests/control_flow.rs

@@ -120,3 +120,23 @@ if not_run {
 }
 "#
 );
+
+run_test!(
+    fizzbuzz,
+    r#"
+let i = 1;
+
+while i < 100 {
+    if i % 15 == 0 {
+        print "FizzBuzz";
+    } else if i % 5 == 0 {
+        print "Buzz";
+    } else if i % 3 == 0 {
+        print "Fizz";
+    } else {
+        print i;
+    }
+    i = i + 1;
+}
+"#
+);

tests/snapshots/control_flow__fizzbuzz.snap

@@ -0,0 +1,7 @@
+---
+source: tests/control_flow.rs
+assertion_line: 124
+expression: output
+
+---
+"1\n2\nFizz\n4\nBuzz\nFizz\n7\n8\nFizz\nBuzz\n11\nFizz\n13\n14\nFizzBuzz\n16\n17\nFizz\n19\nBuzz\nFizz\n22\n23\nFizz\nBuzz\n26\nFizz\n28\n29\nFizzBuzz\n31\n32\nFizz\n34\nBuzz\nFizz\n37\n38\nFizz\nBuzz\n41\nFizz\n43\n44\nFizzBuzz\n46\n47\nFizz\n49\nBuzz\nFizz\n52\n53\nFizz\nBuzz\n56\nFizz\n58\n59\nFizzBuzz\n61\n62\nFizz\n64\nBuzz\nFizz\n67\n68\nFizz\nBuzz\n71\nFizz\n73\n74\nFizzBuzz\n76\n77\nFizz\n79\nBuzz\nFizz\n82\n83\nFizz\nBuzz\n86\nFizz\n88\n89\nFizzBuzz\n91\n92\nFizz\n94\nBuzz\nFizz\n97\n98\nFizz\n"