auditui/main.py

#!/usr/bin/env python3

import sys
from getpass import getpass
from pathlib import Path

try:
    import audible
except ImportError:
    print("Error: audible library not found. Install it with: pip install audible")
    sys.exit(1)


def login_to_audible():
    auth_file = Path.home() / ".config" / "auditui" / "auth.json"
    auth_file.parent.mkdir(parents=True, exist_ok=True)

    if auth_file.exists():
        try:
            auth = audible.Authenticator.from_file(str(auth_file))
            print("Loaded existing authentication.")
            return auth
        except Exception as e:
            print(f"Failed to load existing auth: {e}")
            print("Please re-authenticate.")

    print("Please authenticate with your Audible account.")
    print("You will need to provide:")
    print("  - Your Audible email/username")
    print("  - Your password")
    print("  - Your marketplace locale (e.g., 'US', 'UK', 'DE', 'FR')")

    email = input("Email: ")
    password = getpass("Password: ")
    marketplace = input(
        "Marketplace locale (default: US): ").strip().upper() or "US"

    try:
        auth = audible.Authenticator.from_login(
            username=email,
            password=password,
            locale=marketplace
        )

        auth_file.parent.mkdir(parents=True, exist_ok=True)
        auth.to_file(str(auth_file))
        print("Authentication successful! Credentials saved.")
        return auth
    except Exception as e:
        print(f"Authentication failed: {e}")
        import traceback
        traceback.print_exc()
        sys.exit(1)


def format_runtime(minutes):
    if minutes is None or minutes == 0:
        return "Unknown length"

    minutes = float(minutes)
    if minutes < 60:
        return f"{int(minutes)} minutes"

    hours = int(minutes // 60)
    mins = int(minutes % 60)
    if mins == 0:
        return f"{hours} hour{'s' if hours != 1 else ''}"
    return f"{hours} hour{'s' if hours != 1 else ''} {mins} minute{'s' if mins != 1 else ''}"


def list_library(auth):
    client = audible.Client(auth=auth)

    try:
        print("\nFetching your library...")

        all_items = []
        page = 1
        page_size = 50

        while True:
            library = client.get(
                path="library",
                num_results=page_size,
                page=page,
                response_groups="contributors,media,product_attrs,product_desc,product_details,rating"
            )

            items = library.get("items", [])

            if not items:
                break

            all_items.extend(items)
            print(f"Fetched page {page} ({len(items)} items)...", end="\r")

            if len(items) < page_size:
                break

            page += 1

        print(f"\nFetched {len(all_items)} books total.\n")

        if not all_items:
            print("Your library is empty.")
            return

        print("-" * 80)

        for idx, item in enumerate(all_items, 1):
            product = item.get("product", {})

            title = product.get("title")
            if not title:
                title = item.get("title")
            if not title:
                title = product.get("asin", "Unknown Title")

            authors = product.get("authors", [])
            if not authors:
                authors = product.get("contributors", [])
            if not authors and "authors" in item:
                authors = item.get("authors", [])
            author_names = ", ".join([a.get("name", "")
                                     for a in authors if isinstance(a, dict)])

            runtime = None
            runtime_fields = [
                "runtime_length_min",
                "runtime_length",
                "vLength",
                "length",
                "duration",
                "runtime_length_ms"
            ]

            for field in runtime_fields:
                runtime = product.get(field)
                if runtime is None:
                    runtime = item.get(field)
                if runtime is not None:
                    break

            minutes = None
            if isinstance(runtime, dict):
                if "ms" in runtime:
                    ms = runtime.get("ms", 0)
                    if ms:
                        minutes = ms / 60000
                elif "min" in runtime:
                    minutes = runtime.get("min")
                else:
                    display = runtime.get("display") or runtime.get(
                        "formatted") or runtime.get("value")
                    if display and isinstance(display, (int, float)):
                        minutes = display
            elif isinstance(runtime, (int, float)):
                if runtime > 100000:
                    minutes = runtime / 60000
                else:
                    minutes = runtime

            runtime_str = format_runtime(minutes)

            asin = product.get("asin") or item.get("asin", "")

            print(f"{idx}. {title}")
            if author_names:
                print(f"   Author: {author_names}")
            print(f"   Length: {runtime_str}")
            if asin:
                print(f"   ASIN: {asin}")

            print()

        print("-" * 80)
        print(f"Total: {len(all_items)} books")

    except Exception as e:
        print(f"Error fetching library: {e}")
        import traceback
        traceback.print_exc()


def main():
    auth = login_to_audible()
    list_library(auth)


if __name__ == "__main__":
    main()