package cmd import ( "fmt" "os" "strings" "github.com/dutchcoders/transfer.sh/server" "github.com/fatih/color" "github.com/minio/cli" ) var Version = "0.1" var helpTemplate = `NAME: {{.Name}} - {{.Usage}} DESCRIPTION: {{.Description}} USAGE: {{.Name}} {{if .Flags}}[flags] {{end}}command{{if .Flags}}{{end}} [arguments...] COMMANDS: {{range .Commands}}{{join .Names ", "}}{{ "\t" }}{{.Usage}} {{end}}{{if .Flags}} FLAGS: {{range .Flags}}{{.}} {{end}}{{end}} VERSION: ` + Version + `{{ "\n"}}` var globalFlags = []cli.Flag{ cli.StringFlag{ Name: "listener", Usage: "127.0.0.1:8080", Value: "127.0.0.1:8080", }, // redirect to https? // hostnames cli.StringFlag{ Name: "profile-listener", Usage: "127.0.0.1:6060", Value: "", }, cli.BoolFlag{ Name: "force-https", Usage: "", }, cli.StringFlag{ Name: "tls-listener", Usage: "127.0.0.1:8443", Value: "", }, cli.StringFlag{ Name: "tls-cert-file", Value: "", }, cli.StringFlag{ Name: "tls-private-key", Value: "", }, cli.StringFlag{ Name: "temp-path", Usage: "path to temp files", Value: os.TempDir(), }, cli.StringFlag{ Name: "web-path", Usage: "path to static web files", Value: "", }, cli.StringFlag{ Name: "provider", Usage: "s3|local", Value: "", }, cli.StringFlag{ Name: "aws-access-key", Usage: "", Value: "", EnvVar: "AWS_ACCESS_KEY", }, cli.StringFlag{ Name: "aws-secret-key", Usage: "", Value: "", EnvVar: "AWS_SECRET_KEY", }, cli.StringFlag{ Name: "bucket", Usage: "", Value: "", EnvVar: "BUCKET", }, cli.IntFlag{ Name: "rate-limit", Usage: "requests per minute", Value: 0, EnvVar: "", }, cli.StringFlag{ Name: "lets-encrypt-hosts", Usage: "host1, host2", Value: "", EnvVar: "HOSTS", }, cli.StringFlag{ Name: "log", Usage: "/var/log/transfersh.log", Value: "", }, cli.StringFlag{ Name: "basedir", Usage: "path to storage", Value: "", }, cli.StringFlag{ Name: "clamav-host", Usage: "clamav-host", Value: "", EnvVar: "CLAMAV_HOST", }, cli.StringFlag{ Name: "virustotal-key", Usage: "virustotal-key", Value: "", EnvVar: "VIRUSTOTAL_KEY", }, cli.BoolFlag{ Name: "profiler", Usage: "enable profiling", }, } type Cmd struct { *cli.App } func VersionAction(c *cli.Context) { fmt.Println(color.YellowString(fmt.Sprintf("transfer.sh: Easy file sharing from the command line"))) } func New() *Cmd { app := cli.NewApp() app.Name = "transfer.sh" app.Author = "" app.Usage = "transfer.sh" app.Description = `Easy file sharing from the command line` app.Flags = globalFlags app.CustomAppHelpTemplate = helpTemplate app.Commands = []cli.Command{ { Name: "version", Action: VersionAction, }, } app.Before = func(c *cli.Context) error { return nil } app.Action = func(c *cli.Context) { options := []server.OptionFn{} if v := c.String("listener"); v != "" { options = append(options, server.Listener(v)) } if v := c.String("tls-listener"); v != "" { options = append(options, server.TLSListener(v)) } if v := c.String("profile-listener"); v != "" { options = append(options, server.ProfileListener(v)) } if v := c.String("web-path"); v != "" { options = append(options, server.WebPath(v)) } if v := c.String("temp-path"); v != "" { options = append(options, server.TempPath(v)) } if v := c.String("lets-encrypt-hosts"); v != "" { options = append(options, server.UseLetsEncrypt(strings.Split(v, ","))) } if v := c.String("virustotal-key"); v != "" { options = append(options, server.VirustotalKey(v)) } if v := c.String("clamav-host"); v != "" { options = append(options, server.ClamavHost(v)) } if v := c.Int("rate-limit"); v > 0 { options = append(options, server.RateLimit(v)) } if cert := c.String("tls-cert-file"); cert == "" { } else if pk := c.String("tls-private-key"); pk == "" { } else { options = append(options, server.TLSConfig(cert, pk)) } if c.Bool("profiler") { options = append(options, server.EnableProfiler()) } if c.Bool("force-https") { options = append(options, server.ForceHTTPs()) } switch provider := c.String("provider"); provider { case "s3": if accessKey := c.String("aws-access-key"); accessKey == "" { panic("access-key not set.") } else if secretKey := c.String("aws-secret-key"); secretKey == "" { panic("secret-key not set.") } else if bucket := c.String("bucket"); bucket == "" { panic("bucket not set.") } else if storage, err := server.NewS3Storage(accessKey, secretKey, bucket); err != nil { panic(err) } else { options = append(options, server.UseStorage(storage)) } case "local": if v := c.String("basedir"); v == "" { panic("basedir not set.") } else if storage, err := server.NewLocalStorage(v); err != nil { panic(err) } else { options = append(options, server.UseStorage(storage)) } default: panic("Provider not set or invalid.") } srvr, err := server.New( options..., ) if err != nil { fmt.Println(color.RedString("Error starting server: %s", err.Error())) return } srvr.Run() } return &Cmd{ App: app, } }