feat: enhance search functionality

- Combined `if not_found` and `if not_found.text == "Nothing Found"` into a single condition.
- Added a check to ensure `results` is a list before iterating.
- Appended a book emoji (📖) before each title in the search results.
- Added "🔍 Search again" and "❌ Exit" options to the `titles` and `urls` lists.
- Handled "Search again" selection by calling `search_book()` recursively.
- Handled "Exit" selection by exiting the function gracefully.
- Added the ability to navigate to the next page and previous page.
- Added "➡️ Next page" and "⬅️ Previous page" options to the `titles` and `urls` lists.
- Handled "Next page" and "Previous page" selections to navigate accordingly.
- Implemented caching of previously visited pages to avoid re-fetching them.

Author: rahaaatul

tokysnatcher/search.py

@@ -5,27 +5,71 @@
 from pick import pick
 
 
-def search_book():
-    SEARCH_URL = "https://tokybook.com/?s="
+def fetch_results(query: str, page: int = 1):
+    SEARCH_URL = f"https://tokybook.com/page/{page}/?s="
+    html = get(SEARCH_URL + quote(query))
+    soup = Soup(html)
+    return soup
+
+
+def search_book(query: str = None, page: int = 1, previous_pages: dict = None):
+    if previous_pages is None:
+        previous_pages = {}
+
+    if query is None:
+        query = input("Enter search query: ")
+
     spinner = Halo(text="Searching...", spinner="dots")
-    query = input("Search query: ")
     spinner.start()
 
-    html = get(SEARCH_URL + quote(query))
-    soup = Soup(html)
+    if page in previous_pages:
+        soup = previous_pages[page]
+    else:
+        soup = fetch_results(query, page)
+        previous_pages[page] = soup
 
     not_found = soup.find("h1", {"class": "entry-title"})
 
-    if not_found:
-        if not_found.text == "Nothing Found":
-            print("No results found!")
-            return search_book()
+    if not_found and not_found.text == "Nothing Found":
+        print("No results found!")
+        spinner.stop()
+        return search_book()
 
     results = soup.find("h2", {"class": "entry-title"})
 
-    titles = [x.text for x in results]
+    if not isinstance(results, list):
+        print("No results found. Exiting...")
+        spinner.stop()
+        return None
+
+    titles = [f"📖 {x.text}" for x in results]
     urls = [x.find("a").attrs.get("href") for x in results]
 
+    next_page = soup.find("a", {"class": "next page-numbers"})
+    if next_page:
+        titles.append("➡️ Next page")
+        urls.append("next")
+
+    if page > 1:
+        titles.append("⬅️ Previous page")
+        urls.append("previous")
+
     spinner.stop()
-    _, idx = pick(titles, "Search results:")
+    titles.append("🔍 Search again")
+    urls.append("search_again")
+    titles.append("❌ Exit")
+    urls.append("exit")
+
+    _, idx = pick(titles, "Search results:", indicator="*")
+
+    if urls[idx] == "next":
+        return search_book(query, page + 1, previous_pages)
+    elif urls[idx] == "previous":
+        return search_book(query, page - 1, previous_pages)
+    elif urls[idx] == "search_again":
+        return search_book()
+    elif urls[idx] == "exit":
+        print("Exiting...")
+        return None
+
     return urls[idx]