Replace content CLI with Textual TUI

tools/content/books.py

@@ -0,0 +1,136 @@
+import datetime
+
+import goodreads
+import jsonfile
+
+BOOKS_FILE = "books.json"
+SHELVES = ["read", "reading", "want-to-read"]
+DESCRIPTION_MAX_LENGTH = 600
+GENRE_LIMIT = 5
+
+
+def check_rating(rating: int) -> int:
+    if not 0 <= rating <= 10:
+        raise ValueError("Rating must be between 0 and 10")
+    return rating
+
+
+def check_shelf(shelf: str) -> str:
+    if shelf not in SHELVES:
+        raise ValueError(f"Shelf must be one of {', '.join(SHELVES)}")
+    return shelf
+
+
+def load_books() -> list[dict]:
+    if not jsonfile.content_path(BOOKS_FILE).exists():
+        return []
+    return jsonfile.load(BOOKS_FILE)
+
+
+def save_books(books: list[dict]) -> None:
+    jsonfile.save(BOOKS_FILE, books)
+
+
+def find_index(books: list[dict], text: str) -> int | None:
+    identifier = goodreads.book_id(text) or (text if text.isdigit() else None)
+    for index, book in enumerate(books):
+        if identifier and book.get("goodreadsId") == identifier:
+            return index
+        if book.get("title", "").lower() == text.lower():
+            return index
+    return None
+
+
+def short_description(text: str | None) -> str | None:
+    if not text or len(text) <= DESCRIPTION_MAX_LENGTH:
+        return text
+    return text[:DESCRIPTION_MAX_LENGTH].rsplit(" ", 1)[0] + "..."
+
+
+def stored_fields(scraped: dict) -> dict:
+    return {
+        "goodreadsId": scraped["goodreadsId"],
+        "url": scraped["url"],
+        "title": scraped["title"],
+        "authors": scraped["authors"],
+        "cover": scraped["cover"],
+        "description": short_description(scraped["description"]),
+        "pages": scraped["pages"],
+        "published": scraped["published"],
+        "genres": scraped["genres"][:GENRE_LIMIT],
+        "averageRating": scraped["averageRating"],
+        "ratingCount": scraped["ratingCount"],
+    }
+
+
+def update_book(text: str, change) -> dict:
+    books = load_books()
+    index = find_index(books, text)
+    if index is None:
+        raise ValueError(f"{text} is not listed")
+    books[index] = change(books[index])
+    save_books(books)
+    return books[index]
+
+
+def add_book(url: str, shelf: str = "read", rating: int | None = None) -> dict:
+    check_shelf(shelf)
+    if rating is not None:
+        check_rating(rating)
+    identifier = goodreads.book_id(url)
+    if not identifier:
+        raise ValueError(f"{url} is not a Goodreads book link")
+    if find_index(load_books(), identifier) is not None:
+        raise ValueError(f"{identifier} is already listed")
+
+    scraped = goodreads.scrape_book(url)
+    if not scraped["title"]:
+        raise ValueError("Could not read the book from Goodreads")
+
+    book = stored_fields(scraped)
+    book["shelf"] = shelf
+    if rating is not None:
+        book["rating"] = rating
+    book["added"] = datetime.date.today().isoformat()
+
+    books = load_books()
+    books.append(book)
+    save_books(books)
+    return book
+
+
+def remove_book(text: str) -> dict:
+    books = load_books()
+    index = find_index(books, text)
+    if index is None:
+        raise ValueError(f"{text} is not listed")
+    removed = books.pop(index)
+    save_books(books)
+    return removed
+
+
+def rate_book(text: str, rating: int | None) -> dict:
+    if rating is not None:
+        check_rating(rating)
+
+    def change(book: dict) -> dict:
+        updated = {key: value for key, value in book.items() if key != "rating"}
+        if rating is not None:
+            updated["rating"] = rating
+        return updated
+
+    return update_book(text, change)
+
+
+def move_book(text: str, shelf: str) -> dict:
+    check_shelf(shelf)
+    return update_book(text, lambda book: {**book, "shelf": shelf})
+
+
+def refresh_book(text: str) -> dict:
+    books = load_books()
+    index = find_index(books, text)
+    if index is None:
+        raise ValueError(f"{text} is not listed")
+    scraped = goodreads.scrape_book(books[index]["url"])
+    return update_book(text, lambda book: {**book, **stored_fields(scraped)})