main.go

package main

import (
	"flag"
	"fmt"
	"net"
	"os"
	"time"

	clients "github.com/careyjames/dns-scout/clients"
	constants "github.com/careyjames/dns-scout/constant"
	dnsinformation "github.com/careyjames/dns-scout/dns_information"
	color "github.com/fatih/color"

	"github.com/briandowns/spinner"
	"github.com/chzyer/readline"
)

func main() {
	// Check "version" argument
	args := os.Args[1:]
	if len(args) > 0 && args[0] == "version" {
		fmt.Println("DNS-Scout version:", constants.Version)
		return
	}
	var apiTokenFlag string
	flag.StringVar(&apiTokenFlag, "api-token", "", "IPInfo API token")
	flag.Parse()
	rl, err := readline.NewEx(&readline.Config{
		Prompt:              " \033[38;5;8m\033[38;5;4m",
		HistoryFile:         ".tmp-history",
		AutoComplete:        nil,
		InterruptPrompt:     "^C",
		EOFPrompt:           "exit",
		HistorySearchFold:   true,
		FuncFilterInputRune: nil,
	})
	if err != nil {
		panic(err)
	}
	defer rl.Close()
	apiToken := clients.FetchAPIToken(apiTokenFlag)
	s := spinner.New(spinner.CharSets[9], 100*time.Millisecond) // Use the dots character set and update every 100ms
	for {
		color.New(color.FgWhite).Println(" Enter domain, IP (or 'exit' to quit): ")
		fmt.Println("\033[38;5;8m ------------------------------------\033[0m")
		input, err := rl.Readline()
		if err != nil { // io.EOF, readline.ErrInterrupt
			break
		}
		if input == "exit" {
			return
		}
		s.Start() // Start the spinner
		isIP := net.ParseIP(input) != nil
		_, _, err = net.ParseCIDR(input)
		isCIDR := err == nil
		s.Stop() // Stop the spinner
		promptRunner(isIP, isCIDR, input, apiToken)
	}
}

func promptRunner(isIP bool, isCIDR bool, input string, apiToken string) {
	if !isIP {
		dnsinformation.GetRegistrarPromt(input, isIP)
		dnsinformation.ResolvedIPPrompt(input)
		dnsinformation.GetNSPrompt(input)
		dnsinformation.GetMXPrompt(input)
		dnsinformation.GetTXTPrompt(input)
		dnsinformation.GetSPFPrompt(input)
		dnsinformation.GetDMARCPrompt(input)
		dnsinformation.GetDKIMPrompt(input)
	}
	dnsinformation.GetPTRPrompt(input, isIP)
	if isIP || isCIDR {
		dnsinformation.GetASNInfoPrompt(input, apiToken)
	}
}