fix: support IDNA (#99)

Author: natesales

main.go

@@ -17,6 +17,7 @@ import (
 	"github.com/jessevdk/go-flags"
 	"github.com/miekg/dns"
 	log "github.com/sirupsen/logrus"
+	"golang.org/x/net/idna"
 
 	"github.com/natesales/q/cli"
 	"github.com/natesales/q/output"
@@ -320,6 +321,19 @@ All long form (--) flags can be toggled with the dig-standard +[no]flag notation
 		rrTypes[dns.StringToType["PTR"]] = true
 	}
 
+	// IDNA (punycode) normalize non-ASCII domain names unless reverse lookup
+	if opts.Name != "" && !opts.Reverse {
+		// Skip if already an in-addr.arpa or ip6.arpa name
+		lowerName := strings.ToLower(opts.Name)
+		if !strings.HasSuffix(lowerName, ".in-addr.arpa") && !strings.HasSuffix(lowerName, ".ip6.arpa") {
+			asciiName, err := idna.Lookup.ToASCII(opts.Name)
+			if err != nil {
+				return fmt.Errorf("idna toascii: %s", err)
+			}
+			opts.Name = asciiName
+		}
+	}
+
 	// Log RR types
 	if opts.Verbose {
 		log.Debugf("Name: %s", opts.Name)

main_test.go

@@ -9,6 +9,7 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"golang.org/x/net/idna"
 
 	"github.com/natesales/q/cli"
 	"github.com/natesales/q/transport"
@@ -72,6 +73,19 @@ func TestMainJSONFormat(t *testing.T) {
 	assert.Contains(t, o, `"txt":["v=spf1 -all"`)
 }
 
+func TestMainIDNAQuery(t *testing.T) {
+	out, err := run(
+		"--all",
+		"-q", "www.饭太硬.com",
+		"-t", "A",
+	)
+	assert.Nil(t, err)
+	ascii, err := idna.Lookup.ToASCII("www.饭太硬.com")
+	assert.Nil(t, err)
+	re := regexp.MustCompile(regexp.QuoteMeta(ascii + "."))
+	assert.Regexp(t, re, out.String())
+}
+
 func TestMainInvalidOutputFormat(t *testing.T) {
 	_, err := run(
 		"--all",