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