barak/livemd
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

watching files, listing Markdown files

Browse source
This commit is contained in:
Barak Michener 2015-02-06 19:02:12 -05:00
parent cd29215570
commit 7389d26323
3 changed files with 135 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
.gitignore vendored
View file

@ -3,6 +3,9 @@
*.a *.a
*.so *.so
# Editor cruft
*.swp
# Folders # Folders
_obj _obj
_test _test
@ -20,3 +23,5 @@ _cgo_export.*
_testmain.go _testmain.go
*.exe *.exe
livemd

117
livemd.go Normal file
View file

@ -0,0 +1,117 @@
package main
import (
"fmt"
_ "html/template"
"log"
"net/http"
"os"
"path/filepath"
"strings"
"sync"
txttemplate "text/template"
"github.com/russross/blackfriday"
fsnotify "gopkg.in/fsnotify.v1"
)
var SUFFIXES = [3]string{".md", ".mkd", ".markdown"}
var toc []string
var toc_mutex sync.Mutex
var rootTmpl *txttemplate.Template
var path string
func init() {
var err error
rootTmpl, err = txttemplate.New("root").Parse(rootTemplate)
if err != nil {
log.Fatal(err)
}
}
func HasMarkdownSuffix(s string) bool {
for _, suffix := range SUFFIXES {
if strings.HasSuffix(strings.ToLower(s), suffix) {
return true
}
}
return false
}
func AddWatch(w *fsnotify.Watcher) filepath.WalkFunc {
return func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.IsDir() {
w.Add(path)
} else {
if HasMarkdownSuffix(path) {
toc_mutex.Lock()
toc = append(toc, path)
toc_mutex.Unlock()
log.Println("Found", path)
w.Add(path)
}
}
return nil
}
}
func WatcherEventLoop(w *fsnotify.Watcher, done chan bool) {
for {
select {
case event := <-w.Events:
log.Println("Event:", event)
// TODO(barakmich): On directory creation, stat path if directory, and watch it.
if HasMarkdownSuffix(event.Name) {
}
case err := <-w.Errors:
log.Println("Error:", err)
done <- true
}
}
}
func RootFunc(w http.ResponseWriter, r *http.Request) {
toc_mutex.Lock()
localToc := toc[:]
toc_mutex.Unlock()
for i, s := range localToc {
s = strings.TrimPrefix(s, path)
localToc[i] = "* " + s
}
tocmkd := strings.Join(localToc, "\n")
bytes := blackfriday.MarkdownCommon([]byte(tocmkd))
rootTmpl.Execute(w, string(bytes))
}
func main() {
path = os.Getenv("PWD")
if len(os.Args) > 1 {
path = os.Args[1]
}
watcher, err := fsnotify.NewWatcher()
if err != nil {
log.Fatal(err)
}
defer watcher.Close()
done := make(chan bool)
go WatcherEventLoop(watcher, done)
log.Println("Watching directory", path)
err = filepath.Walk(path, AddWatch(watcher))
if err != nil {
log.Fatal(err)
}
fmt.Println(toc)
fmt.Println(blackfriday.HTML_TOC)
http.HandleFunc("/", RootFunc)
http.ListenAndServe(":8080", nil)
}

12
root_template.go Normal file
View file

@ -0,0 +1,12 @@
package main
var rootTemplate = `
<html>
<head>
<title>LiveMarkdown</title>
</head>
<body>
{{.}}
</body>
</html>
`