awl/cli/cli.go

// SPDX-License-Identifier: BSD-3-Clause

package cli

import (
	"fmt"
	"os"
	"runtime"

	"git.froth.zone/sam/awl/internal/structs"
	"git.froth.zone/sam/awl/util"

	"github.com/miekg/dns"
	flag "github.com/stefansundin/go-zflag"
)

// Parse the arguments passed into awl
func ParseCLI(version string) (Options, error) {

	flag.Usage = func() {
		fmt.Println(`awl - drill, writ small

	Usage: awl name [@server] [record]
	 <name> domain, IP address, phone number
	 <record> defaults to A
	  
	  Arguments may be in any order, including flags.
	  Dig-like +[no]commands are also supported, see dig(1) or dig -h
		  
	Options:`)
		flag.PrintDefaults()
		os.Exit(0)
	}

	// CLI flag
	var (
		port  = flag.Int("port", 0, "`port` to make DNS query (default: 53 for UDP/TCP, 853 for TLS/QUIC", flag.OptShorthand('p'), flag.OptDisablePrintDefault(true))
		query = flag.String("query", ".", "domain name to query", flag.OptShorthand('q'))
		class = flag.String("class", "IN", "DNS class to query", flag.OptShorthand('c'))
		qType = flag.String("qType", "A", "type to query", flag.OptShorthand('t'))

		ipv4    = flag.Bool("4", false, "force IPv4", flag.OptShorthandStr("4"))
		ipv6    = flag.Bool("6", false, "force IPv6", flag.OptShorthand('6'))
		reverse = flag.Bool("reverse", false, "do a reverse lookup", flag.OptShorthand('x'))

		dnssec   = flag.Bool("dnssec", false, "enable DNSSEC", flag.OptShorthand('D'))
		truncate = flag.Bool("no-truncate", false, "ignore truncation if a UDP request truncates (default= retry with TCP)")

		tcp   = flag.Bool("tcp", false, "use TCP")
		tls   = flag.Bool("tls", false, "use DNS-over-TLS", flag.OptShorthand('T'))
		https = flag.Bool("https", false, "use DNS-over-HTTPS", flag.OptShorthand('H'))
		quic  = flag.Bool("quic", false, "use DNS-over-QUIC", flag.OptShorthand('Q'))

		aa = flag.Bool("aa", false, "set/unset AA (Authoratative Answer) flag (default: not set)")
		ad = flag.Bool("ad", false, "set/unset AD (Authenticated Data) flag (default: not set)")
		cd = flag.Bool("cd", false, "set/unset CD (Checking Disabled) flag (default: not set)")
		qr = flag.Bool("qr", false, "set/unset QR (QueRy) flag (default: set)", flag.OptDisablePrintDefault(true))
		rd = flag.Bool("rd", true, "set/unset RD (Recursion Desired) flag (default: set)", flag.OptDisablePrintDefault(true))
		ra = flag.Bool("ra", false, "set/unset RA (Recursion Available) flag (default: set)", flag.OptDisablePrintDefault(true))
		tc = flag.Bool("tc", false, "set/unset TC (TrunCated) flag (default: not set)")
		z  = flag.Bool("z", false, "set/unset Z (Zero) flag (default: not set)", flag.OptShorthand('z'))

		short = flag.Bool("short", false, "print just the results, equivalent to dig +short", flag.OptShorthand('s'))
		json  = flag.Bool("json", false, "print the result(s) as JSON", flag.OptShorthand('j'))
		xml   = flag.Bool("xml", false, "print the result(s) as XML", flag.OptShorthand('X'))
		yaml  = flag.Bool("yaml", false, "print the result(s) as yaml", flag.OptShorthand('y'))

		verbosity   = flag.Int("verbosity", 0, "sets verbosity", flag.OptShorthand('v'), flag.OptNoOptDefVal("3"))
		versionFlag = flag.Bool("version", false, "print version information", flag.OptShorthand('V'))
	)

	// Don't sort the flags when -h is given
	flag.CommandLine.SortFlags = false

	// Parse the flags
	flag.Parse()

	opts := Options{
		Logger:   util.InitLogger(*verbosity),
		Class:    dns.StringToClass[*class],
		Port:     *port,
		IPv4:     *ipv4,
		IPv6:     *ipv6,
		DNSSEC:   *dnssec,
		Short:    *short,
		TCP:      *tcp,
		TLS:      *tls,
		HTTPS:    *https,
		QUIC:     *quic,
		Truncate: *truncate,
		AA:       *aa,
		AD:       *ad,
		TC:       *tc,
		Z:        *z,
		CD:       *cd,
		QR:       *qr,
		RD:       *rd,
		RA:       *ra,
		Reverse:  *reverse,
		JSON:     *json,
		XML:      *xml,
		YAML:     *yaml,
		Request: structs.Request{
			Type: dns.StringToType[*qType],
			Name: *query,
		},
	}

	if *versionFlag {
		fmt.Printf("awl version %s, built with %s\n", version, runtime.Version())
		os.Exit(0)
	}

	// Parse all the arguments that don't start with - or --
	// This includes the dig-style (+) options
	err := ParseMiscArgs(flag.Args(), &opts)
	if err != nil {
		opts.Logger.Fatal(err)
	}

	if opts.Port == 0 {
		if opts.TLS || opts.QUIC {
			opts.Port = 853
		} else {
			opts.Port = 53
		}
	}

	// opts.Request = req

	// Set verbosity to full if just -v is specified
	// flag.Lookup("verbosity").NoOptDefVal = "3"

	return opts, nil
}