workflow-cli 0.1.0__py3-none-any.whl

wfcli/__init__.py

@@ -0,0 +1,3 @@
+"""公司工作流 CLI 工具"""
+
+__version__ = "0.1.0"

wfcli/commands/__init__.py

@@ -0,0 +1 @@
+"""commands 子命令包"""

wfcli/commands/config_cmd.py

@@ -0,0 +1,45 @@
+"""config 子命令 - 配置管理"""
+
+import click
+
+from wfcli import config as cfg
+
+
+@click.group()
+def config():
+    """配置管理"""
+    pass
+
+
+@config.command()
+def show():
+    """查看当前配置"""
+    data = cfg.load_config()
+    _print_dict(data)
+
+
+@config.command()
+@click.argument("key")
+@click.argument("value")
+def set(key: str, value: str):
+    """修改配置项，如: wfcli config set feishu.root_folder_token xxx"""
+    cfg.set_value(key, value)
+    click.echo(f"已更新: {key} = {value}")
+
+
+@config.command()
+def reset():
+    """重置为默认配置"""
+    cfg.save_config(cfg.DEFAULT_CONFIG)
+    click.echo("已重置为默认配置")
+
+
+def _print_dict(data: dict, indent: int = 0) -> None:
+    """递归打印字典内容"""
+    prefix = "  " * indent
+    for k, v in data.items():
+        if isinstance(v, dict):
+            click.echo(f"{prefix}[{k}]")
+            _print_dict(v, indent + 1)
+        else:
+            click.echo(f"{prefix}{k} = {v}")

wfcli/commands/doc.py

@@ -0,0 +1,149 @@
+"""doc 子命令 - 文档生成"""
+from __future__ import annotations
+
+from pathlib import Path
+from datetime import datetime
+
+import click
+
+from wfcli.core.docgen import (
+    select_template,
+    collect_inputs,
+    generate_document,
+    check_and_save,
+    run_checklist,
+)
+from wfcli.core.template import load_template
+from wfcli.core.naming import build_filename
+from wfcli.utils.fs import require_workspace_root, parse_workspace_md
+from wfcli.utils.display import print_success, print_error, print_info
+
+
+@click.group()
+def doc():
+    """文档生成"""
+    pass
+
+
+@doc.command()
+@click.option("--type", "-t", "doc_type", default=None, help="文档类型（跳过交互式选择）")
+@click.option("--topic", default=None, help="文档主题")
+@click.option("--output", "-o", default=None, help="指定输出目录（相对于工作空间根目录）")
+def generate(doc_type: str | None, topic: str | None, output: str | None):
+    """按模板生成交档
+
+    交互式选择模板、填写内容，按命名规范保存到本地对应目录。
+    """
+    ws_root = require_workspace_root()
+    ws_info = parse_workspace_md(ws_root / "WORKSPACE.md")
+
+    # 1. 选择模板
+    if doc_type:
+        from wfcli.core.template import list_templates
+        templates = list_templates(ws_root)
+        template_info = next(
+            (t for t in templates if doc_type in t["type"] or doc_type in t["name"]),
+            None,
+        )
+        if not template_info:
+            print_error(f"未找到匹配的文档类型: {doc_type}")
+            print_info("可用类型:")
+            for t in templates:
+                click.echo(f"  - {t['type']}")
+            return
+    else:
+        template_info = select_template(ws_root)
+
+    print_info(f"使用模板: {template_info['name']}")
+
+    # 2. 加载模板
+    template_data = load_template(template_info["path"])
+
+    # 3. 收集用户输入
+    if not topic:
+        topic = click.prompt("文档主题")
+
+    inputs = collect_inputs(template_data)
+
+    # 4. 生成文档
+    print_info("生成文档...")
+    content = generate_document(
+        template_data=template_data,
+        inputs=inputs,
+        title=topic,
+    )
+
+    # 5. 预览
+    click.echo("\n" + "─" * 50)
+    click.echo(content[:500])
+    if len(content) > 500:
+        click.echo(f"\n... (共 {len(content)} 字符)")
+    click.echo("─" * 50)
+
+    # 6. 自检
+    if not run_checklist(template_data):
+        if not click.confirm("\n自检未全部通过，是否仍要保存？", default=False):
+            return
+
+    # 7. 确定保存位置
+    if output:
+        target_dir = output
+    else:
+        # 根据模板类型推断存放目录
+        target_dir = _infer_target_dir(ws_root, template_info["type"])
+
+    # 8. 保存
+    saved_path = check_and_save(
+        content=content,
+        workspace_root=ws_root,
+        target_dir=target_dir,
+        topic=topic,
+    )
+
+    print_success(f"文档已保存: {saved_path.relative_to(ws_root)}")
+    print_info(f"完整路径: {saved_path}")
+    print_info("提交到云盘: wfcli sync push " + str(saved_path))
+
+
+@doc.command()
+def list():
+    """列出当前工作空间的可用文档类型"""
+    ws_root = require_workspace_root()
+
+    from wfcli.core.template import list_templates
+    templates = list_templates(ws_root)
+
+    if not templates:
+        print_info("当前工作空间没有可用模板")
+        return
+
+    print_info(f"工作空间: {ws_root.name}")
+    click.echo()
+    for t in templates:
+        click.echo(f"  - {t['type']}  ({t['name']}.md)")
+
+
+def _infer_target_dir(ws_root: Path, doc_type: str) -> str:
+    """根据文档类型推断存放目录
+
+    从 WORKSPACE.md 的模板索引表中查找，找不到则用类型名作为目录名。
+    """
+    ws_info = parse_workspace_md(ws_root / "WORKSPACE.md")
+    templates_map = ws_info.get("templates", {})
+
+    # 在模板索引中查找
+    for dtype, tpath in templates_map.items():
+        if doc_type in dtype:
+            # 从模板路径推断存放目录
+            # 模板路径如 "00_模板/01_01_会议纪要模板.md"
+            # 存放目录如 "01_会议纪要"
+            for d in (ws_root).iterdir():
+                if d.is_dir() and doc_type in d.name:
+                    return d.name
+
+    # 兜底：查找包含类型名的目录
+    for d in ws_root.iterdir():
+        if d.is_dir() and doc_type in d.name:
+            return d.name
+
+    return doc_type

wfcli/commands/info.py

@@ -0,0 +1,46 @@
+"""info 子命令 - 版本和状态信息"""
+
+import sys
+from pathlib import Path
+
+import click
+
+from wfcli import __version__
+from wfcli import config as cfg
+from wfcli.utils.fs import find_workspace_root
+from wfcli.utils.display import print_info, print_success
+
+
+@click.command()
+def info():
+    """显示版本和当前工作状态"""
+    click.echo(f"wfcli v{__version__}")
+    click.echo(f"Python {sys.version.split()[0]}")
+    click.echo(f"配置文件: {cfg.CONFIG_FILE}")
+
+    if cfg.CONFIG_FILE.exists():
+        print_success("配置文件已就绪")
+    else:
+        print_info("配置文件不存在，将使用默认配置")
+
+    # 当前工作空间
+    ws_root = find_workspace_root()
+    if ws_root:
+        print_info(f"当前工作空间: {ws_root.name} ({ws_root})")
+    else:
+        print_info("当前不在任何工作空间中")
+
+    # lark-cli 检测
+    import subprocess
+    try:
+        result = subprocess.run(
+            ["lark-cli", "--version"],
+            capture_output=True, text=True, timeout=5,
+        )
+        if result.returncode == 0:
+            ver = result.stdout.strip().split("\n")[0]
+            print_info(f"lark-cli: {ver}")
+        else:
+            print_info("lark-cli: 未检测到")
+    except (FileNotFoundError, subprocess.TimeoutExpired):
+        print_info("lark-cli: 未安装或不可用")

wfcli/commands/sync.py

@@ -0,0 +1,213 @@
+"""sync 子命令 - 文件同步
+
+push: 提交本地文件到云盘
+pull: 从云盘同步更新到本地
+"""
+
+from pathlib import Path
+
+import click
+
+from wfcli.lark import LarkClient, LarkError
+from wfcli import config as cfg
+from wfcli.core.workspace import (
+    list_workspaces,
+    download_workspace,
+    match_cloud_folder,
+)
+from wfcli.utils.fs import require_workspace_root, parse_workspace_md
+from wfcli.utils.display import (
+    print_success,
+    print_error,
+    print_info,
+    print_step,
+)
+
+
+@click.group()
+def sync():
+    """文件同步（云盘 <-> 本地）"""
+    pass
+
+
+# ── push ──────────────────────────────────────────────────────────
+
+@sync.command()
+@click.argument("file", type=click.Path(exists=True))
+def push(file: str):
+    """提交本地文件到云盘
+
+    FILE 为本地文件路径（可以是相对于工作空间根目录的路径）。
+    """
+    ws_root = require_workspace_root()
+    client = LarkClient()
+
+    file_path = Path(file).resolve()
+
+    # 1. 计算相对路径
+    try:
+        rel_path = file_path.relative_to(ws_root)
+    except ValueError:
+        print_error(f"文件不在当前工作空间内: {file_path}")
+        return
+
+    print_info(f"工作空间: {ws_root.name}")
+    print_info(f"相对路径: {rel_path}")
+
+    # 2. 查找工作空间 token
+    workspaces = list_workspaces(client)
+    ws = next((w for w in workspaces if w["name"] == ws_root.name), None)
+    if not ws:
+        print_error(f"云盘上未找到工作空间: {ws_root.name}")
+        return
+
+    ws_token = ws["token"]
+
+    # 3. 匹配云盘目标文件夹
+    print_info("匹配云盘目录...")
+    try:
+        target_folder_token = match_cloud_folder(str(rel_path), ws_token, client)
+    except LarkError:
+        # 目录不存在，尝试上传到工作空间根目录
+        target_folder_token = ws_token
+
+    # 4. 检查是否已有同名文件（覆盖上传）
+    existing_files = client.drive_list_files(target_folder_token)
+    same_name = next(
+        (f for f in existing_files if f.get("name") == file_path.name),
+        None,
+    )
+
+    # 5. 上传
+    if same_name:
+        print_info(f"覆盖上传: {file_path.name}")
+        client.drive_upload(file_path, file_token=same_name.get("token", ""))
+    else:
+        print_info(f"上传新文件: {file_path.name}")
+        client.drive_upload(file_path, folder_token=target_folder_token)
+
+    print_success(f"已提交到云盘: {rel_path}")
+
+
+# ── pull ──────────────────────────────────────────────────────────
+
+@sync.command("pull")
+@click.argument("file", required=False, default=None)
+def pull_cmd(file: str | None):
+    """从云盘同步更新到本地
+
+    不指定文件则全量刷新整个工作空间。
+    指定文件则增量更新单个文件。
+    """
+    ws_root = require_workspace_root()
+    client = LarkClient()
+
+    # 查找工作空间 token
+    workspaces = list_workspaces(client)
+    ws = next((w for w in workspaces if w["name"] == ws_root.name), None)
+    if not ws:
+        print_error(f"云盘上未找到工作空间: {ws_root.name}")
+        return
+
+    if file:
+        # 增量更新单个文件
+        _pull_single_file(ws_root, ws["token"], file, client)
+    else:
+        # 全量刷新
+        _pull_full(ws_root, ws["token"], client)
+
+
+def _pull_full(ws_root: Path, ws_token: str, client: LarkClient) -> None:
+    """全量刷新本地工作空间"""
+    print_info(f"全量刷新工作空间: {ws_root.name}")
+
+    # 1. 重新下载工作空间
+    try:
+        download_workspace(ws_token, ws_root, client)
+        print_success("工作空间已刷新")
+    except LarkError as e:
+        print_error(f"刷新失败: {e}")
+
+
+def _pull_single_file(
+    ws_root: Path,
+    ws_token: str,
+    file_path: str,
+    client: LarkClient,
+) -> None:
+    """增量更新单个文件"""
+    rel_path = file_path
+    print_info(f"增量更新: {rel_path}")
+
+    # 在云盘目录中查找目标文件
+    parts = Path(rel_path).parts
+    current_token = ws_token
+
+    # 逐级进入子目录
+    for part in parts[:-1]:
+        files = client.drive_list_files(current_token)
+        folder = next(
+            (f for f in files if f.get("name") == part and f.get("type") == "folder"),
+            None,
+        )
+        if not folder:
+            print_error(f"云盘上未找到目录: {part}")
+            return
+        current_token = folder.get("token", "")
+
+    # 查找目标文件
+    target_name = parts[-1]
+    files = client.drive_list_files(current_token)
+    target_file = next(
+        (f for f in files if f.get("name") == target_name),
+        None,
+    )
+    if not target_file:
+        print_error(f"云盘上未找到文件: {target_name}")
+        return
+
+    # 下载
+    file_token = target_file.get("token", "")
+    output_dir = ws_root / Path(rel_path).parent
+    output_dir.mkdir(parents=True, exist_ok=True)
+    client.drive_download(file_token, output_dir)
+    print_success(f"已更新: {rel_path}")
+
+
+# ── status ────────────────────────────────────────────────────────
+
+@sync.command()
+def status():
+    """查看本地与云盘的同步状态"""
+    ws_root = require_workspace_root()
+    client = LarkClient()
+
+    # 列出云盘文件
+    workspaces = list_workspaces(client)
+    ws = next((w for w in workspaces if w["name"] == ws_root.name), None)
+    if not ws:
+        print_error(f"云盘上未找到工作空间: {ws_root.name}")
+        return
+
+    cloud_files = client.drive_list_files(ws["token"])
+    cloud_names = {f.get("name", "") for f in cloud_files}
+
+    # 列出本地目录
+    local_dirs = {d.name for d in ws_root.iterdir() if d.is_dir() and not d.name.startswith(".")}
+
+    # 对比
+    only_cloud = cloud_names - local_dirs - {"WORKSPACE.md"}
+    only_local = local_dirs - cloud_names
+
+    if not only_cloud and not only_local:
+        print_success("本地与云盘目录结构一致")
+    else:
+        if only_cloud:
+            print_info("仅云盘存在:")
+            for name in sorted(only_cloud):
+                click.echo(f"  + {name}")
+        if only_local:
+            print_info("仅本地存在:")
+            for name in sorted(only_local):
+                click.echo(f"  - {name}")
+        print_info("运行 'wfcli sync pull' 同步云盘更新")