diff options
| -rw-r--r-- | main.go | 76 |
1 files changed, 60 insertions, 16 deletions
@@ -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 } |