ideas-generator/server/migrations/add-starter-message.js

require('dotenv').config();
const { sequelize } = require('../config/database');

async function addStarterMessageColumn() {
  try {
    console.log('🔧 Adding starterMessage column to assistants table...');
    
    // Add the starterMessage column
    await sequelize.getQueryInterface().addColumn('assistants', 'starterMessage', {
      type: sequelize.Sequelize.DataTypes.TEXT,
      allowNull: true,
    });
    
    console.log('✅ Column added successfully');
    
    // Now extract starter messages from existing systemPrompt data
    console.log('🔍 Extracting starter messages from systemPrompt data...');
    
    const assistants = await sequelize.query(
      'SELECT id, "systemPrompt" FROM assistants WHERE "systemPrompt" IS NOT NULL',
      { type: sequelize.QueryTypes.SELECT }
    );
    
    for (const assistant of assistants) {
      const systemPrompt = assistant.systemPrompt;
      
      // Look for STARTER MESSAGE pattern
      const starterMatch = systemPrompt.match(/STARTER MESSAGE:\s*"([^"]+)"/i);
      
      if (starterMatch) {
        const starterMessage = starterMatch[1];
        
        // Clean the systemPrompt by removing the starter message section
        const cleanSystemPrompt = systemPrompt
          .replace(/\n\nSTARTER MESSAGE:\s*"[^"]+"/i, '')
          .trim();
        
        // Update the database
        await sequelize.query(
          'UPDATE assistants SET "starterMessage" = :starterMessage, "systemPrompt" = :systemPrompt WHERE id = :id',
          {
            replacements: {
              id: assistant.id,
              starterMessage: starterMessage,
              systemPrompt: cleanSystemPrompt
            },
            type: sequelize.QueryTypes.UPDATE
          }
        );
        
        console.log(`✅ Updated assistant: ${starterMessage.substring(0, 50)}...`);
      }
    }
    
    console.log('✅ Migration completed successfully');
    
  } catch (error) {
    console.error('❌ Migration failed:', error);
    throw error;
  } finally {
    await sequelize.close();
  }
}

// Run the migration if this file is executed directly
if (require.main === module) {
  addStarterMessageColumn()
    .then(() => {
      console.log('Migration finished');
      process.exit(0);
    })
    .catch((error) => {
      console.error('Migration error:', error);
      process.exit(1);
    });
}

module.exports = addStarterMessageColumn;