merge old and new dsl

src/main/scala/com/htmlism/mos6502/dsl/AsmDocumentContext.scala

@@ -17,7 +17,12 @@ class AsmDocumentContext {
 
 sealed trait TopLevelAsmDocumentFragment
 
-sealed trait AsmBlockFragment extends TopLevelAsmDocumentFragment
+case class AsmFragment(xs: List[Statement]) extends TopLevelAsmDocumentFragment {
+  def printOut(): Unit = {
+    xs.map(_.toAsm)
+      .foreach(println)
+  }
+}
 
 case class DefinitionGroup(comment: String, xs: List[Definition[_]]) extends TopLevelAsmDocumentFragment
 

src/main/scala/com/htmlism/mos6502/dsl/DslDemo.scala

@@ -126,4 +126,7 @@ class AssemblyContext {
 
   def triplets: List[(String, Option[String], Option[String])] =
     xs.map(_.toTriplet).toList
+
+  def toFragment: AsmFragment =
+    AsmFragment(xs.toList)
 }

src/main/scala/com/htmlism/mos6502/dsl/package.scala

@@ -10,6 +10,16 @@ package object dsl {
     ctx.toDoc
   }
 
+  def asm(f: AssemblyContext => Unit)(implicit ctx: AsmDocumentContext): Unit = {
+    val asmCtx: AssemblyContext =
+      new AssemblyContext
+
+    f(asmCtx)
+
+    ctx
+      .push(asmCtx.toFragment)
+  }
+
   def group[A](s: String)(f: DefinitionGroupContext => A)(implicit ctx: AsmDocumentContext): A = {
     val g: DefinitionGroupContext =
       new DefinitionGroupContext

src/test/scala/com/htmlism/mos6502/dsl/Easy6502Spec.scala

@@ -33,9 +33,26 @@ class Easy6502Spec extends AnyFlatSpec with should.Matchers {
     val doc =
       asmDoc { implicit ctx =>
         enum[Color]
+
+        asm { implicit a =>
+          val scr =
+            IndexedAddressCollection[Color](0x0200, "screen")
+
+          scr.write(0, Color.White)
+          scr.write(1, Color.Green)
+          scr.write(2, Color.Orange)
+        }
       }
 
-    println(doc)
+    doc
+      .xs
+      .foreach {
+        case DefinitionGroup(c, xs) =>
+          println(c)
+          xs.foreach(println)
+        case a: AsmFragment =>
+          a.printOut()
+      }
   }
 
   def withAssemblyContext(f: AssemblyContext => Unit): AssemblyContext = {