Spaces:
Sleeping
Sleeping
File size: 2,398 Bytes
05c5ed5 4862750 05c5ed5 | 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 | import { z } from "zod";
import { passwordSchema } from "lib/validations/password";
import { UserEntity } from "lib/db/pg/schema.pg";
import { getSession } from "auth/server";
export type UserPreferences = {
displayName?: string;
profession?: string; // User's job or profession
responseStyleExample?: string; // Example of preferred response style
botName?: string; // Name of the bot
apiKeys?: Record<string, string>; // Store API keys per provider
};
// user without password
export interface User extends Omit<UserEntity, "password"> {
preferences: UserPreferences | null;
lastLogin?: Date | null;
}
export type BasicUser = Omit<
User,
| "password"
| "preferences"
| "image"
| "role"
| "banned"
| "banReason"
| "banExpires"
> & {
image?: string | null;
role?: string | null;
banned?: boolean | null;
banReason?: string | null;
banExpires?: Date | null;
};
export interface BasicUserWithLastLogin extends BasicUser {
lastLogin: Date | null;
}
export type UserSession = NonNullable<Awaited<ReturnType<typeof getSession>>>;
export type UserSessionUser = UserSession["user"];
export type UserRepository = {
existsByEmail: (email: string) => Promise<boolean>;
updateUserDetails: (data: {
userId: string;
name?: string;
email?: string;
image?: string;
}) => Promise<User>;
updatePreferences: (
userId: string,
preferences: UserPreferences,
) => Promise<User>;
getPreferences: (userId: string) => Promise<UserPreferences | null>;
getUserById: (userId: string) => Promise<BasicUserWithLastLogin | null>;
getUserCount: () => Promise<number>;
getUserStats: (userId: string) => Promise<{
threadCount: number;
messageCount: number;
modelStats: Array<{
model: string;
messageCount: number;
totalTokens: number;
}>;
totalTokens: number;
period: string;
}>;
getUserAuthMethods: (userId: string) => Promise<{
hasPassword: boolean;
oauthProviders: string[];
}>;
};
export const UserZodSchema = z.object({
name: z.string().min(1),
email: z.string().email(),
password: passwordSchema,
});
export const UserPreferencesZodSchema = z.object({
displayName: z.string().optional(),
profession: z.string().optional(),
responseStyleExample: z.string().optional(),
botName: z.string().optional(),
apiKeys: z.record(z.string(), z.string()).optional(),
});
|