__main__.py

"""Command-line entry point.

    python -m copilot login        # interactive sign-in, persists the session
    python -m copilot ask "hi"     # one-shot completion via the browser driver
"""

import sys

from .browser import BrowserCopilot


def main(argv) -> int:
    cmd = argv[0] if argv else "ask"
    if cmd == "login":
        BrowserCopilot(headless=False).login()
        return 0
    if cmd == "ask":
        prompt = " ".join(argv[1:]) or "Hello!"
        with BrowserCopilot() as bot:
            for chunk in bot.create_completion(prompt, stream=True):
                print(chunk, end="", flush=True)
            print()
        return 0
    print(f"Unknown command: {cmd!r}. Use 'login' or 'ask <prompt>'.")
    return 2


if __name__ == "__main__":
    raise SystemExit(main(sys.argv[1:]))