gotty/main.go

128 lines
2.6 KiB
Go
Raw Normal View History

2015-08-16 09:47:23 +00:00
package main
import (
"fmt"
"os"
2015-08-27 06:23:54 +00:00
"os/signal"
"syscall"
2015-08-16 09:47:23 +00:00
"github.com/codegangsta/cli"
"github.com/yudai/gotty/app"
)
func main() {
cmd := cli.NewApp()
cmd.Version = "0.0.4"
2015-08-16 09:47:23 +00:00
cmd.Name = "gotty"
cmd.Usage = "Share your terminal as a web application"
2015-08-21 09:42:04 +00:00
cmd.HideHelp = true
2015-08-27 06:23:54 +00:00
flags := []flag{
flag{"address", "a", "IP address to listen"},
flag{"port", "p", "Port number to listen"},
flag{"permit-write", "w", "Permit clients to write to the TTY (BE CAREFUL)"},
flag{"credential", "c", "Credential for Basic Authentication (ex: user:pass, default disabled)"},
flag{"random-url", "r", "Add a random string to the URL"},
flag{"random-url-length", "", "Random URL length"},
flag{"tls", "t", "Enable TLS/SSL"},
flag{"tls-crt", "", "TLS/SSL crt file path"},
flag{"tls-key", "", "TLS/SSL key file path"},
flag{"profile", "", "Profile file path"},
flag{"title-format", "", "Title format of browser window"},
flag{"reconnect", "", "Enable reconnection"},
flag{"reconnect-time", "", "Time to reconnect"},
flag{"once", "", "Accept only one client and exit on disconnection"},
}
mappingHint := map[string]string{
"profile": "ProfileFile",
"tls": "EnableTLS",
"tls-crt": "TLSCrtFile",
"tls-key": "TLSKeyFile",
"random-url": "EnableRandomUrl",
"reconnect": "EnableReconnect",
}
cliFlags, err := generateFlags(flags, mappingHint)
if err != nil {
exit(err, 3)
}
cmd.Flags = append(
cliFlags,
cli.StringFlag{
2015-08-27 06:23:54 +00:00
Name: "config",
Value: "~/.gotty",
Usage: "Config file path",
EnvVar: "GOTTY_CONFIG",
},
2015-08-27 06:23:54 +00:00
)
2015-08-16 09:47:23 +00:00
cmd.Action = func(c *cli.Context) {
if len(c.Args()) == 0 {
2015-08-17 02:30:44 +00:00
fmt.Println("Error: No command given.\n")
2015-08-16 09:47:23 +00:00
cli.ShowAppHelp(c)
2015-08-27 06:23:54 +00:00
exit(err, 1)
}
options := app.DefaultOptions
configFile := c.String("config")
_, err := os.Stat(configFile)
if configFile != "~/.gotty" || !os.IsNotExist(err) {
if err := app.ApplyConfigFile(&options, configFile); err != nil {
exit(err, 2)
}
2015-08-16 09:47:23 +00:00
}
2015-08-23 11:40:18 +00:00
2015-08-27 06:23:54 +00:00
applyFlags(&options, flags, mappingHint, c)
if c.IsSet("credential") {
options.EnableBasicAuth = true
}
app, err := app.New(c.Args(), &options)
2015-08-16 09:47:23 +00:00
if err != nil {
2015-08-27 06:23:54 +00:00
exit(err, 3)
2015-08-16 09:47:23 +00:00
}
2015-08-23 11:40:18 +00:00
2015-08-24 10:22:25 +00:00
registerSignals(app)
2015-08-23 11:40:18 +00:00
err = app.Run()
if err != nil {
2015-08-27 06:23:54 +00:00
exit(err, 4)
2015-08-23 11:40:18 +00:00
}
2015-08-16 09:47:23 +00:00
}
2015-08-17 02:30:44 +00:00
2015-08-21 09:42:04 +00:00
cli.AppHelpTemplate = helpTemplate
2015-08-17 02:30:44 +00:00
2015-08-16 09:47:23 +00:00
cmd.Run(os.Args)
}
2015-08-24 10:22:25 +00:00
2015-08-27 06:23:54 +00:00
func exit(err error, code int) {
fmt.Println(err)
os.Exit(code)
}
2015-08-24 10:22:25 +00:00
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() {
2015-08-27 06:23:54 +00:00
os.Exit(5)
2015-08-24 10:22:25 +00:00
}
}
}
}()
}