import { router, publicProcedure } from "./trpc";
import { db } from "./db";
import { user, userOpinion } from "./schema";
import { count, eq } from "drizzle-orm";
import { z } from "zod";

export const adminRoute = router({
  totalUser: publicProcedure.query(async () => {
    let rs = await db
      .select({ zone: user.zone, value: count(user.id) })
      .from(user)
      .groupBy(user.zone)
      .execute();
    let zones = await db.query.zone
      .findMany({ with: { province: true } })
      .execute();
    zones.sort((a, b) => a.province.id - b.province.id);
    let summary = zones.map((z) => {
      let num = rs.find((user) => user.zone == z.id)?.value ?? 0;
      return {
        count: num,
        zone: z.name,
        province: z.province.name,
      };
    });
    return summary;
  }),

  removeUser: publicProcedure
    .input(z.object({ cid: z.string(), key: z.string() }))
    .mutation(async ({ input }) => {
      if (input.key !== "3RJjV7Hseo2xiJoVta/x2AJIGw5EK+a5nAwtnAjw37U=") {
        return "Invalid Key";
      }
      let thisUser = await db.query.user
        .findFirst({ where: eq(user.cid, input.cid) })
        .execute();
      if (thisUser === undefined) {
        return "User not found";
      }
      let uoresult = await db
        .delete(userOpinion)
        .where(eq(userOpinion.userId, thisUser.id))
        .execute();
      let rs = await db.delete(user).where(eq(user.cid, input.cid)).execute();
      return { useropinion: uoresult, rs };
    }),
});