50 lines
920 B
Go
50 lines
920 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/ajnavarro/distribyted/cmd/distribyted/command"
|
||
|
|
|
||
|
|
"github.com/jessevdk/go-flags"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
name = "distribyted"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
parser := flags.NewNamedParser(name, flags.Default)
|
||
|
|
|
||
|
|
parser.UnknownOptionHandler = func(option string, arg flags.SplitArgument, args []string) ([]string, error) {
|
||
|
|
if option != "g" {
|
||
|
|
return nil, fmt.Errorf("unknown flag `%s'", option)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(args) == 0 {
|
||
|
|
return nil, fmt.Errorf("unknown flag `%s'", option)
|
||
|
|
}
|
||
|
|
|
||
|
|
return append(append(args, "-d"), args[0]), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err := parser.AddCommand("server",
|
||
|
|
command.ServerDescription,
|
||
|
|
command.ServerHelp,
|
||
|
|
&command.Server{},
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
logrus.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = parser.Parse()
|
||
|
|
if err != nil {
|
||
|
|
if e, ok := err.(*flags.Error); ok && e.Type == flags.ErrCommandRequired {
|
||
|
|
parser.WriteHelp(os.Stdout)
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|