dns/prime.go

package main

import (
	"fmt"
	"log"
	"math/big"
	"regexp"
	"strings"

	"github.com/miekg/dns"
)

var numberRegex = regexp.MustCompile(`^[0-9]+$`)

var maximumPrimeNumber, _ = new(big.Int).SetString("0xFFFF_FFFF_FFFF_FFFF", 0)

func IsNumber(stringToCheck string) bool {
	return numberRegex.MatchString(stringToCheck)
}

func HandlePrimeNumberChecker(question dns.Question) dns.RR {
	log.Println("1")
	mainDomainPartindex := strings.LastIndex(question.Name, "prime.cringe-studios.com")
	log.Println("2")
	if mainDomainPartindex == 0 || mainDomainPartindex == 1 {
		// TODO handle ns page part.. maybe send back a CNAME for a page to enter the number
		log.Println("3")

		return nil
	}

	isQuestionPartIndex := strings.LastIndex(question.Name, "is.prime.cringe-studios.com")
	log.Println("4")
	if isQuestionPartIndex == 0 || isQuestionPartIndex == 1 {
		log.Println("5")
		// handle resource found
		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "tiamat.cringe-studios.com.",
		}
	}

	isnotQuestionPartIndex := strings.LastIndex(question.Name, "isnot.prime.cringe-studios.com")
	log.Println("6")
	if isnotQuestionPartIndex == 0 || isnotQuestionPartIndex == 1 {
		// TODO handle resource found
		log.Println("7")

		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "marduk.cringe-studios.com.",
		}
	}

	maybeQuestionPartIndex := strings.LastIndex(question.Name, "maybe.prime.cringe-studios.com")
	log.Println("8")
	if maybeQuestionPartIndex == 0 || maybeQuestionPartIndex == 1 {
		// TODO handle resource found
		log.Println("9")

		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "server.cringe-studios.com.",
		}
	}

	mainDomainPart := question.Name[0:mainDomainPartindex] // subtract 1 for the p in prime
	log.Println("10:" + mainDomainPart)

	if !strings.HasSuffix(mainDomainPart, ".") {
		// unknown subdomain like 123prime.cringe-studios.com
		log.Println("11")
		return nil
	}

	mainDomainPart = mainDomainPart[0 : len(mainDomainPart)-1] // subtract 1 to accomodate for the length being one bigger then the last index, and subtract another one for the dot at the end
	log.Println("12:" + mainDomainPart)

	if len(mainDomainPart) == 0 {
		// .prime.cringe-studios.com
		log.Println("13")
		return nil
	}

	firstIsIndex := strings.Index(mainDomainPart, "is.")

	log.Println("14")

	if firstIsIndex == 0 || firstIsIndex == 1 {
		// word "is." found at the beginning of the question.. trim
		mainDomainPart = mainDomainPart[len("is."):len(mainDomainPart)]
	}

	log.Println("15: " + mainDomainPart)

	if len(mainDomainPart) == 0 {
		// requested domain was is.prime.cringe-studios.com or something like that...
		// TODO
		log.Println("16")
		return nil
	}

	log.Println("16.5")

	if !IsNumber(mainDomainPart) {
		// return cname to isnot.prime.cringe-studios.com
		log.Println("17")
		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "isnot.prime.cringe-studios.com.",
		}
	}

	log.Println("18")

	bigNumber := new(big.Int)
	fmt.Sscan(mainDomainPart, bigNumber)

	log.Println("19")

	if bigNumber.Cmp(maximumPrimeNumber) == 1 {
		// number is bigger than the maximum limit for the propablyprime function
		// return CNAME for maybe.prime.cringe-studios.com

		log.Println("20")

		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "maybe.prime.cringe-studios.com.",
		}
	}

	if bigNumber.ProbablyPrime(20) {
		log.Println("21")
		// is prime, because the cases bigger than 2^64-1 have been catched above
		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "is.prime.cringe-studios.com.",
		}
	} else {
		log.Println("22")
		// is not prime
		return &dns.CNAME{
			Hdr: dns.RR_Header{
				Name:   question.Name,
				Rrtype: dns.TypeCNAME,
				Class:  dns.ClassINET,
				Ttl:    30,
			},
			Target: "isnot.prime.cringe-studios.com.",
		}
	}
}