feat: add support for common query types

Using `--any` it will query multiple common record types.

Author: mr-karan

cmd/doggo/cli.go

@@ -120,6 +120,8 @@ func setupFlags() *flag.FlagSet {
 	f.String("tls-hostname", "", "Hostname for certificate verification")
 	f.Bool("skip-hostname-verification", false, "Skip TLS Hostname Verification")
 
+	f.Bool("any", false, "Query all supported DNS record types")
+
 	f.BoolP("json", "J", false, "Set the output format as JSON")
 	f.Bool("short", false, "Short output format")
 	f.Bool("time", false, "Display how long the response took")

internal/app/questions.go

@@ -5,13 +5,16 @@ import (
 	"strings"
 
 	"github.com/miekg/dns"
+	"github.com/mr-karan/doggo/pkg/models"
 )
 
 // LoadFallbacks sets fallbacks for options
 // that are not specified by the user but necessary
 // for the resolver.
 func (app *App) LoadFallbacks() {
-	if len(app.QueryFlags.QTypes) == 0 {
+	if app.QueryFlags.QueryAny {
+		app.QueryFlags.QTypes = models.GetCommonRecordTypes()
+	} else if len(app.QueryFlags.QTypes) == 0 {
 		app.QueryFlags.QTypes = append(app.QueryFlags.QTypes, "A")
 	}
 	if len(app.QueryFlags.QClasses) == 0 {

pkg/models/models.go

@@ -1,6 +1,9 @@
 package models
 
-import "time"
+import (
+	"strings"
+	"time"
+)
 
 const (
 	// DefaultTLSPort specifies the default port for a DNS server connecting over TCP over TLS.
@@ -39,6 +42,7 @@ type QueryFlags struct {
 	Strategy           string        `koanf:"strategy" strategy:"-"`
 	InsecureSkipVerify bool          `koanf:"skip-hostname-verification" skip-hostname-verification:"-"`
 	TLSHostname        string        `koanf:"tls-hostname" tls-hostname:"-"`
+	QueryAny           bool          `koanf:"any" json:"any"`
 }
 
 // Nameserver represents the type of Nameserver
@@ -47,3 +51,11 @@ type Nameserver struct {
 	Address string
 	Type    string
 }
+
+// CommonRecordTypes is a string containing all common DNS record types
+const CommonRecordTypes = "A AAAA CNAME MX NS PTR SOA SRV TXT CAA"
+
+// GetCommonRecordTypes returns a slice of common DNS record types
+func GetCommonRecordTypes() []string {
+	return strings.Fields(CommonRecordTypes)
+}