@dujavi/ai-md 0.4.0 → 0.6.0

package/lib/commands.js

@@ -5,60 +5,24 @@ const path = require("path");
 const {
   resolveConfig,
   templatePath,
-  symlinkState,
-  agentSkillTargets,
+  writeMachineConfig,
+  gitClone,
+  gitPull,
+  gitPush,
 } = require("./config");
 const { collectStatus, statusHelp, collectProjects } = require("./status");
-
-function ensureDir(p) {
-  fs.mkdirSync(p, { recursive: true });
-}
-
-function linkPath(target, link, { force = false, dryRun = false } = {}) {
-  ensureDir(path.dirname(link));
-  ensureDir(target);
-  const state = symlinkState(link, target);
-  if (state.state === "ok") {
-    return { path: link, action: "ok", target };
-  }
-  if (state.state === "not_symlink" && !force) {
-    const err = new Error(
-      `${link} exists and is not a symlink (use --force to replace)`
-    );
-    err.code = "EEXIST";
-    throw err;
-  }
-  if (dryRun) {
-    return { path: link, action: state.state === "missing" ? "would_link" : "would_repair", target };
-  }
-  if (fs.existsSync(link) || state.state !== "missing") {
-    try {
-      fs.lstatSync(link);
-      fs.rmSync(link, { recursive: true, force: true });
-    } catch {
-      /* missing */
-    }
-  }
-  fs.symlinkSync(target, link);
-  return {
-    path: link,
-    action: state.state === "missing" ? "linked" : "repaired",
-    target,
-  };
-}
-
-function ensureCursorLinks(cfg, opts = {}) {
-  return [
-    linkPath(cfg.skillsDir, cfg.cursorSkills, opts),
-    linkPath(cfg.rulesDir, cfg.cursorRules, opts),
-  ];
-}
-
-function ensureAgentSkillLinks(cfg, agents, opts = {}) {
-  return agentSkillTargets(cfg.home, agents)
-    .filter((t) => t.name !== "cursor") // cursor handled via ensureCursorLinks
-    .map((t) => linkPath(cfg.skillsDir, t.path, opts));
-}
+const { linkPath, ensureDir, wslWarnings, defaultLinkMode } = require("./link");
+const { runBuild, distRoot, distDirty } = require("./build");
+const { runRescue } = require("./rescue");
+const {
+  selectHarnesses,
+  resolveHarness,
+  listBuiltinIds,
+  getBuiltin,
+  ensureAgentSourceDirs,
+  isHarnessInstalled,
+} = require("./harnesses");
+const { seedSkeleton, initRepo, SKELETON_VERSION } = require("./skeleton");
 
 function ensureGitignore(repoPath, { dryRun = false } = {}) {
   const gitignore = path.join(repoPath, ".gitignore");
@@ -84,6 +48,37 @@ function copyDir(src, dest, { dryRun = false } = {}) {
   return { action: "copied", from: src, to: dest };
 }
 
+function countEntries(dir) {
+  if (!fs.existsSync(dir)) return 0;
+  return fs.readdirSync(dir).filter((n) => !n.startsWith(".")).length;
+}
+
+function mergeTemplate(templateDir, projectDir, { dryRun = false } = {}) {
+  const actions = [];
+  for (const kind of ["rules", "skills"]) {
+    const srcRoot = path.join(templateDir, kind);
+    const destRoot = path.join(projectDir, kind);
+    if (!fs.existsSync(srcRoot)) continue;
+    ensureDir(destRoot);
+    for (const name of fs.readdirSync(srcRoot)) {
+      if (name.startsWith(".")) continue;
+      const src = path.join(srcRoot, name);
+      const dest = path.join(destRoot, name);
+      if (fs.existsSync(dest)) {
+        actions.push({ action: "keep", path: dest });
+        continue;
+      }
+      if (dryRun) {
+        actions.push({ action: "would_add", path: dest, from: src });
+      } else {
+        fs.cpSync(src, dest, { recursive: true });
+        actions.push({ action: "added", path: dest, from: src });
+      }
+    }
+  }
+  return actions;
+}
+
 function initProject({
   repo,
   name,
@@ -92,8 +87,8 @@ function initProject({
   dryRun = false,
 } = {}) {
   const cfg = resolveConfig();
-  if (!fs.existsSync(path.join(cfg.dir, ".git"))) {
-    const err = new Error(`${cfg.dir} is not a git repo; run ai-md install first`);
+  if (!fs.existsSync(path.join(cfg.dir, ".git")) && !fs.existsSync(cfg.dir)) {
+    const err = new Error(`${cfg.dir} missing; run ai-md init or ai-md install first`);
     err.code = "ENOENT";
     throw err;
   }
@@ -131,7 +126,13 @@ function initProject({
   }
 
   const link = path.join(absRepo, ".cursor");
-  actions.push(linkPath(projectDir, link, { force, dryRun }));
+  actions.push(
+    linkPath(projectDir, link, {
+      force,
+      dryRun,
+      linkMode: cfg.linkMode,
+    })
+  );
   actions.push(ensureGitignore(absRepo, { dryRun }));
 
   return {
@@ -144,37 +145,6 @@ function initProject({
   };
 }
 
-function countEntries(dir) {
-  if (!fs.existsSync(dir)) return 0;
-  return fs.readdirSync(dir).filter((n) => !n.startsWith(".")).length;
-}
-
-function mergeTemplate(templateDir, projectDir, { dryRun = false } = {}) {
-  const actions = [];
-  for (const kind of ["rules", "skills"]) {
-    const srcRoot = path.join(templateDir, kind);
-    const destRoot = path.join(projectDir, kind);
-    if (!fs.existsSync(srcRoot)) continue;
-    ensureDir(destRoot);
-    for (const name of fs.readdirSync(srcRoot)) {
-      if (name.startsWith(".")) continue;
-      const src = path.join(srcRoot, name);
-      const dest = path.join(destRoot, name);
-      if (fs.existsSync(dest)) {
-        actions.push({ action: "keep", path: dest });
-        continue;
-      }
-      if (dryRun) {
-        actions.push({ action: "would_add", path: dest, from: src });
-      } else {
-        fs.cpSync(src, dest, { recursive: true });
-        actions.push({ action: "added", path: dest, from: src });
-      }
-    }
-  }
-  return actions;
-}
-
 function applyTemplate({ project, from = "base", dryRun = false } = {}) {
   const cfg = resolveConfig();
   if (!project) {
@@ -200,11 +170,6 @@ function applyTemplate({ project, from = "base", dryRun = false } = {}) {
 
 function linkProject({ repo, name, force = false, dryRun = false } = {}) {
   const cfg = resolveConfig();
-  if (!fs.existsSync(path.join(cfg.dir, ".git"))) {
-    const err = new Error(`${cfg.dir} is not a git repo; run ai-md install first`);
-    err.code = "ENOENT";
-    throw err;
-  }
   if (!repo || !fs.existsSync(repo)) {
     const err = new Error("missing or invalid --repo <path>");
     err.code = "EINVAL";
@@ -218,37 +183,248 @@ function linkProject({ repo, name, force = false, dryRun = false } = {}) {
     ensureDir(path.join(projectDir, "skills"));
   }
   const actions = [
-    linkPath(projectDir, path.join(absRepo, ".cursor"), { force, dryRun }),
+    linkPath(projectDir, path.join(absRepo, ".cursor"), {
+      force,
+      dryRun,
+      linkMode: cfg.linkMode,
+    }),
     ensureGitignore(absRepo, { dryRun }),
   ];
   return { project: projectName, projectDir, repo: absRepo, actions };
 }
 
-function runDoctor({ fix = false, force = false, agents = ["cursor"], dryRun = false } = {}) {
+/** Link dist/<id> → live harness paths. Only for installed harnesses. */
+function linkHarnesses(cfg, agents, opts = {}) {
+  const {
+    force = false,
+    dryRun = false,
+    forceLink = false,
+  } = opts;
+  const { selected, skipped, warnings } = selectHarnesses(cfg, agents, {
+    forceLink,
+  });
+  const links = [];
+  const platformWarn = wslWarnings(cfg, selected);
+
+  const linkedDist = new Set();
+  for (const h of selected) {
+    if (linkedDist.has(h.distId)) continue;
+    linkedDist.add(h.distId);
+    const dist = distRoot(cfg, h.distId);
+    if (!fs.existsSync(dist)) {
+      links.push({
+        harness: h.id,
+        action: "skip",
+        reason: "dist_missing",
+        help: "Run ai-md build first",
+      });
+      continue;
+    }
+    if (h.skills) {
+      links.push({
+        harness: h.id,
+        kind: "skills",
+        ...linkPath(path.join(dist, "skills"), h.skills, {
+          force,
+          dryRun,
+          linkMode: cfg.linkMode,
+        }),
+      });
+    }
+    if (h.rules && h.format !== "skills-only") {
+      links.push({
+        harness: h.id,
+        kind: "rules",
+        ...linkPath(path.join(dist, "rules"), h.rules, {
+          force,
+          dryRun,
+          linkMode: cfg.linkMode,
+        }),
+      });
+    }
+  }
+
+  return {
+    links,
+    skipped,
+    warnings: [...warnings, ...platformWarn],
+  };
+}
+
+function buildAndLink(cfg, opts = {}) {
+  const build = runBuild(cfg, opts);
+  const link = linkHarnesses(cfg, opts.agents, opts);
+  return { build, link };
+}
+
+function runInstall(cfg, opts = {}) {
+  const dir = cfg.dir;
+  if (!fs.existsSync(dir)) {
+    gitClone(cfg.remote, dir, { dryRun: opts.dryRun });
+  } else if (!fs.existsSync(path.join(dir, ".git"))) {
+    const err = new Error(
+      `${dir} exists but is not a git repo. Use ai-md init or remove it.`
+    );
+    err.code = "EEXIST";
+    throw err;
+  }
+  // migrate hint: if legacy flat rules/ exist, status will note
+  return buildAndLink(cfg, {
+    agents: opts.agents,
+    force: opts.force,
+    dryRun: opts.dryRun,
+    forceLink: opts.forceLink,
+  });
+}
+
+function runPull(cfg, opts = {}) {
+  if (!fs.existsSync(path.join(cfg.dir, ".git"))) {
+    const err = new Error(`${cfg.dir} is not a git repo; run ai-md install`);
+    err.code = "ENOENT";
+    throw err;
+  }
+  gitPull(cfg.dir, { dryRun: opts.dryRun });
+  return buildAndLink(cfg, {
+    agents: opts.agents,
+    force: opts.force,
+    dryRun: opts.dryRun,
+    forceLink: opts.forceLink,
+  });
+}
+
+function runPush(cfg, opts = {}) {
+  return gitPush(cfg.dir, {
+    message: opts.message,
+    dryRun: opts.dryRun,
+  });
+}
+
+function runDoctor({
+  fix = false,
+  force = false,
+  agents = ["cursor"],
+  dryRun = false,
+  forceLink = false,
+} = {}) {
   const cfg = resolveConfig();
   const status = collectStatus({ full: true, agents });
   const repairs = [];
   if (fix || force) {
-    repairs.push(...ensureCursorLinks(cfg, { force: true, dryRun }));
-    repairs.push(...ensureAgentSkillLinks(cfg, agents, { force: true, dryRun }));
+    try {
+      const bl = buildAndLink(cfg, {
+        agents,
+        force: true,
+        dryRun,
+        forceLink,
+      });
+      repairs.push(bl);
+    } catch (e) {
+      repairs.push({ error: e.message, code: e.code, dirty: e.dirty });
+    }
   }
   const after = fix || force ? collectStatus({ full: true, agents }) : status;
   return {
-    before: {
-      problems: status.problems,
-      counts: status.counts,
-    },
+    before: { problems: status.problems, counts: status.counts },
     repairs,
     after: {
       problems: after.problems,
       counts: after.counts,
       links: after.links,
+      harnesses: after.harnesses,
       projects: after.projects,
     },
     help: statusHelp(after),
+    wsl: wslWarnings(cfg, []),
+  };
+}
+
+function harnessList(cfg) {
+  const ids = new Set([
+    ...listBuiltinIds(),
+    ...Object.keys((cfg.machineConfig && cfg.machineConfig.harnesses) || {}),
+  ]);
+  return [...ids].sort().map((id) => {
+    const def = resolveHarness(id, cfg);
+    return {
+      id: def.id,
+      format: def.format,
+      emitter: def.emitter,
+      aliasOf: def.aliasOf,
+      skills: def.skills,
+      rules: def.rules,
+      enabled: def.enabled,
+      installed: isHarnessInstalled(def),
+      distId: def.distId,
+      tier: def.tier,
+    };
+  });
+}
+
+function harnessShow(cfg, id) {
+  const def = resolveHarness(id, cfg);
+  return {
+    ...def,
+    installed: isHarnessInstalled(def),
+    builtin: Boolean(getBuiltin(id)),
   };
 }
 
+function harnessSet(cfg, id, { skills, rules, format } = {}) {
+  const key = String(id).toLowerCase();
+  const existing = { ...(cfg.machineConfig || {}) };
+  const harnesses = { ...(existing.harnesses || {}) };
+  const cur = { ...(harnesses[key] || {}) };
+  if (skills != null) cur.skills = skills;
+  if (rules != null) cur.rules = rules;
+  if (format != null) cur.format = format;
+  cur.enabled = cur.enabled !== false;
+  harnesses[key] = cur;
+  const agents = existing.agents || ["cursor"];
+  if (!agents.includes(key) && key !== "codex") {
+    agents.push(key);
+  }
+  const saved = writeMachineConfig({ ...existing, harnesses, agents });
+  ensureAgentSourceDirs(
+    { ...cfg, dir: saved.config.dir || cfg.dir },
+    key === "codex" ? "agents" : key
+  );
+  return { id: key, harness: cur, config: saved };
+}
+
+function harnessUnset(cfg, id, { pathsOnly = false } = {}) {
+  const key = String(id).toLowerCase();
+  const existing = { ...(cfg.machineConfig || {}) };
+  const harnesses = { ...(existing.harnesses || {}) };
+  if (pathsOnly && harnesses[key]) {
+    delete harnesses[key].skills;
+    delete harnesses[key].rules;
+    delete harnesses[key].format;
+  } else {
+    delete harnesses[key];
+  }
+  return writeMachineConfig({ ...existing, harnesses });
+}
+
+function harnessEnable(cfg, id, enabled) {
+  const key = String(id).toLowerCase();
+  const existing = { ...(cfg.machineConfig || {}) };
+  const harnesses = { ...(existing.harnesses || {}) };
+  harnesses[key] = { ...(harnesses[key] || {}), enabled };
+  let agents = [...(existing.agents || ["cursor"])];
+  if (enabled && !agents.includes(key) && key !== "codex") agents.push(key);
+  if (!enabled) agents = agents.filter((a) => a !== key);
+  return writeMachineConfig({ ...existing, harnesses, agents });
+}
+
+// legacy exports expected by older callers
+function ensureCursorLinks(cfg, opts = {}) {
+  return linkHarnesses(cfg, ["cursor"], opts).links;
+}
+
+function ensureAgentSkillLinks(cfg, agents, opts = {}) {
+  return linkHarnesses(cfg, agents, opts).links;
+}
+
 module.exports = {
   ensureCursorLinks,
   ensureAgentSkillLinks,
@@ -258,5 +434,22 @@ module.exports = {
   linkProject,
   runDoctor,
   linkPath,
+  linkHarnesses,
+  buildAndLink,
+  runInstall,
+  runPull,
+  runPush,
+  runBuild,
+  runRescue,
+  seedSkeleton,
+  initRepo,
+  harnessList,
+  harnessShow,
+  harnessSet,
+  harnessUnset,
+  harnessEnable,
   collectProjects,
+  SKELETON_VERSION,
+  distDirty,
+  defaultLinkMode,
 };

package/lib/config-paths.js

@@ -0,0 +1,73 @@
+"use strict";
+
+const fs = require("fs");
+const os = require("os");
+const path = require("path");
+
+function expandHome(p) {
+  if (!p) return p;
+  let s = String(p);
+  if (process.platform === "win32") {
+    s = s.replace(/%([^%]+)%/g, (_, name) => process.env[name] || `%${name}%`);
+  }
+  if (s === "~") return os.homedir();
+  if (s.startsWith("~/") || s.startsWith("~\\")) {
+    return path.join(os.homedir(), s.slice(2));
+  }
+  return s;
+}
+
+function isWsl() {
+  if (process.platform !== "linux") return false;
+  try {
+    const rel = os.release().toLowerCase();
+    if (rel.includes("microsoft") || rel.includes("wsl")) return true;
+  } catch {
+    /* ignore */
+  }
+  try {
+    const v = fs.readFileSync("/proc/version", "utf8").toLowerCase();
+    return v.includes("microsoft") || v.includes("wsl");
+  } catch {
+    return false;
+  }
+}
+
+function defaultLinkMode() {
+  if (process.platform === "win32") return "junction";
+  return "symlink";
+}
+
+function looksLikeWindowsPath(p) {
+  return /^[a-zA-Z]:[\\/]/.test(p) || p.startsWith("\\\\");
+}
+
+function normalizePath(p) {
+  if (!p) return p;
+  try {
+    return fs.realpathSync.native
+      ? fs.realpathSync.native(p)
+      : fs.realpathSync(p);
+  } catch {
+    return path.resolve(p);
+  }
+}
+
+function pathsEqual(a, b) {
+  if (!a || !b) return false;
+  const na = normalizePath(a);
+  const nb = normalizePath(b);
+  if (process.platform === "win32") {
+    return na.toLowerCase() === nb.toLowerCase();
+  }
+  return na === nb;
+}
+
+module.exports = {
+  expandHome,
+  isWsl,
+  defaultLinkMode,
+  looksLikeWindowsPath,
+  normalizePath,
+  pathsEqual,
+};