From 488f1fa17d40d53134a9b132f43222f1fb79bac1 Mon Sep 17 00:00:00 2001
From: Christoph Herb <52382992+chrishrb@users.noreply.github.com>
Date: Mon, 6 Jan 2025 21:58:20 +0100
Subject: [PATCH 1/2] 32 Fix syntax highlighting

---
 defaults/templates/layout.html |  4 ++++
 pkg/parser.go                  | 16 +++++++---------
 pkg/webserver.go               | 27 +++++++++++++++++++++++----
 3 files changed, 34 insertions(+), 13 deletions(-)

diff --git a/defaults/templates/layout.html b/defaults/templates/layout.html
index a8661cb..b639460 100644
--- a/defaults/templates/layout.html
+++ b/defaults/templates/layout.html
@@ -6,14 +6,18 @@
     <link rel="icon" type="image/x-icon" href="/static/images/favicon.ico" />
     {{if eq .Theme "dark" }}
     <link rel="stylesheet" href="/static/css/github-markdown-dark.css" />
+    <style>{{ .CssCodeDark }}</style>
     {{else if eq .Theme "light" }}
     <link rel="stylesheet" href="/static/css/github-markdown-light.css" />
+    <style>{{ .CssCodeLight }}</style>
     {{else}}
     <link
       rel="stylesheet"
       href="/static/css/github-markdown-light.css"
       media="(prefers-color-scheme: light)"
     />
+    <style media="(prefers-color-scheme: light)">{{ .CssCodeLight }}</style>
+    <style media="(prefers-color-scheme: dark)">{{ .CssCodeDark }}</style>
     <link
       rel="stylesheet"
       href="/static/css/github-markdown-dark.css"
diff --git a/pkg/parser.go b/pkg/parser.go
index 5164e9c..fa56871 100644
--- a/pkg/parser.go
+++ b/pkg/parser.go
@@ -10,7 +10,9 @@ import (
 	"regexp"
 	"strings"
 
-	"github.com/alecthomas/chroma/v2/quick"
+	chroma_html "github.com/alecthomas/chroma/v2/formatters/html"
+	"github.com/alecthomas/chroma/v2/lexers"
+	"github.com/alecthomas/chroma/v2/styles"
 	"github.com/chrishrb/go-grip/defaults"
 	"github.com/gomarkdown/markdown"
 	"github.com/gomarkdown/markdown/ast"
@@ -52,13 +54,6 @@ func (client *Client) renderHook(w io.Writer, node ast.Node, entering bool) (ast
 func renderHookCodeBlock(w io.Writer, node ast.Node, theme string) (ast.WalkStatus, bool) {
 	block := node.(*ast.CodeBlock)
 
-	var style string
-	if theme == "dark" {
-		style = "github-dark"
-	} else {
-		style = "github"
-	}
-
 	if string(block.Info) == "mermaid" {
 		m, err := renderMermaid(string(block.Literal), theme)
 		if err != nil {
@@ -68,7 +63,10 @@ func renderHookCodeBlock(w io.Writer, node ast.Node, theme string) (ast.WalkStat
 		return ast.GoToNext, true
 	}
 
-	err := quick.Highlight(w, string(block.Literal), string(block.Info), "html", style)
+	lexer := lexers.Get(string(block.Info))
+	iterator, _ := lexer.Tokenise(nil, string(block.Literal))
+	formatter := chroma_html.New(chroma_html.WithClasses(true))
+	err := formatter.Format(w, styles.Fallback, iterator)
 	if err != nil {
 		log.Println("Error:", err)
 	}
diff --git a/pkg/webserver.go b/pkg/webserver.go
index b2dd333..8c79cc8 100644
--- a/pkg/webserver.go
+++ b/pkg/webserver.go
@@ -9,16 +9,21 @@ import (
 	"net/url"
 	"path"
 	"regexp"
+	"strings"
 	"text/template"
 
 	"github.com/aarol/reload"
+	chroma_html "github.com/alecthomas/chroma/v2/formatters/html"
+	"github.com/alecthomas/chroma/v2/styles"
 	"github.com/chrishrb/go-grip/defaults"
 )
 
 type htmlStruct struct {
-	Content     string
-	Theme       string
-	BoundingBox bool
+	Content      string
+	Theme        string
+	BoundingBox  bool
+	CssCodeLight string
+	CssCodeDark  string
 }
 
 func (client *Client) Serve(file string) error {
@@ -60,7 +65,13 @@ func (client *Client) Serve(file string) error {
 			htmlContent := client.MdToHTML(bytes)
 
 			// Serve
-			err = serveTemplate(w, htmlStruct{Content: string(htmlContent), Theme: client.Theme, BoundingBox: client.BoundingBox})
+			err = serveTemplate(w, htmlStruct{
+				Content:      string(htmlContent),
+				Theme:        client.Theme,
+				BoundingBox:  client.BoundingBox,
+				CssCodeLight: getCssCode("github"),
+				CssCodeDark:  getCssCode("github-dark"),
+			})
 			if err != nil {
 				log.Fatal(err)
 				return
@@ -123,3 +134,11 @@ func serveTemplate(w http.ResponseWriter, html htmlStruct) error {
 	err = tmpl.Execute(w, html)
 	return err
 }
+
+func getCssCode(style string) string {
+	buf := new(strings.Builder)
+	formatter := chroma_html.New(chroma_html.WithClasses(true))
+	s := styles.Get(style)
+	_ = formatter.WriteCSS(buf, s)
+	return buf.String()
+}

From 646a7ce6544d509ece2c2e4af32c9e27760b0365 Mon Sep 17 00:00:00 2001
From: Christoph Herb <52382992+chrishrb@users.noreply.github.com>
Date: Mon, 6 Jan 2025 22:00:25 +0100
Subject: [PATCH 2/2] move style

---
 defaults/templates/layout.html | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/defaults/templates/layout.html b/defaults/templates/layout.html
index b639460..87f4545 100644
--- a/defaults/templates/layout.html
+++ b/defaults/templates/layout.html
@@ -16,13 +16,13 @@
       href="/static/css/github-markdown-light.css"
       media="(prefers-color-scheme: light)"
     />
-    <style media="(prefers-color-scheme: light)">{{ .CssCodeLight }}</style>
-    <style media="(prefers-color-scheme: dark)">{{ .CssCodeDark }}</style>
     <link
       rel="stylesheet"
       href="/static/css/github-markdown-dark.css"
       media="(prefers-color-scheme: dark)"
     />
+    <style media="(prefers-color-scheme: light)">{{ .CssCodeLight }}</style>
+    <style media="(prefers-color-scheme: dark)">{{ .CssCodeDark }}</style>
     {{end}}
     <link rel="stylesheet" href="/static/css/github-print.css" media="print" />
   </head>