main.go

package main

import (
	"carpet/generator"
	"carpet/server"
	"flag"
	"fmt"
	"github.com/fsnotify/fsnotify"
	"gopkg.in/antage/eventsource.v1"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"time"
)

var watcher *fsnotify.Watcher

func compileProject(projectDir string) error {
	var files []string
	err := filepath.Walk(projectDir+"pages", func(path string, info os.FileInfo, e error) error {
		if e != nil {
			return e
		}
		if strings.HasSuffix(path, "html") {
			files = append(files, path)
		}
		return nil
	})
	if err != nil {
		return err
	}

	err = os.RemoveAll(projectDir + "dist")
	if err != nil {
		return err
	}
	err = os.Mkdir(projectDir+"dist", os.ModePerm)
	if err != nil {
		return err
	}

	for _, file := range files {
		if strings.HasSuffix(file, "html") {
			p := generator.Page{
				Filename:   file,
				ProjectDir: projectDir,
			}
			err := p.NewPage()
			if err != nil {
				return err
			}
			err = p.WriteToDisk()
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func watchProject(projectDir string) {
	watcher, _ = fsnotify.NewWatcher()
	defer watcher.Close()

	es := eventsource.New(nil, nil)
	defer es.Close()
	http.Handle("/es-subscribe", es)

	if err := filepath.Walk(projectDir+"pages/", watchDirectory); err != nil {
		log.Fatal(err)
	}
	if err := filepath.Walk(projectDir+"templates/", watchDirectory); err != nil {
		log.Fatal(err)
	}

	done := make(chan bool)

	go func() {
		for {
			select {
			case event := <-watcher.Events:
				if event.Op != fsnotify.Chmod {
					if !strings.HasSuffix(event.Name, "dist") && !strings.HasSuffix(event.Name, "templates") {
						es.SendEventMessage("reload", "reload-event", time.Time{}.String())
						fmt.Println("filesystem change detected, reloading...")
						err := compileProject(projectDir)
						if err != nil {
							log.Fatal(err)
						}
					}
				}
			case err := <-watcher.Errors:
				fmt.Println("ERROR", err)
			}
		}
	}()

	if err := watcher.Add(projectDir); err != nil {
		fmt.Println("ERROR", err)
	}
	<-done
}

func watchDirectory(path string, fi os.FileInfo, err error) error {
	fmt.Println(path)
	if fi.Mode().IsDir() && !strings.HasSuffix(path, ".html") {
		return watcher.Add(path)
	}
	return nil
}

func main() {
	var projectDirectory = flag.String("p", ".", "Project directory")
	var hotReloadFlag = flag.Bool("w", false, "Watch files")
	var serveFlag = flag.Bool("s", false, "Serve output directory")
	var serverPort = flag.String("port", "8090", "Server port")
	flag.Parse()

	projectDir := string(*projectDirectory)
	if projectDir[len(projectDir)-1] != '/' {
		projectDir += "/"
	}

	if *serveFlag {
		go server.Serve(projectDir+"dist/", *serverPort)
	}

	if !*hotReloadFlag {
		err := compileProject(projectDir)
		if err != nil {
			log.Fatal(err)
		}
	} else {
		watchProject(projectDir)
	}
}