const path = require('path');
const mongoose = require('mongoose');
require('module-alias')({ base: path.resolve(__dirname, '..', 'api') });
const { askQuestion, silentExit } = require('./helpers');
const connect = require('./connect');

(async () => {
  await connect();

  console.purple('---------------------------------------');
  console.purple('Reset MeiliSearch Synchronization Flags');
  console.purple('---------------------------------------');
  console.yellow('\nThis script will reset the MeiliSearch indexing flags in MongoDB.');
  console.yellow('Use this when MeiliSearch data has been deleted or corrupted,');
  console.yellow('and you need to trigger a full re-synchronization.\n');

  const confirm = await askQuestion(
    'Are you sure you want to reset all MeiliSearch sync flags? (y/N): ',
  );

  if (confirm.toLowerCase() !== 'y') {
    console.orange('Operation cancelled.');
    silentExit(0);
  }

  try {
    // Reset _meiliIndex flags for messages
    console.cyan('\nResetting message sync flags...');
    const messageResult = await mongoose.connection.db
      .collection('messages')
      .updateMany({ _meiliIndex: true }, { $set: { _meiliIndex: false } });

    console.green(`✓ Reset ${messageResult.modifiedCount} message sync flags`);

    // Reset _meiliIndex flags for conversations
    console.cyan('\nResetting conversation sync flags...');
    const conversationResult = await mongoose.connection.db
      .collection('conversations')
      .updateMany({ _meiliIndex: true }, { $set: { _meiliIndex: false } });

    console.green(`✓ Reset ${conversationResult.modifiedCount} conversation sync flags`);

    // Get current counts
    const totalMessages = await mongoose.connection.db.collection('messages').countDocuments();
    const totalConversations = await mongoose.connection.db
      .collection('conversations')
      .countDocuments();

    console.purple('\n---------------------------------------');
    console.green('MeiliSearch sync flags have been reset successfully!');
    console.cyan(`\nTotal messages to sync: ${totalMessages}`);
    console.cyan(`Total conversations to sync: ${totalConversations}`);
    console.yellow('\nThe next time LibreChat starts or performs a sync check,');
    console.yellow('all data will be re-indexed into MeiliSearch.');
    console.purple('---------------------------------------\n');

    // Ask if user wants to see advanced options
    const showAdvanced = await askQuestion('Show advanced options? (y/N): ');

    if (showAdvanced.toLowerCase() === 'y') {
      console.cyan('\nAdvanced Options:');
      console.yellow('1. To trigger immediate sync, restart LibreChat');
      console.yellow('2. To disable sync, set MEILI_NO_SYNC=true in .env');
      console.yellow(
        '3. To adjust sync batch size, set MEILI_SYNC_BATCH_SIZE in .env (default: 100)',
      );
      console.yellow('4. To adjust sync delay, set MEILI_SYNC_DELAY_MS in .env (default: 100ms)');
      console.yellow(
        '5. To change sync threshold, set MEILI_SYNC_THRESHOLD in .env (default: 1000)\n',
      );
    }

    silentExit(0);
  } catch (error) {
    console.red('\nError resetting MeiliSearch sync flags:');
    console.error(error);
    silentExit(1);
  }
})();

process.on('uncaughtException', (err) => {
  if (!err.message.includes('fetch failed')) {
    console.error('There was an uncaught error:');
    console.error(err);
  }

  if (err.message.includes('fetch failed')) {
    return;
  } else {
    process.exit(1);
  }
});