checkbox-cli 2.0.0 → 2.0.1

package/dist/checkbox.js

@@ -20,7 +20,7 @@ import { join } from 'path';
 // ── Config ──────────────────────────────────────────────────────────
 const CONFIG_DIR = join(homedir(), '.checkbox');
 const CONFIG_FILE = join(CONFIG_DIR, 'config.json');
-const VERSION = '2.0.0';
+const VERSION = '2.0.1';
 function loadConfig() {
     try {
         if (!existsSync(CONFIG_FILE))
@@ -44,7 +44,6 @@ function saveConfig(cfg) {
     mkdirSync(CONFIG_DIR, { recursive: true });
     writeFileSync(CONFIG_FILE, JSON.stringify(cfg, null, 2));
 }
-// Allow env vars to override config file
 function resolveConfig() {
     const envKey = process.env.CHECKBOX_API_KEY || process.env.CHECKBOX_API_TOKEN;
     const fileCfg = loadConfig();
@@ -59,6 +58,20 @@ function resolveConfig() {
     return fileCfg;
 }
 // ── API ─────────────────────────────────────────────────────────────
+/** Extract a human-readable error message from any API error shape */
+function extractError(obj) {
+    if (!obj)
+        return 'Unknown error';
+    if (typeof obj === 'string')
+        return obj;
+    if (typeof obj.error === 'string')
+        return obj.error;
+    if (obj.error?.message)
+        return obj.error.message;
+    if (obj.message)
+        return obj.message;
+    return JSON.stringify(obj);
+}
 function api(cfg) {
     const headers = {
         'Content-Type': 'application/json',
@@ -70,22 +83,58 @@ function api(cfg) {
             headers,
             body: JSON.stringify(body),
         });
+        const json = await res.json().catch(() => null);
         if (!res.ok) {
-            const err = await res.json().catch(() => ({}));
-            throw new Error(err.error || `Request failed (${res.status})`);
+            throw new Error(extractError(json) || `Request failed (${res.status})`);
+        }
+        if (json && json.success === false) {
+            throw new Error(extractError(json) || 'Request failed');
         }
-        return res.json();
+        return json;
     }
     async function get(path) {
         const res = await fetch(`${cfg.apiUrl}${path}`, { headers });
+        const json = await res.json().catch(() => null);
         if (!res.ok) {
-            const err = await res.json().catch(() => ({}));
-            throw new Error(err.error || `Request failed (${res.status})`);
+            throw new Error(extractError(json) || `Request failed (${res.status})`);
         }
-        return res.json();
+        if (json && json.success === false) {
+            throw new Error(extractError(json) || 'Request failed');
+        }
+        return json;
     }
     return { post, get };
 }
+// ── Query helpers ───────────────────────────────────────────────────
+// The actual query types are: list_entities, get_entity, list_my_organizations,
+// get_task_states, workspace_graph, entity_discovery, schema_discovery,
+// permission_discovery, list_approvals.
+// There is NO list_plans or list_tasks — use list_entities with entity_type.
+async function listEntities(post, orgId, entityType) {
+    const res = await post('/api/v2/query', {
+        type: 'list_entities',
+        organization_id: orgId,
+        payload: { entity_type: entityType, organization_id: orgId },
+    });
+    // list_entities returns { success, data: [...], meta }
+    return Array.isArray(res.data) ? res.data : [];
+}
+async function searchEntities(post, orgId, search, limit = 20) {
+    // entity_discovery returns { success, data: { organization_id, entities: [...] } }
+    const res = await post('/api/v2/introspection/entities', {
+        organization_id: orgId,
+        search,
+        limit,
+    });
+    return res.data?.entities || [];
+}
+async function getWorkspaceNodes(post, orgId) {
+    // workspace returns { success, data: { organization_id, nodes: [...], edges: [...] } }
+    const res = await post('/api/v2/introspection/workspace', {
+        organization_id: orgId,
+    });
+    return res.data?.nodes || [];
+}
 // ── Formatters ──────────────────────────────────────────────────────
 function truncate(s, max) {
     return s.length > max ? s.slice(0, max - 1) + '\u2026' : s;
@@ -135,16 +184,13 @@ async function runSetup(existingConfig) {
         orgId: '',
         orgName: '',
     };
-    // Test the key & fetch organizations
     const s = p.spinner();
     s.start('Connecting to Checkbox...');
     let orgs = [];
     try {
         const { post } = api(cfg);
         const res = await post('/api/v2/query', { type: 'list_my_organizations', payload: {} });
-        if (!res.success || !res.data)
-            throw new Error(res.error?.message || 'Failed to fetch organizations');
-        orgs = res.data;
+        orgs = Array.isArray(res.data) ? res.data : [];
         s.stop(`Connected! Found ${orgs.length} workspace${orgs.length === 1 ? '' : 's'}`);
     }
     catch (err) {
@@ -154,7 +200,6 @@ async function runSetup(existingConfig) {
         p.outro(pc.dim('Try again with a valid key.'));
         process.exit(1);
     }
-    // Pick workspace
     if (orgs.length === 0) {
         p.log.warn('No workspaces found. Create one at checkbox.my first.');
         saveConfig(cfg);
@@ -193,7 +238,7 @@ async function runSetup(existingConfig) {
 }
 // ── Interactive Menu ────────────────────────────────────────────────
 async function interactiveMenu(cfg) {
-    const { post, get } = api(cfg);
+    const { post } = api(cfg);
     console.log();
     p.intro(pc.bgCyan(pc.black(' Checkbox ')) +
         pc.dim(` v${VERSION}`) +
@@ -217,20 +262,13 @@ async function interactiveMenu(cfg) {
         return;
     }
     switch (action) {
-        case 'plans':
-            return menuPlans(cfg, post);
-        case 'tasks':
-            return menuTasks(cfg, post);
-        case 'status':
-            return menuStatus(cfg, post);
-        case 'search':
-            return menuSearch(cfg, post);
-        case 'complete':
-            return menuComplete(cfg, post);
-        case 'create-task':
-            return menuCreateTask(cfg, post);
-        case 'switch':
-            return menuSwitch(cfg, post);
+        case 'plans': return menuPlans(cfg, post);
+        case 'tasks': return menuTasks(cfg, post);
+        case 'status': return menuStatus(cfg, post);
+        case 'search': return menuSearch(cfg, post);
+        case 'complete': return menuComplete(cfg, post);
+        case 'create-task': return menuCreateTask(cfg, post);
+        case 'switch': return menuSwitch(cfg, post);
         case 'setup':
             await runSetup(cfg);
             return;
@@ -241,27 +279,19 @@ async function menuPlans(cfg, post) {
     const s = p.spinner();
     s.start('Loading plans...');
     try {
-        const res = await post('/api/v2/query', {
-            type: 'list_plans',
-            organization_id: cfg.orgId,
-            payload: { organization_id: cfg.orgId },
-        });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const plans = res.data || [];
+        const plans = await listEntities(post, cfg.orgId, 'plan');
         s.stop(`Found ${plans.length} plan${plans.length === 1 ? '' : 's'}`);
         if (plans.length === 0) {
             p.log.info('No plans yet. Create one at checkbox.my');
-            p.outro('');
-            return;
+            return interactiveMenu(cfg);
         }
         const lines = plans.map((plan) => {
             const name = truncate(plan.name || 'Untitled', 40);
-            const created = plan.created_at ? formatDate(plan.created_at) : '';
+            const meta = plan.metadata || {};
+            const created = meta.created_at ? formatDate(meta.created_at) : '';
             return `${pc.cyan(pad(name, 42))} ${pc.dim(created)}`;
         });
         p.note(lines.join('\n'), `Plans in ${cfg.orgName}`);
-        // Offer drill-down
         const selected = await p.select({
             message: 'Select a plan to view its tasks',
             options: [
@@ -275,33 +305,36 @@ async function menuPlans(cfg, post) {
         if (p.isCancel(selected) || selected === '__back') {
             return interactiveMenu(cfg);
         }
-        return menuPlanTasks(cfg, post, selected, plans.find((p) => p.id === selected)?.name || '');
+        return menuPlanTasks(cfg, post, selected, plans.find((pl) => pl.id === selected)?.name || '');
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load plans');
+        return interactiveMenu(cfg);
     }
 }
 async function menuPlanTasks(cfg, post, planId, planName) {
     const s = p.spinner();
     s.start('Loading tasks...');
     try {
+        // Search entities by plan — use entity_discovery with search for plan-specific tasks
+        // Or query list_entities for tasks and filter by parent_id
         const res = await post('/api/v2/query', {
-            type: 'list_tasks',
+            type: 'list_entities',
             organization_id: cfg.orgId,
-            payload: { organization_id: cfg.orgId, plan_id: planId },
+            payload: { entity_type: 'task', organization_id: cfg.orgId },
         });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const tasks = res.data || [];
+        const allTasks = Array.isArray(res.data) ? res.data : [];
+        // Filter tasks that belong to this plan via parent_id chain
+        const tasks = allTasks.filter((t) => t.parent_id === planId || t.metadata?.plan_id === planId);
         s.stop(`${tasks.length} task${tasks.length === 1 ? '' : 's'} in ${pc.cyan(planName)}`);
         if (tasks.length === 0) {
-            p.log.info('No tasks in this plan.');
+            p.log.info('No tasks found for this plan.');
             return interactiveMenu(cfg);
         }
         const lines = tasks.map((t) => {
             const name = truncate(t.name || 'Untitled', 50);
-            const freq = t.frequency || '';
+            const freq = t.metadata?.frequency || '';
             return `  ${pc.cyan(pad(name, 52))} ${pc.dim(freq)}`;
         });
         p.note(lines.join('\n'), planName);
@@ -309,7 +342,7 @@ async function menuPlanTasks(cfg, post, planId, planName) {
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load tasks');
         return interactiveMenu(cfg);
     }
 }
@@ -318,14 +351,7 @@ async function menuTasks(cfg, post) {
     const s = p.spinner();
     s.start('Loading tasks...');
     try {
-        const res = await post('/api/v2/query', {
-            type: 'list_tasks',
-            organization_id: cfg.orgId,
-            payload: { organization_id: cfg.orgId },
-        });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const tasks = res.data || [];
+        const tasks = await listEntities(post, cfg.orgId, 'task');
         s.stop(`Found ${tasks.length} task${tasks.length === 1 ? '' : 's'}`);
         if (tasks.length === 0) {
             p.log.info('No tasks yet.');
@@ -337,7 +363,7 @@ async function menuTasks(cfg, post) {
                 ...tasks.slice(0, 25).map((t) => ({
                     value: t.id,
                     label: truncate(t.name || 'Untitled', 60),
-                    hint: t.frequency || '',
+                    hint: t.metadata?.frequency || '',
                 })),
                 ...(tasks.length > 25 ? [{ value: '__more', label: pc.dim(`...and ${tasks.length - 25} more`) }] : []),
                 { value: '__back', label: pc.dim('Back to menu') },
@@ -346,14 +372,13 @@ async function menuTasks(cfg, post) {
         if (p.isCancel(selected) || selected === '__back' || selected === '__more') {
             return interactiveMenu(cfg);
         }
-        // Show task detail
         const task = tasks.find((t) => t.id === selected);
         if (task) {
+            const meta = task.metadata || {};
             const lines = [
                 `${pad('Name', 14)} ${pc.cyan(task.name || 'Untitled')}`,
-                `${pad('Frequency', 14)} ${task.frequency || pc.dim('none')}`,
-                `${pad('Plan', 14)} ${task.plan_name || task.plan_id || pc.dim('none')}`,
-                `${pad('Assigned to', 14)} ${task.assigned_to || pc.dim('unassigned')}`,
+                `${pad('Frequency', 14)} ${meta.frequency || pc.dim('none')}`,
+                `${pad('Assigned to', 14)} ${meta.assigned_to || pc.dim('unassigned')}`,
                 `${pad('ID', 14)} ${pc.dim(task.id)}`,
             ];
             p.note(lines.join('\n'), task.name || 'Task');
@@ -372,7 +397,7 @@ async function menuTasks(cfg, post) {
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load tasks');
         return interactiveMenu(cfg);
     }
 }
@@ -381,30 +406,30 @@ async function menuStatus(cfg, post) {
     const s = p.spinner();
     s.start('Loading workspace...');
     try {
-        const res = await post('/api/v2/introspection/workspace', {
-            organization_id: cfg.orgId,
-        });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const ws = res.data || {};
+        const nodes = await getWorkspaceNodes(post, cfg.orgId);
         s.stop('Workspace loaded');
-        const counts = ws.entity_counts || {};
+        // Count nodes by type
+        const counts = {};
+        for (const node of nodes) {
+            const t = node.type || 'unknown';
+            counts[t] = (counts[t] || 0) + 1;
+        }
         const lines = [
             `${pad('Workspace', 16)} ${pc.cyan(cfg.orgName)}`,
             '',
-            `${pad('Plans', 16)} ${pc.bold(String(counts.plans || 0))}`,
-            `${pad('Tasks', 16)} ${pc.bold(String(counts.tasks || 0))}`,
-            `${pad('Requirements', 16)} ${pc.bold(String(counts.requirements || 0))}`,
-            `${pad('Goals', 16)} ${pc.bold(String(counts.goals || 0))}`,
-            `${pad('Documents', 16)} ${pc.bold(String(counts.documents || 0))}`,
-            `${pad('Tables', 16)} ${pc.bold(String(counts.tables || 0))}`,
-            `${pad('Members', 16)} ${pc.bold(String(counts.members || counts.users || 0))}`,
+            `${pad('Plans', 16)} ${pc.bold(String(counts.plan || 0))}`,
+            `${pad('Tasks', 16)} ${pc.bold(String(counts.task || 0))}`,
+            `${pad('Requirements', 16)} ${pc.bold(String(counts.requirement || 0))}`,
+            `${pad('Goals', 16)} ${pc.bold(String(counts.goal || 0))}`,
+            `${pad('Rules', 16)} ${pc.bold(String(counts.rule || 0))}`,
+            `${pad('Documents', 16)} ${pc.bold(String(counts.document || 0))}`,
+            `${pad('Tables', 16)} ${pc.bold(String(counts.table || 0))}`,
         ];
         p.note(lines.join('\n'), 'Workspace Overview');
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load workspace');
     }
     return interactiveMenu(cfg);
 }
@@ -423,21 +448,14 @@ async function menuSearch(cfg, post) {
     const s = p.spinner();
     s.start('Searching...');
     try {
-        const res = await post('/api/v2/introspection/entities', {
-            organization_id: cfg.orgId,
-            search: query,
-            limit: 20,
-        });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const entities = res.data || [];
+        const entities = await searchEntities(post, cfg.orgId, query);
         s.stop(`Found ${entities.length} result${entities.length === 1 ? '' : 's'}`);
         if (entities.length === 0) {
             p.log.info('No results found.');
         }
         else {
             const lines = entities.map((e) => {
-                const type = pad(e.type || 'unknown', 14);
+                const type = pad(e.entity_type || 'unknown', 14);
                 const name = truncate(e.name || 'Untitled', 50);
                 return `  ${pc.dim(type)} ${pc.cyan(name)}`;
             });
@@ -446,7 +464,7 @@ async function menuSearch(cfg, post) {
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Search failed');
     }
     return interactiveMenu(cfg);
 }
@@ -455,14 +473,7 @@ async function menuComplete(cfg, post) {
     const s = p.spinner();
     s.start('Loading tasks...');
     try {
-        const res = await post('/api/v2/query', {
-            type: 'list_tasks',
-            organization_id: cfg.orgId,
-            payload: { organization_id: cfg.orgId },
-        });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const tasks = res.data || [];
+        const tasks = await listEntities(post, cfg.orgId, 'task');
         s.stop(`${tasks.length} task${tasks.length === 1 ? '' : 's'} available`);
         if (tasks.length === 0) {
             p.log.info('No tasks to complete.');
@@ -474,7 +485,7 @@ async function menuComplete(cfg, post) {
                 ...tasks.slice(0, 30).map((t) => ({
                     value: t.id,
                     label: truncate(t.name || 'Untitled', 60),
-                    hint: t.frequency || '',
+                    hint: t.metadata?.frequency || '',
                 })),
                 { value: '__back', label: pc.dim('Back to menu') },
             ],
@@ -487,7 +498,7 @@ async function menuComplete(cfg, post) {
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load tasks');
     }
     return interactiveMenu(cfg);
 }
@@ -495,33 +506,25 @@ async function doComplete(cfg, post, taskId, taskName) {
     const s = p.spinner();
     s.start('Completing task...');
     try {
-        const res = await post('/api/v2/commands', {
+        await post('/api/v2/commands', {
             type: 'complete_task',
             organization_id: cfg.orgId,
             payload: { task_id: taskId },
         });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
         s.stop(pc.green(`Done! "${taskName}" completed`));
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not complete task');
     }
 }
 // ── Create Task ─────────────────────────────────────────────────────
 async function menuCreateTask(cfg, post) {
-    // First, get plans to attach task to
     const s = p.spinner();
     s.start('Loading plans...');
     let plans = [];
     try {
-        const res = await post('/api/v2/query', {
-            type: 'list_plans',
-            organization_id: cfg.orgId,
-            payload: { organization_id: cfg.orgId },
-        });
-        plans = res.data || [];
+        plans = await listEntities(post, cfg.orgId, 'plan');
         s.stop('');
     }
     catch {
@@ -577,18 +580,16 @@ async function menuCreateTask(cfg, post) {
         };
         if (result.planId)
             payload.plan_id = result.planId;
-        const res = await post('/api/v2/commands', {
+        await post('/api/v2/commands', {
             type: 'create_task',
             organization_id: cfg.orgId,
             payload,
         });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
         spinner.stop(pc.green(`Task "${result.name}" created!`));
     }
     catch (err) {
         spinner.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not create task');
     }
     return interactiveMenu(cfg);
 }
@@ -601,9 +602,7 @@ async function menuSwitch(cfg, post) {
             type: 'list_my_organizations',
             payload: {},
         });
-        if (!res.success)
-            throw new Error(res.error?.message || 'Failed');
-        const orgs = res.data || [];
+        const orgs = Array.isArray(res.data) ? res.data : [];
         s.stop(`${orgs.length} workspace${orgs.length === 1 ? '' : 's'}`);
         const selected = await p.select({
             message: 'Switch to workspace',
@@ -623,58 +622,40 @@ async function menuSwitch(cfg, post) {
     }
     catch (err) {
         s.stop(pc.red('Failed'));
-        p.log.error(err.message);
+        p.log.error(err.message || 'Could not load workspaces');
     }
     return interactiveMenu(cfg);
 }
-// ── Direct Commands (non-interactive, for scripts) ──────────────────
+// ── Direct Commands (non-interactive) ───────────────────────────────
 async function directPlans(cfg) {
     const { post } = api(cfg);
     const s = p.spinner();
     s.start('Loading plans...');
-    const res = await post('/api/v2/query', {
-        type: 'list_plans',
-        organization_id: cfg.orgId,
-        payload: { organization_id: cfg.orgId },
-    });
-    if (!res.success)
-        throw new Error(res.error?.message || 'Failed');
+    const plans = await listEntities(post, cfg.orgId, 'plan');
     s.stop('');
-    const plans = res.data || [];
     if (plans.length === 0) {
         p.log.info('No plans found.');
         return;
     }
     const lines = plans.map((plan) => {
         const name = truncate(plan.name || 'Untitled', 40);
-        const id = plan.id;
-        return `  ${pc.cyan(pad(name, 42))} ${pc.dim(id)}`;
+        return `  ${pc.cyan(pad(name, 42))} ${pc.dim(plan.id)}`;
     });
     p.note(lines.join('\n'), `${plans.length} plan${plans.length === 1 ? '' : 's'}`);
 }
-async function directTasks(cfg, planId) {
+async function directTasks(cfg) {
     const { post } = api(cfg);
     const s = p.spinner();
     s.start('Loading tasks...');
-    const payload = { organization_id: cfg.orgId };
-    if (planId)
-        payload.plan_id = planId;
-    const res = await post('/api/v2/query', {
-        type: 'list_tasks',
-        organization_id: cfg.orgId,
-        payload,
-    });
-    if (!res.success)
-        throw new Error(res.error?.message || 'Failed');
+    const tasks = await listEntities(post, cfg.orgId, 'task');
     s.stop('');
-    const tasks = res.data || [];
     if (tasks.length === 0) {
         p.log.info('No tasks found.');
         return;
     }
     const lines = tasks.map((t) => {
         const name = truncate(t.name || 'Untitled', 50);
-        const freq = t.frequency || '';
+        const freq = t.metadata?.frequency || '';
         return `  ${pc.cyan(pad(name, 52))} ${pc.dim(freq)}  ${pc.dim(t.id)}`;
     });
     p.note(lines.join('\n'), `${tasks.length} task${tasks.length === 1 ? '' : 's'}`);
@@ -683,39 +664,52 @@ async function directComplete(cfg, taskId) {
     const { post } = api(cfg);
     const s = p.spinner();
     s.start('Completing task...');
-    const res = await post('/api/v2/commands', {
+    await post('/api/v2/commands', {
         type: 'complete_task',
         organization_id: cfg.orgId,
         payload: { task_id: taskId },
     });
-    if (!res.success)
-        throw new Error(res.error?.message || 'Failed');
     s.stop(pc.green('Task completed!'));
 }
 async function directStatus(cfg) {
     const { post } = api(cfg);
     const s = p.spinner();
     s.start('Loading workspace...');
-    const res = await post('/api/v2/introspection/workspace', {
-        organization_id: cfg.orgId,
-    });
-    if (!res.success)
-        throw new Error(res.error?.message || 'Failed');
+    const nodes = await getWorkspaceNodes(post, cfg.orgId);
     s.stop('');
-    const counts = res.data?.entity_counts || {};
+    const counts = {};
+    for (const node of nodes) {
+        const t = node.type || 'unknown';
+        counts[t] = (counts[t] || 0) + 1;
+    }
     const lines = [
         `${pad('Workspace', 16)} ${pc.cyan(cfg.orgName)}`,
         '',
-        `${pad('Plans', 16)} ${pc.bold(String(counts.plans || 0))}`,
-        `${pad('Tasks', 16)} ${pc.bold(String(counts.tasks || 0))}`,
-        `${pad('Requirements', 16)} ${pc.bold(String(counts.requirements || 0))}`,
-        `${pad('Goals', 16)} ${pc.bold(String(counts.goals || 0))}`,
-        `${pad('Documents', 16)} ${pc.bold(String(counts.documents || 0))}`,
-        `${pad('Tables', 16)} ${pc.bold(String(counts.tables || 0))}`,
+        `${pad('Plans', 16)} ${pc.bold(String(counts.plan || 0))}`,
+        `${pad('Tasks', 16)} ${pc.bold(String(counts.task || 0))}`,
+        `${pad('Requirements', 16)} ${pc.bold(String(counts.requirement || 0))}`,
+        `${pad('Goals', 16)} ${pc.bold(String(counts.goal || 0))}`,
+        `${pad('Rules', 16)} ${pc.bold(String(counts.rule || 0))}`,
+        `${pad('Documents', 16)} ${pc.bold(String(counts.document || 0))}`,
+        `${pad('Tables', 16)} ${pc.bold(String(counts.table || 0))}`,
     ];
     p.note(lines.join('\n'), 'Workspace Overview');
 }
-// Raw command passthrough for power users and automation
+async function directSearch(cfg, term) {
+    const { post } = api(cfg);
+    const s = p.spinner();
+    s.start('Searching...');
+    const entities = await searchEntities(post, cfg.orgId, term);
+    s.stop('');
+    if (entities.length === 0) {
+        p.log.info('No results found.');
+    }
+    else {
+        const lines = entities.map((e) => `  ${pc.dim(pad(e.entity_type || '', 14))} ${pc.cyan(truncate(e.name || '', 50))}  ${pc.dim(e.id)}`);
+        p.note(lines.join('\n'), `${entities.length} result${entities.length === 1 ? '' : 's'} for "${term}"`);
+    }
+}
+// Raw passthrough for power users
 async function directRaw(cfg, type, args) {
     const { post } = api(cfg);
     const payloadRaw = args[0];
@@ -750,17 +744,14 @@ async function main() {
     const args = process.argv.slice(2);
     const sub = args[0];
     const rest = args.slice(1);
-    // Setup command always available
     if (sub === 'setup' || sub === 'init' || sub === 'login') {
         await runSetup(resolveConfig());
         return;
     }
-    // Version
     if (sub === '--version' || sub === '-v') {
         console.log(`checkbox v${VERSION}`);
         return;
     }
-    // Help
     if (sub === '--help' || sub === '-h' || sub === 'help') {
         console.log();
         p.intro(pc.bgCyan(pc.black(' Checkbox CLI ')) + pc.dim(` v${VERSION}`));
@@ -780,23 +771,19 @@ async function main() {
         p.outro(pc.dim('Run `checkbox setup` to get started'));
         return;
     }
-    // Check if configured
     const cfg = resolveConfig();
     if (!cfg) {
-        // First-time user — run setup
         await runSetup(null);
         return;
     }
-    // Direct commands
     try {
         switch (sub) {
             case 'plans':
                 return await directPlans(cfg);
             case 'tasks':
-                return await directTasks(cfg, getFlag(rest, '--plan'));
+                return await directTasks(cfg);
             case 'complete':
                 if (!rest[0]) {
-                    // Interactive complete
                     const { post } = api(cfg);
                     return await menuComplete(cfg, post);
                 }
@@ -809,24 +796,9 @@ async function main() {
                     const { post } = api(cfg);
                     return await menuSearch(cfg, post);
                 }
-                // Direct search
-                const { post: searchPost } = api(cfg);
-                const searchRes = await searchPost('/api/v2/introspection/entities', {
-                    organization_id: cfg.orgId,
-                    search: rest.join(' '),
-                    limit: 20,
-                });
-                if (searchRes.data?.length) {
-                    const lines = searchRes.data.map((e) => `  ${pc.dim(pad(e.type || '', 14))} ${pc.cyan(truncate(e.name || '', 50))}  ${pc.dim(e.id)}`);
-                    p.note(lines.join('\n'), `Results for "${rest.join(' ')}"`);
-                }
-                else {
-                    p.log.info('No results found.');
-                }
-                return;
+                return await directSearch(cfg, rest.join(' '));
             case 'switch':
                 return await menuSwitch(cfg, api(cfg).post);
-            // Power user commands
             case 'cmd':
             case 'command':
                 return await directRaw(cfg, rest[0], rest.slice(1));
@@ -836,7 +808,6 @@ async function main() {
             case 'organizations':
                 return await directQuery(cfg, 'list_my_organizations', []);
             case undefined:
-                // No args → interactive menu
                 return await interactiveMenu(cfg);
             default:
                 p.log.error(`Unknown command: ${sub}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "checkbox-cli",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "Beautiful, interactive CLI for Checkbox compliance management. Setup wizard, guided workflows, and full workspace control from your terminal.",
   "type": "module",
   "main": "dist/checkbox.js",