figmanage 1.0.1 → 1.2.0

package/dist/tools/compound.js

@@ -1,8 +1,6 @@
 import { z } from 'zod';
-import { internalClient } from '../clients/internal-api.js';
-import { publicClient } from '../clients/public-api.js';
-import { hasPat, hasCookie } from '../auth/client.js';
-import { defineTool, toolResult, toolError, figmaId, requireOrgId } from './register.js';
+import { defineTool, toolResult, toolError, figmaId } from './register.js';
+import { fileSummary, workspaceOverview, openComments, cleanupStaleFiles, organizeProject, setupProjectStructure, seatOptimization, permissionAudit, branchCleanup, } from '../operations/compound.js';
 // -- file_summary --
 defineTool({
     toolset: 'compound',
@@ -15,33 +13,8 @@ defineTool({
             },
         }, async ({ file_key }) => {
             try {
-                const api = publicClient(config);
-                const [fileResult, componentsResult, stylesResult, commentsResult] = await Promise.allSettled([
-                    api.get(`/v1/files/${file_key}`, { params: { depth: '1' } }),
-                    api.get(`/v1/files/${file_key}/components`),
-                    api.get(`/v1/files/${file_key}/styles`),
-                    api.get(`/v1/files/${file_key}/comments`),
-                ]);
-                if (fileResult.status === 'rejected') {
-                    return toolError(`Failed to fetch file: ${fileResult.reason?.response?.status || fileResult.reason?.message}`);
-                }
-                const fileData = fileResult.value.data;
-                const pages = (fileData.document?.children || []).map((c) => c.name);
-                const components = componentsResult.status === 'fulfilled' ? componentsResult.value.data?.meta?.components || [] : [];
-                const styles = stylesResult.status === 'fulfilled' ? stylesResult.value.data?.meta?.styles || [] : [];
-                const comments = commentsResult.status === 'fulfilled' ? commentsResult.value.data?.comments || [] : [];
-                const unresolved = comments.filter((c) => !c.resolved_at);
-                const summary = {
-                    name: fileData.name,
-                    last_modified: fileData.lastModified,
-                    version: fileData.version,
-                    pages,
-                    component_count: components.length,
-                    style_count: styles.length,
-                    comment_count: comments.length,
-                    unresolved_comment_count: unresolved.length,
-                };
-                return toolResult(JSON.stringify(summary, null, 2));
+                const result = await fileSummary(config, { file_key });
+                return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
                 return toolError(`Failed to summarize file: ${e.response?.status || e.message}`);
@@ -61,46 +34,8 @@ defineTool({
             },
         }, async ({ org_id }) => {
             try {
-                let orgId;
-                try {
-                    orgId = requireOrgId(config, org_id);
-                }
-                catch (e) {
-                    return toolError(e.message);
-                }
-                const api = internalClient(config);
-                const [teamsResult, seatsResult, billingResult] = await Promise.allSettled([
-                    api.get(`/api/orgs/${orgId}/teams`, {
-                        params: { include_member_count: true, include_project_count: true, include_top_members: true },
-                    }),
-                    api.get(`/api/orgs/${orgId}/org_users/filter_counts`),
-                    api.get(`/api/orgs/${orgId}/billing_data`),
-                ]);
-                const teamsData = teamsResult.status === 'fulfilled' ? (teamsResult.value.data?.meta || teamsResult.value.data) : null;
-                const teamsRaw = teamsData ? (Array.isArray(teamsData) ? teamsData : (teamsData?.teams || [])) : [];
-                const teams = teamsRaw.map((t) => ({
-                    id: String(t.id),
-                    name: t.name,
-                    members: t.member_count || 0,
-                    projects: t.project_count || 0,
-                }));
-                const seats = seatsResult.status === 'fulfilled' ? (seatsResult.value.data?.meta || seatsResult.value.data) : null;
-                const rawBilling = billingResult.status === 'fulfilled' ? (billingResult.value.data?.meta || billingResult.value.data) : null;
-                // Strip PII from billing data
-                const billing = rawBilling ? { ...rawBilling } : null;
-                if (billing)
-                    delete billing.shipping_address;
-                const errors = [];
-                if (teamsResult.status === 'rejected')
-                    errors.push(`teams: ${teamsResult.reason?.response?.status || teamsResult.reason?.message}`);
-                if (seatsResult.status === 'rejected')
-                    errors.push(`seats: ${seatsResult.reason?.response?.status || seatsResult.reason?.message}`);
-                if (billingResult.status === 'rejected')
-                    errors.push(`billing: ${billingResult.reason?.response?.status || billingResult.reason?.message}`);
-                const overview = { teams, seats, billing };
-                if (errors.length > 0)
-                    overview.errors = errors;
-                return toolResult(JSON.stringify(overview, null, 2));
+                const result = await workspaceOverview(config, { org_id });
+                return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
                 return toolError(`Failed to fetch workspace overview: ${e.response?.status || e.message}`);
@@ -120,49 +55,7 @@ defineTool({
             },
         }, async ({ project_id }) => {
             try {
-                const api = publicClient(config);
-                const filesRes = await api.get(`/v1/projects/${project_id}/files`);
-                const allFiles = filesRes.data?.files || [];
-                const capped = allFiles.length > 20;
-                const files = allFiles.slice(0, 20);
-                const commentResults = [];
-                for (const f of files) {
-                    try {
-                        const commentsRes = await api.get(`/v1/files/${f.key}/comments`);
-                        const comments = (commentsRes.data?.comments || [])
-                            .filter((c) => !c.resolved_at);
-                        commentResults.push({ file_key: f.key, file_name: f.name, comments });
-                    }
-                    catch (e) {
-                        commentResults.push({ file_key: f.key, file_name: f.name, comments: [], error: e.response?.status || e.message });
-                    }
-                }
-                const filesWithComments = commentResults
-                    .filter(f => f.comments.length > 0)
-                    .map(f => ({
-                    file_key: f.file_key,
-                    file_name: f.file_name,
-                    comments: f.comments.map((c) => ({
-                        id: c.id,
-                        author: c.user?.handle || c.user?.email || 'unknown',
-                        message: c.message,
-                        created_at: c.created_at,
-                    })),
-                }));
-                const totalUnresolved = filesWithComments.reduce((sum, f) => sum + f.comments.length, 0);
-                const errors = commentResults
-                    .filter((f) => f.error)
-                    .map((f) => ({ file_key: f.file_key, file_name: f.file_name, error: f.error }));
-                const result = {
-                    total_unresolved: totalUnresolved,
-                    files: filesWithComments,
-                };
-                if (errors.length > 0) {
-                    result.errors = errors;
-                }
-                if (capped) {
-                    result.note = `Project has ${allFiles.length} files; only the first 20 were checked.`;
-                }
+                const result = await openComments(config, { project_id });
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
@@ -187,51 +80,11 @@ defineTool({
             },
         }, async ({ project_id, days_stale, dry_run: rawDryRun }) => {
             try {
-                const dry_run = rawDryRun ?? true;
-                if (!dry_run && !hasCookie(config)) {
-                    return toolError('Cookie auth required to trash files. Run with dry_run=true to preview, or configure cookie auth.');
-                }
-                let files;
-                if (hasPat(config)) {
-                    const res = await publicClient(config).get(`/v1/projects/${project_id}/files`);
-                    files = res.data?.files || [];
-                }
-                else {
-                    const res = await internalClient(config).get(`/api/folders/${project_id}/paginated_files`, { params: { folderId: project_id, sort_column: 'touched_at', sort_order: 'desc', page_size: 100, file_type: '' } });
-                    const meta = res.data?.meta || res.data;
-                    files = meta?.files || meta || [];
-                }
-                const cutoff = Date.now() - days_stale * 86400000;
-                const staleFiles = files.filter((f) => {
-                    const raw = f.last_modified || f.touched_at;
-                    if (!raw)
-                        return false;
-                    const modified = new Date(raw).getTime();
-                    return !isNaN(modified) && modified < cutoff;
+                const result = await cleanupStaleFiles(config, {
+                    project_id,
+                    days_stale: days_stale ?? 90,
+                    dry_run: rawDryRun ?? true,
                 });
-                const MAX_TRASH_BATCH = 25;
-                const result = {
-                    stale_files: staleFiles.map((f) => ({
-                        key: f.key,
-                        name: f.name,
-                        last_modified: f.last_modified || f.touched_at,
-                    })),
-                    total_stale: staleFiles.length,
-                    dry_run,
-                    trashed: false,
-                };
-                if (!dry_run) {
-                    if (staleFiles.length > MAX_TRASH_BATCH) {
-                        return toolError(`${staleFiles.length} stale files exceeds safety limit of ${MAX_TRASH_BATCH}. ` +
-                            `Run with dry_run=true to review, then trash in smaller batches using trash_files.`);
-                    }
-                    if (staleFiles.length > 0) {
-                        await internalClient(config).delete('/api/files_batch', {
-                            data: { files: staleFiles.map((f) => ({ key: f.key })), trashed: true },
-                        });
-                        result.trashed = true;
-                    }
-                }
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
@@ -254,23 +107,7 @@ defineTool({
             },
         }, async ({ file_keys, target_project_id }) => {
             try {
-                const payload = {
-                    files: file_keys.map(key => ({
-                        key,
-                        folder_id: target_project_id,
-                        is_multi_move: true,
-                        restore_files: false,
-                    })),
-                };
-                const res = await internalClient(config).put('/api/files_batch', payload);
-                const data = res.data?.meta || res.data;
-                const moved = Object.keys(data?.success || {}).length;
-                const failed = Object.keys(data?.errors || {}).length;
-                const result = {
-                    moved: moved || (failed === 0 ? file_keys.length : 0),
-                    failed,
-                    errors: data?.errors || {},
-                };
+                const result = await organizeProject(config, { file_keys, target_project_id });
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
@@ -296,34 +133,8 @@ defineTool({
             },
         }, async ({ team_id, projects }) => {
             try {
-                const api = internalClient(config);
-                const created = [];
-                const failed = [];
-                for (const project of projects) {
-                    try {
-                        const createRes = await api.post('/api/folders', {
-                            team_id,
-                            path: project.name,
-                            sharing_audience_control: 'org_view',
-                            team_access: 'team_edit',
-                        });
-                        const meta = createRes.data?.meta;
-                        const p = Array.isArray(meta) ? meta[0] : (meta?.folder || meta || createRes.data);
-                        const folderId = p?.id != null ? String(p.id) : null;
-                        if (!folderId) {
-                            failed.push({ name: project.name, error: 'Could not extract project ID from response' });
-                            continue;
-                        }
-                        if (project.description) {
-                            await api.put(`/api/folders/${folderId}`, { description: project.description });
-                        }
-                        created.push({ id: folderId, name: project.name, description: project.description || null });
-                    }
-                    catch (e) {
-                        failed.push({ name: project.name, error: e.response?.status || e.message });
-                    }
-                }
-                return toolResult(JSON.stringify({ created, failed }, null, 2));
+                const result = await setupProjectStructure(config, { team_id, projects });
+                return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
                 return toolError(`Failed to setup project structure: ${e.response?.status || e.message}`);
@@ -332,11 +143,6 @@ defineTool({
     },
 });
 // -- seat_optimization --
-const SEAT_KEY_MAP = {
-    expert: 'full',
-    developer: 'dev',
-    collaborator: 'collab',
-};
 defineTool({
     toolset: 'compound',
     auth: 'cookie',
@@ -348,106 +154,13 @@ defineTool({
                 days_inactive: z.number().min(1).max(365).optional().default(90).describe('Days without activity to flag as inactive (default: 90)'),
                 include_cost: z.boolean().optional().default(true).describe('Include cost analysis from contract rates (default: true)'),
             },
-        }, async ({ org_id, days_inactive: rawDaysInactive, include_cost: rawIncludeCost }) => {
+        }, async ({ org_id, days_inactive, include_cost }) => {
             try {
-                const days_inactive = rawDaysInactive ?? 90;
-                const include_cost = rawIncludeCost ?? true;
-                let orgId;
-                try {
-                    orgId = requireOrgId(config, org_id);
-                }
-                catch (e) {
-                    return toolError(e.message);
-                }
-                const api = internalClient(config);
-                const cutoff = Date.now() - days_inactive * 86400000;
-                const paidKeys = new Set(['expert', 'developer', 'collaborator']);
-                // Paginate org members (cursor-based, max 500)
-                const allMembers = [];
-                const MAX_PAGES = 20;
-                let cursor;
-                for (let page = 0; page < MAX_PAGES; page++) {
-                    const params = { page_size: 25 };
-                    if (cursor)
-                        params.cursor = cursor;
-                    const res = await api.get(`/api/v2/orgs/${orgId}/org_users`, { params });
-                    const meta = res.data?.meta || {};
-                    const members = meta.users || [];
-                    if (!Array.isArray(members) || members.length === 0)
-                        break;
-                    allMembers.push(...members);
-                    cursor = Array.isArray(meta.cursor) ? meta.cursor[0] : meta.cursor;
-                    if (!cursor || members.length < 25)
-                        break;
-                }
-                // Fetch seat breakdown and optionally contract rates in parallel
-                const parallelCalls = [
-                    api.get(`/api/orgs/${orgId}/org_users/filter_counts`),
-                ];
-                if (include_cost) {
-                    parallelCalls.push(api.get('/api/pricing/contract_rates', {
-                        params: { plan_parent_id: orgId, plan_type: 'organization' },
-                    }));
-                }
-                const [seatsResult, ratesResult] = await Promise.allSettled(parallelCalls);
-                const seats = seatsResult.status === 'fulfilled' ? (seatsResult.value.data?.meta || seatsResult.value.data) : null;
-                // Build cost lookup: seat key -> monthly cents
-                const costMap = {};
-                if (include_cost && ratesResult?.status === 'fulfilled') {
-                    const prices = ratesResult.value.data?.meta?.product_prices || [];
-                    for (const p of prices) {
-                        if (paidKeys.has(p.billable_product_key)) {
-                            costMap[p.billable_product_key] = p.amount;
-                        }
-                    }
-                }
-                // Filter inactive paid members
-                const inactiveUsers = [];
-                let totalPaid = 0;
-                for (const m of allMembers) {
-                    const seatKey = m.active_seat_type?.key;
-                    if (!seatKey || !paidKeys.has(seatKey))
-                        continue;
-                    totalPaid++;
-                    const lastSeen = m.last_seen;
-                    const lastSeenMs = lastSeen ? new Date(lastSeen).getTime() : 0;
-                    const isInactive = !lastSeen || (lastSeenMs > 0 && lastSeenMs < cutoff) || isNaN(lastSeenMs);
-                    if (isInactive) {
-                        inactiveUsers.push({
-                            org_user_id: String(m.id),
-                            user_id: m.user_id,
-                            email: m.user?.email,
-                            name: m.user?.handle,
-                            seat_type: SEAT_KEY_MAP[seatKey] || seatKey,
-                            seat_key: seatKey,
-                            last_active: lastSeen || null,
-                            monthly_cost_cents: costMap[seatKey] || null,
-                        });
-                    }
-                }
-                const monthlyWasteCents = inactiveUsers.reduce((sum, u) => sum + (u.monthly_cost_cents || 0), 0);
-                const recommendations = [];
-                if (inactiveUsers.length > 0) {
-                    recommendations.push(`${inactiveUsers.length} paid seat(s) inactive for ${days_inactive}+ days. Review for downgrade to viewer.`);
-                }
-                const neverActive = inactiveUsers.filter((u) => !u.last_active);
-                if (neverActive.length > 0) {
-                    recommendations.push(`${neverActive.length} paid user(s) have never been active. Likely unused invites.`);
-                }
-                if (monthlyWasteCents > 0) {
-                    recommendations.push(`Potential monthly savings: $${(monthlyWasteCents / 100).toFixed(2)} ($${((monthlyWasteCents * 12) / 100).toFixed(2)}/yr).`);
-                }
-                const result = {
-                    summary: {
-                        total_paid: totalPaid,
-                        inactive_paid: inactiveUsers.length,
-                        monthly_waste_cents: monthlyWasteCents,
-                        annual_savings_cents: monthlyWasteCents * 12,
-                    },
-                    seat_breakdown: seats,
-                    inactive_users: inactiveUsers,
-                    recommendations,
-                };
+                const result = await seatOptimization(config, {
+                    org_id,
+                    days_inactive: days_inactive ?? 90,
+                    include_cost: include_cost ?? true,
+                });
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
@@ -469,161 +182,14 @@ defineTool({
                 flag_external: z.boolean().optional().default(true).describe('Flag external users (default: true)'),
                 org_id: figmaId.optional().describe('Org ID override (defaults to current workspace)'),
             },
-        }, async ({ scope_type, scope_id, flag_external: rawFlagExternal, org_id }) => {
+        }, async ({ scope_type, scope_id, flag_external, org_id }) => {
             try {
-                const flag_external = rawFlagExternal ?? true;
-                const api = internalClient(config);
-                // Resolve org for domain lookup
-                let orgId;
-                let domainCheckSkipped = false;
-                if (flag_external) {
-                    try {
-                        orgId = requireOrgId(config, org_id);
-                    }
-                    catch {
-                        domainCheckSkipped = true;
-                    }
-                }
-                // Fetch org verified domains for external detection
-                let verifiedDomains = new Set();
-                if (flag_external && orgId) {
-                    try {
-                        const domRes = await api.get(`/api/orgs/${orgId}/domains`);
-                        const domains = domRes.data?.meta || [];
-                        if (Array.isArray(domains)) {
-                            for (const d of domains) {
-                                if (d.domain)
-                                    verifiedDomains.add(d.domain.toLowerCase());
-                            }
-                        }
-                    }
-                    catch { /* domain lookup optional, continue without */ }
-                }
-                // Collect file keys to scan
-                let fileKeys = [];
-                if (scope_type === 'team') {
-                    // Fetch team projects, cap at 10
-                    const projectsRes = await api.get(`/api/teams/${scope_id}/folders`);
-                    const rows = projectsRes.data?.meta?.folder_rows || projectsRes.data || [];
-                    const projects = (Array.isArray(rows) ? rows : []).slice(0, 10);
-                    for (const proj of projects) {
-                        if (fileKeys.length >= 25)
-                            break;
-                        try {
-                            const filesRes = await api.get(`/api/folders/${proj.id}/paginated_files`, {
-                                params: { folderId: String(proj.id), page_size: 25, sort_column: 'touched_at', sort_order: 'desc', file_type: '' },
-                            });
-                            const meta = filesRes.data?.meta || filesRes.data;
-                            const files = meta?.files || meta || [];
-                            for (const f of (Array.isArray(files) ? files : [])) {
-                                if (fileKeys.length >= 25)
-                                    break;
-                                fileKeys.push({ key: f.key, name: f.name });
-                            }
-                        }
-                        catch { /* skip inaccessible projects */ }
-                    }
-                }
-                else {
-                    // Project scope
-                    const filesRes = await api.get(`/api/folders/${scope_id}/paginated_files`, {
-                        params: { folderId: scope_id, page_size: 25, sort_column: 'touched_at', sort_order: 'desc', file_type: '' },
-                    });
-                    const meta = filesRes.data?.meta || filesRes.data;
-                    const files = meta?.files || meta || [];
-                    for (const f of (Array.isArray(files) ? files : [])) {
-                        if (fileKeys.length >= 25)
-                            break;
-                        fileKeys.push({ key: f.key, name: f.name });
-                    }
-                }
-                // Fetch permissions and file metadata in parallel, batched 5 at a time
-                const allUsers = new Map();
-                const flags = [];
-                let filesScanned = 0;
-                for (let i = 0; i < fileKeys.length; i += 5) {
-                    const batch = fileKeys.slice(i, i + 5);
-                    const results = await Promise.allSettled(batch.map(async (file) => {
-                        const [rolesRes, fileMetaRes] = await Promise.allSettled([
-                            api.get(`/api/roles/file/${file.key}`),
-                            api.get(`/api/files/${file.key}`),
-                        ]);
-                        const roles = rolesRes.status === 'fulfilled'
-                            ? (Array.isArray(rolesRes.value.data?.meta) ? rolesRes.value.data.meta : [])
-                            : [];
-                        const fileMeta = fileMetaRes.status === 'fulfilled'
-                            ? (fileMetaRes.value.data?.meta || fileMetaRes.value.data || {})
-                            : {};
-                        return { file, roles, fileMeta };
-                    }));
-                    for (const r of results) {
-                        if (r.status === 'rejected')
-                            continue;
-                        filesScanned++;
-                        const { file, roles, fileMeta } = r.value;
-                        // Check link access
-                        const linkAccess = fileMeta.link_access;
-                        if (linkAccess === 'edit' || linkAccess === 'org_edit') {
-                            flags.push({
-                                severity: 'high',
-                                type: 'open_link_access',
-                                details: `${file.name} (${file.key}) has link_access="${linkAccess}"`,
-                            });
-                        }
-                        // Process roles
-                        for (const role of roles) {
-                            const email = role.user?.email || role.pending_email;
-                            const userId = role.user_id ? String(role.user_id) : email;
-                            const level = role.level;
-                            const levelName = level >= 999 ? 'owner' : level >= 300 ? 'editor' : 'viewer';
-                            if (userId && !allUsers.has(userId)) {
-                                allUsers.set(userId, {
-                                    user_id: userId,
-                                    email,
-                                    name: role.user?.handle,
-                                    files_accessed: [],
-                                });
-                            }
-                            if (userId) {
-                                const user = allUsers.get(userId);
-                                user.files_accessed.push({
-                                    file_key: file.key,
-                                    file_name: file.name,
-                                    role: levelName,
-                                });
-                            }
-                            // External editor detection
-                            if (flag_external && email && verifiedDomains.size > 0) {
-                                const domain = email.split('@')[1]?.toLowerCase();
-                                if (domain && !verifiedDomains.has(domain) && (levelName === 'editor' || levelName === 'owner')) {
-                                    flags.push({
-                                        severity: 'high',
-                                        type: 'external_editor',
-                                        details: `${email} (external) has ${levelName} access to ${file.name} (${file.key})`,
-                                    });
-                                }
-                            }
-                        }
-                    }
-                }
-                if (domainCheckSkipped) {
-                    flags.push({
-                        severity: 'info',
-                        type: 'domain_check_skipped',
-                        details: 'Could not resolve org ID; external user detection was skipped. Provide org_id or set FIGMA_ORG_ID.',
-                    });
-                }
-                const result = {
-                    scope: { type: scope_type, id: scope_id },
-                    summary: {
-                        unique_users: allUsers.size,
-                        files_scanned: filesScanned,
-                        total_files: fileKeys.length,
-                        flags_found: flags.length,
-                    },
-                    users: Array.from(allUsers.values()),
-                    flags,
-                };
+                const result = await permissionAudit(config, {
+                    scope_type,
+                    scope_id,
+                    flag_external: flag_external ?? true,
+                    org_id,
+                });
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {
@@ -646,113 +212,13 @@ defineTool({
                 days_stale: z.number().min(1).max(365).optional().default(60).describe('Days since last modification to flag as stale (default: 60)'),
                 dry_run: z.boolean().optional().default(true).describe('Preview only, no archiving (default: true)'),
             },
-        }, async ({ project_id, days_stale: rawDaysStale, dry_run: rawDryRun }) => {
+        }, async ({ project_id, days_stale, dry_run: rawDryRun }) => {
             try {
-                const days_stale = rawDaysStale ?? 60;
-                const dry_run = rawDryRun ?? true;
-                if (!dry_run && !hasCookie(config)) {
-                    return toolError('Cookie auth required to archive branches. Run with dry_run=true to preview, or configure cookie auth.');
-                }
-                // Fetch project files
-                const MAX_FILES = 20;
-                let files;
-                if (hasPat(config)) {
-                    const res = await publicClient(config).get(`/v1/projects/${project_id}/files`);
-                    files = res.data?.files || [];
-                }
-                else {
-                    const res = await internalClient(config).get(`/api/folders/${project_id}/paginated_files`, { params: { folderId: project_id, sort_column: 'touched_at', sort_order: 'desc', page_size: MAX_FILES, file_type: '' } });
-                    const meta = res.data?.meta || res.data;
-                    files = meta?.files || meta || [];
-                }
-                // Cap at 20 files
-                const capped = files.length > 20;
-                files = files.slice(0, 20);
-                // Fetch branch data for each file in parallel
-                const cutoff = Date.now() - days_stale * 86400000;
-                const staleBranches = [];
-                const activeBranches = [];
-                let filesScanned = 0;
-                let totalBranches = 0;
-                const branchResults = await Promise.allSettled(files.map(async (file) => {
-                    let branches;
-                    if (hasPat(config)) {
-                        const res = await publicClient(config).get(`/v1/files/${file.key}`, {
-                            params: { branch_data: 'true', depth: '0' },
-                        });
-                        branches = res.data?.branches || [];
-                    }
-                    else {
-                        const res = await internalClient(config).get(`/api/files/${file.key}`);
-                        const f = res.data?.meta || res.data;
-                        branches = f.branches || [];
-                    }
-                    return { file, branches };
-                }));
-                for (const r of branchResults) {
-                    if (r.status === 'rejected')
-                        continue;
-                    filesScanned++;
-                    const { file, branches } = r.value;
-                    for (const branch of branches) {
-                        totalBranches++;
-                        const lastModified = branch.last_modified;
-                        const lastModifiedMs = lastModified ? new Date(lastModified).getTime() : 0;
-                        const isStale = !lastModified || (lastModifiedMs > 0 && lastModifiedMs < cutoff) || isNaN(lastModifiedMs);
-                        const entry = {
-                            branch_key: branch.key,
-                            branch_name: branch.name,
-                            parent_file_key: file.key,
-                            parent_file_name: file.name,
-                            last_modified: lastModified || null,
-                        };
-                        if (isStale) {
-                            staleBranches.push(entry);
-                        }
-                        else {
-                            activeBranches.push(entry);
-                        }
-                    }
-                }
-                const MAX_ARCHIVE_BATCH = 25;
-                let archived = false;
-                if (!dry_run && staleBranches.length > 0) {
-                    if (staleBranches.length > MAX_ARCHIVE_BATCH) {
-                        return toolError(`${staleBranches.length} stale branches exceeds safety limit of ${MAX_ARCHIVE_BATCH}. ` +
-                            `Run with dry_run=true to review, then archive in smaller batches using delete_branch.`);
-                    }
-                    await internalClient(config).delete('/api/files_batch', {
-                        data: {
-                            files: staleBranches.map(b => ({ key: b.branch_key })),
-                            trashed: true,
-                        },
-                    });
-                    archived = true;
-                }
-                const recommendations = [];
-                if (staleBranches.length > 0) {
-                    recommendations.push(`${staleBranches.length} branch(es) stale for ${days_stale}+ days. ${dry_run ? 'Set dry_run=false to archive.' : 'Archived.'}`);
-                }
-                if (capped) {
-                    recommendations.push(`Project has more than 20 files; only the first 20 were scanned.`);
-                }
-                if (staleBranches.length === 0 && activeBranches.length === 0) {
-                    recommendations.push('No branches found in scanned files.');
-                }
-                const result = {
+                const result = await branchCleanup(config, {
                     project_id,
-                    summary: {
-                        files_scanned: filesScanned,
-                        total_branches: totalBranches,
-                        stale: staleBranches.length,
-                        active: activeBranches.length,
-                    },
-                    stale_branches: staleBranches,
-                    active_branches: activeBranches,
-                    dry_run,
-                    archived,
-                    recommendations,
-                };
+                    days_stale: days_stale ?? 60,
+                    dry_run: rawDryRun ?? true,
+                });
                 return toolResult(JSON.stringify(result, null, 2));
             }
             catch (e) {