#!/usr/bin/env python
# -*- coding: utf-8 -*-
import asyncio
from pathlib import Path

import typer

from metagpt.config import CONFIG

app = typer.Typer()


@app.command()
def startup(
    idea: str = typer.Argument(..., help="Your innovative idea, such as 'Create a 2048 game.'"),
    investment: float = typer.Option(default=3.0, help="Dollar amount to invest in the AI company."),
    n_round: int = typer.Option(default=5, help="Number of rounds for the simulation."),
    code_review: bool = typer.Option(default=True, help="Whether to use code review."),
    run_tests: bool = typer.Option(default=False, help="Whether to enable QA for adding & running tests."),
    implement: bool = typer.Option(default=True, help="Enable or disable code implementation."),
    project_name: str = typer.Option(default="", help="Unique project name, such as 'game_2048'."),
    inc: bool = typer.Option(default=False, help="Incremental mode. Use it to coop with existing repo."),
    project_path: str = typer.Option(
        default="",
        help="Specify the directory path of the old version project to fulfill the " "incremental requirements.",
    ),
    reqa_file: str = typer.Option(default="", help="Specify the source file name for rewriting the quality test code."),
    max_auto_summarize_code: int = typer.Option(
        default=-1,
        help="The maximum number of times the 'SummarizeCode' action is automatically invoked, "
             "with -1 indicating unlimited. This parameter is used for debugging the workflow.",
    ),
    recover_path: str = typer.Option(default=None, help="recover the project from existing serialized storage")
):
    """Run a startup. Be a boss."""
    from metagpt.roles import (
        Architect,
        Engineer,
        ProductManager,
        ProjectManager,
        QaEngineer,
    )
    from metagpt.team import Team

    # Use in the PrepareDocuments action according to Section 2.2.3.5.1 of RFC 135.
    CONFIG.project_path = project_path
    if project_path:
        inc = True
        project_name = project_name or Path(project_path).name
    CONFIG.project_name = project_name
    CONFIG.inc = inc
    CONFIG.reqa_file = reqa_file
    CONFIG.max_auto_summarize_code = max_auto_summarize_code

    if not recover_path:
        company = Team()
        company.hire(
            [
                ProductManager(),
                Architect(),
                ProjectManager(),
            ]
        )

        if implement or code_review:
            company.hire([Engineer(n_borg=5, use_code_review=code_review)])

        if run_tests:
            company.hire([QaEngineer()])
    else:
        # # stg_path = SERDESER_PATH.joinpath("team")
        stg_path = Path(recover_path)
        if not stg_path.exists() or not str(stg_path).endswith("team"):
            raise FileNotFoundError(f"{recover_path} not exists or not endswith `team`")

        company = Team.recover(stg_path=stg_path)
        idea = company.idea  # use original idea

    company.invest(investment)
    company.run_project(idea)
    asyncio.run(company.run(n_round=n_round))


if __name__ == "__main__":
    app()