71 lines
2.5 KiB
TypeScript
71 lines
2.5 KiB
TypeScript
|
import { eq } from 'drizzle-orm';
|
||
|
|
|
||
|
|
import { db } from '@/lib/db';
|
||
|
|
import { env } from '@/lib/env';
|
||
|
|
import { trackedLinks, type NewTrackedLink } from '@/lib/db/schema/tracked-links';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Phase 4c — service-layer helpers for tracked redirect links. Use
|
||
|
|
* `createTrackedLink` from any email-composer flow to wrap an outbound
|
||
|
|
* URL in a `/q/<slug>` short-link that records click-throughs.
|
||
|
|
*
|
||
|
|
* Slug format: random URL-safe ID. Short enough not to overwhelm an
|
||
|
|
* inbox preview pane but long enough that collision probability is
|
||
|
|
* negligible across the lifetime of the system.
|
||
|
|
*/
|
||
|
|
|
||
|
|
function generateSlug(): string {
|
||
|
|
// 8 random bytes → 11-char base64url string. Collision probability
|
||
|
|
// across 1M links: ~1e-7. The DB unique index is the backstop.
|
||
|
|
const bytes = crypto.getRandomValues(new Uint8Array(8));
|
||
|
|
return btoa(String.fromCharCode(...bytes))
|
||
|
|
.replace(/\+/g, '-')
|
||
|
|
.replace(/\//g, '_')
|
||
|
|
.replace(/=+$/, '');
|
||
|
|
}
|
||
|
|
|
||
|
|
export interface CreateTrackedLinkInput {
|
||
|
|
portId: string;
|
||
|
|
targetUrl: string;
|
||
|
|
/** Optional FK to `document_sends.id` so per-email click-throughs are
|
||
|
|
* attributable. Leave null for one-off short links. */
|
||
|
|
sendId?: string;
|
||
|
|
createdByUserId?: string;
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function createTrackedLink(input: CreateTrackedLinkInput) {
|
||
|
|
// Retry on slug collision (extremely rare). Three attempts is more
|
||
|
|
// than enough — at our slug entropy a single collision in 1M links
|
||
|
|
// would be a once-per-century event.
|
||
|
|
for (let attempt = 0; attempt < 3; attempt++) {
|
||
|
|
const slug = generateSlug();
|
||
|
|
try {
|
||
|
|
const values: NewTrackedLink = {
|
||
|
|
portId: input.portId,
|
||
|
|
slug,
|
||
|
|
targetUrl: input.targetUrl,
|
||
|
|
...(input.sendId ? { sendId: input.sendId } : {}),
|
||
|
|
...(input.createdByUserId ? { createdByUserId: input.createdByUserId } : {}),
|
||
|
|
};
|
||
|
|
const [row] = await db.insert(trackedLinks).values(values).returning();
|
||
|
|
return row!;
|
||
|
|
} catch (err) {
|
||
|
|
const msg = err instanceof Error ? err.message : String(err);
|
||
|
|
if (msg.includes('uniq_tracked_links_slug') && attempt < 2) continue;
|
||
|
|
throw err;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
throw new Error('Failed to mint a unique tracked-link slug after 3 attempts');
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Build the public-facing tracked URL for an existing record. */
|
||
|
|
export function buildTrackedUrl(slug: string): string {
|
||
|
|
const base = env.NEXT_PUBLIC_APP_URL.replace(/\/$/, '');
|
||
|
|
return `${base}/q/${slug}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
/** Look up click stats for a single tracked link. */
|
||
|
|
export async function getTrackedLink(id: string) {
|
||
|
|
return db.query.trackedLinks.findFirst({ where: eq(trackedLinks.id, id) });
|
||
|
|
}
|