path: root/main.go

/*
snkt is a static site generator for simple blog-like sites with a focus on simplicity and efficiency.
*/
package main

import (
	"adammathes.com/snkt/config"
	"adammathes.com/snkt/render"
	"adammathes.com/snkt/site"
	"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, 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, "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 !watch && !help && !build && !preview && !version && init_dir == "" {
		flag.Usage()
		return
	}
	if init_dir != "" {
		fmt.Printf("initializing new site in %s\n", init_dir)
		site.Init(init_dir)
		return
	}
	if help {
		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 {
		buildSite()
	}

	if preview {
		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
}