All files / dal lead-sources.dal.ts

100% Statements 16/16
100% Branches 5/5
100% Functions 10/10
100% Lines 16/16

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93              20x           4x 1x         3x                       1x       2x         2x       1x       1x               3x                       6x         6x       2x       2x       3x       3x      
// Data Access Layer for Lead Sources
import { eq, and, sql } from "drizzle-orm";
import type { DrizzleD1Database } from "drizzle-orm/d1";
import * as schema from "../db/schema";
import type { LeadSource, NewLeadSource } from "../db/schema";
 
export class LeadSourcesDal {
	constructor(private db: DrizzleD1Database<typeof schema>) {}
 
	async findByProId(
		proId: string,
		includeInactive = false,
	): Promise<LeadSource[]> {
		if (includeInactive) {
			return this.db
				.select()
				.from(schema.leadSources)
				.where(eq(schema.leadSources.proId, proId));
		}
		return this.db
			.select()
			.from(schema.leadSources)
			.where(
				and(
					eq(schema.leadSources.proId, proId),
					eq(schema.leadSources.isActive, true),
				),
			);
	}
 
	async findActiveByProId(proId: string): Promise<LeadSource[]> {
		return this.findByProId(proId, false);
	}
 
	async findById(id: number): Promise<LeadSource | undefined> {
		const result = await this.db
			.select()
			.from(schema.leadSources)
			.where(eq(schema.leadSources.id, id))
			.limit(1);
		return result[0];
	}
 
	async create(data: NewLeadSource): Promise<LeadSource> {
		const result = await this.db
			.insert(schema.leadSources)
			.values(data)
			.returning();
		return result[0];
	}
 
	/**
	 * Insert if no row exists for this (pro_id, name).
	 * Uses onConflictDoNothing to handle concurrent initialization safely.
	 */
	async createIfNotExists(data: NewLeadSource): Promise<void> {
		await this.db
			.insert(schema.leadSources)
			.values(data)
			.onConflictDoNothing({
				target: [schema.leadSources.proId, schema.leadSources.name],
			});
	}
 
	async update(
		id: number,
		data: Partial<Omit<LeadSource, "id" | "dateCreated">>,
	): Promise<LeadSource | undefined> {
		const result = await this.db
			.update(schema.leadSources)
			.set(data)
			.where(eq(schema.leadSources.id, id))
			.returning();
		return result[0];
	}
 
	async deactivate(id: number): Promise<LeadSource | undefined> {
		return this.update(id, { isActive: false });
	}
 
	async reactivate(id: number): Promise<LeadSource | undefined> {
		return this.update(id, { isActive: true });
	}
 
	async countByProId(proId: string): Promise<number> {
		const result = await this.db
			.select({ count: sql<number>`count(*)` })
			.from(schema.leadSources)
			.where(eq(schema.leadSources.proId, proId));
		return result[0]?.count ?? 0;
	}
}