M README.md → README.md

@@ -14,6 +14,5 @@ ## Features
 - [x] server rendering .md files
 - [x] live reload when .md file changes
 - [x] directory files list on `/` endpoint
-- [ ] on error return error page, instead of panic server
 - [ ] graphviz
 - [ ] colour `<code>` with syntax highlighting

M main.go → main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"html/template"
 	"log"
 	"net/http"
@@ -59,6 +60,11 @@ vars := mux.Vars(r)
 	path := vars["path"]
 	path = strings.Replace(path, "%", "/", -1)
 	log.Println(path)
+
+	if len(strings.Split(path, ".")) < 2 {
+		fmt.Fprintf(w, errTemplate)
+		return
+	}
 
 	if strings.Split(path, ".")[1] != "md" {
 		http.ServeFile(w, r, path)

M md-live-server → md-live-server

M template.go → templates.go

@@ -74,3 +74,24 @@ </script>
 </body>
 </html>
 `
+
+const errTemplate = `
+<!DOCTYPE html>
+<html>
+<title>{{.Title}}</title>
+<style>
+body {
+    font-family: Arial, Helvetica, sans-serif;
+    background:#000000;
+    color:#cccccc;
+}
+</style>
+<body>
+
+<h3>File doesn't exist</h3>
+<br><br>
+<a href="/">Back to dir menu</a>
+
+</body>
+</html>
+`