You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
1.9 KiB
95 lines
1.9 KiB
package cli
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
|
|
"github.com/halverneus/static-file-server/cli/help"
|
|
"github.com/halverneus/static-file-server/cli/server"
|
|
"github.com/halverneus/static-file-server/cli/version"
|
|
"github.com/halverneus/static-file-server/config"
|
|
)
|
|
|
|
var (
|
|
option struct {
|
|
configFile string
|
|
helpFlag bool
|
|
versionFlag bool
|
|
}
|
|
)
|
|
|
|
// Assignments used to simplify testing.
|
|
var (
|
|
selectRoutine = selectionRoutine
|
|
unknownArgsFunc = unknownArgs
|
|
runServerFunc = server.Run
|
|
runHelpFunc = help.Run
|
|
runVersionFunc = version.Run
|
|
loadConfig = config.Load
|
|
)
|
|
|
|
func init() {
|
|
setupFlags()
|
|
}
|
|
|
|
func setupFlags() {
|
|
flag.StringVar(&option.configFile, "config", "", "")
|
|
flag.StringVar(&option.configFile, "c", "", "")
|
|
flag.BoolVar(&option.helpFlag, "help", false, "")
|
|
flag.BoolVar(&option.helpFlag, "h", false, "")
|
|
flag.BoolVar(&option.versionFlag, "version", false, "")
|
|
flag.BoolVar(&option.versionFlag, "v", false, "")
|
|
}
|
|
|
|
// Execute CLI arguments.
|
|
func Execute() (err error) {
|
|
// Parse flag options, then parse commands arguments.
|
|
flag.Parse()
|
|
args := Parse(flag.Args())
|
|
|
|
job := selectRoutine(args)
|
|
return job()
|
|
}
|
|
|
|
func selectionRoutine(args Args) func() error {
|
|
switch {
|
|
|
|
// serve help
|
|
// serve --help
|
|
// serve -h
|
|
case args.Matches("help") || option.helpFlag:
|
|
return runHelpFunc
|
|
|
|
// serve version
|
|
// serve --version
|
|
// serve -v
|
|
case args.Matches("version") || option.versionFlag:
|
|
return runVersionFunc
|
|
|
|
// serve
|
|
case args.Matches():
|
|
return withConfig(runServerFunc)
|
|
|
|
// Unknown arguments.
|
|
default:
|
|
return unknownArgsFunc(args)
|
|
}
|
|
}
|
|
|
|
func unknownArgs(args Args) func() error {
|
|
return func() error {
|
|
return fmt.Errorf(
|
|
"unknown arguments provided [%v], try: 'help'",
|
|
args,
|
|
)
|
|
}
|
|
}
|
|
|
|
func withConfig(routine func() error) func() error {
|
|
return func() (err error) {
|
|
if err = loadConfig(option.configFile); nil != err {
|
|
return
|
|
}
|
|
return routine()
|
|
}
|
|
}
|
|
|