93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
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)
|
|
}
|
|
}
|