Merge pull request #2 from eleksir/gomodules

Gomodules

Author: zaz600

config.go

@@ -0,0 +1,9 @@
+module github.com/zaz600/mdserver
+
+go 1.16
+
+require (
+	github.com/bmizerany/pat v0.0.0-20210406213842-e4b6760bdd6f
+	github.com/russross/blackfriday v1.6.0
+	gopkg.in/yaml.v2 v2.4.0
+)

go.mod

@@ -0,0 +1,8 @@
+github.com/bmizerany/pat v0.0.0-20210406213842-e4b6760bdd6f h1:gOO/tNZMjjvTKZWpY7YnXC72ULNLErRtp94LountVE8=
+github.com/bmizerany/pat v0.0.0-20210406213842-e4b6760bdd6f/go.mod h1:8rLXio+WjiTceGBHIoTvn60HIbs7Hm7bcHjyrSqYB9c=
+github.com/russross/blackfriday v1.6.0 h1:KqfZb0pUVN2lYqZUYRddxF4OR8ZMURnJIG5Y3VRLtww=
+github.com/russross/blackfriday v1.6.0/go.mod h1:ti0ldHuxg49ri4ksnFxlkCfN+hvslNlmVHqNRXXJNAY=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=

go.sum

@@ -1,19 +1,30 @@
 package main
 
 import (
+	"fmt"
 	"html/template"
+	"io/ioutil"
 	"log"
 	"net/http"
+	"os"
 	"path"
 	"strings"
+	"sync"
 
 	"github.com/bmizerany/pat"
+	"github.com/russross/blackfriday"
+	"gopkg.in/yaml.v2"
 )
 
 const (
 	configFileName = "mdserver.yaml"
 )
 
+// Config - структура для считывания конфигурационного файла
+type Config struct {
+	Listen string `yaml:"listen"`
+}
+
 var (
 	// компилируем шаблоны, если не удалось, то выходим
 	postTemplate  = template.Must(template.ParseFiles(path.Join("templates", "layout.html"), path.Join("templates", "post.html")))
@@ -23,9 +34,11 @@ var (
 
 func main() {
 	cfg, err := readConfig(configFileName)
+
 	if err != nil {
 		log.Fatalln(err)
 	}
+
 	// для отдачи сервером статичных файлов из папки public/static
 	fs := noDirListing(http.FileServer(http.Dir("./public/static")))
 	http.Handle("/static/", http.StripPrefix("/static/", fs))
@@ -54,6 +67,7 @@ func postHandler(w http.ResponseWriter, r *http.Request) {
 	p := path.Join("posts", page)
 
 	var postMD string
+
 	if page != "" {
 		// если page не пусто, то считаем, что запрашивается файл
 		// получим posts/p1.md
@@ -64,11 +78,16 @@ func postHandler(w http.ResponseWriter, r *http.Request) {
 	}
 
 	post, status, err := posts.Get(postMD)
+
 	if err != nil {
 		errorHandler(w, r, status)
+
 		return
 	}
-	if err := postTemplate.ExecuteTemplate(w, "layout", post); err != nil {
+
+	err := postTemplate.ExecuteTemplate(w, "layout", post)
+
+	if err != nil {
 		log.Println(err.Error())
 		errorHandler(w, r, 500)
 	}
@@ -77,9 +96,13 @@ func postHandler(w http.ResponseWriter, r *http.Request) {
 func errorHandler(w http.ResponseWriter, r *http.Request, status int) {
 	log.Printf("error %d %s %s\n", status, r.RemoteAddr, r.URL.Path)
 	w.WriteHeader(status)
-	if err := errorTemplate.ExecuteTemplate(w, "layout", map[string]interface{}{"Error": http.StatusText(status), "Status": status}); err != nil {
+
+	err := errorTemplate.ExecuteTemplate(w, "layout", map[string]interface{}{"Error": http.StatusText(status), "Status": status})
+
+	if err != nil {
 		log.Println(err.Error())
 		http.Error(w, http.StatusText(500), 500)
+
 		return
 	}
 }
@@ -92,8 +115,84 @@ func noDirListing(h http.Handler) http.HandlerFunc {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if strings.HasSuffix(r.URL.Path, "/") || r.URL.Path == "" {
 			http.NotFound(w, r)
+
 			return
 		}
+
 		h.ServeHTTP(w, r)
 	})
 }
+
+func readConfig(ConfigName string) (conf *Config, err error) {
+	var file []byte
+
+	file, err = ioutil.ReadFile(ConfigName)
+
+	if err != nil {
+		return nil, err
+	}
+
+	err = yaml.Unmarshal(file, conf)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return conf, nil
+}
+
+type post struct {
+	Title   string
+	Body    template.HTML
+	ModTime int64
+}
+
+type postArray struct {
+	Items map[string]post
+	sync.RWMutex
+}
+
+func newPostArray() *postArray {
+	p := postArray{}
+	p.Items = make(map[string]post)
+
+	return &p
+}
+
+// Get Загружает markdown-файл и конвертирует его в HTML
+// Возвращает объект типа Post
+// Если путь не существует или является каталогом, то возвращаем ошибку
+func (p *postArray) Get(md string) (post, int, error) {
+	info, err := os.Stat(md)
+
+	if err != nil {
+		if os.IsNotExist(err) {
+			// файл не существует
+			return post{}, 404, err
+		}
+
+		return post{}, 500, err
+	}
+
+	if info.IsDir() {
+		// не файл, а папка
+		return post{}, 404, fmt.Errorf("dir")
+	}
+
+	val, ok := p.Items[md]
+
+	if !ok || (ok && val.ModTime != info.ModTime().UnixNano()) {
+		p.RLock()
+		defer p.RUnlock()
+		fileread, _ := ioutil.ReadFile(md)
+		lines := strings.Split(string(fileread), "\n")
+		title := string(lines[0])
+		body := strings.Join(lines[1:], "\n")
+		body = string(blackfriday.MarkdownCommon([]byte(body)))
+		p.Items[md] = post{title, template.HTML(body), info.ModTime().UnixNano()}
+	}
+
+	post := p.Items[md]
+
+	return post, 200, nil
+}

mdserver.go renamed to main.go

@@ -1 +1 @@
-listen: 0.0.0.0:8890
+listen: 0.0.0.0:8890

mdserver.yaml.example

@@ -1,4 +1,4 @@
-body { padding-bottom: 70px; padding-top: 70px; }
+body { padding-bottom: 70px; padding-top: 70px; }
 #container
 {
 	background:#fff;