main.go

package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/fsnotify/fsnotify"

	shellwords "github.com/mattn/go-shellwords"
)

var shellParser shellwords.Parser

func init() {
	shellParser.ParseEnv = true
	shellParser.ParseBacktick = true
}

func main() {
	configFile := flag.String("c", "", "configuration file path")
	flag.Parse()

	mainLogger := NewLogFunc("main")
	var (
		config *Config
		err    error
	)
	if *configFile != "" {
		config, err = NewConfig(*configFile)
		if err != nil {
			mainLogger(fmt.Sprintf("failed to create config: %s", err))
			os.Exit(1)
		}
	} else {
		config = NewDefaultConfig()
	}

	watcher, err := NewWatcher(config)
	if err != nil {
		mainLogger(fmt.Sprintf("failed to initialize watcher: %s", err))
		os.Exit(1)
	}
	runner, err := NewRunner(config)
	if err != nil {
		mainLogger(fmt.Sprintf("failed to initialize runner: %s", err))
		os.Exit(1)
	}
	builder, err := NewBuilder(config)
	if err != nil {
		mainLogger(fmt.Sprintf("failed to initialize builder: %s", err))
		os.Exit(1)
	}

	if err := builder.Build(); err != nil {
		mainLogger(fmt.Sprintf("failed to build: %s", err))
		os.Exit(1)
	}
	err = runner.Serve()
	if err != nil {
		mainLogger(fmt.Sprintf("failed to start server: %s", err))
		os.Exit(1)
	}
	defer runner.Stop()

	done := make(chan bool)
	go func() {
		for {
			select {
			case event, ok := <-watcher.w.Events:
				if !ok {
					mainLogger("failed to catch an event")
				}
				mainLogger(fmt.Sprintf("event: %s", event))
				if event.Op&fsnotify.Write == fsnotify.Write {
					e := event.String()
					mainLogger(fmt.Sprintf("file modified: %s", e))
					if config.RestartProcess {
						runner.Stop()
					}
					builder.Build()
					runner.Serve()
				}
			case err := <-watcher.w.Errors:
				mainLogger(fmt.Sprintf("error: %s", err))
			}
		}
	}()
	<-done
}