@wrongstack/plugins 0.281.1 → 0.282.0

package/dist/pr-drafter.js

@@ -0,0 +1,334 @@
+import { execFileSync } from 'child_process';
+import { mkdirSync, writeFileSync } from 'fs';
+import { dirname, resolve, isAbsolute, relative } from 'path';
+
+// src/pr-drafter/index.ts
+var API_VERSION = "^0.1.10";
+var state = {
+  commits: [],
+  files: /* @__PURE__ */ new Set(),
+  models: /* @__PURE__ */ new Set(),
+  totalInputTokens: 0,
+  totalOutputTokens: 0,
+  toolCalls: 0,
+  draftsWritten: 0,
+  draftErrors: 0,
+  stopInvocations: 0,
+  stopHookUnregister: null,
+  eventUnsubscribers: []
+};
+var DEFAULTS = {
+  enabled: true,
+  outputPath: ".wrongstack/PR_DRAFT.md",
+  writeOnStop: true,
+  includeDiff: true,
+  includeCommits: true,
+  includeFiles: true,
+  aiSummary: false
+};
+function readConfig(raw) {
+  if (!raw || typeof raw !== "object") return { ...DEFAULTS };
+  const r = raw;
+  return {
+    enabled: r["enabled"] !== false,
+    outputPath: typeof r["outputPath"] === "string" ? r["outputPath"] : DEFAULTS.outputPath,
+    writeOnStop: r["writeOnStop"] !== false,
+    includeDiff: r["includeDiff"] !== false,
+    includeCommits: r["includeCommits"] !== false,
+    includeFiles: r["includeFiles"] !== false,
+    aiSummary: r["aiSummary"] === true
+  };
+}
+function resolveProjectPath(rawPath, cwd = process.cwd()) {
+  if (typeof rawPath !== "string" || rawPath.length === 0) return null;
+  const root = resolve(cwd);
+  const resolved = isAbsolute(rawPath) ? resolve(rawPath) : resolve(root, rawPath);
+  const rel = relative(root, resolved);
+  if (rel === "" || !rel.startsWith("..") && !isAbsolute(rel)) return resolved;
+  return null;
+}
+function gitBranch() {
+  try {
+    const out = execFileSync("git", ["rev-parse", "--abbrev-ref", "HEAD"], {
+      encoding: "utf-8",
+      cwd: process.cwd(),
+      stdio: ["pipe", "pipe", "ignore"],
+      timeout: 5e3
+    });
+    return out.trim();
+  } catch {
+    return null;
+  }
+}
+function gitDiff() {
+  try {
+    return execFileSync("git", ["diff", "--stat"], {
+      encoding: "utf-8",
+      cwd: process.cwd(),
+      stdio: ["pipe", "pipe", "ignore"],
+      timeout: 1e4
+    }).trim();
+  } catch {
+    return null;
+  }
+}
+async function buildDraft(cfg, llm) {
+  const branch = gitBranch();
+  const diffStat = cfg.includeDiff ? gitDiff() : null;
+  const lines = [];
+  lines.push("# PR Draft");
+  if (branch) lines.push(`**Branch:** \`${branch}\``);
+  lines.push(`**Generated:** ${(/* @__PURE__ */ new Date()).toISOString()}`);
+  lines.push("");
+  if (cfg.aiSummary && llm) {
+    try {
+      const context = `Files changed: ${[...state.files].join(", ") || "none"}
+Commits: ${state.commits.join("; ") || "none"}
+Tool calls: ${state.toolCalls}
+Tokens: ${state.totalInputTokens} in / ${state.totalOutputTokens} out`;
+      const result = await llm.complete(
+        "Write a concise PR title and one-paragraph summary for the changes below. Format exactly as:\nTITLE: <title>\nSUMMARY: <summary>\n\n" + context,
+        { system: "You write terse engineering PR descriptions.", maxTokens: 250 }
+      );
+      const text = result.text.trim();
+      const titleMatch = /TITLE:\s*(.+)/i.exec(text);
+      const summaryMatch = /SUMMARY:\s*(.+)/is.exec(text);
+      if (titleMatch?.[1]) lines.push(`## ${titleMatch[1].trim()}`);
+      if (summaryMatch?.[1]) lines.push(summaryMatch[1].trim(), "");
+    } catch {
+    }
+  }
+  if (cfg.includeCommits && state.commits.length > 0) {
+    lines.push("## Commits");
+    for (const c of state.commits) lines.push(`- ${c}`);
+    lines.push("");
+  }
+  if (cfg.includeFiles && state.files.size > 0) {
+    lines.push("## Files changed");
+    for (const f of [...state.files].sort()) lines.push(`- \`${f}\``);
+    lines.push("");
+  }
+  if (diffStat) {
+    lines.push("## Diff summary");
+    lines.push("```");
+    lines.push(diffStat);
+    lines.push("```");
+    lines.push("");
+  }
+  lines.push("## Session metrics");
+  lines.push(`- Tool calls: ${state.toolCalls}`);
+  lines.push(`- Models used: ${[...state.models].join(", ") || "unknown"}`);
+  lines.push(`- Tokens: ${state.totalInputTokens} in / ${state.totalOutputTokens} out`);
+  lines.push("");
+  const title = state.commits[0] ?? `Changes on ${branch ?? "current branch"}`;
+  return { title, body: lines.join("\n") };
+}
+async function writeDraft(cfg, llm) {
+  const resolved = resolveProjectPath(cfg.outputPath);
+  if (!resolved) {
+    state.draftErrors += 1;
+    return;
+  }
+  const draft = await buildDraft(cfg, llm);
+  try {
+    mkdirSync(dirname(resolved), { recursive: true });
+    writeFileSync(resolved, draft.body);
+    state.draftsWritten += 1;
+  } catch {
+    state.draftErrors += 1;
+  }
+}
+var plugin = {
+  name: "pr-drafter",
+  version: "0.1.0",
+  description: "Collects session work (commits, edited files, diff) and drafts a pull-request description",
+  apiVersion: API_VERSION,
+  capabilities: { tools: true, hooks: true },
+  defaultConfig: { ...DEFAULTS },
+  configSchema: {
+    type: "object",
+    properties: {
+      enabled: { type: "boolean", default: true, description: "Master switch." },
+      outputPath: {
+        type: "string",
+        default: ".wrongstack/PR_DRAFT.md",
+        description: "Path where the markdown PR draft is written."
+      },
+      writeOnStop: {
+        type: "boolean",
+        default: true,
+        description: "Automatically write the draft when the session stops."
+      },
+      includeDiff: {
+        type: "boolean",
+        default: true,
+        description: "Include git diff --stat in the draft."
+      },
+      includeCommits: {
+        type: "boolean",
+        default: true,
+        description: "Include recorded commit messages."
+      },
+      includeFiles: {
+        type: "boolean",
+        default: true,
+        description: "Include files touched by write/edit tools."
+      },
+      aiSummary: {
+        type: "boolean",
+        default: false,
+        description: "Use api.llm to generate a PR title and summary."
+      },
+      llm: {
+        type: "object",
+        description: "Optional { provider, model } override for the AI summary."
+      }
+    }
+  },
+  setup(api) {
+    state.commits = [];
+    state.files = /* @__PURE__ */ new Set();
+    state.models = /* @__PURE__ */ new Set();
+    state.totalInputTokens = 0;
+    state.totalOutputTokens = 0;
+    state.toolCalls = 0;
+    state.draftsWritten = 0;
+    state.draftErrors = 0;
+    state.stopInvocations = 0;
+    state.stopHookUnregister = null;
+    for (const off of state.eventUnsubscribers) {
+      try {
+        off();
+      } catch {
+      }
+    }
+    state.eventUnsubscribers = [];
+    const cfg = readConfig(api.config.extensions?.["pr-drafter"]);
+    if (api.onPattern) {
+      const offTool = api.onPattern("tool.completed", (_event, payload) => {
+        const p = payload;
+        const toolName = p?.tool;
+        state.toolCalls += 1;
+        if (toolName === "git_autocommit" && p?.result?.committed) {
+          const msg = p.result.commitMessage ?? p.input?.message ?? "commit";
+          state.commits.push(msg);
+        }
+        if ((toolName === "write" || toolName === "edit") && p?.input?.path) {
+          state.files.add(p.input.path);
+        }
+      });
+      state.eventUnsubscribers.push(offTool);
+    }
+    if (api.onEvent) {
+      const offUsage = api.onEvent("provider.response", (payload) => {
+        const p = payload;
+        if (p?.model) state.models.add(p.model);
+        state.totalInputTokens += p?.usage?.input ?? 0;
+        state.totalOutputTokens += p?.usage?.output ?? 0;
+      });
+      state.eventUnsubscribers.push(offUsage);
+    }
+    const stopHook = async () => {
+      if (!cfg.enabled || !cfg.writeOnStop) return;
+      state.stopInvocations += 1;
+      await writeDraft(cfg, api.llm);
+    };
+    state.stopHookUnregister = api.registerHook("Stop", void 0, stopHook);
+    api.tools.register({
+      name: "pr_draft",
+      description: "Generate or refresh the pull-request draft for the current session. Writes the markdown file and returns its path + title.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          preview: {
+            type: "boolean",
+            description: "When true, return the draft body without writing to disk."
+          }
+        }
+      },
+      permission: "auto",
+      category: "Workflow",
+      mutating: false,
+      async execute(input) {
+        if (!cfg.enabled) return { ok: false, error: "pr-drafter is disabled" };
+        const draft = await buildDraft(cfg, api.llm);
+        if (input.preview) {
+          return { ok: true, preview: true, title: draft.title, body: draft.body };
+        }
+        const resolved = resolveProjectPath(cfg.outputPath);
+        if (!resolved) return { ok: false, error: "outputPath resolves outside project" };
+        try {
+          mkdirSync(dirname(resolved), { recursive: true });
+          writeFileSync(resolved, draft.body);
+          state.draftsWritten += 1;
+          return {
+            ok: true,
+            path: cfg.outputPath,
+            resolvedPath: resolved,
+            title: draft.title
+          };
+        } catch (err) {
+          state.draftErrors += 1;
+          return {
+            ok: false,
+            error: err instanceof Error ? err.message : String(err)
+          };
+        }
+      }
+    });
+    api.log.info("pr-drafter plugin loaded", {
+      version: "0.1.0",
+      outputPath: cfg.outputPath,
+      writeOnStop: cfg.writeOnStop
+    });
+  },
+  teardown(api) {
+    if (state.stopHookUnregister) {
+      try {
+        state.stopHookUnregister();
+      } catch {
+      }
+      state.stopHookUnregister = null;
+    }
+    for (const off of state.eventUnsubscribers) {
+      try {
+        off();
+      } catch {
+      }
+    }
+    state.eventUnsubscribers = [];
+    const final = {
+      commits: state.commits.length,
+      files: state.files.size,
+      toolCalls: state.toolCalls,
+      draftsWritten: state.draftsWritten,
+      draftErrors: state.draftErrors
+    };
+    state.commits = [];
+    state.files = /* @__PURE__ */ new Set();
+    state.models = /* @__PURE__ */ new Set();
+    state.totalInputTokens = 0;
+    state.totalOutputTokens = 0;
+    state.toolCalls = 0;
+    state.draftsWritten = 0;
+    state.draftErrors = 0;
+    state.stopInvocations = 0;
+    api.log.info("pr-drafter: teardown complete", { final });
+  },
+  async health() {
+    return {
+      ok: state.draftErrors === 0,
+      message: `pr-drafter: ${state.commits.length} commit(s), ${state.files.size} file(s), ${state.draftsWritten} draft(s) written, ${state.draftErrors} error(s)`,
+      counters: {
+        commits: state.commits.length,
+        files: state.files.size,
+        toolCalls: state.toolCalls,
+        draftsWritten: state.draftsWritten,
+        draftErrors: state.draftErrors
+      }
+    };
+  }
+};
+var pr_drafter_default = plugin;
+
+export { pr_drafter_default as default };

package/dist/prompt-firewall.js

@@ -157,6 +157,11 @@ var plugin = {
     }
     const cfg = readConfig(api.config.extensions?.["prompt-firewall"]);
     if (cfg.enabled) {
+      api.emitCustom?.("provider.wrap:loaded", {
+        plugin: "prompt-firewall",
+        kind: "security",
+        wraps: ["request", "response"]
+      });
       state.extensionUnregister = api.extensions.register({
         name: "prompt-firewall",
         owner: "prompt-firewall",

package/dist/refactor-suggester.d.ts

@@ -0,0 +1,38 @@
+import { Plugin } from '@wrongstack/core';
+
+/**
+ * refactor-suggester plugin — regex-based smell detector that suggests
+ * refactoring opportunities in source files.
+ *
+ * Tools registered:
+ * - suggest_refactors : Scan a file or directory for smells.
+ * - refactor_status : Report config + counters.
+ *
+ * Hooks registered:
+ * - PostToolUse with matcher `write|edit` to source files, injecting
+ *   suggestions for the changed file.
+ *
+ * Config (`config.extensions['refactor-suggester']`):
+ *
+ * ```jsonc
+ * {
+ *   "enabled": true,
+ *   "extensions": [".ts", ".tsx", ".js", ".jsx"],
+ *   "maxSuggestions": 20,
+ *   "rules": { "longFunctionLines": 50, "maxParams": 5, "maxNesting": 3 }
+ * }
+ * ```
+ *
+ * @public
+ */
+
+type SmellType = 'long-function' | 'deep-nesting' | 'many-parameters' | 'magic-number' | 'console-log';
+interface RefactorSuggestion {
+    file: string;
+    line: number;
+    type: SmellType;
+    message: string;
+}
+declare const plugin: Plugin;
+
+export { type RefactorSuggestion, type SmellType, plugin as default };