gotty/main.go

package main

import (
	"fmt"
	"os"

	"github.com/codegangsta/cli"

	"github.com/yudai/gotty/app"
	"os/signal"
	"syscall"
)

func main() {
	cmd := cli.NewApp()
	cmd.Version = "0.0.3"
	cmd.Name = "gotty"
	cmd.Usage = "Share your terminal as a web application"
	cmd.HideHelp = true
	cmd.Flags = []cli.Flag{
		cli.StringFlag{
			Name:   "addr, a",
			Value:  "",
			Usage:  "IP address to listen",
			EnvVar: "GOTTY_ADDR",
		},
		cli.StringFlag{
			Name:   "port, p",
			Value:  "8080",
			Usage:  "Port number to listen",
			EnvVar: "GOTTY_PORT",
		},
		cli.BoolFlag{
			Name:   "permit-write, w",
			Usage:  "Permit clients to write to the TTY (BE CAREFUL)",
			EnvVar: "GOTTY_PERMIT_WRITE",
		},
		cli.StringFlag{
			Name:   "credential, c",
			Usage:  "Credential for Basic Authentication (ex: user:pass)",
			EnvVar: "GOTTY_CREDENTIAL",
		},
		cli.BoolFlag{
			Name:   "random-url, r",
			Usage:  "Add a random string to the URL",
			EnvVar: "GOTTY_RANDOM_URL",
		},
		cli.StringFlag{
			Name:   "profile-file, f",
			Value:  app.DefaultProfileFilePath,
			Usage:  "Path to profile file",
			EnvVar: "GOTTY_PROFILE_FILE",
		},
		cli.BoolFlag{
			Name:   "enable-tls, t",
			Usage:  "Enable TLS/SSL",
			EnvVar: "GOTTY_ENABLE_TLS",
		},
		cli.StringFlag{
			Name:   "tls-cert",
			Value:  app.DefaultTLSCertPath,
			Usage:  "TLS/SSL cert",
			EnvVar: "GOTTY_TLS_CERT",
		},
		cli.StringFlag{
			Name:   "tls-key",
			Value:  app.DefaultTLSKeyPath,
			Usage:  "TLS/SSL key",
			EnvVar: "GOTTY_TLS_KEY",
		},
		cli.StringFlag{
			Name:   "title-format",
			Value:  "GoTTY - {{ .Command }} ({{ .Hostname }})",
			Usage:  "Title format of browser window",
			EnvVar: "GOTTY_TITLE_FORMAT",
		},
		cli.IntFlag{
			Name:   "auto-reconnect",
			Value:  -1,
			Usage:  "Seconds to automatically reconnect to the server when the connection is closed (default: disabled)",
			EnvVar: "GOTTY_AUTO_RECONNECT",
		},
		cli.BoolFlag{
			Name:   "once",
			Usage:  "Accept only one client and exit on disconnection",
			EnvVar: "GOTTY_ONCE",
		},
	}
	cmd.Action = func(c *cli.Context) {
		if len(c.Args()) == 0 {
			fmt.Println("Error: No command given.\n")
			cli.ShowAppHelp(c)
			os.Exit(1)
		}

		app, err := app.New(
			app.Options{
				c.String("addr"),
				c.String("port"),
				c.Bool("permit-write"),
				c.String("credential"),
				c.Bool("random-url"),
				c.String("profile-file"),
				c.Bool("enable-tls"),
				c.String("tls-cert"),
				c.String("tls-key"),
				c.String("title-format"),
				c.Int("auto-reconnect"),
				c.Bool("once"),
				c.Args(),
			},
		)
		if err != nil {
			fmt.Println(err)
			os.Exit(2)
		}

		registerSignals(app)

		err = app.Run()
		if err != nil {
			fmt.Println(err)
			os.Exit(3)
		}
	}

	cli.AppHelpTemplate = helpTemplate

	cmd.Run(os.Args)
}

func registerSignals(app *app.App) {
	sigChan := make(chan os.Signal, 1)
	signal.Notify(
		sigChan,
		syscall.SIGINT,
		syscall.SIGTERM,
	)

	go func() {
		for {
			s := <-sigChan
			switch s {
			case syscall.SIGINT, syscall.SIGTERM:
				if !app.Exit() {
					os.Exit(4)
				}
			}
		}
	}()
}