reCoreD-UI/cmd/config/dns.go

package config

import (
	"reCoreD-UI/controllers"
	"reCoreD-UI/database"

	"github.com/urfave/cli/v2"
	"github.com/urfave/cli/v2/altsrc"
)

var DNSCommand = &cli.Command{
	Name:  "dns",
	Usage: "Config DNS Settings",
	Flags: []cli.Flag{
		altsrc.NewStringSliceFlag(&cli.StringSliceFlag{
			Name:     "servers",
			Usage:    "dns servers",
			Aliases:  []string{"s"},
			Required: true,
		}),
	},
	Action: setDNS,
}

func setDNS(c *cli.Context) error {
	if err := database.Connect(c.String("mysql-dsn")); err != nil {
		return err
	}
	if c.Bool("debug") {
		database.Client = database.Client.Debug()
	}
	return controllers.SetupDNS(c.StringSlice("servers")...)
}