main.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"net/http"
	"os"

	log "github.com/Sirupsen/logrus"
	ug "github.com/boivie/undergang/app"
	"github.com/urfave/cli"
)

var version = "(locally built)"

func main() {
	log.SetFormatter(&log.TextFormatter{})
	log.SetLevel(log.DebugLevel)

	app := cli.NewApp()
	app.Name = "undergang"
	app.Version = version
	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:  "listen",
			Value: ":8002",
			Usage: "Listening address (ip and port)",
		},
		cli.StringFlag{
			Name:  "pathinfo",
			Usage: "URL for pathinfo service",
		},
		cli.StringFlag{
			Name:  "sshproxy",
			Usage: "Optional utility for proxying SSH connections",
		},
		cli.StringFlag{
			Name:  "config",
			Usage: "Configuration file",
		},
		cli.BoolFlag{
			Name:  "json-log",
			Usage: "Log in JSON format",
		},
	}
	app.Action = func(c *cli.Context) {
		if c.Bool("json-log") {
			log.SetFormatter(&log.JSONFormatter{})
		}

		log.Info(`
                __
.--.--.-----.--|  |.-----.----.-----.---.-.-----.-----.
|  |  |     |  _  ||  -__|   _|  _  |  _  |     |  _  |
|_____|__|__|_____||_____|__| |___  |___._|__|__|___  |
                              |_____|           |_____|
`)
		log.Info("Version " + version)

		ug.Init(c.String("pathinfo"), c.String("sshproxy"), version)

		if c.String("config") != "" {
			var config struct {
				Paths []ug.PathInfo
			}
			buf, err := ioutil.ReadFile(c.String("config"))
			if err != nil {
				panic(err)
			}
			if err = json.Unmarshal(buf, &config); err != nil {
				panic(err)
			}
			for _, path := range config.Paths {
				ug.AddPath(path)
			}
		}

		log.Infof("Accepting requests on %s", c.String("listen"))
		err := http.ListenAndServe(c.String("listen"), nil)
		if err != nil {
			panic(err)
		}
	}

	app.Run(os.Args)
}