add filesystem watcher

1 files changed, 60 insertions, 16 deletions

diff --git a/main.go b/main.go
index 5536742..97aad6f 100644
--- a/main.go
+++ b/main.go
@@ -10,23 +10,25 @@ import (
 	"adammathes.com/snkt/vlog"
 	"adammathes.com/snkt/web"
 	"fmt"
+	"github.com/fsnotify/fsnotify"
 	flag "github.com/ogier/pflag"
 )
 
 func main() {
 
 	var configFile, init_dir string
-	var build, preview, version, verbose, help bool
+	var build, preview, version, verbose, help, watch bool
 
 	flag.StringVarP(&configFile, "config", "c", "config.yml", "`configuration` file")
 	flag.StringVarP(&init_dir, "init", "i", "", "initialize new site at `directory`")
-	flag.BoolVarP(&build, "build", "b", false, "build site")
-	flag.BoolVarP(&preview, "preview", "p", false, "preview site with local HTTP server")
-	flag.BoolVarP(&help, "help", "h", false, "print help message")
-	flag.BoolVarP(&verbose, "verbose", "v", false, "log actions during build to STDOUT")
+	flag.BoolVarP(&build, "build", "b", false, "generates site from input files and templates")
+	flag.BoolVarP(&preview, "preview", "p", false, "preview site via spawned HTTP server")
+	flag.BoolVarP(&help, "help", "h", false, "print usage information")
+	flag.BoolVarP(&verbose, "verbose", "v", false, "log operations during build to STDOUT")
+	flag.BoolVarP(&watch, "watch", "w", false, "watch configured input text dir, rebuild on changes")
 	flag.Parse()
 
-	if !help && !build && !preview && !version && init_dir == "" {
+	if !watch && !help && !build && !preview && !version && init_dir == "" {
 		flag.Usage()
 		return
 	}
@@ -36,26 +38,68 @@ func main() {
 		return
 	}
 	if help {
-		fmt.Printf("please see README.md\n")
+		flag.Usage()
+		fmt.Printf("code and docs: https://github.com/adammathes/snkt \n")
 		return
 	}
+
 	config.Init(configFile)
 	if verbose {
 		config.Config.Verbose = true
 	}
-
 	render.Init()
+
 	if build {
-		var s site.Site
-		vlog.Printf("reading posts...\n")
-		s.Read()
-		vlog.Printf("writing posts and archives...\n")
-		s.Write()
+		buildSite()
 	}
 
 	if preview {
-		fmt.Printf("spawning preview at [%s] of [%s]\n",
-			config.Config.PreviewDir, config.Config.PreviewServer)
-		web.Serve(config.Config.PreviewServer, config.Config.PreviewDir)
+		fmt.Printf("spawning preview server [%s] of [%s]\n",
+			config.Config.PreviewServer, config.Config.PreviewDir)
+		go func() {
+			web.Serve(config.Config.PreviewServer, config.Config.PreviewDir)
+		}()
+	}
+
+	if watch {
+		fmt.Printf("watching directory %s\n", config.Config.TxtDir)
+		watchSite()
+	}
+}
+
+func buildSite() {
+	var s site.Site
+	vlog.Printf("reading posts...\n")
+	s.Read()
+	vlog.Printf("writing posts and archives...\n")
+	s.Write()
+}
+
+func watchSite() {
+	watcher, err := fsnotify.NewWatcher()
+	if err != nil {
+		panic(err)
+	}
+	defer watcher.Close()
+
+	done := make(chan bool)
+	go func() {
+		for {
+			select {
+			case event := <-watcher.Events:
+				if event.Op&fsnotify.Write == fsnotify.Write {
+					fmt.Printf("rebuilding\n")
+					buildSite()
+				}
+			case err := <-watcher.Errors:
+				vlog.Printf("error: %v", err)
+			}
+		}
+	}()
+
+	err = watcher.Add(config.Config.TxtDir)
+	if err != nil {
+		panic(err)
 	}
+	<-done
 }