workflow-cli 0.1.0__py3-none-any.whl

wfcli/commands/workspace.py

@@ -0,0 +1,368 @@
+"""workspace 子命令 - 工作空间管理
+
+包含: list / create / pull / add-type / update-template
+"""
+from __future__ import annotations
+
+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,
+    check_workspace_exists,
+    download_workspace,
+)
+from wfcli.utils.display import (
+    print_success,
+    print_error,
+    print_info,
+    print_step,
+    print_file_list,
+    print_table,
+)
+
+
+@click.group()
+def workspace():
+    """工作空间管理"""
+    pass
+
+
+# ── list ──────────────────────────────────────────────────────────
+
+@workspace.command()
+def list():
+    """列出云盘上所有工作空间"""
+    try:
+        client = LarkClient()
+        workspaces = list_workspaces(client)
+        if not workspaces:
+            print_info("云盘上暂无工作空间")
+            return
+
+        rows = [[ws["name"], ws["token"][:16] + "..."] for ws in workspaces]
+        print_table(["名称", "Token"], rows, title="云盘工作空间列表")
+        print_info(f"共 {len(workspaces)} 个工作空间")
+    except LarkError as e:
+        print_error(str(e))
+
+
+# ── create ────────────────────────────────────────────────────────
+
+@workspace.command()
+@click.argument("name", required=False)
+def create(name: str | None):
+    """新建工作空间"""
+    client = LarkClient()
+
+    # 1. 获取名称
+    if not name:
+        name = click.prompt("请输入工作空间名称")
+
+    # 2. 检查重名
+    print_info("检查重名...")
+    existing = check_workspace_exists(name, client)
+    if existing:
+        print_error(f"工作空间 '{name}' 已存在，请更换名称。")
+        return
+
+    # 3. 收集需求
+    usage = click.prompt(
+        "工作空间用途",
+        type=click.Choice(["综合行政管理", "项目管理", "部门日常管理", "其他"]),
+        default="综合行政管理",
+    )
+
+    doc_types_raw = click.prompt(
+        "文档类型（多选，用逗号分隔序号）\n"
+        "  1.会议纪要  2.通知公告  3.工作报告  4.请示批复\n"
+        "  5.工作总结  6.工作计划  7.简报  8.备忘录",
+        default="1,3,5",
+    )
+    doc_type_map = {
+        "1": "会议纪要", "2": "通知公告", "3": "工作报告", "4": "请示批复",
+        "5": "工作总结", "6": "工作计划", "7": "简报", "8": "备忘录",
+    }
+    doc_types = [doc_type_map.get(s.strip(), s.strip()) for s in doc_types_raw.split(",") if s.strip()]
+
+    # 4. 创建工作空间
+    print_step(1, 5, f"创建云盘文件夹: {name}")
+    root_token = cfg.get("feishu.root_folder_token")
+    ws_token = client.drive_create_folder(name, root_token)
+
+    # 5. 设计目录结构并创建
+    print_step(2, 5, "创建目录结构")
+    dirs_to_create = ["00_模板"]
+    for i, dt in enumerate(doc_types, 1):
+        dirs_to_create.append(f"{i:02d}_{dt}")
+
+    dir_tokens = {}
+    for d in dirs_to_create:
+        token = client.drive_create_folder(d, ws_token)
+        dir_tokens[d] = token
+
+    # 6. 生成 WORKSPACE.md
+    print_step(3, 5, "生成 WORKSPACE.md")
+    ws_content = _generate_workspace_md(name, usage, doc_types, dirs_to_create)
+
+    # 7. 上传 WORKSPACE.md
+    print_step(4, 5, "上传 WORKSPACE.md 到云盘")
+    import tempfile
+    with tempfile.NamedTemporaryFile(
+        mode="w", suffix=".md", delete=False, encoding="utf-8",
+        prefix="WORKSPACE",
+    ) as tmp:
+        tmp.write(ws_content)
+        tmp_path = tmp.name
+
+    try:
+        client.drive_upload(tmp_path, folder_token=ws_token)
+    finally:
+        Path(tmp_path).unlink(missing_ok=True)
+
+    # 8. 完成
+    print_step(5, 5, "完成")
+    print_success(f"工作空间 '{name}' 创建成功！")
+    print_info(f"云盘文件夹 token: {ws_token}")
+
+
+# ── pull ──────────────────────────────────────────────────────────
+
+@workspace.command()
+@click.argument("name", required=False)
+@click.option("--output", "-o", type=click.Path(), default=None, help="本地输出目录")
+def pull(name: str | None, output: str | None):
+    """下载工作空间到本地"""
+    client = LarkClient()
+
+    # 1. 获取工作空间列表
+    workspaces = list_workspaces(client)
+    if not workspaces:
+        print_error("云盘上暂无工作空间")
+        return
+
+    # 2. 选择工作空间
+    if name:
+        ws = next((w for w in workspaces if w["name"] == name), None)
+        if not ws:
+            print_error(f"未找到工作空间: {name}")
+            return
+    else:
+        click.echo("\n可用工作空间：")
+        for i, w in enumerate(workspaces, 1):
+            click.echo(f"  [{i}] {w['name']}")
+        choice = click.prompt("请选择", type=click.IntRange(1, len(workspaces)))
+        ws = workspaces[choice - 1]
+
+    # 3. 确定本地目录
+    if output:
+        local_dir = Path(output) / ws["name"]
+    else:
+        default_dir = Path(cfg.get("workspace.default_local_dir", "~/workspaces")).expanduser()
+        local_dir = default_dir / ws["name"]
+
+    if local_dir.exists():
+        if not click.confirm(f"目录 {local_dir} 已存在，是否覆盖？", default=True):
+            return
+
+    # 4. 下载
+    print_info(f"正在下载工作空间: {ws['name']}")
+    try:
+        download_workspace(ws["token"], local_dir, client)
+        print_success(f"工作空间已下载到: {local_dir}")
+        print_info("使用以下命令开始工作:")
+        click.echo(f"  cd {local_dir}")
+        click.echo(f"  wfcli doc generate")
+    except LarkError as e:
+        print_error(f"下载失败: {e}")
+
+
+# ── add-type ──────────────────────────────────────────────────────
+
+@workspace.command("add-type")
+@click.argument("type_name")
+def add_type(type_name: str):
+    """新增文档类型（创建目录 + 模板 + 同步云盘）"""
+    from wfcli.utils.fs import require_workspace_root
+    ws_root = require_workspace_root()
+    client = LarkClient()
+
+    # 确定编号
+    existing_dirs = [d.name for d in ws_root.iterdir() if d.is_dir()]
+    max_num = 0
+    for d in existing_dirs:
+        parts = d.split("_")
+        if parts[0].isdigit():
+            max_num = max(max_num, int(parts[0]))
+    new_num = max_num + 1
+    dir_name = f"{new_num:02d}_{type_name}"
+
+    # 创建本地目录
+    (ws_root / dir_name).mkdir(exist_ok=True)
+    print_success(f"已创建目录: {dir_name}")
+
+    # 创建模板
+    template_name = f"{new_num:02d}_01_{type_name}模板.md"
+    template_path = ws_root / "00_模板" / template_name
+    template_content = _generate_template_stub(type_name)
+    template_path.write_text(template_content, encoding="utf-8")
+    print_success(f"已创建模板: {template_name}")
+
+    # 同步到云盘
+    print_info("同步到云盘...")
+    try:
+        ws_token = _find_workspace_token(ws_root.name, client)
+        # 创建云盘目录
+        client.drive_create_folder(dir_name, ws_token)
+        # 上传模板
+        templates_token = _find_folder_token(ws_token, "00_模板", client)
+        client.drive_upload(template_path, folder_token=templates_token)
+        print_success("已同步到云盘")
+    except LarkError as e:
+        print_error(f"云盘同步失败（本地文件已保存）: {e}")
+
+
+# ── update-template ───────────────────────────────────────────────
+
+@workspace.command("update-template")
+@click.argument("template_name")
+def update_template(template_name: str):
+    """更新模板文件并同步云盘"""
+    from wfcli.utils.fs import require_workspace_root
+    ws_root = require_workspace_root()
+    client = LarkClient()
+
+    template_path = ws_root / "00_模板" / template_name
+    if not template_path.exists():
+        # 尝试模糊匹配
+        matches = list((ws_root / "00_模板").glob(f"*{template_name}*"))
+        if len(matches) == 1:
+            template_path = matches[0]
+        elif len(matches) > 1:
+            print_error("匹配到多个模板，请指定完整名称：")
+            for m in matches:
+                click.echo(f"  - {m.name}")
+            return
+        else:
+            print_error(f"未找到模板: {template_name}")
+            return
+
+    print_info(f"模板路径: {template_path}")
+    click.echo(template_path.read_text(encoding="utf-8"))
+    click.echo("─" * 40)
+
+    if click.confirm("确认同步此模板到云盘？", default=True):
+        try:
+            ws_token = _find_workspace_token(ws_root.name, client)
+            templates_token = _find_folder_token(ws_token, "00_模板", client)
+
+            # 查找是否已有同名文件（覆盖上传）
+            existing = client.drive_list_files(templates_token)
+            same_name = next(
+                (f for f in existing if f.get("name") == template_path.name),
+                None
+            )
+            if same_name:
+                client.drive_upload(template_path, file_token=same_name.get("token", ""))
+            else:
+                client.drive_upload(template_path, folder_token=templates_token)
+
+            print_success("模板已同步到云盘")
+        except LarkError as e:
+            print_error(f"同步失败: {e}")
+
+
+# ── 内部工具函数 ──────────────────────────────────────────────────
+
+def _generate_workspace_md(name: str, usage: str, doc_types: list[str], dirs: list[str]) -> str:
+    """生成 WORKSPACE.md 内容"""
+    lines = [
+        f"# {name}",
+        "",
+        f"> 用途：{usage}",
+        f"> 协作流程：https://ocn860ugrb9t.feishu.cn/file/M56mb02a5ohz7nxsjG3cMgEinXb",
+        "",
+        "## 目录结构",
+        "",
+        "```",
+        f"{name}/",
+    ]
+    for d in dirs:
+        lines.append(f"├── {d}/")
+    lines.append("```")
+
+    lines.extend([
+        "",
+        "## 文档类型与模板索引",
+        "",
+        "| 文档类型 | 模板路径 | 存放目录 |",
+        "|---------|---------|---------|",
+    ])
+    for i, dt in enumerate(doc_types, 1):
+        dir_name = f"{i:02d}_{dt}"
+        template_name = f"{i:02d}_01_{dt}模板.md"
+        lines.append(f"| {dt} | 00_模板/{template_name} | {dir_name}/ |")
+
+    lines.extend([
+        "",
+        "## 文件命名规范",
+        "",
+        "- 时间标识格式：YYYY-MM-DD",
+        "- 文件名格式：{日期}_{主题描述}.md",
+        "- 示例：2026-06-24_季度经营分析会.md",
+    ])
+
+    return "\n".join(lines) + "\n"
+
+
+def _generate_template_stub(type_name: str) -> str:
+    """生成模板骨架"""
+    return f"""# {type_name}
+
+## 基本信息
+
+- 主题：[待填写]
+- 日期：[待填写]
+- 参与人：[待填写]
+
+## 正文内容
+
+[待填写]
+
+## AI 执行指令
+
+- 请根据主题和参与人信息撰写正文
+- 语言简洁、条理清晰
+- 篇幅控制在 500-1000 字
+
+## 自检清单
+
+- [ ] 标题格式正确
+- [ ] 基本信息已填写完整
+- [ ] 正文内容条理清晰
+- [ ] 篇幅符合要求
+"""
+
+
+def _find_workspace_token(ws_name: str, client: LarkClient) -> str:
+    """查找工作空间的 folder_token"""
+    workspaces = list_workspaces(client)
+    ws = next((w for w in workspaces if w["name"] == ws_name), None)
+    if not ws:
+        raise LarkError(f"云盘上未找到工作空间: {ws_name}")
+    return ws["token"]
+
+
+def _find_folder_token(parent_token: str, folder_name: str, client: LarkClient) -> str:
+    """在父目录下查找指定子文件夹的 token"""
+    files = client.drive_list_files(parent_token)
+    folder = next(
+        (f for f in files if f.get("name") == folder_name and f.get("type") == "folder"),
+        None
+    )
+    if not folder:
+        raise LarkError(f"云盘上未找到目录: {folder_name}")
+    return folder.get("token", "")

wfcli/config.py

@@ -0,0 +1,162 @@
+"""配置管理模块
+
+配置文件路径：~/.wfcli/config.toml
+格式：TOML
+
+提供 load_config / save_config / get / set 等方法。
+"""
+
+# Python 3.11+ 内置 tomllib，3.10 需要 tomli 作为 fallback
+from pathlib import Path
+from typing import Any
+
+try:
+    import tomllib
+except ImportError:
+    import tomli as tomllib  # type: ignore[no-redef]
+
+try:
+    import tomli_w
+except ImportError:
+    tomli_w = None  # type: ignore[assignment]
+
+import click
+
+# ── 默认配置 ──────────────────────────────────────────────────────
+
+DEFAULT_CONFIG: dict[str, Any] = {
+    "feishu": {
+        "root_folder_token": "KpXtfJAx2lalrMdOz3UcafbCnQb",  # 云盘工作空间根目录
+    },
+    "workspace": {
+        "default_local_dir": "~/workspaces",  # 默认本地工作空间存放路径
+    },
+    "update": {
+        "enabled": True,
+        "check_interval_days": 1,
+    },
+}
+
+# ── 配置路径 ──────────────────────────────────────────────────────
+
+CONFIG_DIR = Path.home() / ".wfcli"
+CONFIG_FILE = CONFIG_DIR / "config.toml"
+
+
+# ── 读写工具 ──────────────────────────────────────────────────────
+
+def _ensure_config_dir() -> None:
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def load_config() -> dict[str, Any]:
+    """加载配置文件，不存在则返回默认配置"""
+    if not CONFIG_FILE.exists():
+        return DEFAULT_CONFIG.copy()
+    try:
+        with open(CONFIG_FILE, "rb") as f:
+            data = tomllib.load(f)
+        # 合并默认值，确保新增字段不会缺失
+        merged = _deep_merge(DEFAULT_CONFIG, data)
+        return merged
+    except Exception as e:
+        click.echo(f"[warn] 读取配置失败，使用默认配置: {e}", err=True)
+        return DEFAULT_CONFIG.copy()
+
+
+def save_config(config: dict[str, Any]) -> None:
+    """保存配置到文件"""
+    _ensure_config_dir()
+    try:
+        if tomli_w is not None:
+            with open(CONFIG_FILE, "wb") as f:
+                tomli_w.dump(config, f)
+        else:
+            # 没有 tomli_w，手动序列化为 TOML 格式
+            with open(CONFIG_FILE, "w", encoding="utf-8") as f:
+                f.write(_dict_to_toml(config))
+    except Exception as e:
+        raise click.ClickException(f"保存配置失败: {e}")
+
+
+def get(key: str, default: Any = None) -> Any:
+    """获取配置值，支持点分隔的嵌套 key，如 'feishu.root_folder_token'"""
+    config = load_config()
+    keys = key.split(".")
+    current: Any = config
+    for k in keys:
+        if isinstance(current, dict):
+            current = current.get(k)
+        else:
+            return default
+        if current is None:
+            return default
+    return current
+
+
+def set_value(key: str, value: Any) -> None:
+    """设置配置值，支持点分隔的嵌套 key"""
+    config = load_config()
+    keys = key.split(".")
+    current = config
+    for k in keys[:-1]:
+        if k not in current or not isinstance(current[k], dict):
+            current[k] = {}
+        current = current[k]
+
+    # 尝试类型推断
+    if isinstance(value, str):
+        if value.lower() == "true":
+            value = True
+        elif value.lower() == "false":
+            value = False
+        else:
+            try:
+                value = int(value)
+            except ValueError:
+                try:
+                    value = float(value)
+                except ValueError:
+                    pass
+
+    current[keys[-1]] = value
+    save_config(config)
+
+
+def init_config_if_missing() -> None:
+    """如果配置文件不存在，写入默认配置"""
+    if not CONFIG_FILE.exists():
+        save_config(DEFAULT_CONFIG)
+
+
+# ── 工具函数 ──────────────────────────────────────────────────────
+
+def _deep_merge(base: dict, override: dict) -> dict:
+    """深度合并两个字典，override 优先"""
+    result = base.copy()
+    for k, v in override.items():
+        if k in result and isinstance(result[k], dict) and isinstance(v, dict):
+            result[k] = _deep_merge(result[k], v)
+        else:
+            result[k] = v
+    return result
+
+
+def _dict_to_toml(data: dict, prefix: str = "") -> str:
+    """简单字典序列化为 TOML（仅支持基本类型）"""
+    lines = []
+    sections = []
+    for key, value in data.items():
+        if isinstance(value, dict):
+            sections.append((key, value))
+        elif isinstance(value, bool):
+            lines.append(f"{key} = {'true' if value else 'false'}")
+        elif isinstance(value, (int, float)):
+            lines.append(f"{key} = {value}")
+        else:
+            lines.append(f'{key} = "{value}"')
+    for key, value in sections:
+        section_name = f"{prefix}.{key}" if prefix else key
+        lines.append(f"\n[{section_name}]")
+        lines.append(_dict_to_toml(value, section_name))
+    return "\n".join(lines)

wfcli/core/__init__.py

@@ -0,0 +1 @@
+"""core 业务逻辑包"""