feat: Init

This commit is contained in:
beilunyang
2024-12-16 01:35:08 +08:00
commit cc7e5003c5
73 changed files with 15001 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
import { GET, POST } from "@/lib/auth"
export { GET, POST }
export const runtime = 'edge'

View File

@@ -0,0 +1,43 @@
import { NextResponse } from "next/server"
import { createDb } from "@/lib/db"
import { messages } from "@/lib/schema"
import { and, eq } from "drizzle-orm"
export const runtime = "edge"
export async function GET(request: Request, { params }: { params: Promise<{ id: string; messageId: string }> }) {
try {
const { id, messageId } = await params
const db = createDb()
const message = await db.query.messages.findFirst({
where: and(
eq(messages.id, messageId),
eq(messages.emailId, id)
)
})
if (!message) {
return NextResponse.json(
{ error: "Message not found" },
{ status: 404 }
)
}
return NextResponse.json({
message: {
id: message.id,
from_address: message.fromAddress,
subject: message.subject,
content: message.content,
html: message.html,
received_at: message.receivedAt.getTime()
}
})
} catch (error) {
console.error('Failed to fetch message:', error)
return NextResponse.json(
{ error: "Failed to fetch message" },
{ status: 500 }
)
}
}

View File

@@ -0,0 +1,80 @@
import { NextResponse } from "next/server"
import { createDb } from "@/lib/db"
import { messages } from "@/lib/schema"
import { and, eq, lt, or, sql } from "drizzle-orm"
import { encodeCursor, decodeCursor } from "@/lib/cursor"
export const runtime = "edge"
const PAGE_SIZE = 20
export async function GET(
request: Request,
{ params }: { params: Promise<{ id: string }> }
) {
const { searchParams } = new URL(request.url)
const cursorStr = searchParams.get('cursor')
try {
const db = createDb()
const { id } = await params
const baseConditions = eq(messages.emailId, id)
const totalResult = await db.select({ count: sql<number>`count(*)` })
.from(messages)
.where(baseConditions)
const totalCount = Number(totalResult[0].count)
const conditions = [baseConditions]
if (cursorStr) {
const { timestamp, id } = decodeCursor(cursorStr)
conditions.push(
// @ts-expect-error "ignore the error"
or(
lt(messages.receivedAt, new Date(timestamp)),
and(
eq(messages.receivedAt, new Date(timestamp)),
lt(messages.id, id)
)
)
)
}
const results = await db.query.messages.findMany({
where: and(...conditions),
orderBy: (messages, { desc }) => [
desc(messages.receivedAt),
desc(messages.id)
],
limit: PAGE_SIZE + 1
})
const hasMore = results.length > PAGE_SIZE
const nextCursor = hasMore
? encodeCursor(
results[PAGE_SIZE - 1].receivedAt.getTime(),
results[PAGE_SIZE - 1].id
)
: null
const messageList = hasMore ? results.slice(0, PAGE_SIZE) : results
return NextResponse.json({
messages: messageList.map(msg => ({
id: msg.id,
from_address: msg.fromAddress,
subject: msg.subject,
received_at: msg.receivedAt.getTime()
})),
nextCursor,
total: totalCount
})
} catch (error) {
console.error('Failed to fetch messages:', error)
return NextResponse.json(
{ error: "Failed to fetch messages" },
{ status: 500 }
)
}
}

View File

@@ -0,0 +1,85 @@
import { NextResponse } from "next/server"
import { nanoid } from "nanoid"
import { auth } from "@/lib/auth"
import { createDb } from "@/lib/db"
import { emails } from "@/lib/schema"
import { eq, and, gt, sql } from "drizzle-orm"
import { EXPIRY_OPTIONS } from "@/types/email"
import { EMAIL_CONFIG } from "@/config"
export const runtime = "edge"
export async function POST(request: Request) {
const db = createDb()
const session = await auth()
try {
// Check current number of active emails for user
const activeEmailsCount = await db
.select({ count: sql<number>`count(*)` })
.from(emails)
.where(
and(
eq(emails.userId, session!.user!.id!),
gt(emails.expiresAt, new Date())
)
)
if (Number(activeEmailsCount[0].count) >= EMAIL_CONFIG.MAX_ACTIVE_EMAILS) {
return NextResponse.json(
{ error: `Reached the maximum email limit (${EMAIL_CONFIG.MAX_ACTIVE_EMAILS})` },
{ status: 403 }
)
}
const { name, expiryTime } = await request.json<{
name: string
expiryTime: number
}>()
// Validate expiry time
if (!EXPIRY_OPTIONS.some(option => option.value === expiryTime)) {
return NextResponse.json(
{ error: "Invalid expiry time" },
{ status: 400 }
)
}
const address = `${name || nanoid(8)}@${EMAIL_CONFIG.DOMAIN}`
const existingEmail = await db.query.emails.findFirst({
where: eq(emails.address, address)
})
if (existingEmail) {
return NextResponse.json(
{ error: "Email already exists" },
{ status: 409 }
)
}
const now = new Date()
const expires = new Date(now.getTime() + expiryTime)
const emailData: typeof emails.$inferInsert = {
address,
createdAt: now,
expiresAt: expires,
userId: session!.user!.id
}
const result = await db.insert(emails)
.values(emailData)
.returning({ id: emails.id, address: emails.address })
return NextResponse.json({
id: result[0].id,
email: result[0].address
})
} catch (error) {
console.error('Failed to generate email:', error)
return NextResponse.json(
{ error: "Failed to generate email" },
{ status: 500 }
)
}
}

79
app/api/emails/route.ts Normal file
View File

@@ -0,0 +1,79 @@
import { auth } from "@/lib/auth"
import { createDb } from "@/lib/db"
import { and, eq, gt, lt, or, sql } from "drizzle-orm"
import { NextResponse } from "next/server"
import { emails } from "@/lib/schema"
import { encodeCursor, decodeCursor } from "@/lib/cursor"
export const runtime = "edge"
const PAGE_SIZE = 20
export async function GET(request: Request) {
const session = await auth()
const { searchParams } = new URL(request.url)
const cursor = searchParams.get('cursor')
if (!session?.user?.email) {
return NextResponse.json({ emails: [], nextCursor: null, total: 0 })
}
const db = createDb()
try {
const baseConditions = and(
eq(emails.userId, session.user.id!),
gt(emails.expiresAt, new Date())
)
const totalResult = await db.select({ count: sql<number>`count(*)` })
.from(emails)
.where(baseConditions)
const totalCount = Number(totalResult[0].count)
const conditions = [baseConditions]
if (cursor) {
const { timestamp, id } = decodeCursor(cursor)
conditions.push(
or(
lt(emails.createdAt, new Date(timestamp)),
and(
eq(emails.createdAt, new Date(timestamp)),
lt(emails.id, id)
)
)
)
}
const results = await db.query.emails.findMany({
where: and(...conditions),
orderBy: (emails, { desc }) => [
desc(emails.createdAt),
desc(emails.id)
],
limit: PAGE_SIZE + 1
})
const hasMore = results.length > PAGE_SIZE
const nextCursor = hasMore
? encodeCursor(
results[PAGE_SIZE - 1].createdAt.getTime(),
results[PAGE_SIZE - 1].id
)
: null
const emailList = hasMore ? results.slice(0, PAGE_SIZE) : results
return NextResponse.json({
emails: emailList,
nextCursor,
total: totalCount
})
} catch (error) {
console.error('Failed to fetch user emails:', error)
return NextResponse.json(
{ error: "Failed to fetch emails" },
{ status: 500 }
)
}
}