From 34c7edaaec449e287530537302d3c7459d93fdec Mon Sep 17 00:00:00 2001 From: Eltrac Date: Fri, 28 Nov 2025 14:49:00 +0800 Subject: feat(cli): new command `wthis list` --- brewget.go | 101 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++- main.go | 24 +++++++++++++-- 2 files changed, 121 insertions(+), 4 deletions(-) diff --git a/brewget.go b/brewget.go index 5bc38f2..b5c8c8a 100644 --- a/brewget.go +++ b/brewget.go @@ -2,9 +2,11 @@ package main import ( "encoding/json" - "github.com/fatih/color" "os/exec" "strings" + "time" + + "github.com/fatih/color" ) type FormulaInfo struct { @@ -34,6 +36,9 @@ type CaskInfo struct { Version string `json:"version"` } +const brewDepCacheTTL = 3*24*time.Hour +const brewListCacheTTL = 24*time.Hour + // GetInfo fetches package info for both formulae and casks. // It returns one of the two structs depending on the package type. func GetBrewInfo(pkg string) (*FormulaInfo, *CaskInfo) { @@ -76,3 +81,97 @@ func GetBrewUses(pkg string) []string { return lines } + +func GetBrewList() []string { + const cachepath = "brewlist.json" + var list []string + + // Try reading from cache + list_raw, cache_hit := ReadCache(cachepath, brewListCacheTTL) + if cache_hit { + err := json.Unmarshal(list_raw, &list) + if err != nil { + color.Red("❌ Error unmarshaling cache of brew list") + return []string{} + } + } else { + out, err := exec. + Command("brew", "list", "--installed-on-request"). + Output() + if err != nil { + color.Red("❌ Error fetching brew list.") + return []string{} + } + + list = strings.Split(strings.TrimSpace(string(out)), "\n") + } + + // Write cache + defer func() { + brewlist_json, err := json.Marshal(list) + if err != nil { + color.Red("❌ Error marshaling brew list data to write cache.") + } + WriteCache(cachepath, brewlist_json) + }() + + return list +} + +func GetBrewDeps() map[string][]string { + depmap := map[string][]string{} + + // Try reading from cache + const cachepath = "depmap.json" + cache_data, cache_hit := ReadCache(cachepath, brewDepCacheTTL) + + if cache_hit { + // if cache hit, decode + err := json.Unmarshal(cache_data, &depmap) + if err != nil { + color.Red("❌ Error unmarshaling cache of dependency map.") + } + } else { + // if not, try fetch + out, err := exec. + Command("brew", "deps", "--installed", "--direct"). + Output() + if err != nil { + color.Red("❌ Error fetching brew dependencies.") + return map[string][]string{} + } + + // parse raw data + // a list of items like "package: depA depB depC..." + dep_strs := strings.FieldsFunc(string(out), func(r rune) bool { + return r == '\n' || r == '\r' + }) + + for _, dep_str := range dep_strs { + parts := strings.Split(dep_str, ":") + pkg := parts[0] + pkg_deps := parts[1] + + deps := strings.Split(pkg_deps, " ") + filtered := []string{} + // remove empty elements in slice + for _, dep := range deps { + if dep != "" { + filtered = append(filtered, dep) + } + } + depmap[pkg] = filtered + } + } + + // Writing cache + defer func() { + depmap_json, err := json.Marshal(depmap) + if err != nil { + color.Red("Error marshaling JSON when writing cache:", err) + } + WriteCache("depmap.json", depmap_json) + }() + + return depmap +} diff --git a/main.go b/main.go index 8ec0015..a2af868 100644 --- a/main.go +++ b/main.go @@ -3,10 +3,11 @@ package main import ( "context" "fmt" - "github.com/fatih/color" - "github.com/urfave/cli/v3" "log" "os" + + "github.com/fatih/color" + "github.com/urfave/cli/v3" ) var NoCache, DoClearCache bool @@ -51,7 +52,24 @@ func main() { Aliases: []string{"l", "ls"}, Usage: "Better brew list.", Action: func(ctx context.Context, cmd *cli.Command) error { - // TODO + depmap := GetBrewDeps() + list := GetBrewList() + + for _, pkg := range list { + deps := depmap[pkg] + count := len(deps) + + color.New(color.FgBlue).Printf("%s", pkg) + if count > 1 { + fmt.Printf(" (%d dependencies): \n", count) + } else { + fmt.Printf(" (%d dependency): \n", count) + } + + for _, dep := range deps { + fmt.Printf(" ► %s\n", dep) + } + } return nil }, }, -- cgit v1.2.3