akm-opencode-v2 0.10.261106-beta → 0.10.261107-beta

package/dist/index.js

@@ -45,7 +45,8 @@ function readPluginVersion(moduleUrl, packageName) {
 // ../claude/shared/ref-extraction.ts
 import { existsSync, statSync } from "fs";
 import path2 from "path";
-var AKM_REF_STRICT = /^(?:[A-Za-z0-9@._+-]+\/\/)?(?:agents|commands|env|facts|instructions|knowledge|lessons|memories|scripts|secrets|sessions|skills|tasks|workflows)\/[A-Za-z0-9._/-]+(?:#[A-Za-z0-9._~!$&'()*+,;=:@%/?-]+)?$/;
+var AKM_REF_STRICT = /^(?:[A-Za-z0-9@._+-]+\/\/)?[A-Za-z0-9@._+-]+(?:\/[A-Za-z0-9._+@-]+)*(?:#[A-Za-z0-9._~!$&'()*+,;=:@%/?-]+)?$/;
+var CONCEPT_ROOT_REF = /^(?:[A-Za-z0-9@._+-]+\/\/)?(?:agents|commands|env|facts|instructions|knowledge|lessons|memories|scripts|secrets|sessions|skills|tasks|workflows)\/[A-Za-z0-9._/-]+(?:#[A-Za-z0-9._~!$&'()*+,;=:@%/?-]+)?$/;
 var EDGE_PUNCTUATION = new Set([".", ",", ";", ":", "!", "?", "(", ")", "[", "]", "{", "}", "'", '"', "`"]);
 function normalizeToken(token) {
   let start = 0;
@@ -60,14 +61,39 @@ function extractAkmRefsFromString(text) {
   const refs = new Set;
   for (const token of text.split(/\s+/)) {
     const normalized = normalizeToken(token);
-    if (normalized && AKM_REF_STRICT.test(normalized))
+    if (normalized && CONCEPT_ROOT_REF.test(normalized))
       refs.add(normalized);
   }
   return [...refs];
 }
+function akmRefFromValue(value) {
+  if (typeof value !== "string")
+    return;
+  const ref = value.trim();
+  return AKM_REF_STRICT.test(ref) && !ref.split("#", 1)[0].split("//").pop().split("/").some((s) => s === "." || s === "..") ? ref : undefined;
+}
+function refsOfAkmResult(result) {
+  if (!result || typeof result !== "object")
+    return [];
+  const r = result;
+  const entries = [r, ...Array.isArray(r.hits) ? r.hits : [], ...Array.isArray(r.items) ? r.items : []];
+  const refs = entries.map((e) => e && typeof e === "object" ? akmRefFromValue(e.ref) : undefined);
+  return [...new Set(refs.filter((ref) => !!ref))];
+}
+function extractRefsFromAkmOutput(text) {
+  let parsed;
+  try {
+    parsed = JSON.parse(text);
+  } catch {
+    parsed = undefined;
+  }
+  const refLines = [...text.matchAll(/^\s*ref:\s*(\S+)\s*$/gm)].map((m) => akmRefFromValue(m[1]));
+  return [...new Set([...refsOfAkmResult(parsed), ...refLines.filter((r) => !!r), ...extractAkmRefsFromString(text)])];
+}
 function normalizeCandidate(candidate) {
   const canonical = normalizeToken(candidate);
-  if (!AKM_REF_STRICT.test(canonical))
+  const ref = akmRefFromValue(canonical);
+  if (!ref)
     return null;
   const withoutFragment = canonical.split("#", 1)[0] ?? "";
   const separator = withoutFragment.indexOf("//");
@@ -103,39 +129,35 @@ var SCRIPT_EXTENSIONS = new Set([
   ".kts"
 ]);
 function markdownFile(directPath) {
+  if (path2.basename(directPath).replace(/\.md$/, "") === "SKILL")
+    return false;
   return isFile(`${directPath}.md`) || directPath.endsWith(".md") && isFile(directPath);
 }
-function isAsset(type, directPath) {
-  if (type === "skills")
-    return isFile(path2.join(directPath, "SKILL.md"));
-  if (type === "scripts")
-    return SCRIPT_EXTENSIONS.has(path2.extname(directPath).toLowerCase()) && isFile(directPath);
-  if (type === "secrets")
-    return !/\.(lock|sensitive)$/.test(directPath) && isFile(directPath);
-  if (type === "tasks")
-    return isFile(`${directPath}.yml`);
-  if (type === "env") {
-    return isFile(`${directPath}.env`) || path2.basename(directPath) === "default" && isFile(path2.join(path2.dirname(directPath), ".env"));
-  }
-  return markdownFile(directPath);
-}
-var OUTSIDE_THEIR_DIRECTORY = new Set(["scripts", "agents", "commands", "facts", "instructions", "knowledge", "lessons", "memories", "sessions"]);
+function isAsset(directPath, ref) {
+  if (isFile(path2.join(directPath, "SKILL.md")) || markdownFile(directPath))
+    return true;
+  if (SCRIPT_EXTENSIONS.has(path2.extname(directPath).toLowerCase()) && isFile(directPath))
+    return true;
+  const folders = ref.split("/").slice(0, -1);
+  if (folders.includes("env")) {
+    if (isFile(`${directPath}.env`))
+      return true;
+    if (path2.basename(directPath) === "default" && isFile(path2.join(path2.dirname(directPath), ".env")))
+      return true;
+  }
+  if (folders.includes("tasks") && isFile(`${directPath}.yml`))
+    return true;
+  return folders.includes("secrets") && !/\.(lock|sensitive)$/.test(directPath) && isFile(directPath);
+}
 function conceptExistsInAnyBundle(conceptId, bundleRoots) {
-  const type = conceptId.split("/", 1)[0];
-  const rest = conceptId.slice(type.length + 1);
   for (const root of bundleRoots) {
     if (!root)
       continue;
     const resolvedRoot = path2.resolve(root);
+    const inside = (file) => file !== resolvedRoot && file.startsWith(`${resolvedRoot}${path2.sep}`);
     const directPath = path2.resolve(resolvedRoot, conceptId);
-    if (directPath !== resolvedRoot && !directPath.startsWith(`${resolvedRoot}${path2.sep}`))
-      continue;
-    if (isAsset(type, directPath))
+    if (inside(directPath) && isAsset(directPath, conceptId))
       return true;
-    if (OUTSIDE_THEIR_DIRECTORY.has(type) && rest.includes("/") && !rest.startsWith(`${type}/`)) {
-      if (isAsset(type, path2.resolve(resolvedRoot, rest)))
-        return true;
-    }
   }
   return false;
 }
@@ -802,35 +824,9 @@ function classifyToolFeedback(value) {
     return "positive";
   return;
 }
-function extractToolRefs(toolName, args, output) {
-  const refs = new Set;
-  const addMatches = (value) => {
-    if (typeof value !== "string")
-      return;
-    for (const ref of extractAkmRefsFromString(value))
-      refs.add(ref);
-  };
-  for (const key of ["ref", "package_ref"])
-    addMatches(args[key]);
-  if (output && typeof output === "object") {
-    const o = output;
-    addMatches(o.ref);
-    if (Array.isArray(o.hits)) {
-      for (const hit of o.hits) {
-        if (hit && typeof hit === "object")
-          addMatches(hit.ref);
-      }
-    }
-    if (Array.isArray(o.items)) {
-      for (const item of o.items) {
-        if (item && typeof item === "object")
-          addMatches(item.ref);
-      }
-    }
-    if (toolName === "akm_remember" && typeof o.ref === "string")
-      addMatches(o.ref);
-  }
-  return [...refs];
+function extractToolRefs(args, output) {
+  const named = [args.ref, args.package_ref].map(akmRefFromValue).filter((ref) => !!ref);
+  return [...new Set([...named, ...refsOfAkmResult(output)])];
 }
 function extractAkmRefsFromAllArgs(args) {
   if (!args || typeof args !== "object")
@@ -909,7 +905,7 @@ function createFeedbackTracker(options) {
     const isAkmTool = input.tool.startsWith("akm_");
     const candidateRefs = [...new Set([...extractAkmRefsFromAllArgs(input.args), ...extractAkmRefsFromString(input.outputText)])];
     const parsed = isAkmTool ? parseToolOutput(input.outputText) : null;
-    const refs = isAkmTool && parsed ? extractToolRefs(input.tool, input.args, parsed) : validateRefCandidates(candidateRefs, [await input.getBundleDir() ?? ""]);
+    const refs = isAkmTool && parsed ? extractToolRefs(input.args, parsed) : validateRefCandidates(candidateRefs, [await input.getBundleDir() ?? ""]);
     const failedCall = classifyToolFeedback(parsed) === "negative";
     if (refs.length > 0) {
       input.writeEvent({
@@ -929,7 +925,7 @@ function createFeedbackTracker(options) {
   }
   function observeAkmToolResult(input) {
     const outcome = classifyToolFeedback(input.parsed);
-    const refs = extractToolRefs(input.tool, input.args, input.parsed);
+    const refs = extractToolRefs(input.args, input.parsed);
     const failedCall = outcome === "negative";
     if (!failedCall)
       noteRecentRefs(input.sessionID, refs);
@@ -3017,7 +3013,7 @@ ${NEED_MORE_CONTEXT_HINT}` : NEED_MORE_CONTEXT_HINT;
           sessionID,
           query: truncateLogText(decision.query, 200),
           reason: decision.reason,
-          refs: extractAkmRefsFromString(curated ?? ""),
+          refs: extractRefsFromAkmOutput(curated ?? ""),
           outcome: curated ? "ok" : "skipped"
         });
       });
@@ -3223,7 +3219,7 @@ ${st.hints}` : AKM_HINTS_PREFIX,
       log(outcome.ok ? "info" : "error", outcome.ok ? "AKM tool call completed" : "AKM tool call failed", {
         toolName: spec,
         sessionID: context.sessionID,
-        refs: extractAkmRefsFromString(outcome.text),
+        refs: extractRefsFromAkmOutput(outcome.text),
         ...outcome.ok ? {} : { error: failureMessage(outcome.text) }
       });
       return outcome;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "akm-opencode-v2",
-  "version": "0.10.261106-beta",
+  "version": "0.10.261107-beta",
   "type": "module",
   "description": "OpenCode 2 plugin for AKM: the five AKM tools, automatic recall, and automatic session extraction. For OpenCode 1, use akm-opencode.",
   "keywords": [
@@ -39,7 +39,7 @@
     "typecheck": "tsc -p tsconfig.json"
   },
   "dependencies": {
-    "akm-cli": "0.10.261106-beta"
+    "akm-cli": "0.10.261107-beta"
   },
   "peerDependencies": {
     "@opencode/plugin": "2.1.4",