From 41e2b2a3d74a2232477c0eacbc5ec15069840cc1 Mon Sep 17 00:00:00 2001
From: halotukozak <bartlomiejkozak@proton.me>
Date: Sun, 12 May 2024 16:41:11 +0200
Subject: [PATCH] table enhacements

---
 .idea/sbt.xml                                 |  2 +-
 .idea/scala_compiler.xml                      |  2 +-
 build.sbt                                     |  2 +-
 src/main/scala/tables/Table.scala             | 73 +++++++++++++++++++
 src/main/scala/typography/Headings.scala      | 11 ---
 .../scala/typography/macros/Headings.scala    |  5 +-
 .../scala/integration/IntegrationTest.scala   | 20 ++++-
 src/test/scala/tables/TableTest.scala         | 59 +++++++++++++++
 src/test/scala/typography/HeadingTest.scala   | 16 ----
 9 files changed, 155 insertions(+), 35 deletions(-)
 create mode 100644 src/main/scala/tables/Table.scala
 create mode 100644 src/test/scala/tables/TableTest.scala

diff --git a/.idea/sbt.xml b/.idea/sbt.xml
index c90073e..89f461f 100644
--- a/.idea/sbt.xml
+++ b/.idea/sbt.xml
@@ -3,7 +3,7 @@
   <component name="ScalaSbtSettings">
     <option name="linkedExternalProjectsSettings">
       <SbtProjectSettings>
-        <option name="converterVersion" value="1" />
+        <option name="converterVersion" value="2" />
         <option name="externalProjectPath" value="$PROJECT_DIR$" />
         <option name="modules">
           <set>
diff --git a/.idea/scala_compiler.xml b/.idea/scala_compiler.xml
index 6d5ef30..3b66412 100644
--- a/.idea/scala_compiler.xml
+++ b/.idea/scala_compiler.xml
@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ScalaCompilerConfiguration">
-    <profile name="sbt 1" modules="Smarkdown" />
+    <profile name="sbt 1" modules="Smark" />
   </component>
 </project>
\ No newline at end of file
diff --git a/build.sbt b/build.sbt
index 79897fe..c5089ce 100644
--- a/build.sbt
+++ b/build.sbt
@@ -6,5 +6,5 @@ lazy val root = (project in file("."))
   .settings(
     name := "Smark",
     idePackagePrefix := Some("halotukozak.smark"),
-    libraryDependencies += "org.scalatest" %% "scalatest" % "3.2.17" % Test,
+    libraryDependencies += "org.scalatest" %% "scalatest" % "3.2.18" % Test,
   )
diff --git a/src/main/scala/tables/Table.scala b/src/main/scala/tables/Table.scala
new file mode 100644
index 0000000..b2311f7
--- /dev/null
+++ b/src/main/scala/tables/Table.scala
@@ -0,0 +1,73 @@
+package halotukozak.smark
+package tables
+
+
+import tables.Column.symbol
+
+import scala.collection.mutable.ArrayBuffer
+
+private final class Table(private[tables] val rows: ArrayBuffer[Row | Header] = new ArrayBuffer[Row | Header]) extends AnyVal:
+  extension (ab: Iterable[String])
+    private def mkRow = ab.mkString("| ", " | ", " |")
+
+  private[tables] def eval = rows.map {
+    case r: Row => r.cells.map(_.elem).mkRow
+    case h: Header => h.elements.map(_.name).mkRow + "\n" + h.elements.map(symbol).mkRow
+  }.mkString("\n")
+
+
+private class Row(private[tables] val cells: ArrayBuffer[Cell] = new ArrayBuffer[Cell]) extends AnyVal
+
+private class Header(private[tables] val elements: Seq[Column]) extends AnyVal
+
+
+sealed abstract class Column(val name: String)
+
+private object Column:
+  final case class Left(override val name: String) extends Column(name)
+
+  final case class Right(override val name: String) extends Column(name)
+
+  final case class Center(override val name: String) extends Column(name)
+
+  final case class None(override val name: String) extends Column(name)
+
+  def symbol: Column => String = (_: Column) match
+    case Left(name) => ":---"
+    case Right(name) => "---:"
+    case Center(name) => ":---:"
+    case None(name) => "---"
+
+end Column
+
+type Left = Column.Left
+type Right = Column.Right
+type Center = Column.Center
+type None = Column.None
+given Conversion[String, Left] = Column.Left(_)
+given Conversion[String, Right] = Column.Right(_)
+given Conversion[String, Center] = Column.Center(_)
+given Conversion[String, None] = Column.None(_)
+
+
+private final class Cell(private[tables] val elem: String) extends AnyVal
+
+def table(init: Table ?=> Unit): String =
+  given t: Table = Table()
+
+  init(using t)
+  t.eval
+
+def row(init: Row ?=> Unit)(using t: Table): Unit =
+  given r: Row = Row()
+
+  init(using r)
+  t.rows += r
+
+def header(columns: (Column | String)*)(using t: Table): Unit =
+  t.rows += Header(columns.map {
+    case c: Column => c
+    case s: String => s: None
+  })
+
+def cell(str: String)(using r: Row): Unit = r.cells += Cell(str)
diff --git a/src/main/scala/typography/Headings.scala b/src/main/scala/typography/Headings.scala
index 16e2dfb..3d21594 100644
--- a/src/main/scala/typography/Headings.scala
+++ b/src/main/scala/typography/Headings.scala
@@ -1,16 +1,5 @@
 package halotukozak.smark
 package typography
 
-import typography.macros.heading
-
 
 private[typography] type HeadingLevel = 1 | 2 | 3 | 4 | 5 | 6
-
-def headingUnsafe(n: Int, inner: String): String = n match
-  case 1 => heading[1](inner)
-  case 2 => heading[2](inner)
-  case 3 => heading[3](inner)
-  case 4 => heading[4](inner)
-  case 5 => heading[5](inner)
-  case 6 => heading[6](inner)
-  case _ => throw IllegalArgumentException(s"Invalid heading level: $n")
diff --git a/src/main/scala/typography/macros/Headings.scala b/src/main/scala/typography/macros/Headings.scala
index 753d3ed..89d5bc3 100644
--- a/src/main/scala/typography/macros/Headings.scala
+++ b/src/main/scala/typography/macros/Headings.scala
@@ -3,7 +3,4 @@ package typography.macros
 
 import typography.HeadingLevel
 
-import scala.quoted.{Expr, Quotes}
-
-inline def heading[N <: HeadingLevel : ValueOf](inline inner: String): String = ${ headingImpl('{ inner }, '{ valueOf[N] }) }
-private def headingImpl(inner: Expr[String], n: Expr[Int])(using Quotes): Expr[String] = '{ "#" * $n + $inner }
+inline def heading[N <: HeadingLevel : ValueOf](inline inner: String): String = "#" * valueOf[N] + inner
\ No newline at end of file
diff --git a/src/test/scala/integration/IntegrationTest.scala b/src/test/scala/integration/IntegrationTest.scala
index 2337649..1ba1b52 100644
--- a/src/test/scala/integration/IntegrationTest.scala
+++ b/src/test/scala/integration/IntegrationTest.scala
@@ -1,6 +1,7 @@
 package halotukozak.smark
 package integration
 
+import tables.*
 import typography.*
 import typography.macros.*
 
@@ -37,6 +38,20 @@ class IntegrationTest extends AnyWordSpec with Matchers {
         ),
         image["Google"](url = "https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png"),
         link("https://www.google.com", "Google"),
+        table {
+          row {
+            cell {
+              text[Bold]("a")
+            }
+            cell("b")
+          }
+          row {
+            cell {
+              text[Code](scala"val x = 1")
+            }
+            cell("d")
+          }
+        },
       ) shouldEqual
         """hello
           |
@@ -71,7 +86,10 @@ class IntegrationTest extends AnyWordSpec with Matchers {
           |
           |![Google](https://www.google.com/images/branding/googlelogo/1x/googlelogo_color_272x92dp.png)
           |
-          |[Google](https://www.google.com)""".stripMargin
+          |[Google](https://www.google.com)
+          |
+          || **a** | b |
+          || `val x = 1` | d |""".stripMargin
     }
   }
 
diff --git a/src/test/scala/tables/TableTest.scala b/src/test/scala/tables/TableTest.scala
new file mode 100644
index 0000000..bfb2894
--- /dev/null
+++ b/src/test/scala/tables/TableTest.scala
@@ -0,0 +1,59 @@
+package halotukozak.smark
+package tables
+
+import typography.macros.text
+import typography.{Bold, Code, ScalaHelper}
+
+import org.scalatest.matchers.should.Matchers
+import org.scalatest.wordspec.AnyWordSpec
+
+
+final class TableTest extends AnyWordSpec with Matchers {
+
+  "table" should {
+    "be evaluated" in {
+      table {
+        row {}
+      } shouldBe "|  |"
+
+      table {
+        row {
+          cell("a")
+          cell("b")
+        }
+      } shouldBe "| a | b |"
+
+      table {
+        row {
+          cell {
+            text[Bold]("a")
+          }
+          cell("b")
+        }
+        row {
+          cell {
+            text[Code](scala"val x = 1")
+          }
+          cell("d")
+        }
+      } shouldBe "| **a** | b |\n| `val x = 1` | d |"
+
+      table {
+        header("a": Left, "b")
+        row {
+          cell("1")
+          cell("2")
+        }
+      } shouldBe "| a | b |\n| :--- | --- |\n| 1 | 2 |"
+
+      table {
+        header("a": Right, "b": Center)
+        row {
+          cell("1")
+          cell("2")
+        }
+      } shouldBe "| a | b |\n| ---: | :---: |\n| 1 | 2 |"
+    }
+  }
+
+}
diff --git a/src/test/scala/typography/HeadingTest.scala b/src/test/scala/typography/HeadingTest.scala
index c50b958..62b408a 100644
--- a/src/test/scala/typography/HeadingTest.scala
+++ b/src/test/scala/typography/HeadingTest.scala
@@ -2,7 +2,6 @@ package halotukozak.smark
 package typography
 
 
-import typography.headingUnsafe
 import typography.macros.*
 
 import org.scalatest.matchers.should.Matchers
@@ -11,7 +10,6 @@ import org.scalatest.wordspec.AnyWordSpec
 final class HeadingTest extends AnyWordSpec with Matchers {
 
   "Heading" should {
-
     "be evaluated" when {
       "level 1" in {
         heading[1]("Hello") shouldBe "#Hello"
@@ -40,19 +38,5 @@ final class HeadingTest extends AnyWordSpec with Matchers {
         assertDoesNotCompile("""heading[-5]("Hello")""")
       }
     }
-
-    "unsafe should be evaluated" when {
-      "in range" in {
-        for (i <- 1 to 6) do
-          headingUnsafe(i, "Hello") shouldBe "#" * i + "Hello"
-      }
-
-      "out of range" in {
-        for (i <- 7 to 10) do
-          assertThrows[IllegalArgumentException](headingUnsafe(i, "Hello"))
-        for (i <- -5 to 0) do
-          assertThrows[IllegalArgumentException](headingUnsafe(i, "Hello"))
-      }
-    }
   }
 }
\ No newline at end of file