n8n-mcp/src/scripts/fetch-templates-robust.ts

#!/usr/bin/env node
import { createDatabaseAdapter } from '../database/database-adapter';
import { TemplateRepository } from '../templates/template-repository';
import { TemplateFetcher } from '../templates/template-fetcher';
import * as fs from 'fs';
import * as path from 'path';

async function fetchTemplatesRobust() {
  console.log('🌐 Fetching n8n workflow templates (last year)...\n');
  
  // Ensure data directory exists
  const dataDir = './data';
  if (!fs.existsSync(dataDir)) {
    fs.mkdirSync(dataDir, { recursive: true });
  }
  
  // Initialize database
  const db = await createDatabaseAdapter('./data/nodes.db');
  
  // Drop existing templates table to ensure clean schema
  try {
    db.exec('DROP TABLE IF EXISTS templates');
    db.exec('DROP TABLE IF EXISTS templates_fts');
    console.log('🗑️  Dropped existing templates tables\n');
  } catch (error) {
    // Ignore errors if tables don't exist
  }
  
  // Apply schema with updated constraint
  const schema = fs.readFileSync(path.join(__dirname, '../../src/database/schema.sql'), 'utf8');
  db.exec(schema);
  
  // Create repository and fetcher
  const repository = new TemplateRepository(db);
  const fetcher = new TemplateFetcher();
  
  // Progress tracking
  let lastMessage = '';
  const startTime = Date.now();
  
  try {
    // Fetch template list
    console.log('📋 Phase 1: Fetching template list from n8n.io API\n');
    const templates = await fetcher.fetchTemplates((current, total) => {
      // Clear previous line
      if (lastMessage) {
        process.stdout.write('\r' + ' '.repeat(lastMessage.length) + '\r');
      }
      
      const progress = Math.round((current / total) * 100);
      lastMessage = `📊 Fetching template list: ${current}/${total} (${progress}%)`;
      process.stdout.write(lastMessage);
    });
    
    console.log('\n');
    console.log(`✅ Found ${templates.length} templates from last year\n`);
    
    // Fetch details and save incrementally
    console.log('📥 Phase 2: Fetching details and saving to database\n');
    let saved = 0;
    let errors = 0;
    
    for (let i = 0; i < templates.length; i++) {
      const template = templates[i];
      
      try {
        // Clear previous line
        if (lastMessage) {
          process.stdout.write('\r' + ' '.repeat(lastMessage.length) + '\r');
        }
        
        const progress = Math.round(((i + 1) / templates.length) * 100);
        lastMessage = `📊 Processing: ${i + 1}/${templates.length} (${progress}%) - Saved: ${saved}, Errors: ${errors}`;
        process.stdout.write(lastMessage);
        
        // Fetch detail
        const detail = await fetcher.fetchTemplateDetail(template.id);
        
        // Save immediately
        repository.saveTemplate(template, detail);
        saved++;
        
        // Rate limiting
        await new Promise(resolve => setTimeout(resolve, 200));
      } catch (error: any) {
        errors++;
        console.error(`\n❌ Error processing template ${template.id} (${template.name}): ${error.message}`);
        // Continue with next template
      }
    }
    
    console.log('\n');
    
    // Get stats
    const elapsed = Math.round((Date.now() - startTime) / 1000);
    const stats = await repository.getTemplateStats();
    
    console.log('✅ Template fetch complete!\n');
    console.log('📈 Statistics:');
    console.log(`   - Templates found: ${templates.length}`);
    console.log(`   - Templates saved: ${saved}`);
    console.log(`   - Errors: ${errors}`);
    console.log(`   - Success rate: ${Math.round((saved / templates.length) * 100)}%`);
    console.log(`   - Time elapsed: ${elapsed} seconds`);
    console.log(`   - Average time per template: ${(elapsed / saved).toFixed(2)} seconds`);
    
    if (stats.topUsedNodes && stats.topUsedNodes.length > 0) {
      console.log('\n🔝 Top used nodes:');
      stats.topUsedNodes.slice(0, 10).forEach((node: any, index: number) => {
        console.log(`   ${index + 1}. ${node.node} (${node.count} templates)`);
      });
    }
    
  } catch (error) {
    console.error('\n❌ Fatal error:', error);
    process.exit(1);
  }
  
  // Close database
  if ('close' in db && typeof db.close === 'function') {
    db.close();
  }
}

// Run if called directly
if (require.main === module) {
  fetchTemplatesRobust().catch(console.error);
}

export { fetchTemplatesRobust };