package common

import (
	"fmt"
	"os"
)

var (
	ColorReset     = "\033[0m"
	ColorRed       = "\033[31m"
	ColorPurple    = "\033[35m"
	ColorLightBlue = "\033[34m"
	ColorCyan      = "\033[36m"
	ColorGreen     = "\033[32m"
	ColorOrange    = "\033[91m"
	ColorGray      = "\033[90m"
	ColorYellow    = "\033[93m"
)

func Banner() {
	fmt.Printf(`%s 
    _  __________=__
    \\@([____]_____()
   _/\|-[____]
  /     /(( )     ___    __   _____  ___   ___   _     _  _   _  
 /____|'----'    | |_)  / /\   | |  | | \ / / \ \ \    / | |\ | 
 \____/          |_|   /_/--\  |_|  |_|_/ \_\_/  \_\/\/  |_| \| 
%s%s
 sincerely,
     ~ delorean%s

`, ColorRed, ColorReset, ColorGray, ColorReset)
}

var Vendors = map[string]string{
	"Microsoft Defender for Endpoint": "\033[34mMicrosoft Defender for Endpoint\033[0m",
	"VMWare Carbon Black":             "\033[36mVMware\033[0m \033[90mCarbon Black\033[0m",
	"CrowdStrike Falcon":              "\033[31mCrowdStrike\033[0m \033[1mFalcon\033[0m",
	"CheckPoint Harmony":              "\033[35mCheckPoint\033[0m \033[1mHarmony\033[0m",
	"Cybereason":                      "\033[93mCybereason\033[0m",
	"Trellix":                         "\033[32mTrellix\033[0m",
	"Palo Alto Networks":              "\033[91mPalo Alto Networks\033[0m",
	"SentinelOne":                     "\033[35mSentinelOne\033[0m",
	"Symantec":                        "\033[93mSymantec\033[0m",
	"Tanium":                          "\033[31mTanium\033[0m",
	"Nextron Aurora":                  "\033[36mNextron\033[0m \033[90mAurora\033[0m",
	"Trend Micro":                     "\033[31mTrend\033[0m \033[1mMicro\033[0m",
}

func Success(msg string) {
	fmt.Printf(" %s~+~%s %s\n", ColorGreen, ColorReset, msg)
}

func Info(msg string) {
	fmt.Printf(" %s~i~%s %s\n", ColorCyan, ColorReset, msg)
}

func Warning(msg string) {
	fmt.Printf(" %s~!~%s %s\n", ColorYellow, ColorReset, msg)
}

func Error(msg string) {
	fmt.Printf(" %s~x~%s %s\n", ColorRed, ColorReset, msg)
}

func Fatal(msg string) {
	fmt.Printf(" %s~f~%s %s\n", ColorRed, ColorReset, msg)
	os.Exit(-1)
}

func Usage() {
	fmt.Printf(" %s~u~%s usage:\npatdown -t <domain>\npatdown -n ns1.target.com -n ns2.target.com", ColorOrange, ColorReset)
}