130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/codegangsta/cli"
|
|
|
|
"github.com/yudai/gotty/app"
|
|
)
|
|
|
|
func main() {
|
|
cmd := cli.NewApp()
|
|
cmd.Version = "0.0.6"
|
|
cmd.Name = "gotty"
|
|
cmd.Usage = "Share your terminal as a web application"
|
|
cmd.HideHelp = true
|
|
|
|
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{"index", "", "Custom index file"},
|
|
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{
|
|
"index": "IndexFile",
|
|
"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{
|
|
Name: "config",
|
|
Value: "~/.gotty",
|
|
Usage: "Config file path",
|
|
EnvVar: "GOTTY_CONFIG",
|
|
},
|
|
)
|
|
|
|
cmd.Action = func(c *cli.Context) {
|
|
if len(c.Args()) == 0 {
|
|
fmt.Println("Error: No command given.\n")
|
|
cli.ShowAppHelp(c)
|
|
exit(err, 1)
|
|
}
|
|
|
|
options := app.DefaultOptions
|
|
|
|
configFile := c.String("config")
|
|
_, err := os.Stat(app.ExpandHomeDir(configFile))
|
|
if configFile != "~/.gotty" || !os.IsNotExist(err) {
|
|
if err := app.ApplyConfigFile(&options, configFile); err != nil {
|
|
exit(err, 2)
|
|
}
|
|
}
|
|
|
|
applyFlags(&options, flags, mappingHint, c)
|
|
|
|
if c.IsSet("credential") {
|
|
options.EnableBasicAuth = true
|
|
}
|
|
|
|
app, err := app.New(c.Args(), &options)
|
|
if err != nil {
|
|
exit(err, 3)
|
|
}
|
|
|
|
registerSignals(app)
|
|
|
|
err = app.Run()
|
|
if err != nil {
|
|
exit(err, 4)
|
|
}
|
|
}
|
|
|
|
cli.AppHelpTemplate = helpTemplate
|
|
|
|
cmd.Run(os.Args)
|
|
}
|
|
|
|
func exit(err error, code int) {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
os.Exit(code)
|
|
}
|
|
|
|
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(5)
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
}
|