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
|
package formatter
import (
"encoding/json"
"fmt"
"git.db.org.ai/dborg/internal/models"
"strings"
)
func FormatAccountStats(stats *models.AccountStatsResponse, jsonOutput bool) (string, error) {
if jsonOutput {
data, err := json.Marshal(stats)
if err != nil {
return "", fmt.Errorf("failed to marshal account stats: %w", err)
}
return string(data), nil
}
var output strings.Builder
if stats.Account != nil {
output.WriteString("Account Information:\n")
output.WriteString(fmt.Sprintf(" Name: %s\n", stats.Account.Name))
output.WriteString(fmt.Sprintf(" API Key: %s\n", stats.Account.APIKey))
output.WriteString(fmt.Sprintf(" Credits: %d\n", stats.Account.Credits))
output.WriteString(fmt.Sprintf(" Unlimited: %t\n", stats.Account.Unlimited))
output.WriteString(fmt.Sprintf(" Premium: %t\n", stats.Account.IsPremium))
output.WriteString(fmt.Sprintf(" Admin: %t\n", stats.Account.IsAdmin))
output.WriteString("\n")
}
if stats.UsageStats != nil && len(stats.UsageStats) > 0 {
output.WriteString("Usage Statistics:\n")
for service, count := range stats.UsageStats {
output.WriteString(fmt.Sprintf(" %s: %d\n", service, count))
}
output.WriteString("\n")
}
if stats.TotalRequests > 0 {
output.WriteString(fmt.Sprintf("Total Requests: %d\n", stats.TotalRequests))
}
if stats.CreditsUsed > 0 {
output.WriteString(fmt.Sprintf("Credits Used: %d\n", stats.CreditsUsed))
}
if stats.Message != "" {
output.WriteString(fmt.Sprintf("Message: %s\n", stats.Message))
}
return output.String(), nil
}
|
