summaryrefslogtreecommitdiffstats
path: root/internal/formatter/shortlinks.go
blob: d6073cb000df6d99103255de162c360000aad3b4 (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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package formatter

import (
	"fmt"
	"strings"

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

func FormatShortlinksResults(response *models.ShortlinksSearchResponse, asJSON bool) error {
	if asJSON {
		return utils.PrintJSON(response)
	}

	if response.Results == nil {
		PrintWarning("No results found")
		return nil
	}

	resultsMap, ok := response.Results.(map[string]interface{})
	if !ok {
		return utils.PrintJSON(response)
	}

	hits, ok := resultsMap["hits"].([]interface{})
	if !ok || len(hits) == 0 {
		PrintWarning("No results found")
		return nil
	}

	numHits, _ := resultsMap["num_hits"].(float64)
	elapsed, _ := resultsMap["elapsed_time_micros"].(float64)

	fmt.Printf("%s %s %s\n",
		Cyan("Found"),
		Bold(Yellow(fmt.Sprintf("%d", int(numHits)))),
		Cyan(fmt.Sprintf("credentials (%.2fms)", elapsed/1000)))
	fmt.Println()

	for i, hit := range hits {
		hitMap, ok := hit.(map[string]interface{})
		if !ok {
			continue
		}

		if i > 0 {
			fmt.Println(Dim(strings.Repeat("─", 80)))
		}

		username, _ := hitMap["username"].(string)
		password, _ := hitMap["password"].(string)
		url, _ := hitMap["url"].(string)
		filename, _ := hitMap["filename"].(string)

		fmt.Printf("%s %s\n", Bold(Green("●")), Bold(username))

		if password != "" {
			fmt.Printf("  %s %s\n", Dim("Password:"), Yellow(password))
		}

		if url != "" {
			fmt.Printf("  %s %s\n", Dim("URL:"), Blue(url))
		}

		if filename != "" {
			fmt.Printf("  %s %s\n", Dim("Source:"), Magenta(filename))
		}

		fmt.Println()
	}

	if response.Credits.Unlimited {
		fmt.Printf("%s: %s\n", Dim("Credits"), Green("Unlimited"))
	} else {
		fmt.Printf("%s: %s\n", Dim("Credits Remaining"), FormatCredits(int64(response.Credits.Remaining)))
	}

	return nil
}