search.py

"""
MobileAPI Interactive Search Demo

A command-line tool demonstrating MobileAPI's fuzzy search and autocomplete
capabilities. Type a query to see matching devices with confidence scores.

Usage:
    export MOBILEAPI_KEY="your_api_key_here"
    python search.py

    # Or run a single search:
    python search.py "iPhone 16"

Get your free API key: https://mobileapi.dev/signup/
API docs: https://mobileapi.dev/docs/
"""

import os
import sys

# Add parent directory to path so we can import the SDK
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "sdk"))

from mobileapi import MobileAPI, MobileAPIError, RateLimitError


def print_header():
    """Print the welcome header."""
    print()
    print("=" * 60)
    print("  MobileAPI Search Demo")
    print("  Search 27,000+ devices with fuzzy matching")
    print("=" * 60)
    print()


def print_search_results(data: dict) -> None:
    """Format and print search results with match confidence."""
    results = data.get("results", [])
    total = data.get("total", 0)
    page = data.get("page", 1)
    total_pages = data.get("total_pages", 1)

    if not results:
        print("  No devices found. Try a different search term.")
        print()
        return

    print(f"  Found {total} device(s) (page {page}/{total_pages})")
    print("-" * 60)
    print()

    for i, device in enumerate(results, 1):
        name = device.get("name", "Unknown")
        certainty = device.get("match_certainty", "N/A")
        match_type = device.get("match_type", "")
        device_id = device.get("id", "?")

        # Certainty color indicator
        if isinstance(certainty, (int, float)):
            if certainty == 100:
                indicator = "[EXACT]"
            elif certainty >= 85:
                indicator = "[HIGH] "
            elif certainty >= 70:
                indicator = "[FAIR] "
            else:
                indicator = "[LOW]  "
            certainty_str = f"{certainty}%"
        else:
            indicator = "       "
            certainty_str = str(certainty)

        print(f"  {i:2d}. {name}")
        print(f"      ID: {device_id}  |  Match: {certainty_str} {indicator}", end="")
        if match_type:
            print(f"  |  Type: {match_type}", end="")
        print()
        print()

    print("-" * 60)


def print_autocomplete_results(data: dict) -> None:
    """Format and print autocomplete suggestions."""
    results = data.get("results", [])

    if not results:
        print("  No suggestions found.")
        print()
        return

    print(f"  {len(results)} suggestion(s):")
    print()

    for i, device in enumerate(results, 1):
        name = device.get("name", "Unknown")
        device_id = device.get("id", "?")
        print(f"  {i:2d}. {name}  (ID: {device_id})")

    print()


def interactive_mode(api: MobileAPI) -> None:
    """Run the interactive search loop."""
    print("Commands:")
    print("  Type a device name to search (e.g., 'Samsung Galaxy S25')")
    print("  Prefix with 'a:' for autocomplete (e.g., 'a:sam gal')")
    print("  Type 'quit' or 'q' to exit")
    print()

    while True:
        try:
            query = input("search> ").strip()
        except (EOFError, KeyboardInterrupt):
            print("\nGoodbye!")
            break

        if not query:
            continue

        if query.lower() in ("quit", "q", "exit"):
            print("Goodbye!")
            break

        try:
            if query.startswith("a:"):
                # Autocomplete mode
                term = query[2:].strip()
                if not term:
                    print("  Usage: a:<search term>  (e.g., a:iphone)")
                    print()
                    continue
                print(f"\n  Autocomplete for: \"{term}\"")
                data = api.autocomplete(name=term, limit=10)
                print_autocomplete_results(data)

            else:
                # Full search mode
                print(f"\n  Searching for: \"{query}\"")
                print()
                data = api.search_devices(name=query, limit=10)
                print_search_results(data)

        except RateLimitError as exc:
            print(f"\n  Rate limit reached. Please wait and try again.")
            if exc.retry_after:
                print(f"  Retry after: {exc.retry_after} seconds")
            print()

        except MobileAPIError as exc:
            print(f"\n  API error: {exc}")
            print()


def single_search(api: MobileAPI, query: str) -> None:
    """Run a single search and print results."""
    print(f"  Searching for: \"{query}\"")
    print()

    try:
        data = api.search_devices(name=query, limit=10)
        print_search_results(data)
    except MobileAPIError as exc:
        print(f"  Error: {exc}")
        sys.exit(1)


def main():
    """Entry point."""
    print_header()

    # Initialize API client
    try:
        api = MobileAPI()
    except Exception as exc:
        print(f"  Error: {exc}")
        print()
        print("  Set your API key:")
        print("    export MOBILEAPI_KEY='your_api_key_here'")
        print()
        print("  Get a free key at: https://mobileapi.dev/signup/")
        sys.exit(1)

    print(f"  API key loaded. Ready to search!")
    print()

    # Check if a query was passed as a command-line argument
    if len(sys.argv) > 1:
        query = " ".join(sys.argv[1:])
        single_search(api, query)
    else:
        interactive_mode(api)


if __name__ == "__main__":
    main()