fix max num of results

laureanray · Aug 16, 2023 · 3203278 · 3203278
1 parent c2e39f3
commit 3203278
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/cmd/search.go b/cmd/search.go
@@ -29,7 +29,7 @@ func getExtension(s string) string {
 var (
   selectedFilter  string
 	outputDirectory string
-	numberOfResults = 10
+	numberOfResults int
 
 	searchCmd = &cobra.Command{
 		Use:   "search",
@@ -44,6 +44,9 @@ var (
 
       m := mirror.NewLegacyMirror(libgen.IS)
 
+      // Set Defaults
+      m.SetNumberOfResults(numberOfResults)
+
 			var books []book.Book
 
 			switch selectedFilter {
@@ -71,6 +74,7 @@ var (
 			prompt := promptui.Select{
 				Label: "Select Title",
 				Items: titles,
+        Size: 10,
 			}
 
 			resultInt, _, err := prompt.Run()

diff --git a/internal/mirror/legacy_mirror.go b/internal/mirror/legacy_mirror.go
@@ -25,11 +25,15 @@ func NewLegacyMirror(domain libgen.Domain) *LegacyMirror {
 		domain: domain,
 		// TODO: Make this configurable
 		config: Configuration{
-			numberOfResults: 5,
+			numberOfResults: 10,
 		},
 	}
 }
 
+func (m *LegacyMirror) SetNumberOfResults(numberOfResults int) {
+  m.config.numberOfResults = numberOfResults
+}
+
 func (m *LegacyMirror) SearchByTitle(query string) ([]book.Book, error) {
 	fmt.Println("Searching for: ", console.Higlight(query))
 	var document *goquery.Document

diff --git a/internal/mirror/mirror.go b/internal/mirror/mirror.go
@@ -20,5 +20,3 @@ type Configuration struct {
 type NewMirror struct {
 	domain libgen.Domain
 }
-
-