package main

import (
	"os"

	"github.com/sirupsen/logrus"
	"github.com/urfave/cli/v2"
)

var (
	// Version and date of the build. This is injected at build-time.
	buildVersion   = "unknown"
	buildDate      = "unknown"
	verboseEnabled = false
)

// initLogger initializes logger
func initLogger(verbose bool) *logrus.Logger {
	logger := logrus.New()
	logger.SetFormatter(&logrus.TextFormatter{
		DisableTimestamp:       true,
		DisableLevelTruncation: true,
	})
	// Set logger level
	if verbose {
		logger.SetLevel(logrus.DebugLevel)
		logger.Debug("verbose logging enabled")
	} else {
		logger.SetLevel(logrus.InfoLevel)
	}
	return logger
}

func main() {
	var (
		logger = initLogger(verboseEnabled)
		app    = cli.NewApp()
	)
	// Initialize hub.
	hub := NewHub(logger, buildVersion)

	// Configure CLI app.
	app.Name = "doggo"
	app.Usage = "Command-line DNS Client"
	app.Version = buildVersion

	var qFlags QueryFlags
	// Register command line flags.
	app.Flags = []cli.Flag{
		&cli.StringSliceFlag{
			Name:        "query",
			Usage:       "Domain name to query",
			Destination: qFlags.QNames,
		},
		&cli.StringSliceFlag{
			Name:        "type",
			Usage:       "Type of DNS record to be queried (A, AAAA, MX etc)",
			Destination: qFlags.QTypes,
		},
		&cli.StringSliceFlag{
			Name:        "nameserver",
			Usage:       "Address of the nameserver to send packets to",
			Destination: qFlags.Nameservers,
		},
		&cli.StringSliceFlag{
			Name:        "class",
			Usage:       "Network class of the DNS record to be queried (IN, CH, HS etc)",
			Destination: qFlags.QClasses,
		},
		&cli.BoolFlag{
			Name:        "verbose",
			Usage:       "Enable verbose logging",
			Destination: &verboseEnabled,
			DefaultText: "false",
		},
	}

	app.Before = hub.loadQueryArgs
	app.Action = func(c *cli.Context) error {
		if len(hub.QueryFlags.QNames.Value()) == 0 {
			cli.ShowAppHelpAndExit(c, 0)
		}
		hub.Lookup(c)
		return nil
	}
	// Run the app.
	hub.Logger.Debug("Starting doggo...")
	err := app.Run(os.Args)
	if err != nil {
		logger.Errorf("oops! we encountered an issue: %s", err)
	}
}