from dataclasses import dataclass
from functools import wraps
from textwrap import dedent
from typing import Any, Callable, Optional

import click


def require_workspace():
    def require_workspace_inner(f: Callable[..., None]):
        @wraps(f)
        @click.pass_context
        def new_func(ctx: click.Context, *args: Any, **kwargs: Any):
            if ctx.obj.is_workspace_valid:
                return ctx.invoke(f, *args, **kwargs)
            else:
                click.echo(
                    "The current directory is not a valid Task Weaver project directory. "
                    "There needs to be a `taskweaver-config.json` in the root of the project directory. "
                    "Please change the working directory to a valid project directory or initialize a new one. "
                    "Refer to --help for more information.",
                )
                ctx.exit(1)

        return new_func

    return require_workspace_inner


@dataclass
class CliContext:
    workspace: Optional[str]
    workspace_param: Optional[str]
    is_workspace_valid: bool
    is_workspace_empty: bool


def get_ascii_banner() -> str:
    return dedent(
        r"""
        =========================================================
         _____         _     _       __
        |_   _|_ _ ___| | _ | |     / /__  ____ __   _____  _____
          | |/ _` / __| |/ /| | /| / / _ \/ __ `/ | / / _ \/ ___/
          | | (_| \__ \   < | |/ |/ /  __/ /_/ /| |/ /  __/ /
          |_|\__,_|___/_|\_\|__/|__/\___/\__,_/ |___/\___/_/
        =========================================================
        """,
    ).strip()