@tekmidian/pai 0.3.1 → 0.4.0

package/dist/cli/index.mjs

@@ -4099,6 +4099,42 @@ function mergeStatusLine(settings, incoming, report) {
 	return true;
 }
 /**
+* Merge permissions — append allow/deny entries, deduplicating.
+*/
+function mergePermissions(settings, incoming, report) {
+	let changed = false;
+	const perms = typeof settings["permissions"] === "object" && settings["permissions"] !== null ? settings["permissions"] : {};
+	for (const list of ["allow", "deny"]) {
+		const entries = incoming[list];
+		if (!entries || entries.length === 0) continue;
+		const existing = Array.isArray(perms[list]) ? perms[list] : [];
+		const existingSet = new Set(existing);
+		for (const entry of entries) if (existingSet.has(entry)) report.push(chalk.dim(`  Skipped: permissions.${list} "${entry}" already present`));
+		else {
+			existing.push(entry);
+			existingSet.add(entry);
+			report.push(chalk.green(`  Added permissions.${list}: ${entry}`));
+			changed = true;
+		}
+		perms[list] = existing;
+	}
+	settings["permissions"] = perms;
+	return changed;
+}
+/**
+* Merge flags — set keys only if not already present, never overwrite.
+*/
+function mergeFlags(settings, incoming, report) {
+	let changed = false;
+	for (const [key, value] of Object.entries(incoming)) if (Object.prototype.hasOwnProperty.call(settings, key)) report.push(chalk.dim(`  Skipped: ${key} already set`));
+	else {
+		settings[key] = value;
+		report.push(chalk.green(`  Added flag: ${key}`));
+		changed = true;
+	}
+	return changed;
+}
+/**
 * Merge env vars, hooks, and/or a statusLine entry into ~/.claude/settings.json.
 * Never overwrites existing values — only adds what is missing.
 *
@@ -4117,6 +4153,12 @@ function mergeSettings(opts) {
 	if (opts.statusLine !== void 0) {
 		if (mergeStatusLine(settings, opts.statusLine, report)) changed = true;
 	}
+	if (opts.permissions !== void 0) {
+		if (mergePermissions(settings, opts.permissions, report)) changed = true;
+	}
+	if (opts.flags !== void 0 && Object.keys(opts.flags).length > 0) {
+		if (mergeFlags(settings, opts.flags, report)) changed = true;
+	}
 	if (changed) writeSettingsJson(settings);
 	return {
 		changed,
@@ -4231,7 +4273,7 @@ function getDockerDir() {
 	const candidates = [
 		join(process.cwd(), "docker"),
 		join(homedir(), "dev", "ai", "PAI", "docker"),
-		join("/", "usr", "local", "lib", "node_modules", "@mnott", "pai", "docker")
+		join("/", "usr", "local", "lib", "node_modules", "@tekmidian", "pai", "docker")
 	];
 	for (const c of candidates) if (existsSync(join(c, "docker-compose.yml"))) return c;
 	return join(process.cwd(), "docker");
@@ -4240,7 +4282,7 @@ function getTemplatesDir$1() {
 	const candidates = [
 		join(process.cwd(), "templates"),
 		join(homedir(), "dev", "ai", "PAI", "templates"),
-		join("/", "usr", "local", "lib", "node_modules", "@mnott", "pai", "templates")
+		join("/", "usr", "local", "lib", "node_modules", "@tekmidian", "pai", "templates")
 	];
 	for (const c of candidates) if (existsSync(join(c, "claude-md.template.md"))) return c;
 	return join(process.cwd(), "templates");
@@ -4254,6 +4296,18 @@ function getHooksDir() {
 	for (const c of candidates) if (existsSync(join(c, "session-stop.sh"))) return c;
 	return join(process.cwd(), "src", "hooks");
 }
+function getDistHooksDir() {
+	const moduleDir = new URL(".", import.meta.url).pathname;
+	const fromModule = join(moduleDir, "..", "hooks");
+	const candidates = [
+		fromModule,
+		join(process.cwd(), "dist", "hooks"),
+		join(homedir(), "dev", "ai", "PAI", "dist", "hooks"),
+		join("/", "usr", "local", "lib", "node_modules", "@tekmidian", "pai", "dist", "hooks")
+	];
+	for (const candidate of candidates) if (existsSync(join(candidate, "stop-hook.mjs"))) return candidate;
+	return fromModule;
+}
 function getStatuslineScript() {
 	const candidates = [
 		join(process.cwd(), "statusline-command.sh"),
@@ -4263,6 +4317,15 @@ function getStatuslineScript() {
 	for (const c of candidates) if (existsSync(c)) return c;
 	return null;
 }
+function getTabColorScript() {
+	const candidates = [
+		join(process.cwd(), "tab-color-command.sh"),
+		join(homedir(), "dev", "ai", "PAI", "tab-color-command.sh"),
+		join("/", "usr", "local", "lib", "node_modules", "@tekmidian", "pai", "tab-color-command.sh")
+	];
+	for (const c of candidates) if (existsSync(c)) return c;
+	return null;
+}
 async function startDocker(rl) {
 	const dockerDir = getDockerDir();
 	if (!existsSync(join(dockerDir, "docker-compose.yml"))) {
@@ -4583,10 +4646,57 @@ async function stepPaiSkill(rl) {
 	return true;
 }
 /**
-* Step 6: Hook scripts (pre-compact, session-stop, statusline)
+* Step 6: AI Steering Rules installation (~/.claude/skills/PAI/AI-STEERING-RULES.md)
+*/
+async function stepAiSteeringRules(rl) {
+	section("Step 6: AI Steering Rules");
+	line$1();
+	line$1("  PAI ships a set of universal behavioral rules for AI assistants:");
+	line$1("  surgical fixes, verification before assertion, root cause analysis,");
+	line$1("  and honest failure modes. These rules load at startup as a skill.");
+	line$1();
+	const templatePath = join(getTemplatesDir$1(), "ai-steering-rules.template.md");
+	if (!existsSync(templatePath)) {
+		console.log(c.warn("AI steering rules template not found: " + templatePath));
+		console.log(c.dim("  Skipping AI steering rules installation."));
+		return false;
+	}
+	const skillDir = join(homedir(), ".claude", "skills", "PAI");
+	const skillFile = join(skillDir, "AI-STEERING-RULES.md");
+	if (existsSync(skillFile)) {
+		const content = readFileSync(skillFile, "utf-8");
+		const isGenerated = content.includes("Generated by PAI Setup");
+		if (isGenerated) console.log(c.dim("  Found existing PAI-generated AI-STEERING-RULES.md."));
+		else {
+			console.log(c.yellow("  Found existing AI-STEERING-RULES.md (not PAI-generated)."));
+			console.log(c.dim("  A backup will be created before overwriting."));
+		}
+		line$1();
+		if (!await promptYesNo(rl, "Update ~/.claude/skills/PAI/AI-STEERING-RULES.md with the latest rules?", isGenerated)) {
+			console.log(c.dim("  Keeping existing AI-STEERING-RULES.md unchanged."));
+			return false;
+		}
+		if (!isGenerated) {
+			const backupPath = skillFile + ".backup";
+			writeFileSync(backupPath, content, "utf-8");
+			console.log(c.ok(`Backed up existing AI-STEERING-RULES.md to ${backupPath}`));
+		}
+	} else if (!await promptYesNo(rl, "Install AI steering rules to ~/.claude/skills/PAI/AI-STEERING-RULES.md?", true)) {
+		console.log(c.dim("  Skipping AI steering rules installation."));
+		return false;
+	}
+	const template = readFileSync(templatePath, "utf-8");
+	if (!existsSync(skillDir)) mkdirSync(skillDir, { recursive: true });
+	writeFileSync(skillFile, template, "utf-8");
+	line$1();
+	console.log(c.ok("Installed ~/.claude/skills/PAI/AI-STEERING-RULES.md"));
+	return true;
+}
+/**
+* Step 7: Hook scripts (pre-compact, session-stop, statusline)
 */
 async function stepHooks(rl) {
-	section("Step 6: Lifecycle Hooks");
+	section("Step 7: Lifecycle Hooks");
 	line$1();
 	line$1("  PAI hooks fire on session stop and context compaction to save state,");
 	line$1("  update notes, and display live statusline information.");
@@ -4597,6 +4707,7 @@ async function stepHooks(rl) {
 	}
 	const hooksDir = getHooksDir();
 	const statuslineSrc = getStatuslineScript();
+	const tabColorSrc = getTabColorScript();
 	const claudeDir = join(homedir(), ".claude");
 	const hooksTarget = join(claudeDir, "Hooks");
 	if (!existsSync(hooksTarget)) mkdirSync(hooksTarget, { recursive: true });
@@ -4623,38 +4734,240 @@ async function stepHooks(rl) {
 	installFile(join(hooksDir, "session-stop.sh"), join(hooksTarget, "pai-session-stop.sh"), "pai-session-stop.sh");
 	if (statuslineSrc) installFile(statuslineSrc, join(claudeDir, "statusline-command.sh"), "statusline-command.sh");
 	else console.log(c.warn("  statusline-command.sh not found — skipping statusline."));
+	if (tabColorSrc) installFile(tabColorSrc, join(claudeDir, "tab-color-command.sh"), "tab-color-command.sh");
+	else console.log(c.warn("  tab-color-command.sh not found — skipping tab color."));
 	return anyInstalled;
 }
 /**
-* Step 7: Patch ~/.claude/settings.json with PAI hooks, env vars, and statusline
+* Step 7b: Install compiled TypeScript hooks (.mjs) to ~/.claude/Hooks/
+*
+* Finds all .mjs files in the package's dist/hooks/ directory and copies them
+* to ~/.claude/Hooks/. Content is compared before copying — identical files are
+* skipped for idempotent re-runs. Each installed file gets chmod 755.
 */
-async function stepSettings(rl) {
-	section("Step 7: Settings Patch");
+async function stepTsHooks(rl) {
+	section("Step 7b: TypeScript Hooks Installation");
+	line$1();
+	line$1("  PAI ships 14 compiled TypeScript hooks (.mjs) that fire on session events,");
+	line$1("  tool use, and context compaction to capture context and update notes.");
+	line$1();
+	if (!await promptYesNo(rl, "Install PAI TypeScript hooks to ~/.claude/Hooks/?", true)) {
+		console.log(c.dim("  Skipping TypeScript hooks installation."));
+		return false;
+	}
+	const distHooksDir = getDistHooksDir();
+	if (!existsSync(distHooksDir)) {
+		console.log(c.warn(`  dist/hooks/ directory not found at: ${distHooksDir}`));
+		console.log(c.dim("  Build the package first: bun run build"));
+		return false;
+	}
+	const hooksTarget = join(join(homedir(), ".claude"), "Hooks");
+	if (!existsSync(hooksTarget)) mkdirSync(hooksTarget, { recursive: true });
+	let allFiles;
+	try {
+		allFiles = readdirSync(distHooksDir).filter((f) => f.endsWith(".mjs"));
+	} catch (e) {
+		console.log(c.warn(`  Could not read dist/hooks/: ${e}`));
+		return false;
+	}
+	if (allFiles.length === 0) {
+		console.log(c.warn("  No .mjs files found in dist/hooks/. Build first: bun run build"));
+		return false;
+	}
 	line$1();
-	line$1("  PAI will add env vars, hook registrations, and the statusline command");
+	let copiedCount = 0;
+	let skippedCount = 0;
+	for (const filename of allFiles) {
+		const src = join(distHooksDir, filename);
+		const dest = join(hooksTarget, filename);
+		const srcContent = readFileSync(src, "utf-8");
+		if (existsSync(dest)) {
+			if (srcContent === readFileSync(dest, "utf-8")) {
+				console.log(c.dim(`  Unchanged: ${filename}`));
+				skippedCount++;
+				continue;
+			}
+		}
+		copyFileSync(src, dest);
+		chmodSync(dest, 493);
+		console.log(c.ok(`Installed: ${filename}`));
+		copiedCount++;
+	}
+	line$1();
+	if (copiedCount > 0) console.log(c.ok(`${copiedCount} hook(s) installed, ${skippedCount} unchanged.`));
+	else console.log(c.dim(`  All ${skippedCount} hook(s) already up-to-date.`));
+	return copiedCount > 0;
+}
+/**
+* Step 8b: Prompt for the DA (Digital Assistant) name
+*
+* The name appears in tab titles, session notes, and hook output.
+* Stored in env.DA via settings merge.
+*/
+async function stepDaName(rl) {
+	section("Step 8b: Assistant Name");
+	line$1();
+	line$1("  Choose a name for your AI assistant. This name appears in tab titles");
+	line$1("  and session notes when hooks are active.");
+	line$1();
+	const daName = await prompt(rl, chalk.bold("  Assistant name [PAI]: ")) || "PAI";
+	line$1();
+	console.log(c.ok(`Assistant name set to: ${daName}`));
+	return daName;
+}
+/**
+* Step 8: Patch ~/.claude/settings.json with PAI hooks, env vars, permissions, and flags
+*
+* Registers all 17 hook entries across 8 event types, adds env vars including DA name,
+* sets the statusline command, adds tool permissions (allow/deny), and sets flags.
+*/
+async function stepSettings(rl, daName) {
+	section("Step 8: Settings Patch");
+	line$1();
+	line$1("  PAI will add env vars, all hook registrations, permissions, and flags");
 	line$1("  to ~/.claude/settings.json. Existing values are never overwritten.");
 	line$1();
-	if (!await promptYesNo(rl, "Patch ~/.claude/settings.json with PAI hooks, env vars, and statusline?", true)) {
+	if (!await promptYesNo(rl, "Patch ~/.claude/settings.json with PAI hooks, env vars, and settings?", true)) {
 		console.log(c.dim("  Skipping settings patch."));
 		return false;
 	}
 	const result = mergeSettings({
 		env: {
 			PAI_DIR: join(homedir(), ".claude"),
-			CLAUDE_AUTOCOMPACT_PCT_OVERRIDE: "80"
+			CLAUDE_AUTOCOMPACT_PCT_OVERRIDE: "80",
+			CLAUDE_CODE_MAX_OUTPUT_TOKENS: "64000",
+			DA: daName
 		},
-		hooks: [{
-			hookType: "PreCompact",
-			matcher: "",
-			command: "${PAI_DIR}/Hooks/pai-pre-compact.sh"
-		}, {
-			hookType: "Stop",
-			command: "${PAI_DIR}/Hooks/pai-session-stop.sh"
-		}],
+		hooks: [
+			{
+				hookType: "SessionStart",
+				command: "${PAI_DIR}/Hooks/load-core-context.mjs"
+			},
+			{
+				hookType: "SessionStart",
+				command: "${PAI_DIR}/Hooks/load-project-context.mjs"
+			},
+			{
+				hookType: "SessionStart",
+				command: "${PAI_DIR}/Hooks/initialize-session.mjs"
+			},
+			{
+				hookType: "SessionStart",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type SessionStart"
+			},
+			{
+				hookType: "UserPromptSubmit",
+				command: "${PAI_DIR}/Hooks/cleanup-session-files.mjs"
+			},
+			{
+				hookType: "UserPromptSubmit",
+				command: "${PAI_DIR}/Hooks/update-tab-titles.mjs"
+			},
+			{
+				hookType: "UserPromptSubmit",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type UserPromptSubmit"
+			},
+			{
+				hookType: "PreToolUse",
+				matcher: "Bash",
+				command: "${PAI_DIR}/Hooks/security-validator.mjs"
+			},
+			{
+				hookType: "PreToolUse",
+				matcher: "*",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type PreToolUse"
+			},
+			{
+				hookType: "PostToolUse",
+				matcher: "TodoWrite",
+				command: "${PAI_DIR}/Hooks/sync-todo-to-md.mjs"
+			},
+			{
+				hookType: "PostToolUse",
+				matcher: "*",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type PostToolUse"
+			},
+			{
+				hookType: "PostToolUse",
+				matcher: "*",
+				command: "${PAI_DIR}/Hooks/capture-tool-output.mjs"
+			},
+			{
+				hookType: "PostToolUse",
+				matcher: "*",
+				command: "${PAI_DIR}/Hooks/update-tab-on-action.mjs"
+			},
+			{
+				hookType: "Stop",
+				command: "${PAI_DIR}/Hooks/stop-hook.mjs"
+			},
+			{
+				hookType: "Stop",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type Stop"
+			},
+			{
+				hookType: "Stop",
+				command: "${PAI_DIR}/Hooks/pai-session-stop.sh"
+			},
+			{
+				hookType: "SubagentStop",
+				command: "${PAI_DIR}/Hooks/subagent-stop-hook.mjs"
+			},
+			{
+				hookType: "SubagentStop",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type SubagentStop"
+			},
+			{
+				hookType: "SessionEnd",
+				command: "${PAI_DIR}/Hooks/capture-session-summary.mjs"
+			},
+			{
+				hookType: "SessionEnd",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type SessionEnd"
+			},
+			{
+				hookType: "PreCompact",
+				command: "${PAI_DIR}/Hooks/context-compression-hook.mjs"
+			},
+			{
+				hookType: "PreCompact",
+				command: "${PAI_DIR}/Hooks/capture-all-events.mjs --event-type PreCompact"
+			},
+			{
+				hookType: "PreCompact",
+				matcher: "",
+				command: "${PAI_DIR}/Hooks/pai-pre-compact.sh"
+			}
+		],
 		statusLine: {
 			type: "command",
 			command: "bash ${PAI_DIR}/statusline-command.sh"
-		}
+		},
+		permissions: {
+			allow: [
+				"Bash",
+				"Read",
+				"Write",
+				"Edit",
+				"Glob",
+				"Grep",
+				"WebFetch",
+				"WebSearch",
+				"NotebookEdit",
+				"TodoWrite",
+				"ExitPlanMode",
+				"mcp__pai"
+			],
+			deny: [
+				"Bash(rm -rf /)",
+				"Bash(rm -rf /*)",
+				"Bash(rm -rf ~)",
+				"Bash(rm -rf $HOME)",
+				"Bash(sudo rm -rf /)",
+				"Bash(sudo rm -rf /*)"
+			]
+		},
+		flags: { enableAllProjectMcpServers: true }
 	});
 	line$1();
 	for (const r of result.report) console.log(r);
@@ -4662,10 +4975,10 @@ async function stepSettings(rl) {
 	return result.changed;
 }
 /**
-* Step 8: Daemon install (launchd plist)
+* Step 9: Daemon install (launchd plist)
 */
 async function stepDaemon(rl) {
-	section("Step 8: Daemon Install");
+	section("Step 9: Daemon Install");
 	line$1();
 	line$1("  The PAI daemon indexes your projects every 5 minutes in the background.");
 	line$1();
@@ -4689,10 +5002,10 @@ async function stepDaemon(rl) {
 	return true;
 }
 /**
-* Step 9: MCP registration in ~/.claude.json
+* Step 10: MCP registration in ~/.claude.json
 */
 async function stepMcp(rl) {
-	section("Step 9: MCP Registration");
+	section("Step 10: MCP Registration");
 	line$1();
 	line$1("  Registering the PAI MCP server lets Claude Code call PAI tools directly.");
 	line$1();
@@ -4719,10 +5032,10 @@ async function stepMcp(rl) {
 	return true;
 }
 /**
-* Step 10: Directory scanning configuration
+* Step 11: Directory scanning configuration
 */
 async function stepDirectories(rl) {
-	section("Step 10: Directories to Index");
+	section("Step 11: Directories to Index");
 	line$1();
 	line$1("  PAI indexes files in your registered projects. You can register projects");
 	line$1("  individually with `pai project add <path>`, or let the registry scanner");
@@ -4749,10 +5062,10 @@ async function stepDirectories(rl) {
 	stepDirectories._runScan = runScan;
 }
 /**
-* Step 11: Initial index
+* Step 12: Initial index
 */
 async function stepInitialIndex(rl) {
-	section("Step 11: Initial Index");
+	section("Step 12: Initial Index");
 	line$1();
 	line$1("  Indexing scans your registered projects and builds the search index.");
 	line$1("  The daemon runs indexing automatically every 5 minutes once started.");
@@ -4794,9 +5107,9 @@ async function stepInitialIndex(rl) {
 	}
 }
 /**
-* Step 12: Summary and next steps
+* Step 13: Summary and next steps
 */
-function stepSummary(configUpdates, claudeMdGenerated, paiSkillInstalled, hooksInstalled, settingsPatched, daemonInstalled, mcpRegistered) {
+function stepSummary(configUpdates, claudeMdGenerated, paiSkillInstalled, aiSteeringRulesInstalled, hooksInstalled, tsHooksInstalled, settingsPatched, daName, daemonInstalled, mcpRegistered) {
 	section("Setup Complete");
 	line$1();
 	console.log(c.ok("PAI Knowledge OS is configured!"));
@@ -4809,8 +5122,11 @@ function stepSummary(configUpdates, claudeMdGenerated, paiSkillInstalled, hooksI
 	console.log(chalk.dim("  Embedding model:  ") + chalk.cyan(model && model !== "none" ? model : "(none — keyword search only)"));
 	console.log(chalk.dim("  CLAUDE.md:        ") + chalk.cyan(claudeMdGenerated ? "~/.claude/CLAUDE.md (generated)" : "(unchanged)"));
 	console.log(chalk.dim("  PAI skill:        ") + chalk.cyan(paiSkillInstalled ? "~/.claude/skills/PAI/SKILL.md (installed)" : "(unchanged)"));
-	console.log(chalk.dim("  Hooks:            ") + chalk.cyan(hooksInstalled ? "pai-pre-compact.sh, pai-session-stop.sh (installed)" : "(unchanged)"));
-	console.log(chalk.dim("  Settings:         ") + chalk.cyan(settingsPatched ? "env vars, hooks, statusline (patched)" : "(unchanged)"));
+	console.log(chalk.dim("  Steering rules:   ") + chalk.cyan(aiSteeringRulesInstalled ? "~/.claude/skills/PAI/AI-STEERING-RULES.md (installed)" : "(unchanged)"));
+	console.log(chalk.dim("  Hooks (shell):    ") + chalk.cyan(hooksInstalled ? "pai-pre-compact.sh, pai-session-stop.sh (installed)" : "(unchanged)"));
+	console.log(chalk.dim("  Hooks (TS):       ") + chalk.cyan(tsHooksInstalled ? "14 .mjs hooks installed to ~/.claude/Hooks/" : "(unchanged)"));
+	console.log(chalk.dim("  Assistant name:   ") + chalk.cyan(daName));
+	console.log(chalk.dim("  Settings:         ") + chalk.cyan(settingsPatched ? "env vars, hooks, permissions, flags (patched)" : "(unchanged)"));
 	console.log(chalk.dim("  Daemon:           ") + chalk.cyan(daemonInstalled ? "com.pai.pai-daemon (installed)" : "(unchanged)"));
 	console.log(chalk.dim("  MCP:              ") + chalk.cyan(mcpRegistered ? "registered in ~/.claude.json" : "(unchanged)"));
 	line$1();
@@ -4866,8 +5182,11 @@ async function runSetup() {
 		const embeddingConfig = await stepEmbedding(rl);
 		const claudeMdGenerated = await stepClaudeMd(rl);
 		const paiSkillInstalled = await stepPaiSkill(rl);
+		const aiSteeringRulesInstalled = await stepAiSteeringRules(rl);
 		const hooksInstalled = await stepHooks(rl);
-		const settingsPatched = await stepSettings(rl);
+		const tsHooksInstalled = await stepTsHooks(rl);
+		const daName = await stepDaName(rl);
+		const settingsPatched = await stepSettings(rl, daName);
 		const daemonInstalled = await stepDaemon(rl);
 		const mcpRegistered = await stepMcp(rl);
 		await stepDirectories(rl);
@@ -4879,7 +5198,7 @@ async function runSetup() {
 		line$1();
 		console.log(c.ok("Configuration saved."));
 		await stepInitialIndex(rl);
-		stepSummary(allUpdates, claudeMdGenerated, paiSkillInstalled, hooksInstalled, settingsPatched, daemonInstalled, mcpRegistered);
+		stepSummary(allUpdates, claudeMdGenerated, paiSkillInstalled, aiSteeringRulesInstalled, hooksInstalled, tsHooksInstalled, settingsPatched, daName, daemonInstalled, mcpRegistered);
 	} finally {
 		rl.close();
 	}