Function main

stream/main.py:37–85

Main entry point

()

async def main() -> None:
    """Main entry point"""
    args = parse_args()

    # Set up logging with GridFormatter for consistent output
    set_source("PROXY")

    console_handler = logging.StreamHandler(sys.stderr)
    console_handler.setFormatter(GridFormatter(show_tree=True, colorize=True))
    console_handler.setLevel(logging.INFO)

    # Configure proxy_server logger specifically (not root)
    logger = logging.getLogger("proxy_server")
    logger.handlers.clear()  # Remove any existing handlers
    logger.addHandler(console_handler)
    logger.setLevel(logging.INFO)
    logger.propagate = False  # Prevent double logging

    logging.getLogger("asyncio").setLevel(logging.ERROR)
    logging.getLogger("websockets").setLevel(logging.ERROR)

    # Create certs directory
    cert_dir = Path("certs")
    cert_dir.mkdir(exist_ok=True)

    # Print startup information
    logger.info(f"Starting proxy server on {args.host}:{args.port}")
    logger.info(f"Intercepting domains: {args.domains}")
    if args.proxy:
        logger.info(f"Using upstream proxy: {args.proxy}")

    # Create and start the proxy server
    proxy_server = ProxyServer(
        host=args.host,
        port=args.port,
        intercept_domains=args.domains,
        upstream_proxy=args.proxy,
        queue=None,
    )

    try:
        await proxy_server.start()
    except KeyboardInterrupt:
        logger.info("Shutting down proxy server")
    except asyncio.CancelledError:
        raise
    except Exception as e:
        logger.error(f"Error starting proxy server: {e}", exc_info=True)
        sys.exit(1)


async def builtin(