[add] better search error handling

api/app-routes.go

@@ -209,7 +209,12 @@ func (api *API) createAppResourcesRoute(routeName string, args ...map[string]any
 				}
 
 				// Search
-				searchResults := search.SearchBook(*sParams.Query, bType)
+				searchResults, err := search.SearchBook(*sParams.Query, bType)
+				if err != nil {
+					errorPage(c, http.StatusInternalServerError, fmt.Sprintf("Search Error: %v", err))
+					return
+				}
+
 				templateVars["Data"] = searchResults
 				templateVars["BookType"] = *sParams.BookType
 			}

search/search.go

@@ -38,29 +38,37 @@ type SearchItem struct {
 	UploadDate string
 }
 
-func SearchBook(query string, bookType BookType) (allEntries []SearchItem) {
+func SearchBook(query string, bookType BookType) ([]SearchItem, error) {
-	log.Info(query)
 	if bookType == BOOK_FICTION {
 		// Search Fiction
 		url := "https://libgen.is/fiction/?q=" + url.QueryEscape(query) + "&language=English&format=epub"
-		body := getPage(url)
+		body, err := getPage(url)
-		allEntries = parseLibGenFiction(body)
+		if err != nil {
+			return nil, err
+		}
+		return parseLibGenFiction(body)
 	} else if bookType == BOOK_NON_FICTION {
 		// Search NonFiction
 		url := "https://libgen.is/search.php?req=" + url.QueryEscape(query)
-		body := getPage(url)
+		body, err := getPage(url)
-		allEntries = parseLibGenNonFiction(body)
+		if err != nil {
+			return nil, err
+		}
+		return parseLibGenNonFiction(body)
+	} else {
+		return nil, errors.New("Invalid Book Type")
 	}
-
-	return
 }
 
-func GoodReadsMostRead(c Cadence) []SearchItem {
+func GoodReadsMostRead(c Cadence) ([]SearchItem, error) {
-	body := getPage("https://www.goodreads.com/book/most_read?category=all&country=US&duration=" + string(c))
+	body, err := getPage("https://www.goodreads.com/book/most_read?category=all&country=US&duration=" + string(c))
+	if err != nil {
+		return nil, err
+	}
 	return parseGoodReads(body)
 }
 
-func GetBookURL(id string, bookType BookType) string {
+func GetBookURL(id string, bookType BookType) (string, error) {
 	// Derive Info URL
 	var infoURL string
 	if bookType == BOOK_FICTION {
@@ -70,7 +78,10 @@ func GetBookURL(id string, bookType BookType) string {
 	}
 
 	// Parse & Derive Download URL
-	body := getPage(infoURL)
+	body, err := getPage(infoURL)
+	if err != nil {
+		return "", err
+	}
 
 	// downloadURL := parseLibGenDownloadURL(body)
 	return parseLibGenDownloadURL(body)
@@ -86,8 +97,15 @@ func SaveBook(id string, bookType BookType) (string, error) {
 	}
 
 	// Parse & Derive Download URL
-	body := getPage(infoURL)
+	body, err := getPage(infoURL)
-	bookURL := parseLibGenDownloadURL(body)
+	if err != nil {
+		return "", err
+	}
+	bookURL, err := parseLibGenDownloadURL(body)
+	if err != nil {
+		log.Error("[SaveBook] Parse Download URL Error: ", err)
+		return "", errors.New("Download Failure")
+	}
 
 	// Create File
 	tempFile, err := os.CreateTemp("", "book")
@@ -119,15 +137,29 @@ func SaveBook(id string, bookType BookType) (string, error) {
 	return tempFile.Name(), nil
 }
 
-func getPage(page string) io.ReadCloser {
+func getPage(page string) (io.ReadCloser, error) {
-	resp, _ := http.Get(page)
+	// Set 10s Timeout
-	return resp.Body
+	client := http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	// Get Page
+	resp, err := client.Get(page)
+	if err != nil {
+		return nil, err
+	}
+
+	// Return Body
+	return resp.Body, err
 }
 
-func parseLibGenFiction(body io.ReadCloser) []SearchItem {
+func parseLibGenFiction(body io.ReadCloser) ([]SearchItem, error) {
 	// Parse
 	defer body.Close()
-	doc, _ := goquery.NewDocumentFromReader(body)
+	doc, err := goquery.NewDocumentFromReader(body)
+	if err != nil {
+		return nil, err
+	}
 
 	// Normalize Results
 	var allEntries []SearchItem
@@ -171,13 +203,16 @@ func parseLibGenFiction(body io.ReadCloser) []SearchItem {
 	})
 
 	// Return Results
-	return allEntries
+	return allEntries, nil
 }
 
-func parseLibGenNonFiction(body io.ReadCloser) []SearchItem {
+func parseLibGenNonFiction(body io.ReadCloser) ([]SearchItem, error) {
 	// Parse
 	defer body.Close()
-	doc, _ := goquery.NewDocumentFromReader(body)
+	doc, err := goquery.NewDocumentFromReader(body)
+	if err != nil {
+		return nil, err
+	}
 
 	// Normalize Results
 	var allEntries []SearchItem
@@ -213,25 +248,31 @@ func parseLibGenNonFiction(body io.ReadCloser) []SearchItem {
 	})
 
 	// Return Results
-	return allEntries
+	return allEntries, nil
 }
 
-func parseLibGenDownloadURL(body io.ReadCloser) string {
+func parseLibGenDownloadURL(body io.ReadCloser) (string, error) {
 	// Parse
 	defer body.Close()
 	doc, _ := goquery.NewDocumentFromReader(body)
 
 	// Return Download URL
 	// downloadURL, _ := doc.Find("#download [href*=cloudflare]").Attr("href")
-	downloadURL, _ := doc.Find("#download h2 a").Attr("href")
+	downloadURL, exists := doc.Find("#download h2 a").Attr("href")
+	if exists == false {
+		return "", errors.New("Download URL not found")
+	}
 
-	return downloadURL
+	return downloadURL, nil
 }
 
-func parseGoodReads(body io.ReadCloser) []SearchItem {
+func parseGoodReads(body io.ReadCloser) ([]SearchItem, error) {
 	// Parse
 	defer body.Close()
-	doc, _ := goquery.NewDocumentFromReader(body)
+	doc, err := goquery.NewDocumentFromReader(body)
+	if err != nil {
+		return nil, err
+	}
 
 	// Normalize Results
 	var allEntries []SearchItem
@@ -249,5 +290,5 @@ func parseGoodReads(body io.ReadCloser) []SearchItem {
 	})
 
 	// Return Results
-	return allEntries
+	return allEntries, nil
 }