101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
import json
|
|
from dataclasses import asdict, dataclass, field
|
|
from pathlib import Path
|
|
from typing import Dict, List, Set, Tuple
|
|
|
|
import click
|
|
import oyaml as yaml
|
|
import yaml as pyyaml
|
|
|
|
from devchat.utils import get_logger
|
|
from devchat.workflow.namespace import get_prioritized_namespace_path
|
|
from devchat.workflow.path import COMMAND_FILENAMES
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class WorkflowMeta:
|
|
name: str
|
|
namespace: str
|
|
active: bool
|
|
command_conf: Dict = field(
|
|
default_factory=dict
|
|
) # content of command.yml, excluding "steps" field
|
|
|
|
def __str__(self):
|
|
return f"{'*' if self.active else ' '} {self.name} ({self.namespace})"
|
|
|
|
|
|
def iter_namespace(ns_path: str, existing_names: Set[str]) -> Tuple[List[WorkflowMeta], Set[str]]:
|
|
"""
|
|
Get all workflows under the namespace path.
|
|
|
|
Args:
|
|
ns_path: the namespace path
|
|
existing_names: the existing workflow names to check if the workflow is the first priority
|
|
|
|
Returns:
|
|
List[WorkflowMeta]: the workflows
|
|
Set[str]: the updated existing workflow names
|
|
"""
|
|
root = Path(ns_path)
|
|
interest_files = set(COMMAND_FILENAMES)
|
|
result = []
|
|
unique_names = set(existing_names)
|
|
for file in root.rglob("*"):
|
|
try:
|
|
if file.is_file() and file.name in interest_files:
|
|
rel_path = file.relative_to(root)
|
|
parts = rel_path.parts
|
|
workflow_name = ".".join(parts[:-1])
|
|
is_first = workflow_name not in unique_names
|
|
|
|
# load the config content from file
|
|
with open(file, "r", encoding="utf-8") as file_handle:
|
|
yaml_content = file_handle.read()
|
|
command_conf = yaml.safe_load(yaml_content)
|
|
# pop the "steps" field
|
|
command_conf.pop("steps", None)
|
|
|
|
workflow = WorkflowMeta(
|
|
name=workflow_name,
|
|
namespace=root.name,
|
|
active=is_first,
|
|
command_conf=command_conf,
|
|
)
|
|
unique_names.add(workflow_name)
|
|
result.append(workflow)
|
|
except pyyaml.scanner.ScannerError as err:
|
|
logger.error("Failed to load %s: %s", rel_path, err)
|
|
except Exception as err:
|
|
logger.error("Unknown error when loading %s: %s", rel_path, err)
|
|
|
|
return result, unique_names
|
|
|
|
|
|
@click.command(help="List all local workflows.", name="list")
|
|
@click.option("--json", "in_json", is_flag=True, help="Output in json format.")
|
|
def list_cmd(in_json: bool):
|
|
namespace_paths = get_prioritized_namespace_path()
|
|
|
|
workflows: List[WorkflowMeta] = []
|
|
visited_names = set()
|
|
for ns_path in namespace_paths:
|
|
ws_names, visited_names = iter_namespace(ns_path, visited_names)
|
|
workflows.extend(ws_names)
|
|
|
|
if not in_json:
|
|
# print basic info
|
|
active_count = len([workflow for workflow in workflows if workflow.active])
|
|
total_count = len(workflows)
|
|
click.echo(f"workflows (active/total): {active_count}/{total_count}")
|
|
for workflow in workflows:
|
|
click.echo(workflow)
|
|
|
|
else:
|
|
# convert workflows to json
|
|
data = [asdict(workflow) for workflow in workflows]
|
|
json_format = json.dumps(data)
|
|
click.echo(json_format)
|