moemail/app/api/roles/init-emperor/route.ts

import { auth, assignRoleToUser } from "@/lib/auth";
import { createDb } from "@/lib/db";
import { roles, userRoles } from "@/lib/schema";
import { ROLES } from "@/lib/permissions";
import { eq } from "drizzle-orm";

export const runtime = "edge";

export async function GET() {
  const session = await auth();
  if (!session?.user?.id) {
    return Response.json({ error: "未授权" }, { status: 401 });
  }

  const db = createDb();

  const emperorRole = await db.query.roles.findFirst({
    where: eq(roles.name, ROLES.EMPEROR),
    with: {
      userRoles: true,
    },
  });

  if (emperorRole && emperorRole.userRoles.length > 0) {
    return Response.json({ error: "已存在皇帝, 谋反将被处死" }, { status: 400 });
  }

  try {
    const currentUserRole = await db.query.userRoles.findFirst({
      where: eq(userRoles.userId, session.user.id),
      with: {
        role: true,
      },
    });

    if (currentUserRole?.role.name === ROLES.EMPEROR) {
      return Response.json({ message: "你已经是皇帝了" });
    }

    let roleId = emperorRole?.id;
    if (!roleId) {
      const [newRole] = await db.insert(roles)
        .values({
          name: ROLES.EMPEROR,
          description: "皇帝（网站所有者）",
        })
        .returning({ id: roles.id });
      roleId = newRole.id;
    }

    await assignRoleToUser(db, session.user.id, roleId);

    return Response.json({ message: "登基成功，你已成为皇帝" });
  } catch (error) {
    console.error("Failed to initialize emperor:", error);
    return Response.json(
      { error: "登基称帝失败" },
      { status: 500 }
    );
  }
}