Function generateTestData

server/scripts/generateStudyTestData.js:23–163

()

const STREAK_BREAK_PROBABILITY = 0.15; // 15% chance of breaking a streak

async function generateTestData() {
  try {
    console.log('Connecting to database...');
    await connectToDatabase();
    const db = getDb();
    
    // Parse command line arguments
    const args = process.argv.slice(2);
    if (args.length < 2) {
      console.error('Usage: node generateStudyTestData.js <userId> <deckId> [daysOfHistory]');
      process.exit(1);
    }
    
    const userId = parseInt(args[0]);
    const deckId = parseInt(args[1]);
    const daysOfHistory = args[2] ? parseInt(args[2]) : 30;
    
    // Verify user and deck exist
    const user = await db.collection('users').findOne({ userId });
    if (!user) {
      console.error(`User with ID ${userId} not found.`);
      process.exit(1);
    }
    
    const deck = await db.collection('flashcard_decks').findOne({ deckId });
    if (!deck) {
      console.error(`Deck with ID ${deckId} not found.`);
      process.exit(1);
    }
    
    console.log(`Generating ${daysOfHistory} days of study data for user ${userId} (${user.name}) and deck ${deckId} (${deck.name}).`);
    
    // Clear existing data for this user and deck
    await db.collection('study_progress').deleteMany({ userId, deckId });
    await db.collection('study_streaks').deleteMany({ userId, deckId });
    
    const today = new Date();
    today.setHours(0, 0, 0, 0);
    
    // Initialize variables for tracking streak data
    let currentStreak = 0;
    let maxStreak = 0;
    let maxStreakStartDate = null;
    let maxStreakEndDate = null;
    let currentStreakStartDate = null;
    let lastStudyDate = null;
    const studyDates = [];
    
    // Generate daily progress entries
    for (let i = 0; i < daysOfHistory; i++) {
      const date = new Date(today);
      date.setDate(date.getDate() - (daysOfHistory - i - 1));
      
      // Decide if user studied on this day (with higher probability for recent days)
      const recencyBoost = i / daysOfHistory; // 0 to 1 factor increasing for more recent days
      const didStudy = Math.random() < (STUDY_PROBABILITY + (recencyBoost * 0.25));
      
      // Add a chance to break a streak for more realistic data