66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"reCoreD-UI/cmd/config"
|
||
|
|
"reCoreD-UI/cmd/server"
|
||
|
|
_ "reCoreD-UI/docs"
|
||
|
|
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"github.com/urfave/cli/v2/altsrc"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
logrus.SetReportCaller(true)
|
||
|
|
}
|
||
|
|
|
||
|
|
// @title reCoreD-UI API
|
||
|
|
// @version 1.0
|
||
|
|
// @description APIs for reCoreD-UI
|
||
|
|
// @BasePath /api/v1
|
||
|
|
// @securityDefinitions.basic BasicAuth
|
||
|
|
func main() {
|
||
|
|
flags := []cli.Flag{
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "config",
|
||
|
|
Usage: "config yaml file",
|
||
|
|
Aliases: []string{"c"},
|
||
|
|
EnvVars: []string{"RECORED_CONFIG_FILE"},
|
||
|
|
},
|
||
|
|
altsrc.NewStringFlag(&cli.StringFlag{
|
||
|
|
Name: "mysql-dsn",
|
||
|
|
Usage: "mysql dsn",
|
||
|
|
EnvVars: []string{"RECORED_MYSQL_DSN"},
|
||
|
|
}),
|
||
|
|
altsrc.NewBoolFlag(&cli.BoolFlag{
|
||
|
|
Name: "debug",
|
||
|
|
Usage: "enable debug mode",
|
||
|
|
Value: false,
|
||
|
|
Action: func(ctx *cli.Context, b bool) error {
|
||
|
|
if b {
|
||
|
|
logrus.SetLevel(logrus.DebugLevel)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
}),
|
||
|
|
}
|
||
|
|
|
||
|
|
app := &cli.App{
|
||
|
|
Name: "reCoreD-UI",
|
||
|
|
Usage: "Web UI for CoreDNS",
|
||
|
|
Before: altsrc.InitInputSourceWithContext(
|
||
|
|
flags, altsrc.NewYamlSourceFromFlagFunc("config"),
|
||
|
|
),
|
||
|
|
Flags: flags,
|
||
|
|
Commands: []*cli.Command{
|
||
|
|
server.Command,
|
||
|
|
config.Command,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := app.Run(os.Args); err != nil {
|
||
|
|
logrus.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|