baton-issue-tracker 1.13.4 → 1.14.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "baton-issue-tracker",
-  "version": "1.13.4",
+  "version": "1.14.0",
   "description": "A CLI issue tracker for AI agents",
   "type": "module",
   "bin": {

package/source/cli.js

@@ -31,6 +31,7 @@ import { run as runSubmit } from './commands/submit.js';
 import { run as runUnassign } from './commands/unassign.js';
 import { run as runUnclaim } from './commands/unclaim.js';
 import { run as runClaim } from './commands/claim.js';
+import { run as runAssign} from './commands/assign.js';
 
 import { authenticateContext } from './services/authService.js';
 
@@ -163,6 +164,7 @@ async function main() {
     submit: () => runSubmit(args),
     unassign: () => runUnassign(args),
     unclaim: () => runUnclaim(args),
+    assign: () => runAssign(args),
   };
 
   const handler = handlers[command];

package/source/commands/assign.js

@@ -0,0 +1,66 @@
+// commands/assign.js
+// assign command for the issue tracker
+// usage: baton assign <id> <agent-name> [--json]
+// AI was consulted for a portion of this file
+
+import { getAgentByName } from '../services/agentsService.js';
+import { assignIssue } from '../services/issuesService.js';
+import { getCurrentActor } from '../services/authService.js';
+import { hasFlag, renderOutput } from '../util.js';
+
+const DECIMAL_BASE = 10;
+
+/**
+ * Runs the assign command to associate an issue with a registered agent.
+ * Usage: baton assign <id> <agent-name> [--json]
+ * @param {string[]} args - The command line arguments
+ * @returns {Promise<number>} The exit code: 0 is success, 1 is error.
+ */
+export async function run(args = []) {
+  // Check if the user requested JSON output formatting via the --json flag
+  const isJson = hasFlag(args, '--json');
+  
+  // Strip out any flags starting with '-' to isolate the raw [issueIdStr, agentName] positional arguments
+  const [issueIdStr, agentName] = args.filter(arg => !arg.startsWith('-'));
+
+  // Guard Clause: Ensure both the issue ID and the agent name were provided
+  if (!issueIdStr || !agentName) {
+    console.error('Usage Error: Missing arguments.\nUsage: baton assign <id> <agent-name> [--json]');
+    return 1;
+  }
+
+  // Convert the extracted issue ID string into a safe decimal integer
+  const issueId = parseInt(issueIdStr, DECIMAL_BASE);
+  
+  // Guard Clause: If parsing failed (e.g., the user typed letters instead of an ID), reject it
+  if (isNaN(issueId)) {
+    console.error(`Error: Invalid issue ID "${issueIdStr}". Must be an integer.`);
+    return 1;
+  }
+
+  try {
+    // Look up the current operator. If null, error
+    const actor = getCurrentActor() || Object.assign(new Error('No authenticated context found.'), { isAuthErr: true });
+    // If that identifier property exists, it means the lookup failed; error
+    if (actor.isAuthErr) throw actor;
+
+    // Look up the target assignee by name. If not found, error
+    const target = getAgentByName(agentName) || Object.assign(new Error(`Agent/User "${agentName}" is not registered.`), { isTargetErr: true });
+    // If the lookup failed, throw the target error
+    if (target.isTargetErr) throw target;
+
+    // Execute the assignment in the database layer, passing the parsed ID, assignee ID, and operator ID
+    const issue = assignIssue(issueId, target.id, actor.id);
+
+    // Render the final output payload. If --json is on, it prints raw data. Otherwise, executes callback.
+    renderOutput(isJson, { status: 'success', issueId: issue.id, assignee: { id: target.id, name: target.name, type: target.type } }, () => {
+      console.log(`Success: Issue #${issueId} assigned to "${agentName}"`);
+    });
+    
+    return 0; // Command completed successfully
+  } catch (error) {
+    // Intercepts any thrown operational error (or database crash) and outputs a clean message
+    console.error(`Error: ${error.message}`);
+    return 1; // Return error exit code
+  }
+}

package/source/commands/init.js

@@ -14,6 +14,8 @@
 import { readFileSync, writeFileSync, existsSync } from 'node:fs';
 import { join } from 'node:path';
 import { initDB } from '../db/index.js';
+import os from 'node:os';
+import { registerAgent } from '../services/agentsService.js';
 import { Priority } from '../models/issue.js';
 import {
   createIssue,
@@ -156,6 +158,22 @@ export async function run(args = []) {
     clearAllIssues();
   }
 
+  // auto-register default deployment human user
+  let autoRegisterMessage = '';
+  try {
+    const activeName = process.env.BATON_AGENT || os.userInfo().username;
+    registerAgent(activeName, 'human');
+    autoRegisterMessage = `Auto-registered default human user: "${activeName}"`;
+  } catch (error){
+    if (!error.message?.includes('UNIQUE constraint failed')) {
+      autoRegisterMessage = `Warning: Could not auto-register default user: ${error.message}`;
+    }
+  }
+
+  if (autoRegisterMessage && !isJson) {
+    console.log(autoRegisterMessage);
+  }
+
   const templatePath = join('docs', 'BATON_AGENT_RULES.md');
 
   let rulesContent = '';

package/source/services/issuesService.js

@@ -232,6 +232,7 @@ export function updateIssue(id, oldIssue, { title, description, tokenLimit, stat
     updates.status = toUpdate || status;
   }
 
+
   // Normalize priority argument
   if (priority !== undefined) {
     const priorityValues = Object.values(Priority);
@@ -276,6 +277,30 @@ export function unassignIssue(issueId){
   return getIssue(issueId);
 }
 
+/**
+ * Assigns an issue to a specific registered agent or human.
+ * Logs an edit event.
+ * @param {number} issueId 
+ * @param {number} assigneeId 
+ * @returns {Issue} - the issue that matches the ID
+ */
+export function assignIssue(issueId, assigneeId) {
+  const db = getDB();
+  
+  // Verify the issue exists first (will throw if not found)
+  findById(db, issueId);
+  
+  db.update(issuesTable)
+    .set({ assigneeId: assigneeId })
+    .where(eq(issuesTable.id, issueId))
+    .run();
+
+  // Pass actorId directly instead of hacking the global module variable
+  logActivity(db, issueId, Action.EDIT, `Issue #${issueId} was assigned.`);
+  
+  return getIssue(issueId);
+}
+
 /**
  * Change the status of an issue from in-review to closed
  * Logs a closed event.