All files / dal pro-websites.dal.ts

100% Statements 17/17
100% Branches 0/0
100% Functions 10/10
100% Lines 17/17

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 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112              31x     2x         2x       2x         2x       2x         2x       2x       2x             2x               2x             2x               2x       1x             1x                             2x                     2x      
// Data Access Layer for Pro Website Configuration
import { eq } from "drizzle-orm";
import type { DrizzleD1Database } from "drizzle-orm/d1";
import * as schema from "../db/schema";
import type { ProWebsite, NewProWebsite } from "../db/schema";
 
export class ProWebsitesDal {
	constructor(private db: DrizzleD1Database<typeof schema>) {}
 
	async findByProId(proId: string): Promise<ProWebsite | undefined> {
		const result = await this.db
			.select()
			.from(schema.proWebsites)
			.where(eq(schema.proWebsites.proId, proId))
			.limit(1);
		return result[0];
	}
 
	async findByCustomDomain(domain: string): Promise<ProWebsite | undefined> {
		const result = await this.db
			.select()
			.from(schema.proWebsites)
			.where(eq(schema.proWebsites.customDomain, domain))
			.limit(1);
		return result[0];
	}
 
	async findById(id: number): Promise<ProWebsite | undefined> {
		const result = await this.db
			.select()
			.from(schema.proWebsites)
			.where(eq(schema.proWebsites.id, id))
			.limit(1);
		return result[0];
	}
 
	async create(data: NewProWebsite): Promise<ProWebsite | undefined> {
		const result = await this.db
			.insert(schema.proWebsites)
			.values(data)
			.returning();
		return result[0];
	}
 
	async update(
		id: number,
		data: Partial<NewProWebsite>,
	): Promise<ProWebsite | undefined> {
		const result = await this.db
			.update(schema.proWebsites)
			.set({
				...data,
				dateUpdated: new Date(),
			})
			.where(eq(schema.proWebsites.id, id))
			.returning();
		return result[0];
	}
 
	async updateByProId(
		proId: string,
		data: Partial<NewProWebsite>,
	): Promise<ProWebsite | undefined> {
		const result = await this.db
			.update(schema.proWebsites)
			.set({
				...data,
				dateUpdated: new Date(),
			})
			.where(eq(schema.proWebsites.proId, proId))
			.returning();
		return result[0];
	}
 
	async delete(id: number): Promise<void> {
		await this.db
			.delete(schema.proWebsites)
			.where(eq(schema.proWebsites.id, id));
	}
 
	// Mark content as changed (sets hasUnpublishedChanges flag)
	async markContentChanged(proId: string): Promise<void> {
		await this.db
			.update(schema.proWebsites)
			.set({
				hasUnpublishedChanges: true,
				contentLastModifiedAt: new Date(),
				dateUpdated: new Date(),
			})
			.where(eq(schema.proWebsites.proId, proId));
	}
 
	// Mark as published (clears hasUnpublishedChanges flag)
	async markPublished(
		id: number,
		version: number,
	): Promise<ProWebsite | undefined> {
		const result = await this.db
			.update(schema.proWebsites)
			.set({
				status: "published",
				hasUnpublishedChanges: false,
				lastPublishedAt: new Date(),
				publishedVersion: version,
				dateUpdated: new Date(),
			})
			.where(eq(schema.proWebsites.id, id))
			.returning();
		return result[0];
	}
}