cleanup_old_agents.js

Go to the documentation of this file.

#!/usr/bin/env node
// Script to clean up old/test agents from the database
const pool = require('../services/db');
 
/**
 *
 */
async function cleanupOldAgents() {
  console.log('[Cleanup] Starting agent cleanup...');
  
  try {
    // Find agents older than 30 days with no recent activity
    const oldAgentsQuery = `
      SELECT agent_id, agent_uuid, hostname, last_seen, created_at
      FROM agents
      WHERE last_seen < NOW() - INTERVAL '30 days'
         OR (last_seen IS NULL AND created_at < NOW() - INTERVAL '7 days')
      ORDER BY last_seen DESC NULLS LAST
    `;
    
    const result = await pool.query(oldAgentsQuery);
    console.log(`[Cleanup] Found ${result.rows.length} old agents`);
    
    if (result.rows.length === 0) {
      console.log('[Cleanup] No old agents to clean up');
      return;
    }
    
    // Display agents
    console.log('\nOld agents:');
    result.rows.forEach((agent, idx) => {
      console.log(`${idx + 1}. ${agent.hostname} (${agent.agent_uuid})`);
      console.log(`   Last seen: ${agent.last_seen || 'Never'}`);
      console.log(`   Created: ${agent.created_at}`);
    });
    
    // Confirm deletion
    const readline = require('readline').createInterface({
      input: process.stdin,
      output: process.stdout
    });
    
    readline.question('\nDelete these agents? (yes/no): ', async (answer) => {
      if (answer.toLowerCase() === 'yes' || answer.toLowerCase() === 'y') {
        // Delete agents
        for (const agent of result.rows) {
          try {
            await pool.query('DELETE FROM agents WHERE agent_id = $1', [agent.agent_id]);
            console.log(`✓ Deleted ${agent.hostname} (${agent.agent_uuid})`);
          } catch (err) {
            console.error(`✗ Failed to delete ${agent.hostname}:`, err.message);
          }
        }
        console.log('\n[Cleanup] Agent cleanup complete');
      } else {
        console.log('[Cleanup] Cancelled');
      }
      
      readline.close();
      process.exit(0);
    });
    
  } catch (err) {
    console.error('[Cleanup] Error:', err);
    process.exit(1);
  }
}
 
cleanupOldAgents();