From 58afd5bff63319f6a2041fb44d07ea78275dcc78 Mon Sep 17 00:00:00 2001
From: Tanmoy Sarkar <57363826+tanmoysrt@users.noreply.github.com>
Date: Wed, 22 Nov 2023 03:56:48 +0530
Subject: [PATCH] chore: add AAAA and NS support

---
 main.go | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

diff --git a/main.go b/main.go
index 2f06310..8158844 100644
--- a/main.go
+++ b/main.go
@@ -10,17 +10,14 @@ import (
 
 func parseQuery(m *dns.Msg, qdns *QuickDNSResolver) {
 	for _, q := range m.Question {
+		println(q.Name, q.Qtype)
 		switch q.Qtype {
-		case dns.TypeAAAA:
-			{
-				isQDNS, ip := qdns.ResolveARecord(q.Name)
-				if isQDNS {
-					rr, err := dns.NewRR(fmt.Sprintf("%s AAAA %s", q.Name, ip))
-					if err == nil {
-						m.Answer = append(m.Answer, rr)
-					}
-				}
-			}
+		case dns.TypeNone:
+			fallthrough
+		case dns.TypeANY:
+			fallthrough
+		case dns.TypeNS:
+			fallthrough
 		case dns.TypeA:
 			{
 				isQDNS, ip := qdns.ResolveARecord(q.Name)
@@ -30,9 +27,6 @@ func parseQuery(m *dns.Msg, qdns *QuickDNSResolver) {
 						m.Answer = append(m.Answer, rr)
 					}
 				}
-			}
-		case dns.TypeNS:
-			{
 				// ns1.swiftwave.xyz and ns2.swiftwave.xyz
 				rr, err := dns.NewRR(fmt.Sprintf("%s NS ns1.swiftwave.xyz", q.Name))
 				if err == nil {