bookstack

#!/usr/bin/env python3

from pathlib import Path
import argparse
from bookshelfmanager import BookshelfManager
import sys


def get_default_data_path() -> Path:
    """Get default data file path in user's home directory."""
    return Path.home() / '.local' / 'share' / 'bookshelf' / 'books.json'


def parse_arguments() -> argparse.Namespace:
    """Parse command line arguments."""
    parser = argparse.ArgumentParser(
        description="Visualize your personal bookshelf in ASCII art!",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  %(prog)s --display                    # Show bookshelf
  %(prog)s --add                        # Add a new book
  %(prog)s --display --out shelf.txt    # Save to file
  %(prog)s --data ~/my-books.json --add # Use custom data file
        """
    )

    parser.add_argument(
        '--display', action='store_true',
        help='Display the bookshelf'
    )
    parser.add_argument(
        '--add', action='store_true',
        help='Add a new book interactively'
    )
    parser.add_argument(
        '--out', metavar='FILE',
        help='Save output to file instead of displaying'
    )
    parser.add_argument(
        '--data', metavar='FILE', type=Path,
        default=get_default_data_path(),
        help='Data file path (default: %(default)s)'
    )
    parser.add_argument(
        '--reset', action='store_true',
        help='Reset to default bookshelf'
    )

    args = parser.parse_args()

    if not any([args.display, args.add, args.reset]):
        parser.print_help()
        sys.exit(0)

    return args


def main():
    """Main application entry point."""
    args = parse_arguments()
    manager = BookshelfManager(args.data)

    manager.load_books()

    if args.reset:
        if manager.reset_to_default():
            print("Bookshelf reset to default books")
        else:
            print("Failed to reset bookshelf")

    elif args.add:
        manager.add_book_interactive()

    elif args.display:
        manager.display_bookshelf(args.out)


if __name__ == "__main__":
    main()