rp/pr/__main__.py

import argparse
import sys
from pr.core import Assistant
from pr import __version__

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()