Fix markdown file hot reload when layout changes

richcorbs · Oct 8, 2024 · e04dff9 · e04dff9
1 parent 2280ced
commit e04dff9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/build.go b/build.go
@@ -200,7 +200,7 @@ func initializeDependencies() error {
 		file := filepath.Base(path)
 		ext := filepath.Ext(file)
 
-		if ext == ".html" {
+		if ext == ".html" || ext == ".md" {
 			content, err := os.ReadFile(path)
 			if err != nil {
 				return err
@@ -209,7 +209,7 @@ func initializeDependencies() error {
 			foundLayout := false
 
 			for _, layout := range layouts {
-				if strings.HasPrefix(string(content), "<"+layout.Name+"Layout>") && strings.HasSuffix(string(content), "</"+layout.Name+"Layout>") {
+				if strings.HasPrefix(string(content), "<"+layout.Name+"Layout>") && (strings.HasSuffix(string(content), "</"+layout.Name+"Layout>") || strings.HasSuffix(string(content), "</"+layout.Name+"Layout>\n")) {
 					foundLayout = true
 					if !sliceContains(path, dependencies[layout.Path]) {
 						dependencies[layout.Path] = append(dependencies[layout.Path], path)

diff --git a/main.go b/main.go
@@ -34,7 +34,7 @@ var PORT = "8080"
 
 const DIST = "dist"
 const SRC = "./src"
-const DEFAULT_LAYOUT = "./src/layouts/Default.html"
+const DEFAULT_LAYOUT = "src/layouts/Default.html"
 
 // src
 // └── assets