import typer
import os
from speckle_project_data import SpeckleProjectData
from automate_function import FunctionInputs, automate_function
from typing_extensions import Annotated
from typing import Optional


def main(
    speckle_project_data: str,
    function_inputs: str,
    speckle_token: Annotated[Optional[str], typer.Argument()] = None,
):
    speckle_token = (
        speckle_token if speckle_token else os.environ.get("SPECKLE_TOKEN", None)
    )
    if not speckle_token:
        raise ValueError("The supplied speckle token is not valid")

    project_data = SpeckleProjectData.model_validate_json(speckle_project_data)
    inputs = FunctionInputs.model_validate_json(function_inputs)
    automate_function(project_data, inputs, speckle_token)


if __name__ == "__main__":
    typer.run(main)