first

packages/database/drizzle/queries/organizations.ts

@@ -0,0 +1,105 @@
+import { and, eq } from "drizzle-orm";
+import type { z } from "zod";
+import { db } from "../client";
+import { member, organization } from "../schema/postgres";
+import type { OrganizationUpdateSchema } from "../zod";
+
+export async function getOrganizations({
+	limit,
+	offset,
+	query,
+}: {
+	limit: number;
+	offset: number;
+	query?: string;
+}) {
+	return db.query.organization.findMany({
+		where: query
+			? (org, { like }) => like(org.name, `%${query}%`)
+			: undefined,
+		limit,
+		offset,
+		extras: {
+			membersCount: db
+				.$count(member, eq(member.organizationId, organization.id))
+				.as("membersCount"),
+		},
+	});
+}
+
+export async function countAllOrganizations() {
+	return db.$count(organization);
+}
+
+export async function getOrganizationById(id: string) {
+	return db.query.organization.findFirst({
+		where: (org, { eq }) => eq(org.id, id),
+		with: {
+			members: true,
+			invitations: true,
+		},
+	});
+}
+
+export async function getInvitationById(id: string) {
+	return db.query.invitation.findFirst({
+		where: (invitation, { eq }) => eq(invitation.id, id),
+		with: {
+			organization: true,
+		},
+	});
+}
+
+export async function getOrganizationBySlug(slug: string) {
+	return db.query.organization.findFirst({
+		where: (org, { eq }) => eq(org.slug, slug),
+	});
+}
+
+export async function getOrganizationMembership(
+	organizationId: string,
+	userId: string,
+) {
+	return db.query.member.findFirst({
+		where: (member, { eq }) =>
+			and(
+				eq(member.organizationId, organizationId),
+				eq(member.userId, userId),
+			),
+		with: {
+			organization: true,
+		},
+	});
+}
+
+export async function getOrganizationWithPurchasesAndMembersCount(
+	organizationId: string,
+) {
+	return db.query.organization.findFirst({
+		where: (org, { eq }) => eq(org.id, organizationId),
+		with: {
+			purchases: true,
+		},
+		extras: {
+			membersCount: db
+				.$count(member, eq(member.organizationId, organization.id))
+				.as("membersCount"),
+		},
+	});
+}
+
+export async function getPendingInvitationByEmail(email: string) {
+	return db.query.invitation.findFirst({
+		where: (invitation, { eq }) =>
+			and(eq(invitation.email, email), eq(invitation.status, "pending")),
+	});
+}
+
+export async function updateOrganization(
+	updatedOrganization: z.infer<typeof OrganizationUpdateSchema>,
+) {
+	return db
+		.update(organization)
+		.set(updatedOrganization)
+		.where(eq(organization.id, updatedOrganization.id));
+}