feat: Init

scripts/generate-test-data.ts

@@ -0,0 +1,76 @@
+import { drizzle } from 'drizzle-orm/d1'
+import { emails, messages } from '../app/lib/schema'
+import { nanoid } from 'nanoid'
+import { EMAIL_CONFIG} from '../app/config'
+
+const TEST_USER_ID = '4e4c1d5d-a3c9-407a-8808-2a2424b38c62'
+
+interface Env {
+  DB: D1Database
+  NEXT_PUBLIC_EMAIL_DOMAIN: string
+}
+
+const MAX_EMAIL_COUNT = 5
+const MAX_MESSAGE_COUNT = 50
+const BATCH_SIZE = 10 // SQLite 变量限制
+
+async function generateTestData(env: Env) {
+  const db = drizzle(env.DB)
+  const now = new Date()
+
+  try {
+    // 生成测试邮箱
+    const testEmails = Array.from({ length: MAX_EMAIL_COUNT }).map(() => ({
+      id: crypto.randomUUID(),
+      address: `${nanoid(6)}@${EMAIL_CONFIG.DOMAIN}`,
+      userId: TEST_USER_ID,
+      createdAt: now,
+      expiresAt: new Date(now.getTime() + 24 * 60 * 60 * 1000),
+    }))
+
+    // 插入测试邮箱
+    const emailResults = await db.insert(emails).values(testEmails).returning()
+    console.log('Created test emails:', emailResults)
+
+    // 为每个邮箱生成测试消息
+    for (const email of emailResults) {
+      const allMessages = Array.from({ length: MAX_MESSAGE_COUNT }).map((_, index) => ({
+        id: crypto.randomUUID(),
+        emailId: email.id,
+        fromAddress: `sender${index + 1}@example.com`,
+        subject: `Test Message ${index + 1} - ${nanoid(6)}`,
+        content: `This is test message ${index + 1} content.\n\nBest regards,\nSender ${index + 1}`,
+        html: `<div>
+          <h1>Test Message ${index + 1}</h1>
+          <p>This is test message ${index + 1} content.</p>
+          <p>With some <strong>HTML</strong> formatting.</p>
+          <br>
+          <p>Best regards,<br>Sender ${index + 1}</p>
+        </div>`,
+        receivedAt: new Date(now.getTime() - index * 60 * 60 * 1000),
+      }))
+
+      // 分批插入消息
+      for (let i = 0; i < allMessages.length; i += BATCH_SIZE) {
+        const batch = allMessages.slice(i, i + BATCH_SIZE)
+        await db.insert(messages).values(batch)
+        console.log(`Created batch of ${batch.length} messages for email ${email.address}`)
+      }
+    }
+
+    console.log('Test data generation completed successfully!')
+  } catch (error) {
+    console.error('Failed to generate test data:', error)
+  }
+}
+
+// eslint-disable-next-line import/no-anonymous-default-export
+export default {
+  async fetch(request: Request, env: Env) {
+    if (request.method === 'GET') {
+      await generateTestData(env)
+      return new Response('Test data generated successfully', { status: 200 })
+    }
+    return new Response('Method not allowed', { status: 405 })
+  }
+}