import {
  and,
  asc,
  count,
  desc,
  eq,
  getOrderByFromSort,
  ilike,
  isNull,
  or,
  sql,
} from "@turbostarter/db";
import { invite, mesh, meshMember } from "@turbostarter/db/schema";
import { db } from "@turbostarter/db/server";

import type { GetMyMeshesInput } from "../../schema";

export const getMyMeshes = async ({
  userId,
  ...input
}: GetMyMeshesInput & { userId: string }) => {
  const offset = (input.page - 1) * input.perPage;

  // User sees: meshes they own OR meshes where they have a meshMember row
  const baseWhere = or(
    eq(mesh.ownerUserId, userId),
    sql`EXISTS (SELECT 1 FROM mesh.member mm WHERE mm.mesh_id = ${mesh.id} AND mm.user_id = ${userId} AND mm.revoked_at IS NULL)`,
  );

  const where = and(
    baseWhere,
    input.q
      ? or(ilike(mesh.name, `%${input.q}%`), ilike(mesh.slug, `%${input.q}%`))
      : undefined,
  );

  const orderBy = input.sort
    ? getOrderByFromSort({ sort: input.sort, defaultSchema: mesh })
    : [desc(mesh.createdAt)];

  return db.transaction(async (tx) => {
    const data = await tx
      .select({
        id: mesh.id,
        name: mesh.name,
        slug: mesh.slug,
        visibility: mesh.visibility,
        transport: mesh.transport,
        tier: mesh.tier,
        createdAt: mesh.createdAt,
        archivedAt: mesh.archivedAt,
        isOwner: sql<boolean>`${mesh.ownerUserId} = ${userId}`,
        myRole: sql<"admin" | "member">`CASE WHEN ${mesh.ownerUserId} = ${userId} THEN 'admin'::text ELSE COALESCE((SELECT role::text FROM mesh.member mm2 WHERE mm2.mesh_id = ${mesh.id} AND mm2.user_id = ${userId} AND mm2.revoked_at IS NULL LIMIT 1), 'member') END`,
        memberCount: sql<number>`(SELECT COUNT(*)::int FROM mesh.member mm3 WHERE mm3.mesh_id = ${mesh.id} AND mm3.revoked_at IS NULL)`,
      })
      .from(mesh)
      .where(where)
      .limit(input.perPage)
      .offset(offset)
      .orderBy(...orderBy);

    const total = await tx
      .select({ count: count() })
      .from(mesh)
      .where(where)
      .execute()
      .then((res) => res[0]?.count ?? 0);

    return { data, total };
  });
};

export const getMyMeshById = async ({
  userId,
  meshId,
}: {
  userId: string;
  meshId: string;
}) => {
  const [m] = await db
    .select({
      id: mesh.id,
      name: mesh.name,
      slug: mesh.slug,
      visibility: mesh.visibility,
      transport: mesh.transport,
      tier: mesh.tier,
      maxPeers: mesh.maxPeers,
      createdAt: mesh.createdAt,
      archivedAt: mesh.archivedAt,
      ownerUserId: mesh.ownerUserId,
    })
    .from(mesh)
    .where(eq(mesh.id, meshId))
    .limit(1);

  if (!m) return null;

  // Authz: user must own OR be a non-revoked member
  const isOwner = m.ownerUserId === userId;
  if (!isOwner) {
    const [membership] = await db
      .select({ id: meshMember.id, role: meshMember.role })
      .from(meshMember)
      .where(
        and(
          eq(meshMember.meshId, meshId),
          eq(meshMember.userId, userId),
          isNull(meshMember.revokedAt),
        ),
      )
      .limit(1);
    if (!membership) return null;
  }

  const members = await db
    .select({
      id: meshMember.id,
      displayName: meshMember.displayName,
      role: meshMember.role,
      joinedAt: meshMember.joinedAt,
      lastSeenAt: meshMember.lastSeenAt,
      revokedAt: meshMember.revokedAt,
      userId: meshMember.userId,
    })
    .from(meshMember)
    .where(eq(meshMember.meshId, meshId))
    .orderBy(asc(meshMember.joinedAt));

  const invites = await db
    .select({
      id: invite.id,
      token: invite.token,
      maxUses: invite.maxUses,
      usedCount: invite.usedCount,
      role: invite.role,
      expiresAt: invite.expiresAt,
      createdAt: invite.createdAt,
      revokedAt: invite.revokedAt,
    })
    .from(invite)
    .where(eq(invite.meshId, meshId))
    .orderBy(desc(invite.createdAt))
    .limit(50);

  // Derive myRole for the mesh top-level field
  const myRole: "admin" | "member" = isOwner
    ? "admin"
    : (members.find((mem) => mem.userId === userId)?.role ?? "member");

  return {
    mesh: { ...m, isOwner, myRole },
    members: members.map((mem) => ({
      id: mem.id,
      displayName: mem.displayName,
      role: mem.role,
      joinedAt: mem.joinedAt,
      lastSeenAt: mem.lastSeenAt,
      revokedAt: mem.revokedAt,
      isMe: mem.userId === userId,
    })),
    invites,
  };
};

export const getMyInvitesSent = async ({ userId }: { userId: string }) =>
  db
    .select({
      id: invite.id,
      meshId: invite.meshId,
      meshName: mesh.name,
      meshSlug: mesh.slug,
      token: invite.token,
      role: invite.role,
      maxUses: invite.maxUses,
      usedCount: invite.usedCount,
      expiresAt: invite.expiresAt,
      createdAt: invite.createdAt,
      revokedAt: invite.revokedAt,
    })
    .from(invite)
    .leftJoin(mesh, eq(invite.meshId, mesh.id))
    .where(eq(invite.createdBy, userId))
    .orderBy(desc(invite.createdAt))
    .limit(100);