summaryrefslogtreecommitdiffstats
path: root/internal/formatter/dns.go
blob: 6abd5b05a3adc681f7141d59d86b035f314b522e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package formatter

import (
	"encoding/json"
	"fmt"
	"strings"

	"git.db.org.ai/dborg/internal/models"
)

func FormatDNSResults(result *models.DomainResult, asJSON bool) (string, error) {
	if asJSON {
		data, err := json.Marshal(result)
		if err != nil {
			return "", fmt.Errorf("failed to marshal JSON: %w", err)
		}
		return string(data), nil
	}

	var sb strings.Builder

	sb.WriteString("\n")

	domainColor := result.Domain
	if result.Status == "ACTIVE" {
		domainColor = Green(result.Domain)
	} else if result.Status == "INACTIVE" {
		domainColor = Red(result.Domain)
	} else {
		domainColor = Yellow(result.Domain)
	}

	sb.WriteString(fmt.Sprintf("%s %s\n", Cyan("Domain:"), Bold(domainColor)))

	statusIndicator := StatusPending
	if result.Status == "ACTIVE" {
		statusIndicator = StatusSuccess
	} else if result.Status == "INACTIVE" {
		statusIndicator = StatusError
	} else {
		statusIndicator = StatusWarning
	}
	sb.WriteString(fmt.Sprintf("%s %s %s\n", Cyan("Status:"), statusIndicator.String(), result.Status))

	if result.Title != "" {
		sb.WriteString(fmt.Sprintf("%s %s\n", Cyan("Title:"), result.Title))
	}

	if len(result.Tech) > 0 {
		sb.WriteString(fmt.Sprintf("%s\n", Cyan("Tech Stack:")))
		for _, tech := range result.Tech {
			sb.WriteString(fmt.Sprintf("  %s %s\n", Dim("•"), Green(tech)))
		}
	}

	return sb.String(), nil
}