update mirros to have default outputDir

laureanray · Jul 13, 2023 · 4a638e8 · 4a638e8
1 parent b59e597
commit 4a638e8
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/internal/mirror/current_mirror.go b/internal/mirror/current_mirror.go
@@ -123,8 +123,12 @@ func (m *CurrentMirror) searchSite(query string) (*goquery.Document, error) {
 	return document, e
 }
 
-func (m *CurrentMirror) DownloadSelection(selectedBook book.Book) {
+func (m *CurrentMirror) DownloadSelection(selectedBook book.Book, outputDirectory string) {
   fmt.Println(console.Info("Downloading book..."))
+
   directLink := documentparser.GetDirectDownloadLinkFromCurrent(selectedBook.Mirrors[0])
-  downloader.NewDownloader(selectedBook, directLink).Download()
+  if outputDirectory == "" {
+    outputDirectory = "./"
+  }
+  downloader.NewDownloader(selectedBook, directLink, outputDirectory).Download()
 }
diff --git a/internal/mirror/legacy_mirror.go b/internal/mirror/legacy_mirror.go
@@ -121,8 +121,13 @@ func (m *LegacyMirror) searchSite(query string) (*goquery.Document, error) {
 	return document, e
 }
 
-func (m *LegacyMirror) DownloadSelection(selectedBook book.Book) {
+func (m *LegacyMirror) DownloadSelection(selectedBook book.Book, outputDirectory string) {
   fmt.Println(console.Info("Downloading book..."))
   directLink := documentparser.GetDirectDownloadLinkFromLegacy(selectedBook.Mirrors[0])
-  downloader.NewDownloader(selectedBook, directLink).Download()
+
+  if outputDirectory == "" {
+    outputDirectory = "./"
+  }
+
+  downloader.NewDownloader(selectedBook, directLink, outputDirectory).Download()
 }
diff --git a/internal/mirror/mirror.go b/internal/mirror/mirror.go
@@ -10,7 +10,7 @@ type Mirror interface {
   SearchByAuthor(author string) ([]book.Book, error)
   // SearchByISBN(isbn string) []book.Book
   // 1GetDownloadLink(book book.Book) string
-  DownloadSelection(book book.Book)
+  DownloadSelection(book book.Book, outputDirectory string)
 }
 
 // TODO: Make this persistent