all: embed CSS and index HTML template using memfs

Previously, with direct embed, every time we change the CSS or index
template, we need to restart the "ciigo serve" command.

Using memfs make us easy to update and see the changes directly, without
restarting the server.

1 files changed, 51 insertions, 16 deletions

diff --git a/converter.go b/converter.go
index 57695e7..ea63510 100644
--- a/converter.go
+++ b/converter.go
@@ -1,5 +1,5 @@
-// SPDX-FileCopyrightText: 2019 Shulhan <ms@kilabit.info>
 // SPDX-License-Identifier: GPL-3.0-or-later
+// SPDX-FileCopyrightText: 2019 Shulhan <ms@kilabit.info>
 
 package ciigo
 
@@ -12,6 +12,7 @@ import (
 	"time"
 
 	"git.sr.ht/~shulhan/asciidoctor-go"
+	"git.sr.ht/~shulhan/pakakeh.go/lib/memfs"
 	"github.com/yuin/goldmark"
 	meta "github.com/yuin/goldmark-meta"
 	"github.com/yuin/goldmark/parser"
@@ -31,18 +32,20 @@ type Converter struct {
 // NewConverter create and initialize Converter with HTML template.
 // If htmlTemplate is empty, it will use the internal, predefined template.
 func NewConverter(htmlTemplate string) (converter *Converter, err error) {
-	var (
-		logp = `NewConverter`
-
-		tmplContent string
-	)
+	var logp = `NewConverter`
 
 	converter = &Converter{}
 
 	converter.tmpl = template.New(``)
 
+	var bhtml []byte
 	if len(htmlTemplate) == 0 {
-		tmplContent = templateIndexHTML
+		var node *memfs.Node
+		node, err = staticfs.Get(`/index.gohtml`)
+		if err != nil {
+			return nil, fmt.Errorf(`%s: %w`, logp, err)
+		}
+		bhtml = node.Content
 	} else {
 		converter.htmlTemplate = filepath.Clean(htmlTemplate)
 
@@ -55,15 +58,12 @@ func NewConverter(htmlTemplate string) (converter *Converter, err error) {
 
 		converter.htmlTemplateModtime = fi.ModTime()
 
-		var bhtml []byte
-
 		bhtml, err = os.ReadFile(converter.htmlTemplate)
 		if err != nil {
 			return nil, fmt.Errorf(`%s: %w`, logp, err)
 		}
-
-		tmplContent = string(bhtml)
 	}
+	var tmplContent = string(bhtml)
 
 	converter.tmpl, err = converter.tmpl.Parse(tmplContent)
 	if err != nil {
@@ -90,7 +90,7 @@ func (converter *Converter) convertFileMarkups(fileMarkups map[string]*FileMarku
 
 	for _, fmarkup = range fileMarkups {
 		if !isForce {
-			if !converter.shouldConvert(fmarkup) {
+			if !converter.shouldConvert(fmarkup, isForce) {
 				continue
 			}
 		}
@@ -105,7 +105,12 @@ func (converter *Converter) convertFileMarkups(fileMarkups map[string]*FileMarku
 }
 
 func (converter *Converter) htmlTemplateUseInternal() (err error) {
-	converter.tmpl, err = converter.tmpl.Parse(templateIndexHTML)
+	var node *memfs.Node
+	node, err = staticfs.Get(`/index.gohtml`)
+	if err != nil {
+		return err
+	}
+	converter.tmpl, err = converter.tmpl.Parse(string(node.Content))
 	if err != nil {
 		return err
 	}
@@ -217,9 +222,14 @@ func (converter *Converter) markdownToHTML(fmarkup *FileMarkup) (fhtml *fileHTML
 
 // shouldConvert will return true if the file markup fmarkup needs to be
 // converted to HTML.
-// It return true if the HTML file not exist or the template or markup file is
-// newer than the HTML file.
-func (converter *Converter) shouldConvert(fmarkup *FileMarkup) bool {
+// It return true if,
+//   - the HTML file not exist, or
+//   - the markup file is newer than the HTML file,
+//
+// In development mode, it will return true if,
+//   - the template file is newer than the HTML file, or
+//   - the CSS file is newer than the HTML file.
+func (converter *Converter) shouldConvert(fmarkup *FileMarkup, isDevelopment bool) bool {
 	var fi os.FileInfo
 	fi, _ = os.Stat(fmarkup.pathHTML)
 	if fi == nil {
@@ -252,5 +262,30 @@ func (converter *Converter) shouldConvert(fmarkup *FileMarkup) bool {
 		fmarkup.info = fi
 		return true
 	}
+
+	if !isDevelopment {
+		return false
+	}
+
+	// Check with staticfs for CSS and index HTML files.
+	if staticfs == nil {
+		return false
+	}
+	var node *memfs.Node
+	node, _ = staticfs.Get(`/index.gohtml`)
+	if node != nil && node.ModTime().After(htmlModtime) {
+		var newtmpl = template.New(``)
+		newtmpl, err = newtmpl.Parse(string(node.Content))
+		if err != nil {
+			return false
+		}
+		converter.tmpl = newtmpl
+		return true
+	}
+	node, _ = staticfs.Get(`/ciigo.css`)
+	if node != nil && node.ModTime().After(htmlModtime) {
+		return true
+	}
+
 	return false
 }