rp/pr/__main__.py

import argparse
import sys

from pr import __version__
from pr.core import Assistant


def main():
    parser = argparse.ArgumentParser(
        description="PR Assistant - Professional CLI AI assistant with autonomous execution",
        epilog="""
Examples:
  pr "What is Python?"                    # Single query
  pr -i                                   # Interactive mode
  pr -i --model gpt-4                     # Use specific model
  pr --save-session my-task -i            # Save session
  pr --load-session my-task               # Load session
  pr --list-sessions                      # List all sessions
  pr --usage                              # Show token usage stats

Commands in interactive mode:
  /auto [task]    - Enter autonomous mode
  /reset          - Clear message history
  /verbose        - Toggle verbose output
  /models         - List available models
  /tools          - List available tools
  /usage          - Show usage statistics
  /save <name>    - Save current session
  exit, quit, q   - Exit the program
        """,
        formatter_class=argparse.RawDescriptionHelpFormatter,
    )

    parser.add_argument("message", nargs="?", help="Message to send to assistant")
    parser.add_argument("--version", action="version", version=f"PR Assistant {__version__}")
    parser.add_argument("-m", "--model", help="AI model to use")
    parser.add_argument("-u", "--api-url", help="API endpoint URL")
    parser.add_argument("--model-list-url", help="Model list endpoint URL")
    parser.add_argument("-i", "--interactive", action="store_true", help="Interactive mode")
    parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
    parser.add_argument(
        "--debug", action="store_true", help="Enable debug mode with detailed logging"
    )
    parser.add_argument("--no-syntax", action="store_true", help="Disable syntax highlighting")
    parser.add_argument(
        "--include-env",
        action="store_true",
        help="Include environment variables in context",
    )
    parser.add_argument("-c", "--context", action="append", help="Additional context files")
    parser.add_argument(
        "--api-mode", action="store_true", help="API mode for specialized interaction"
    )

    parser.add_argument(
        "--output",
        choices=["text", "json", "structured"],
        default="text",
        help="Output format",
    )
    parser.add_argument("--quiet", action="store_true", help="Minimal output")

    parser.add_argument("--save-session", metavar="NAME", help="Save session with given name")
    parser.add_argument("--load-session", metavar="NAME", help="Load session with given name")
    parser.add_argument("--list-sessions", action="store_true", help="List all saved sessions")
    parser.add_argument("--delete-session", metavar="NAME", help="Delete a saved session")
    parser.add_argument(
        "--export-session",
        nargs=2,
        metavar=("NAME", "FILE"),
        help="Export session to file",
    )

    parser.add_argument("--usage", action="store_true", help="Show token usage statistics")
    parser.add_argument(
        "--create-config", action="store_true", help="Create default configuration file"
    )
    parser.add_argument("--plugins", action="store_true", help="List loaded plugins")

    args = parser.parse_args()

    if args.create_config:
        from pr.core.config_loader import create_default_config

        if create_default_config():
            print("Configuration file created at ~/.prrc")
        else:
            print("Error creating configuration file", file=sys.stderr)
        return

    if args.list_sessions:
        from pr.core.session import SessionManager

        sm = SessionManager()
        sessions = sm.list_sessions()
        if not sessions:
            print("No saved sessions found")
        else:
            print(f"Found {len(sessions)} saved sessions:\n")
            for sess in sessions:
                print(f"  {sess['name']}")
                print(f"    Created: {sess['created_at']}")
                print(f"    Messages: {sess['message_count']}")
                print()
        return

    if args.delete_session:
        from pr.core.session import SessionManager

        sm = SessionManager()
        if sm.delete_session(args.delete_session):
            print(f"Session '{args.delete_session}' deleted")
        else:
            print(f"Error deleting session '{args.delete_session}'", file=sys.stderr)
        return

    if args.export_session:
        from pr.core.session import SessionManager

        sm = SessionManager()
        name, output_file = args.export_session
        format_type = "json"
        if output_file.endswith(".md"):
            format_type = "markdown"
        elif output_file.endswith(".txt"):
            format_type = "txt"

        if sm.export_session(name, output_file, format_type):
            print(f"Session exported to {output_file}")
        else:
            print(f"Error exporting session", file=sys.stderr)
        return

    if args.usage:
        from pr.core.usage_tracker import UsageTracker

        usage = UsageTracker.get_total_usage()
        print(f"\nTotal Usage Statistics:")
        print(f"  Requests: {usage['total_requests']}")
        print(f"  Tokens: {usage['total_tokens']:,}")
        print(f"  Estimated Cost: ${usage['total_cost']:.4f}")
        return

    if args.plugins:
        from pr.plugins.loader import PluginLoader

        loader = PluginLoader()
        loader.load_plugins()
        plugins = loader.list_loaded_plugins()
        if not plugins:
            print("No plugins loaded")
        else:
            print(f"Loaded {len(plugins)} plugins:")
            for plugin in plugins:
                print(f"  - {plugin}")
        return

    assistant = Assistant(args)
    assistant.run()


if __name__ == "__main__":
    main()