feat: fetch data from multiple sources and update when new one comes

1 files changed, 66 insertions, 28 deletions

diff --git a/main.go b/main.go
index 8796971..b30b665 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"slices"
 	"sort"
+	"sync"
 
 	"github.com/fatih/color"
 	"github.com/urfave/cli/v3"
@@ -45,11 +46,49 @@ func main() {
 			}
 			pkgName := cmd.Args().First()
 
-			// search and print
+			fmt.Print("\033[s")
 			showLoadingPrompt(fmt.Sprintf("What the heck is \"%s\"", pkgName))
-			stat := fetchToCreateStatistics(pkgName)
-			hideLoadingPrompt()
-			stat.Print()
+
+			// try reading from cache
+			var stat *statistics
+			if NoCache {
+				stat = &statistics{PkgName: pkgName}
+			} else {
+				stat = NewStatisticsFromCache(pkgName)
+				if stat == nil {
+					stat = &statistics{PkgName: pkgName}
+				}
+			}
+			fmlChan := make(chan *FormulaInfo)
+			caskChan := make(chan *CaskInfo)
+			rvsChan := make(chan []string)
+			fetchData(pkgName, fmlChan, caskChan, rvsChan)
+			var wg sync.WaitGroup
+			wg.Add(1)
+			go func(){
+				defer wg.Done()
+				var fml *FormulaInfo
+				var cask *CaskInfo
+				var rvs []string
+				remaining := 3
+				for remaining > 0 {
+					select {
+					case v := <-fmlChan:
+						fml = v
+						remaining--
+					case v := <-caskChan:
+						cask = v
+						remaining--
+					case v := <-rvsChan:
+						rvs = v
+						remaining--
+					}
+					fmt.Print("\033[u\033[J")
+					stat = NewStatistics(fml, cask, pkgName, rvs)
+					stat.Print()
+				}
+			}()
+			wg.Wait()
 
 			// cache
 			if !NoCache {
@@ -153,30 +192,29 @@ func main() {
 	}
 }
 
-func fetchToCreateStatistics(pkgName string) *statistics {
-	// try reading from cache
-	var stat *statistics
-	if !NoCache {
-		stat = NewStatisticsFromCache(pkgName)
-	}
-	// if cache not hit or explicitly requested no cache
-	if NoCache || stat == nil {
-		fmlChan := make(chan *FormulaInfo)
-		caskChan := make(chan *CaskInfo)
-		rvsChan := make(chan []string)
-		// start 2 goroutines, fetching formula/cask info and uses
-		go func() {
-			formula, cask := GetBrewInfo(pkgName)
-			fmlChan <- formula
-			caskChan <- cask
-		}()
-		go func() {
-			rvsChan <- GetBrewUses(pkgName)
-		}()
-		// create fine-grained data out of raw
-		stat = NewStatistics(<-fmlChan, <-caskChan, pkgName, <-rvsChan)
-	}
-	return stat
+func fetchData(pkgName string, 
+			   fmlChan chan *FormulaInfo, 
+			   caskChan chan *CaskInfo, 
+			   rvsChan chan []string) {
+	// fetching formula/cask info and uses
+	// async from brew CLI and HTTP API
+	// use whoever comes first
+	go func() {
+		formula, cask := GetBrewInfo(pkgName)
+		fmlChan <- formula
+		caskChan <- cask
+	}()
+	go func() {
+		fmlChan <- GetFormulaFromAPI(pkgName)
+	}()
+	go func() { 
+		caskChan <- GetCaskFromAPI(pkgName)
+	}()
+	
+	// fetching brew uses
+	go func() {
+		rvsChan <- GetBrewUses(pkgName)
+	}()
 }
 
 func printErrorAndExit(msg string) {