pk-agent 0.7.3 → 0.7.4

package/dist/index.js

@@ -5577,7 +5577,7 @@ var init_parser2 = __esm(() => {
 });
 
 // package.json
-var version = "0.7.3";
+var version = "0.7.4";
 var init_package = () => {};
 
 // src/storage/storage.ts
@@ -11086,7 +11086,7 @@ var init_session = __esm(() => {
 });
 
 // src/index.ts
-import { Command as Command14 } from "commander";
+import { Command as Command15 } from "commander";
 import chalk10 from "chalk";
 
 // src/cli/auth.ts
@@ -20301,14 +20301,305 @@ function createTestCommand() {
   return cmd;
 }
 
+// src/cli/tray.ts
+import { Command as Command14 } from "commander";
+import matter5 from "gray-matter";
+import YAML2 from "yaml";
+import { homedir as homedir21 } from "os";
+import { basename as basename8, join as join24, resolve as resolve25 } from "path";
+import { existsSync as existsSync17 } from "fs";
+import { mkdir as mkdir7, readFile as readFile11, readdir, writeFile as writeFile8 } from "fs/promises";
+import { spawn as spawn7 } from "child_process";
+var TOGGLEABLE_ABILITIES = [
+  "filesystem",
+  "bash",
+  "codexSdk",
+  "agentsSdk",
+  "captureWindow",
+  "fullstack",
+  "pkAgentBuilder"
+];
+function getGlobalAgentsDir() {
+  return join24(homedir21(), ".pk-agent", "agents");
+}
+function getAbilityListFromFrontmatter(frontmatter) {
+  const abilities = [];
+  const tools = frontmatter.tools ?? {};
+  for (const ability of TOGGLEABLE_ABILITIES) {
+    if (tools[ability] !== undefined) {
+      abilities.push(ability);
+    }
+  }
+  if (frontmatter.subagents !== undefined) {
+    abilities.push("subagents");
+  }
+  if (frontmatter.mcpServers !== undefined) {
+    abilities.push("mcpServers");
+  }
+  if (frontmatter.schedule !== undefined) {
+    abilities.push("schedule");
+  }
+  return abilities;
+}
+function normalizeEnabledToolValue(ability) {
+  if (ability === "filesystem") {
+    return [{ paths: ["${root}"], permissions: ["read"] }];
+  }
+  if (ability === "bash") {
+    return { commands: ["echo *"] };
+  }
+  return true;
+}
+function serializePkAgent2(frontmatter, body) {
+  const yaml = YAML2.stringify(frontmatter, {
+    indent: 2,
+    lineWidth: 0
+  }).trimEnd();
+  const normalizedBody = body.trimStart();
+  if (normalizedBody.length === 0) {
+    return `---
+${yaml}
+---
+`;
+  }
+  return `---
+${yaml}
+---
+
+${normalizedBody}
+`;
+}
+async function discoverGlobalAgents() {
+  const agentsDir = getGlobalAgentsDir();
+  const results = [];
+  if (!existsSync17(agentsDir)) {
+    return results;
+  }
+  const entries = await readdir(agentsDir, { withFileTypes: true });
+  for (const entry of entries) {
+    if (!entry.isFile() || !entry.name.endsWith(".pk-agent")) {
+      continue;
+    }
+    const filePath = join24(agentsDir, entry.name);
+    try {
+      const raw = await readFile11(filePath, "utf-8");
+      const parsed = matter5(raw);
+      const frontmatter = parsed.data;
+      results.push({
+        name: basename8(entry.name, ".pk-agent"),
+        file: entry.name,
+        path: filePath,
+        abilities: getAbilityListFromFrontmatter(frontmatter)
+      });
+    } catch {}
+  }
+  results.sort((a, b) => a.name.localeCompare(b.name));
+  return results;
+}
+async function toggleGlobalAgentAbility(agentName, ability) {
+  if (!TOGGLEABLE_ABILITIES.includes(ability)) {
+    throw new Error(`Unsupported ability '${ability}'. Supported: ${TOGGLEABLE_ABILITIES.join(", ")}`);
+  }
+  const agentsDir = getGlobalAgentsDir();
+  const filePath = resolve25(agentsDir, `${agentName}.pk-agent`);
+  if (!existsSync17(filePath)) {
+    throw new Error(`Global agent not found: ${agentName}`);
+  }
+  const raw = await readFile11(filePath, "utf-8");
+  const parsed = matter5(raw);
+  const frontmatter = parsed.data;
+  const tools = { ...frontmatter.tools ?? {} };
+  const typedAbility = ability;
+  let enabled = false;
+  if (tools[typedAbility] !== undefined) {
+    delete tools[typedAbility];
+    enabled = false;
+  } else {
+    tools[typedAbility] = normalizeEnabledToolValue(typedAbility);
+    enabled = true;
+  }
+  if (Object.keys(tools).length === 0) {
+    delete frontmatter.tools;
+  } else {
+    frontmatter.tools = tools;
+  }
+  const updated = serializePkAgent2(frontmatter, parsed.content);
+  await writeFile8(filePath, updated, "utf-8");
+  return { enabled };
+}
+function buildTrayScript(cliExecPath, cliScriptPath) {
+  const safeCliExecPath = cliExecPath.replace(/'/g, "''");
+  const safeCliScriptPath = (cliScriptPath ?? "").replace(/'/g, "''");
+  return `
+Add-Type -AssemblyName System.Windows.Forms
+Add-Type -AssemblyName System.Drawing
+
+$ErrorActionPreference = 'Stop'
+$cliExec = '${safeCliExecPath}'
+$cliScript = '${safeCliScriptPath}'
+
+function Invoke-PkAgent {
+  param([string[]]$Args)
+  if ([string]::IsNullOrWhiteSpace($cliScript)) {
+    & $cliExec @Args
+  } else {
+    & $cliExec $cliScript @Args
+  }
+}
+
+function Build-Menu {
+  param([System.Windows.Forms.ContextMenuStrip]$Menu, [System.Windows.Forms.NotifyIcon]$Notify)
+
+  $Menu.Items.Clear()
+  $header = New-Object System.Windows.Forms.ToolStripMenuItem
+  $header.Text = 'PK-Agent Tray'
+  $header.Enabled = $false
+  $Menu.Items.Add($header) | Out-Null
+  $Menu.Items.Add('-') | Out-Null
+
+  try {
+    $json = Invoke-PkAgent @('tray', '--list-global-json')
+    $data = $json | ConvertFrom-Json
+    if ($data.agents.Count -eq 0) {
+      $empty = New-Object System.Windows.Forms.ToolStripMenuItem
+      $empty.Text = 'No global agents found'
+      $empty.Enabled = $false
+      $Menu.Items.Add($empty) | Out-Null
+    } else {
+      foreach ($agent in $data.agents) {
+        $agentItem = New-Object System.Windows.Forms.ToolStripMenuItem
+        $agentItem.Text = $agent.name
+        $agentItem.ToolTipText = $agent.path
+        foreach ($ability in @('filesystem','bash','codexSdk','agentsSdk','captureWindow','fullstack','pkAgentBuilder')) {
+          $agentName = [string]$agent.name
+          $abilityName = [string]$ability
+          $abilityItem = New-Object System.Windows.Forms.ToolStripMenuItem
+          $abilityItem.Text = $abilityName
+          $abilityItem.Checked = $agent.abilities -contains $abilityName
+          $abilityItem.Add_Click(({
+            try {
+              Invoke-PkAgent @('tray', '--toggle-global-ability', $agentName, $abilityName) | Out-Null
+              Build-Menu -Menu $Menu -Notify $Notify
+            } catch {
+              [System.Windows.Forms.MessageBox]::Show($_.Exception.Message, 'pk-agent tray')
+            }
+          }).GetNewClosure())
+          $agentItem.DropDownItems.Add($abilityItem) | Out-Null
+        }
+
+        $details = New-Object System.Windows.Forms.ToolStripMenuItem
+        $details.Enabled = $false
+        if ($agent.abilities.Count -gt 0) {
+          $details.Text = 'abilities: ' + ($agent.abilities -join ', ')
+        } else {
+          $details.Text = 'abilities: none'
+        }
+        $agentItem.DropDownItems.Add('-') | Out-Null
+        $agentItem.DropDownItems.Add($details) | Out-Null
+        $Menu.Items.Add($agentItem) | Out-Null
+      }
+    }
+  } catch {
+    $err = New-Object System.Windows.Forms.ToolStripMenuItem
+    $err.Text = 'Failed to load agents'
+    $err.Enabled = $false
+    $Menu.Items.Add($err) | Out-Null
+  }
+
+  $Menu.Items.Add('-') | Out-Null
+
+  $refresh = New-Object System.Windows.Forms.ToolStripMenuItem
+  $refresh.Text = 'Refresh'
+  $refresh.Add_Click({ Build-Menu -Menu $Menu -Notify $Notify })
+  $Menu.Items.Add($refresh) | Out-Null
+
+  $openAgents = New-Object System.Windows.Forms.ToolStripMenuItem
+  $openAgents.Text = 'Open global agents folder'
+  $openAgents.Add_Click({
+    $dir = Join-Path $HOME '.pk-agent\\agents'
+    New-Item -ItemType Directory -Force -Path $dir | Out-Null
+    Start-Process explorer.exe $dir
+  })
+  $Menu.Items.Add($openAgents) | Out-Null
+
+  $quit = New-Object System.Windows.Forms.ToolStripMenuItem
+  $quit.Text = 'Quit'
+  $quit.Add_Click({
+    $Notify.Visible = $false
+    [System.Windows.Forms.Application]::Exit()
+  })
+  $Menu.Items.Add($quit) | Out-Null
+}
+
+$menu = New-Object System.Windows.Forms.ContextMenuStrip
+$notify = New-Object System.Windows.Forms.NotifyIcon
+$notify.Icon = [System.Drawing.SystemIcons]::Application
+$notify.Visible = $true
+$notify.Text = 'PK-Agent'
+$notify.ContextMenuStrip = $menu
+
+Build-Menu -Menu $menu -Notify $notify
+
+[System.Windows.Forms.Application]::Run()
+`;
+}
+async function launchWindowsTray() {
+  const agentsDir = getGlobalAgentsDir();
+  await mkdir7(agentsDir, { recursive: true });
+  const cliExecPath = process.execPath;
+  const cliScriptPath = process.argv[1];
+  const script = buildTrayScript(cliExecPath, cliScriptPath);
+  const scriptPath = join24(homedir21(), ".pk-agent", "cache", "pk-agent-tray.ps1");
+  await mkdir7(join24(homedir21(), ".pk-agent", "cache"), { recursive: true });
+  await writeFile8(scriptPath, script, "utf-8");
+  spawn7("powershell", [
+    "-NoProfile",
+    "-ExecutionPolicy",
+    "Bypass",
+    "-File",
+    scriptPath
+  ], {
+    detached: true,
+    stdio: "ignore",
+    windowsHide: true
+  }).unref();
+}
+function createTrayCommand() {
+  return new Command14("tray").description("Launch Windows system tray manager for global pk-agent agents").option("--list-global-json", "List global agents and abilities as JSON").option("--toggle-global-ability <values...>", "Toggle ability for a global agent: <agent> <ability>").action(async (options) => {
+    if (options.listGlobalJson) {
+      const agents = await discoverGlobalAgents();
+      process.stdout.write(`${JSON.stringify({ agents }, null, 2)}
+`);
+      return;
+    }
+    const toggleArgs = options.toggleGlobalAbility;
+    if (toggleArgs && toggleArgs.length > 0) {
+      if (toggleArgs.length !== 2) {
+        throw new Error("--toggle-global-ability requires exactly 2 values: <agent> <ability>");
+      }
+      const [agentName, ability] = toggleArgs;
+      const result = await toggleGlobalAgentAbility(agentName, ability);
+      process.stdout.write(`${JSON.stringify({ agent: agentName, ability, enabled: result.enabled })}
+`);
+      return;
+    }
+    if (process.platform !== "win32") {
+      throw new Error("Systray is currently supported on Windows only.");
+    }
+    await launchWindowsTray();
+    process.stdout.write(`[OK] PK-Agent tray launched.
+`);
+  });
+}
+
 // src/cli/worker.ts
 init_parser2();
 init_mcp();
 init_runner();
 init_logger();
 init_redaction();
-import { resolve as resolve25, dirname as dirname17 } from "path";
-import { existsSync as existsSync17 } from "fs";
+import { resolve as resolve26, dirname as dirname17 } from "path";
+import { existsSync as existsSync18 } from "fs";
 import * as dotenv5 from "dotenv";
 function toSafeChunk(chunk) {
   if (chunk.type === "tool-call") {
@@ -20341,8 +20632,8 @@ async function executeAgent(req, emit) {
   const startTime = Date.now();
   let mcp = [];
   try {
-    const agentPath = resolve25(req.projectRoot, req.agentPath);
-    if (!existsSync17(agentPath)) {
+    const agentPath = resolve26(req.projectRoot, req.agentPath);
+    if (!existsSync18(agentPath)) {
       return {
         id: req.id,
         type: "done",
@@ -20350,11 +20641,11 @@ async function executeAgent(req, emit) {
         error: { code: "AGENT_NOT_FOUND", message: `Agent file not found: ${req.agentPath}` }
       };
     }
-    const envFile = resolve25(req.projectRoot, ".env");
-    const envLocalFile = resolve25(req.projectRoot, ".env.local");
-    if (existsSync17(envLocalFile)) {
+    const envFile = resolve26(req.projectRoot, ".env");
+    const envLocalFile = resolve26(req.projectRoot, ".env.local");
+    if (existsSync18(envLocalFile)) {
       dotenv5.config({ path: envLocalFile });
-    } else if (existsSync17(envFile)) {
+    } else if (existsSync18(envFile)) {
       dotenv5.config({ path: envFile });
     }
     try {
@@ -20557,7 +20848,7 @@ async function runInternalWorker() {
 init_cli_helpers();
 init_pk_agent_dirs();
 var version2 = getVersionString();
-var program = new Command14;
+var program = new Command15;
 try {
   ensurePkAgentUserDirs();
   ensurePkAgentProjectDirs(process.cwd());
@@ -20607,6 +20898,7 @@ program.addCommand(createRunCommand());
 program.addCommand(createLintCommand());
 program.addCommand(createConfigCommand());
 program.addCommand(createTestCommand());
+program.addCommand(createTrayCommand());
 program.addCommand(createBenchmarkCommand(), { hidden: true });
 if (process.argv[2] === "--internal-worker") {
   runInternalWorker();
@@ -20617,5 +20909,5 @@ if (process.argv[2] === "--internal-worker") {
   }
 }
 
-//# debugId=7A021AEF99F4959564756E2164756E21
+//# debugId=18D15260DA895C9E64756E2164756E21
 //# sourceMappingURL=index.js.map