#!/usr/bin/env bash

brt_dns() {
  local host="${1:-}"
  local rtype="${2:-A}"

  if [[ -z "$host" ]]; then
    brt_error "Usage: brt -dns <host> [A|AAAA|MX|TXT|NS|CNAME|SOA|PTR|ANY]"
    exit 1
  fi

  rtype=$(echo "$rtype" | tr '[:lower:]' '[:upper:]')

  if command -v dig &>/dev/null; then
    brt_info "DNS lookup: ${host} (${rtype})"
    dig +short "$host" "$rtype" 2>/dev/null || dig "$host" "$rtype"
  elif command -v nslookup &>/dev/null; then
    brt_info "DNS lookup: ${host}"
    nslookup -type="$rtype" "$host" 2>/dev/null || nslookup "$host"
  elif command -v host &>/dev/null; then
    host -t "$rtype" "$host"
  else
    brt_error "No DNS tool found (dig, nslookup, or host required)"
    exit 1
  fi
}