#!/bin/bash

get_tool_name() {
    echo "Domain TXT Record Check"
}

get_tool_description() {
    echo "Check SPF, DMARC, DKIM, and other TXT records of a domain"
}

execute_tool() {
    result=""
    if [ -n "$domain" ]; then
        if [[ "$domain" == *$'\n'* ]]; then
            for single_domain in $domain; do
                process_domain "$single_domain"
            done
        else
            process_domain "$domain"
        fi
    else
        result="No domains specified."
    fi

    echo -e "$result"
}

process_domain() {
    local single_domain="$1"
    spf_record=$(dig +short TXT "$single_domain" | grep -i "v=spf1")
    dmarc_record=$(dig +short TXT "_dmarc.$single_domain")
    
    dkim_records=$(dig +short TXT "$single_domain" | grep -i "v=dkim1")
    
    result+="\e[1mDomain:\e[0m $single_domain\n"
    
    if [ -n "$spf_record" ]; then
        result+="\e[1mSPF Record:\e[0m $spf_record\n"
    else
        result+="\e[1mSPF Record:\e[0m \e[31mFailed\e[0m\n"
    fi
    
    if [ -n "$dmarc_record" ]; then
        result+="\e[1mDMARC Record:\e[0m $dmarc_record\n"
    else
        result+="\e[1mDMARC Record:\e[0m \e[31mFailed\e[0m\n"
    fi
    
    if [ -n "$dkim_records" ]; then
        result+="\e[1mDKIM Records:\e[0m\n$dkim_records\n"
    else
        result+="\e[1mDKIM Records:\e[0m \e[31mFailed\e[0m\n"
    fi
    
    other_txt_records=$(dig +short TXT "$single_domain" | grep -v -iE "v=spf1|_dmarc|v=dkim1")
    if [ -n "$other_txt_records" ]; then
        result+="\e[1mOther TXT Records:\e[0m\n$other_txt_records\n"
    fi
    
    result+="\n"
}