main.go

package main

import (
	"flag"
	"fmt"
	"github.com/kylelemons/go-gypsy/yaml"
	"os"
	"os/exec"
	"path"
	"runtime"
)

const ConfigName = ".tower.yml"

func main() {
	appMainFile := flag.String("m", "main.go", "path to your app's main file.")
	appPort := flag.String("p", "5000", "port of your app.")
	verbose := flag.Bool("v", false, "show more stuff.")

	flag.Parse()

	args := flag.Args()
	if len(args) == 1 && args[0] == "init" {
		generateExampleConfig()
		return
	}

	startTower(*appMainFile, *appPort, *verbose)
}

func generateExampleConfig() {
	_, file, _, _ := runtime.Caller(0)
	exampleConfig := path.Dir(file) + "/tower.yml"
	exec.Command("cp", exampleConfig, ConfigName).Run()
	fmt.Println("== Generated config file " + ConfigName)
}

var (
	app App
)

func startTower(appMainFile, appPort string, verbose bool) {
	watchedFiles := ""

	config, err := yaml.ReadFile(ConfigName)
	if err == nil {
		if verbose {
			fmt.Println("== Load config from " + ConfigName)
		}
		appMainFile, _ = config.Get("main")
		appPort, _ = config.Get("port")
		watchedFiles, _ = config.Get("watch")
	}

	err = dialAddress("127.0.0.1:"+appPort, 1)
	if err == nil {
		fmt.Println("Error: port (" + appPort + ") already in used.")
		os.Exit(1)
	}

	if verbose {
		fmt.Println("== Application Info")
		fmt.Printf("  build app with: %s\n", appMainFile)
		fmt.Printf("  redirect requests from localhost:%s to localhost:%s\n\n", ProxyPort, appPort)
	}

	app = NewApp(appMainFile, appPort)
	watcher := NewWatcher(app.Root, watchedFiles)
	proxy := NewProxy(&app, &watcher)

	go func() {
		mustSuccess(watcher.Watch())
	}()
	mustSuccess(proxy.Listen())
}