@rex_koh/subagent-budget-guard 0.1.0 → 0.1.2

package/.claude-plugin/plugin.json

@@ -2,7 +2,7 @@
   "name": "subagent-budget-guard",
   "displayName": "Subagent Budget Guard",
   "description": "Hard-deny subagent launches, record verified subagent usage, and enforce a session budget against Claude Code's 5-hour rate-limit percentage.",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "author": {
     "name": "ClaudeSubAgentSuppressor"
   },
@@ -15,14 +15,6 @@
     "hooks"
   ],
   "userConfig": {
-    "max_subagents_per_session": {
-      "type": "number",
-      "title": "Max subagents per session",
-      "description": "Maximum normal Agent tool subagents allowed in one Claude Code session. The default 0 blocks all new subagents.",
-      "default": 0,
-      "min": 0,
-      "required": true
-    },
     "max_concurrent_subagents": {
       "type": "number",
       "title": "Max concurrent subagents",
@@ -31,14 +23,6 @@
       "min": 0,
       "required": true
     },
-    "max_agent_team_tasks_per_session": {
-      "type": "number",
-      "title": "Max agent-team tasks per session",
-      "description": "Maximum agent-team tasks that may be created in one session. The default 0 suppresses agent-team task creation.",
-      "default": 0,
-      "min": 0,
-      "required": true
-    },
     "max_subagent_tokens_per_session": {
       "type": "number",
       "title": "Max verified subagent tokens per session",
@@ -68,7 +52,7 @@
     "enforcement_enabled": {
       "type": "boolean",
       "title": "Enable enforcement",
-      "description": "When true, hooks deny over-budget subagents, agent-team tasks, and prompts. When false, the plugin records usage without blocking.",
+      "description": "When true, hooks deny over-budget subagents and prompts. When false, the plugin records usage without blocking.",
       "default": true,
       "required": true
     }

package/README.md

@@ -43,4 +43,4 @@ Offline verification:
 node bin/verify.js --offline
 ```
 
-The plugin is strict by default: `max_subagents_per_session`, `max_concurrent_subagents`, and `max_agent_team_tasks_per_session` all default to `0`.
+The plugin is strict by default: `max_concurrent_subagents` defaults to `0`, so normal subagent launches are blocked unless raised.

package/bin/verify.js

@@ -4,6 +4,9 @@ import {
   runLiveVerification,
   runOfflineVerification
 } from '../lib/verifier.js';
+import { pathExists } from '../lib/guard.js';
+import { fileURLToPath } from 'node:url';
+import path from 'node:path';
 
 function modeFromArgs() {
   if (process.argv.includes('--live')) return 'live';
@@ -12,10 +15,16 @@ function modeFromArgs() {
 
 async function main() {
   const mode = modeFromArgs();
+  const packageRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '..');
+  const cwd = process.cwd();
+  const cwdLooksLikeMarketplace =
+    (await pathExists(path.join(cwd, '.claude-plugin', 'marketplace.json'))) ||
+    (await pathExists(path.join(cwd, 'plugins', 'subagent-budget-guard', '.claude-plugin', 'plugin.json')));
+  const repoRoot = cwdLooksLikeMarketplace ? cwd : packageRoot;
   const result =
     mode === 'live'
-      ? await runLiveVerification({ repoRoot: process.cwd(), env: process.env })
-      : await runOfflineVerification({ repoRoot: process.cwd(), env: process.env });
+      ? await runLiveVerification({ repoRoot, env: process.env })
+      : await runOfflineVerification({ repoRoot, env: process.env });
 
   process.stdout.write(`${formatVerificationResult(result)}\n`);
   process.exitCode = result.ok ? 0 : 1;

package/lib/guard.js

@@ -16,9 +16,7 @@ import path from 'node:path';
 export const PLUGIN_NAME = 'subagent-budget-guard';
 
 export const DEFAULT_CONFIG = Object.freeze({
-  max_subagents_per_session: 0,
   max_concurrent_subagents: 0,
-  max_agent_team_tasks_per_session: 0,
   max_subagent_tokens_per_session: 0,
   session_five_hour_budget_percent: 25,
   absolute_five_hour_ceiling_percent: 95,
@@ -306,14 +304,6 @@ function agentDenyReason(state, config) {
   const budgetReason = fiveHourBudgetDecision(state, config);
   if (budgetReason) return budgetReason;
 
-  if (config.max_subagents_per_session === 0) {
-    return 'Subagent launch denied: max_subagents_per_session is 0.';
-  }
-
-  if (state.subagents.allowed >= config.max_subagents_per_session) {
-    return `Subagent launch denied: max_subagents_per_session ${config.max_subagents_per_session} already reached.`;
-  }
-
   if (config.max_concurrent_subagents === 0) {
     return 'Subagent launch denied: max_concurrent_subagents is 0.';
   }
@@ -474,14 +464,6 @@ function taskDenyReason(state, config) {
   const budgetReason = fiveHourBudgetDecision(state, config);
   if (budgetReason) return budgetReason;
 
-  if (config.max_agent_team_tasks_per_session === 0) {
-    return 'Agent-team task denied: max_agent_team_tasks_per_session is 0.';
-  }
-
-  if (state.agentTeam.created >= config.max_agent_team_tasks_per_session) {
-    return `Agent-team task denied: max_agent_team_tasks_per_session ${config.max_agent_team_tasks_per_session} already reached.`;
-  }
-
   return null;
 }
 
@@ -614,9 +596,7 @@ export async function buildReport(sessionId, env = process.env) {
     state,
     summary: {
       verifiedTokenLabel: `${state.subagents.verifiedTokens.toLocaleString('en-US')} verified tokens`,
-      subagentLaunches: `${state.subagents.allowed}/${config.max_subagents_per_session}`,
       activeSubagents: `${state.subagents.active}/${config.max_concurrent_subagents}`,
-      agentTeamTasks: `${state.agentTeam.created}/${config.max_agent_team_tasks_per_session}`,
       fiveHourBudget:
         consumed === null
           ? '5-hour usage unavailable'

package/lib/verifier.js

@@ -20,8 +20,31 @@ async function readJson(filePath) {
   return JSON.parse(text.replace(/^\uFEFF/, ''));
 }
 
-function pluginRoot(repoRoot) {
-  return path.join(repoRoot, 'plugins', 'subagent-budget-guard');
+async function resolveLayout(repoRoot) {
+  const packagePluginRoot = repoRoot;
+  const marketplacePluginRoot = path.join(repoRoot, 'plugins', 'subagent-budget-guard');
+
+  if (await pathExists(path.join(repoRoot, '.claude-plugin', 'marketplace.json'))) {
+    return {
+      repoRoot,
+      pluginRoot: marketplacePluginRoot,
+      hasMarketplace: true
+    };
+  }
+
+  if (await pathExists(path.join(packagePluginRoot, '.claude-plugin', 'plugin.json'))) {
+    return {
+      repoRoot,
+      pluginRoot: packagePluginRoot,
+      hasMarketplace: false
+    };
+  }
+
+  return {
+    repoRoot,
+    pluginRoot: marketplacePluginRoot,
+    hasMarketplace: false
+  };
 }
 
 async function withCheck(result, name, fn) {
@@ -48,23 +71,36 @@ export async function runOfflineVerification({
     checks: [],
     failures: []
   };
-  const root = pluginRoot(repoRoot);
+  const layout = await resolveLayout(repoRoot);
+  const root = layout.pluginRoot;
 
-  await withCheck(result, 'marketplace-manifest', async () => {
-    const marketplacePath = path.join(repoRoot, '.claude-plugin', 'marketplace.json');
-    const marketplace = await readJson(marketplacePath);
-    assert(marketplace.name === 'subagent-budget-tools', 'marketplace name mismatch');
-    assert(Array.isArray(marketplace.plugins), 'marketplace.plugins must be an array');
-    const entry = marketplace.plugins.find((plugin) => plugin.name === 'subagent-budget-guard');
-    assert(entry, 'subagent-budget-guard entry missing');
-    assert(entry.source?.source === 'npm', 'marketplace source must use npm');
-    assert(
-      entry.source?.package === '@rex_koh/subagent-budget-guard',
-      'marketplace npm package mismatch'
-    );
-    assert(entry.source?.version === '0.1.0', 'marketplace npm version mismatch');
-    return marketplacePath;
-  });
+  if (layout.hasMarketplace) {
+    await withCheck(result, 'marketplace-manifest', async () => {
+      const marketplacePath = path.join(repoRoot, '.claude-plugin', 'marketplace.json');
+      const marketplace = await readJson(marketplacePath);
+      assert(marketplace.name === 'subagent-budget-tools', 'marketplace name mismatch');
+      assert(Array.isArray(marketplace.plugins), 'marketplace.plugins must be an array');
+      const entry = marketplace.plugins.find((plugin) => plugin.name === 'subagent-budget-guard');
+      assert(entry, 'subagent-budget-guard entry missing');
+      assert(entry.source?.source === 'npm', 'marketplace source must use npm');
+      assert(
+        entry.source?.package === '@rex_koh/subagent-budget-guard',
+        'marketplace npm package mismatch'
+      );
+      assert(entry.source?.version === '0.1.2', 'marketplace npm version mismatch');
+      return marketplacePath;
+    });
+  } else {
+    await withCheck(result, 'package-root', async () => {
+      const packageJsonPath = path.join(root, 'package.json');
+      const packageJson = await readJson(packageJsonPath);
+      assert(
+        packageJson.name === '@rex_koh/subagent-budget-guard',
+        'package root name mismatch'
+      );
+      return packageJsonPath;
+    });
+  }
 
   await withCheck(result, 'plugin-manifest-user-config', async () => {
     const manifestPath = path.join(root, '.claude-plugin', 'plugin.json');
@@ -303,7 +339,8 @@ export async function runLiveVerification({
   result.checks.push(...offline.checks);
   result.failures.push(...offline.failures);
 
-  const root = pluginRoot(repoRoot);
+  const layout = await resolveLayout(repoRoot);
+  const root = layout.pluginRoot;
   const hasClaude = await commandExists('claude');
   if (!hasClaude) {
     result.warnings.push('claude executable was not found on PATH; skipped claude plugin validate and install-state checks.');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rex_koh/subagent-budget-guard",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Claude Code plugin that blocks subagents by default, records verified subagent usage, and enforces 5-hour usage budgets.",
   "license": "MIT",
   "author": "ClaudeSubAgentSuppressor",