auth.ts RawHistoryFolder

1import { query, mutation } from './_generated/server';
2import { customQuery, customMutation } from 'convex-helpers/server/customFunctions';
3import { v } from 'convex/values';
4import type { Doc } from './_generated/dataModel';
5import type { QueryCtx, MutationCtx } from './_generated/server';
6
7// Helper to validate session and get user
8async function validateSession(
9	ctx: QueryCtx | MutationCtx,
10	tokenHash: string
11): Promise<{ session: Doc<'sessions'>; user: Doc<'users'> }> {
12	const session = await ctx.db
13		.query('sessions')
14		.withIndex('by_tokenHash', (q) => q.eq('tokenHash', tokenHash))
15		.first();
16
17	if (!session) {
18		throw new Error('Invalid session');
19	}
20
21	if (session.expiresAt < Date.now()) {
22		throw new Error('Session expired');
23	}
24
25	const user = await ctx.db.get(session.userId);
26	if (!user) {
27		throw new Error('User not found');
28	}
29
30	return { session, user };
31}
32
33// Authenticated query - requires tokenHash, provides user in context
34export const authQuery = customQuery(query, {
35	args: { tokenHash: v.string() },
36	input: async (ctx, { tokenHash }) => {
37		const { session, user } = await validateSession(ctx, tokenHash);
38		return { ctx: { session, user }, args: {} };
39	}
40});
41
42// Authenticated mutation - requires tokenHash, provides user in context
43export const authMutation = customMutation(mutation, {
44	args: { tokenHash: v.string() },
45	input: async (ctx, { tokenHash }) => {
46		const { session, user } = await validateSession(ctx, tokenHash);
47		return { ctx: { session, user }, args: {} };
48	}
49});
50
51// Type helpers for use in handlers
52export type AuthQueryCtx = QueryCtx & { session: Doc<'sessions'>; user: Doc<'users'> };
53export type AuthMutationCtx = MutationCtx & { session: Doc<'sessions'>; user: Doc<'users'> };