march-control-cli 0.1.3

package/core/desktop-state.js

@@ -0,0 +1,511 @@
+import { SUPPORTED_PLATFORMS } from "./constants.js";
+import { getMarchDesktopStatePath } from "./paths.js";
+import { readJson, writeJson } from "./utils.js";
+
+const SUPPORTED_APP_TYPES = ["global", ...SUPPORTED_PLATFORMS];
+
+function now() {
+  return new Date().toISOString();
+}
+
+function normalizeText(value, fallback = "") {
+  const text = `${value ?? ""}`.trim();
+  return text || fallback;
+}
+
+function uniqueStrings(values) {
+  const result = [];
+  const seen = new Set();
+
+  for (const value of values) {
+    const text = normalizeText(value);
+    if (!text) {
+      continue;
+    }
+
+    const key = text.toLowerCase();
+    if (seen.has(key)) {
+      continue;
+    }
+
+    seen.add(key);
+    result.push(text);
+  }
+
+  return result;
+}
+
+function createId(prefix, seed) {
+  const slug = normalizeText(seed)
+    .toLowerCase()
+    .replace(/[^a-z0-9]+/g, "-")
+    .replace(/^-+|-+$/g, "")
+    .slice(0, 24);
+
+  const rand = Math.random().toString(36).slice(2, 8);
+  return `${prefix}-${slug || "item"}-${rand}`;
+}
+
+function sanitizeEnabledPlatforms(raw, fallbackPlatforms) {
+  const base = Array.isArray(fallbackPlatforms) && fallbackPlatforms.length > 0 ? fallbackPlatforms : SUPPORTED_PLATFORMS;
+  const values = Array.isArray(raw) ? raw : base;
+  const seen = new Set();
+  const result = [];
+
+  for (const value of values) {
+    if (!SUPPORTED_PLATFORMS.includes(value)) {
+      continue;
+    }
+
+    if (seen.has(value)) {
+      continue;
+    }
+
+    seen.add(value);
+    result.push(value);
+  }
+
+  return result.length > 0 ? result : [...base];
+}
+
+function sanitizeTags(raw, fallback = []) {
+  const values = Array.isArray(raw) ? raw : fallback;
+  const tags = uniqueStrings(values);
+  return tags.length > 0 ? tags : [];
+}
+
+function sanitizeAppType(raw, fallback = "global") {
+  const value = normalizeText(raw, fallback);
+  return SUPPORTED_APP_TYPES.includes(value) ? value : fallback;
+}
+
+function createDefaultDesktopState() {
+  return {
+    mcpServers: [
+      {
+        id: "filesystem",
+        name: "文件系统",
+        description: "已挂载工作区根目录，可读写",
+        tags: ["files", "storage"],
+        enabledPlatforms: ["codex", "opencode", "openclaw"]
+      },
+      {
+        id: "browser",
+        name: "浏览器",
+        description: "可访问文档、版本发布和参考资料",
+        tags: ["web", "docs"],
+        enabledPlatforms: ["codex", "openclaw"]
+      },
+      {
+        id: "git",
+        name: "Git",
+        description: "可读取仓库元数据与分支关系",
+        tags: ["code", "history"],
+        enabledPlatforms: ["codex", "opencode", "openclaw"]
+      },
+      {
+        id: "shell",
+        name: "命令行",
+        description: "可在工作区边界内执行本地命令",
+        tags: ["terminal"],
+        enabledPlatforms: ["codex", "opencode"]
+      }
+    ],
+    prompts: [
+      {
+        id: "default-coding",
+        appType: "global",
+        name: "默认编码",
+        description: "用于开发实现、调试和仓库级改动。",
+        content: "优先做可落地实现、Provider 流程和直接代码修改，减少空泛规划。",
+        enabled: true,
+        updatedAt: now()
+      },
+      {
+        id: "ui-review",
+        appType: "codex",
+        name: "界面评审",
+        description: "用于视觉比对、布局评估和细节打磨。",
+        content: "按目标参考评审当前界面，识别间距、密度、交互和品牌表达差距。",
+        enabled: false,
+        updatedAt: now()
+      },
+      {
+        id: "provider-setup",
+        appType: "openclaw",
+        name: "Provider 配置",
+        description: "用于中转配置、路由切换和安装流程。",
+        content: "重点检查 Provider 连通性、路由健康度、代理设置和目标配置文件。",
+        enabled: true,
+        updatedAt: now()
+      }
+    ],
+    skills: [
+      {
+        id: "openai-docs",
+        name: "openai-docs",
+        description: "查询 OpenAI 官方文档",
+        directory: "~/.codex/skills/.system/openai-docs",
+        enabledPlatforms: ["codex"]
+      },
+      {
+        id: "plugin-creator",
+        name: "plugin-creator",
+        description: "创建并生成 Codex 插件骨架",
+        directory: "~/.codex/skills/.system/plugin-creator",
+        enabledPlatforms: ["codex"]
+      },
+      {
+        id: "skill-installer",
+        name: "skill-installer",
+        description: "从精选来源安装技能",
+        directory: "~/.codex/skills/.system/skill-installer",
+        enabledPlatforms: ["codex"]
+      },
+      {
+        id: "xianyu-keyword-research",
+        name: "xianyu-keyword-research",
+        description: "构建闲鱼关键词结构化词池",
+        directory: "~/.codex/skills/xianyu-keyword-research",
+        enabledPlatforms: ["codex", "openclaw"]
+      }
+    ],
+    skillRepos: [
+      {
+        id: "anthropics-skills",
+        owner: "anthropics",
+        name: "skills",
+        branch: "main",
+        enabled: true
+      },
+      {
+        id: "composio-awesome-claude-skills",
+        owner: "ComposioHQ",
+        name: "awesome-claude-skills",
+        branch: "master",
+        enabled: true
+      },
+      {
+        id: "cexll-myclaude",
+        owner: "cexll",
+        name: "myclaude",
+        branch: "master",
+        enabled: true
+      },
+      {
+        id: "jimliu-baoyu-skills",
+        owner: "JimLiu",
+        name: "baoyu-skills",
+        branch: "main",
+        enabled: true
+      }
+    ]
+  };
+}
+
+function normalizeState(raw, fallback) {
+  const mcpServers = Array.isArray(raw?.mcpServers)
+    ? raw.mcpServers
+        .map((server, index) => {
+          const fallbackServer = fallback.mcpServers[index] || fallback.mcpServers[0];
+          const id = normalizeText(server?.id);
+          if (!id) {
+            return null;
+          }
+
+          return {
+            id,
+            name: normalizeText(server?.name, fallbackServer.name),
+            description: normalizeText(server?.description, fallbackServer.description),
+            tags: sanitizeTags(server?.tags, fallbackServer.tags),
+            homepage: normalizeText(server?.homepage) || undefined,
+            enabledPlatforms: sanitizeEnabledPlatforms(server?.enabledPlatforms, fallbackServer.enabledPlatforms)
+          };
+        })
+        .filter(Boolean)
+    : fallback.mcpServers;
+
+  const prompts = Array.isArray(raw?.prompts)
+    ? raw.prompts
+        .map((prompt, index) => {
+          const fallbackPrompt = fallback.prompts[index] || fallback.prompts[0];
+          const id = normalizeText(prompt?.id);
+          if (!id) {
+            return null;
+          }
+
+          return {
+            id,
+            appType: sanitizeAppType(prompt?.appType, fallbackPrompt.appType),
+            name: normalizeText(prompt?.name, fallbackPrompt.name),
+            description: normalizeText(prompt?.description, fallbackPrompt.description),
+            content: normalizeText(prompt?.content, fallbackPrompt.content),
+            enabled: typeof prompt?.enabled === "boolean" ? prompt.enabled : fallbackPrompt.enabled,
+            updatedAt: normalizeText(prompt?.updatedAt, now())
+          };
+        })
+        .filter(Boolean)
+    : fallback.prompts;
+
+  const skills = Array.isArray(raw?.skills)
+    ? raw.skills
+        .map((skill, index) => {
+          const fallbackSkill = fallback.skills[index] || fallback.skills[0];
+          const id = normalizeText(skill?.id);
+          if (!id) {
+            return null;
+          }
+
+          return {
+            id,
+            name: normalizeText(skill?.name, fallbackSkill.name),
+            description: normalizeText(skill?.description, fallbackSkill.description),
+            directory: normalizeText(skill?.directory, fallbackSkill.directory),
+            enabledPlatforms: sanitizeEnabledPlatforms(skill?.enabledPlatforms, fallbackSkill.enabledPlatforms)
+          };
+        })
+        .filter(Boolean)
+    : fallback.skills;
+
+  const skillRepos = Array.isArray(raw?.skillRepos)
+    ? raw.skillRepos
+        .map((repo, index) => {
+          const fallbackRepo = fallback.skillRepos[index] || fallback.skillRepos[0];
+          const id = normalizeText(repo?.id);
+          if (!id) {
+            return null;
+          }
+
+          return {
+            id,
+            owner: normalizeText(repo?.owner, fallbackRepo.owner),
+            name: normalizeText(repo?.name, fallbackRepo.name),
+            branch: normalizeText(repo?.branch, fallbackRepo.branch),
+            enabled: typeof repo?.enabled === "boolean" ? repo.enabled : fallbackRepo.enabled
+          };
+        })
+        .filter(Boolean)
+    : fallback.skillRepos;
+
+  return {
+    mcpServers: mcpServers.length > 0 ? mcpServers : fallback.mcpServers,
+    prompts: prompts.length > 0 ? prompts : fallback.prompts,
+    skills: skills.length > 0 ? skills : fallback.skills,
+    skillRepos: skillRepos.length > 0 ? skillRepos : fallback.skillRepos
+  };
+}
+
+function loadDesktopState() {
+  const fallback = createDefaultDesktopState();
+  const stored = readJson(getMarchDesktopStatePath(), null);
+  const normalized = normalizeState(stored, fallback);
+
+  if (!stored) {
+    writeJson(getMarchDesktopStatePath(), normalized);
+  }
+
+  return normalized;
+}
+
+function saveDesktopState(state) {
+  writeJson(getMarchDesktopStatePath(), state);
+}
+
+export function getDesktopState() {
+  return loadDesktopState();
+}
+
+export function toggleMcpServerForPlatform(serverId, platform) {
+  if (!SUPPORTED_PLATFORMS.includes(platform)) {
+    throw new Error(`Unsupported platform: ${platform}`);
+  }
+
+  const state = loadDesktopState();
+  const server = state.mcpServers.find((item) => item.id === serverId);
+  if (!server) {
+    throw new Error(`MCP server not found: ${serverId}`);
+  }
+
+  server.enabledPlatforms = server.enabledPlatforms.includes(platform)
+    ? server.enabledPlatforms.filter((item) => item !== platform)
+    : [...server.enabledPlatforms, platform];
+
+  saveDesktopState(state);
+  return state;
+}
+
+export function upsertMcpServerFromDesktop(input) {
+  const state = loadDesktopState();
+  const id = normalizeText(input?.id);
+  const name = normalizeText(input?.name);
+  const description = normalizeText(input?.description);
+  if (!name) {
+    throw new Error("MCP 名称不能为空");
+  }
+  if (!description) {
+    throw new Error("MCP 描述不能为空");
+  }
+
+  const payload = {
+    id: id || createId("mcp", name),
+    name,
+    description,
+    tags: sanitizeTags(input?.tags),
+    homepage: normalizeText(input?.homepage) || undefined,
+    enabledPlatforms: sanitizeEnabledPlatforms(input?.enabledPlatforms, SUPPORTED_PLATFORMS)
+  };
+
+  const index = state.mcpServers.findIndex((item) => item.id === payload.id);
+  if (index >= 0) {
+    state.mcpServers[index] = payload;
+  } else {
+    state.mcpServers.unshift(payload);
+  }
+
+  saveDesktopState(state);
+  return state;
+}
+
+export function deleteMcpServerFromDesktop(serverId) {
+  const id = normalizeText(serverId);
+  if (!id) {
+    throw new Error("MCP ID 不能为空");
+  }
+
+  const state = loadDesktopState();
+  const next = state.mcpServers.filter((item) => item.id !== id);
+  if (next.length === state.mcpServers.length) {
+    throw new Error(`MCP server not found: ${id}`);
+  }
+
+  state.mcpServers = next;
+  saveDesktopState(state);
+  return state;
+}
+
+export function togglePromptFromDesktop(promptId) {
+  const state = loadDesktopState();
+  const prompt = state.prompts.find((item) => item.id === promptId);
+  if (!prompt) {
+    throw new Error(`Prompt not found: ${promptId}`);
+  }
+
+  prompt.enabled = !prompt.enabled;
+  prompt.updatedAt = now();
+  saveDesktopState(state);
+  return state;
+}
+
+export function upsertPromptFromDesktop(input) {
+  const state = loadDesktopState();
+  const id = normalizeText(input?.id);
+  const name = normalizeText(input?.name);
+  const description = normalizeText(input?.description);
+  const content = normalizeText(input?.content);
+
+  if (!name) {
+    throw new Error("提示词名称不能为空");
+  }
+  if (!content) {
+    throw new Error("提示词内容不能为空");
+  }
+
+  const payload = {
+    id: id || createId("prompt", name),
+    appType: sanitizeAppType(input?.appType, "global"),
+    name,
+    description,
+    content,
+    enabled: typeof input?.enabled === "boolean" ? input.enabled : true,
+    updatedAt: now()
+  };
+
+  const index = state.prompts.findIndex((item) => item.id === payload.id);
+  if (index >= 0) {
+    state.prompts[index] = payload;
+  } else {
+    state.prompts.unshift(payload);
+  }
+
+  saveDesktopState(state);
+  return state;
+}
+
+export function deletePromptFromDesktop(promptId) {
+  const id = normalizeText(promptId);
+  if (!id) {
+    throw new Error("提示词 ID 不能为空");
+  }
+
+  const state = loadDesktopState();
+  const next = state.prompts.filter((item) => item.id !== id);
+  if (next.length === state.prompts.length) {
+    throw new Error(`Prompt not found: ${id}`);
+  }
+
+  state.prompts = next;
+  saveDesktopState(state);
+  return state;
+}
+
+export function upsertSkillFromDesktop(input) {
+  const state = loadDesktopState();
+  const id = normalizeText(input?.id);
+  const name = normalizeText(input?.name);
+  const description = normalizeText(input?.description);
+  const directory = normalizeText(input?.directory);
+
+  if (!name) {
+    throw new Error("技能名称不能为空");
+  }
+  if (!directory) {
+    throw new Error("技能目录不能为空");
+  }
+
+  const payload = {
+    id: id || createId("skill", name),
+    name,
+    description,
+    directory,
+    enabledPlatforms: sanitizeEnabledPlatforms(input?.enabledPlatforms, SUPPORTED_PLATFORMS)
+  };
+
+  const index = state.skills.findIndex((item) => item.id === payload.id);
+  if (index >= 0) {
+    state.skills[index] = payload;
+  } else {
+    state.skills.unshift(payload);
+  }
+
+  saveDesktopState(state);
+  return state;
+}
+
+export function deleteSkillFromDesktop(skillId) {
+  const id = normalizeText(skillId);
+  if (!id) {
+    throw new Error("技能 ID 不能为空");
+  }
+
+  const state = loadDesktopState();
+  const next = state.skills.filter((item) => item.id !== id);
+  if (next.length === state.skills.length) {
+    throw new Error(`Skill not found: ${id}`);
+  }
+
+  state.skills = next;
+  saveDesktopState(state);
+  return state;
+}
+
+export function toggleSkillRepoFromDesktop(repoId) {
+  const state = loadDesktopState();
+  const repo = state.skillRepos.find((item) => item.id === repoId);
+  if (!repo) {
+    throw new Error(`Skill repo not found: ${repoId}`);
+  }
+
+  repo.enabled = !repo.enabled;
+  saveDesktopState(state);
+  return state;
+}