wehub-resource-sync
48 lines
1.1 KiB
TypeScript
48 lines
1.1 KiB
TypeScript
import { prisma } from "~/db.server";
|
|
import { env } from "~/env.server";
|
|
import { FEATURE_FLAG } from "~/v3/featureFlags";
|
|
import { makeFlag } from "~/v3/featureFlags.server";
|
|
|
|
export async function canAccessAi(options: {
|
|
userId: string;
|
|
isAdmin: boolean;
|
|
isImpersonating: boolean;
|
|
organizationSlug: string;
|
|
}): Promise<boolean> {
|
|
const { userId, isAdmin, isImpersonating, organizationSlug } = options;
|
|
|
|
// 1. If env var is set then globally enabled
|
|
if (env.AI_FEATURES_ENABLED === "1") {
|
|
return true;
|
|
}
|
|
|
|
// 2. Admins always have access
|
|
if (isAdmin || isImpersonating) {
|
|
return true;
|
|
}
|
|
|
|
// 3. Check if org/global feature flag is on
|
|
const org = await prisma.organization.findFirst({
|
|
where: {
|
|
slug: organizationSlug,
|
|
members: { some: { userId } },
|
|
},
|
|
select: {
|
|
featureFlags: true,
|
|
},
|
|
});
|
|
|
|
const flag = makeFlag();
|
|
const flagResult = await flag({
|
|
key: FEATURE_FLAG.hasAiAccess,
|
|
defaultValue: false,
|
|
overrides: (org?.featureFlags as Record<string, unknown>) ?? {},
|
|
});
|
|
if (flagResult) {
|
|
return true;
|
|
}
|
|
|
|
// 4. Not enabled anywhere
|
|
return false;
|
|
}
|