#!/usr/bin/env node

/**
 * 创建付费测试用户的脚本
 * 
 * 使用方法：
 * node scripts/create-premium-user.js [email] [password] [displayName]
 * 
 * 例子：
 * node scripts/create-premium-user.js premium@test.com TestPass123 "Premium User"
 */

const { Client } = require('pg');
const path = require('path');
require('dotenv').config({ path: path.join(__dirname, '..', '.env') });

async function createPremiumUser(email, password, displayName) {
  const client = new Client({
    connectionString: process.env.DATABASE_URL
  });

  try {
    await client.connect();
    console.log('✅ Connected to database');

    // 1. 首先注册普通用户
    console.log(`\n📝 Registering user: ${email}`);
    
    const registerResponse = await fetch('http://localhost:3001/api/v1/auth/register-email', {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        email,
        password,
        displayName
      })
    });

    if (!registerResponse.ok) {
      const error = await registerResponse.json();
      throw new Error(`Registration failed: ${error.message}`);
    }

    const registerData = await registerResponse.json();
    const userId = registerData.userId;
    
    console.log(`✅ User registered successfully with ID: ${userId}`);

    // 2. 更新用户为付费用户
    const fakeCustomerId = 'cus_test_premium_' + Date.now();
    const fakeSubscriptionId = 'sub_test_premium_' + Date.now();

    const updateQuery = `
      UPDATE user_profiles 
      SET 
        payment_provider_customer_id = $1,
        payment_provider_subscription_id = $2,
        updated_at = NOW()
      WHERE id = $3
    `;

    const result = await client.query(updateQuery, [fakeCustomerId, fakeSubscriptionId, userId]);
    
    if (result.rowCount > 0) {
      console.log('✅ User upgraded to premium successfully!');
      
      // 3. 测试登录和profile获取
      console.log('\n🔐 Testing login...');
      const loginResponse = await fetch('http://localhost:3001/api/v1/auth/login-email', {
        method: 'POST',
        headers: {
          'Content-Type': 'application/json',
        },
        body: JSON.stringify({
          email,
          password
        })
      });

      if (loginResponse.ok) {
        const loginData = await loginResponse.json();
        console.log('✅ Login successful');
        
        // 测试profile
        const profileResponse = await fetch('http://localhost:3001/api/v1/auth/profile', {
          method: 'GET',
          headers: {
            'Authorization': `Bearer ${loginData.accessToken}`,
            'Content-Type': 'application/json',
          }
        });

        if (profileResponse.ok) {
          const profileData = await profileResponse.json();
          console.log('✅ Profile retrieved successfully');
          
          console.log('\n🎉 Premium user created successfully!');
          console.log('='.repeat(50));
          console.log(`👤 User Details:`);
          console.log(`   Email: ${email}`);
          console.log(`   Password: ${password}`);
          console.log(`   Display Name: ${displayName}`);
          console.log(`   User ID: ${userId}`);
          console.log(`   Customer ID: ${fakeCustomerId}`);
          console.log(`   Subscription ID: ${fakeSubscriptionId}`);
          console.log(`   Subscription Status: ${profileData.subscriptionStatus}`);
          console.log(`   Has Active Subscription: ${profileData.hasActiveSubscription}`);
          console.log('='.repeat(50));
        }
      }
    } else {
      console.log('❌ Failed to upgrade user to premium');
    }

  } catch (error) {
    console.error('❌ Error:', error.message);
    process.exit(1);
  } finally {
    await client.end();
  }
}

// 命令行参数处理
const args = process.argv.slice(2);
const email = args[0] || 'premium' + Date.now() + '@test.com';
const password = args[1] || 'TestPassword123';
const displayName = args[2] || 'Premium User';

console.log('🚀 Creating Premium User...');
console.log(`Email: ${email}`);
console.log(`Password: ${password}`);
console.log(`Display Name: ${displayName}`);

createPremiumUser(email, password, displayName);