diff --git a/cmd/cmd.go b/cmd/cmd.go index 40a16af..9163071 100644 --- a/cmd/cmd.go +++ b/cmd/cmd.go @@ -117,6 +117,12 @@ var globalFlags = []cli.Flag{ Value: "", EnvVar: "CLAMAV_HOST", }, + cli.StringFlag{ + Name: "virustotal-key", + Usage: "virustotal-key", + Value: "", + EnvVar: "VIRUSTOTAL_KEY", + }, cli.BoolFlag{ Name: "profiler", Usage: "enable profiling", @@ -176,6 +182,10 @@ func New() *Cmd { 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)) } diff --git a/server/server.go b/server/server.go index 2d5ad59..a1c37a5 100644 --- a/server/server.go +++ b/server/server.go @@ -66,6 +66,12 @@ func ClamavHost(s string) OptionFn { } } +func VirustotalKey(s string) OptionFn { + return func(srvr *Server) { + srvr.VirusTotalKey = s + } +} + func Listener(s string) OptionFn { return func(srvr *Server) { srvr.ListenerString = s