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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
package formatter
import (
"bytes"
"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
}
func FormatDNSSite(resp *models.DNSResponse, asJSON bool) (string, error) {
if asJSON {
data, err := json.MarshalIndent(resp, "", " ")
if err != nil {
return "", fmt.Errorf("failed to marshal JSON: %w", err)
}
return string(data), nil
}
var buf bytes.Buffer
if resp.Error != "" {
fmt.Fprintf(&buf, "%s\n", Red(fmt.Sprintf("Error: %s", resp.Error)))
return buf.String(), nil
}
if resp.Message != "" {
fmt.Fprintf(&buf, "%s\n", Bold(Cyan(resp.Message)))
}
if resp.Response != "" {
fmt.Fprintf(&buf, "%s\n", resp.Response)
}
if resp.Data != nil && len(resp.Data) > 0 {
dataJSON, err := json.MarshalIndent(resp.Data, "", " ")
if err == nil {
fmt.Fprintf(&buf, "\n%s\n", string(dataJSON))
}
}
return buf.String(), nil
}
|
