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) } } } }() }