gotty/main.go
2015-08-24 19:50:22 +09:00

152 lines
3.0 KiB
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)
}
}
}
}()
}