server.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"path"
	"regexp"

	"github.com/togostanza/ts/provider"
)

var cmdServer = &Command{
	Run:       runServer,
	Name:      "server",
	Short:     "run server",
	UsageLine: "server [-port port] [-stanza-base-dir dir] [-development]",
	Long:      "Run ts server for development",
}

var REGEXP_STANZA_PATH = regexp.MustCompile(`^/stanza/([^/]+)/`)
var flagServerDevelopment bool

func init() {
	cmdServer.Flag.IntVar(&flagPort, "port", 8080, "port to listen on")
	cmdServer.Flag.BoolVar(&flagServerDevelopment, "development", true, "development mode")
	addBuildFlags(cmdServer)
}

func runServer(cmd *Command, args []string) {
	sp, err := provider.New(flagStanzaBaseDir)
	if err != nil {
		log.Fatal(err)
	}

	distPath := path.Join(flagStanzaBaseDir, "dist")
	distStanzaPath := path.Join(distPath, "stanza")
	if err := sp.Build(distStanzaPath, flagServerDevelopment); err != nil {
		log.Fatal(err)
	}

	mux := http.NewServeMux()
	assetsHandler := http.FileServer(http.Dir(distPath))

	mux.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
		w.Header().Set("Access-Control-Allow-Origin", "*")
		if req.URL.Path == "/" {
			http.Redirect(w, req, "/stanza/", http.StatusFound)
			return
		}
		if m := REGEXP_STANZA_PATH.FindStringSubmatch(req.URL.Path); len(m) > 0 {
			if m[1] != "assets" {
				err := sp.RebuildIfRequired(distStanzaPath, flagServerDevelopment)
				if err != nil {
					log.Println("ERROR during rebuild:", err)
				}
			}
		}
		assetsHandler.ServeHTTP(w, req)
	})

	addr := fmt.Sprintf(":%d", flagPort)
	log.Println("listening on", addr)

	if err := http.ListenAndServe(addr, mux); err != nil {
		log.Fatal(err)
	}
}