move things to ee

Author: brendan-kellam

packages/shared/src/entitlements.ts

@@ -38,13 +38,14 @@ const entitlements = [
     "audit",
     "analytics",
     "permission-syncing",
-    "github-app"
+    "github-app",
+    "chat-sharing"
 ] as const;
 export type Entitlement = (typeof entitlements)[number];
 
 const entitlementsByPlan: Record<Plan, Entitlement[]> = {
     oss: [
-        "anonymous-access"
+        "anonymous-access",
     ],
     "self-hosted:enterprise": [
         "search-contexts",
@@ -53,17 +54,19 @@ const entitlementsByPlan: Record<Plan, Entitlement[]> = {
         "audit",
         "analytics",
         "permission-syncing",
-        "github-app"
+        "github-app",
+        "chat-sharing",
     ],
     "self-hosted:enterprise-unlimited": [
+        "anonymous-access",
         "search-contexts",
         "sso",
         "code-nav",
         "audit",
         "analytics",
         "permission-syncing",
         "github-app",
-        "anonymous-access"
+        "chat-sharing",
     ],
 } as const;
 

packages/web/src/app/[domain]/chat/[id]/page.tsx

@@ -17,6 +17,7 @@ import { getOrgFromDomain } from '@/data/org';
 import { ChatVisibility } from '@sourcebot/db';
 import { Metadata } from 'next';
 import { SBChatMessage } from '@/features/chat/types';
+import { env, hasEntitlement } from '@sourcebot/shared';
 
 interface PageProps {
     params: Promise<{
@@ -126,6 +127,8 @@ export default async function Page(props: PageProps) {
 
     const indexedRepos = repos.filter((repo) => repo.indexedAt !== undefined);
 
+    const hasChatSharingEntitlement = hasEntitlement('chat-sharing');
+
     return (
         <div className="flex flex-col h-screen w-screen">
             <TopBar
@@ -145,6 +148,10 @@ export default async function Page(props: PageProps) {
                         visibility={visibility}
                         currentUser={session?.user}
                         sharedWithUsers={sharedWithUsers}
+                        isChatSharingEnabledInCurrentPlan={hasChatSharingEntitlement}
+                        // Disable chat sharing for the askgh experiment since we
+                        // don't want to allow users to search other members.
+                        isChatSharingEnabled={env.EXPERIMENT_ASK_GH_ENABLED === 'false'}
                     />
                 ) : undefined}
             />

…ponents/shareChatPopover/invitePanel.tsx …ents/shareChatPopover/ee/invitePanel.tsxpackages/web/src/app/[domain]/chat/components/shareChatPopover/invitePanel.tsx renamed to packages/web/src/app/[domain]/chat/components/shareChatPopover/ee/invitePanel.tsx packages/web/src/app/[domain]/chat/components/shareChatPopover/invitePanel.tsx renamed to packages/web/src/app/[domain]/chat/components/shareChatPopover/ee/invitePanel.tsx

@@ -1,7 +1,7 @@
 'use client';
 
 import { searchChatShareableMembers } from "@/app/api/(client)/client";
-import { SearchChatShareableMembersResponse } from "@/app/api/(server)/chat/[chatId]/searchMembers/route";
+import { SearchChatShareableMembersResponse } from "@/app/api/(server)/ee/chat/[chatId]/searchMembers/route";
 import { SessionUser } from "@/auth";
 import { Avatar, AvatarFallback, AvatarImage } from "@/components/ui/avatar";
 import { Badge } from "@/components/ui/badge";

packages/web/src/app/[domain]/chat/components/shareChatPopover/index.tsx

@@ -14,13 +14,15 @@ import { useToast } from "@/components/hooks/use-toast";
 import { isServiceError } from "@/lib/utils";
 import { Link2Icon, LockIcon } from "lucide-react";
 import { SessionUser } from "@/auth";
-import { InvitePanel } from "./invitePanel";
+import { InvitePanel } from "./ee/invitePanel";
 
 interface ShareChatPopoverProps {
     chatId: string;
     visibility: ChatVisibility;
     currentUser?: SessionUser;
     sharedWithUsers: SessionUser[];
+    isChatSharingEnabledInCurrentPlan: boolean;
+    isChatSharingEnabled: boolean;
 }
 
 type View = 'main' | 'invite';
@@ -29,7 +31,9 @@ export const ShareChatPopover = ({
     chatId,
     visibility: _visibility,
     currentUser,
-    sharedWithUsers: _sharedWithUsers
+    sharedWithUsers: _sharedWithUsers,
+    isChatSharingEnabledInCurrentPlan,
+    isChatSharingEnabled,
 }: ShareChatPopoverProps) => {
     const [visibility, setVisibility] = useState(_visibility);
     const [sharedWithUsers, setSharedWithUsers] = useState(_sharedWithUsers);
@@ -131,14 +135,16 @@ export const ShareChatPopover = ({
                         currentUser={currentUser}
                         sharedWithUsers={sharedWithUsers}
                         onOpenInviteView={() => setView('invite')}
+                        isChatSharingEnabledInCurrentPlan={isChatSharingEnabledInCurrentPlan}
+                        isChatSharingEnabled={isChatSharingEnabled}
                     />
-                ) : (
+                ) : (isChatSharingEnabledInCurrentPlan && isChatSharingEnabled)? (
                     <InvitePanel
                         chatId={chatId}
                         onBack={() => setView('main')}
                         onShareChatWithUsers={onShareChatWithUsers}
                     />
-                )}
+                ) : null}
             </PopoverContent>
         </Popover>
     );

packages/web/src/app/[domain]/chat/components/shareChatPopover/shareSettings.tsx

@@ -12,6 +12,7 @@ import {
     SelectValue,
 } from "@/components/ui/select";
 import { Separator } from "@/components/ui/separator";
+import { cn } from "@/lib/utils";
 import placeholderAvatar from "@/public/placeholder_avatar.png";
 import { ChatVisibility } from "@sourcebot/db";
 import { Info, Link2Icon, Loader2, Lock, X } from "lucide-react";
@@ -26,6 +27,8 @@ interface ShareSettingsProps {
     onOpenInviteView: () => void;
     currentUser?: SessionUser;
     sharedWithUsers: SessionUser[];
+    isChatSharingEnabledInCurrentPlan: boolean;
+    isChatSharingEnabled: boolean;
 }
 
 export const ShareSettings = ({
@@ -35,6 +38,8 @@ export const ShareSettings = ({
     onOpenInviteView,
     currentUser,
     sharedWithUsers,
+    isChatSharingEnabledInCurrentPlan,
+    isChatSharingEnabled,
 }: ShareSettingsProps) => {
     const [isVisibilityUpdating, setIsVisibilityUpdating] = useState(false);
     const [removingUserIds, setRemovingUserIds] = useState<Set<string>>(new Set());
@@ -65,18 +70,26 @@ export const ShareSettings = ({
             <Separator className="-mx-4 w-auto mt-2 mb-4" />
 
             {/* Fake Search Bar - Click to open invite view */}
-            {isAuthenticated && (
+            {(isAuthenticated && isChatSharingEnabled) && (
                 <>
-                    <Button
-                        variant="outline"
-                        className="w-full justify-start text-muted-foreground font-normal"
-                        onClick={onOpenInviteView}
-                    >
-                        Search for a user
-                    </Button>
+                    <span className={cn({ "cursor-not-allowed": !isChatSharingEnabledInCurrentPlan })}>
+                        <Button
+                            variant="outline"
+                            className={cn("w-full justify-start text-muted-foreground font-normal", {
+                                "opacity-50 pointer-events-none": !isChatSharingEnabledInCurrentPlan,
+                            })}
+                            onClick={onOpenInviteView}
+                            disabled={!isChatSharingEnabledInCurrentPlan}
+                        >
+                            Search for a user
+                        </Button>
+                    </span>
 
                     {/* People with access */}
-                    <div className="mt-4">
+                    <div className={cn("mt-4", {
+                        "opacity-50 cursor-not-allowed": !isChatSharingEnabledInCurrentPlan,
+                        "[&_*]:pointer-events-none": !isChatSharingEnabledInCurrentPlan,
+                    })}>
                         <label className="text-sm text-muted-foreground mb-2 block">
                             People with access
                         </label>
@@ -143,6 +156,13 @@ export const ShareSettings = ({
                         </div>
                     </div>
 
+                    {!isChatSharingEnabledInCurrentPlan && (
+                        <p className="text-xs text-muted-foreground mt-2.5">
+                            <Info className="h-3 w-3 inline-block mr-1.5 align-middle" />
+                            <span className="align-middle">Sharing with specific users is not available on your current <Link href={'https://sourcebot.dev/pricing'} target="_blank" rel="noopener noreferrer" className="underline">plan</Link>.</span>
+                        </p>
+                    )}
+
                     <Separator className="-mx-4 w-auto my-4" />
                 </>
             )}

packages/web/src/app/api/(client)/client.ts

@@ -23,7 +23,7 @@ import { PermissionSyncStatusResponse } from "../(server)/ee/permissionSyncStatu
 import {
     SearchChatShareableMembersQueryParams,
     SearchChatShareableMembersResponse,
-} from "../(server)/chat/[chatId]/searchMembers/route";
+} from "../(server)/ee/chat/[chatId]/searchMembers/route";
 
 export const search = async (body: SearchRequest): Promise<SearchResponse | ServiceError> => {
     const result = await fetch("/api/search", {
@@ -144,7 +144,7 @@ export const searchChatShareableMembers = async (
     params: SearchChatShareableMembersQueryParams & { chatId: string },
     signal?: AbortSignal
 ): Promise<SearchChatShareableMembersResponse | ServiceError> => {
-    const url = new URL(`/api/chat/${params.chatId}/searchMembers`, window.location.origin);
+    const url = new URL(`/api/ee/chat/${params.chatId}/searchMembers`, window.location.origin);
     if (params.query) {
         url.searchParams.set('query', params.query);
     }

…er)/chat/[chatId]/searchMembers/route.ts …/ee/chat/[chatId]/searchMembers/route.tspackages/web/src/app/api/(server)/chat/[chatId]/searchMembers/route.ts renamed to packages/web/src/app/api/(server)/ee/chat/[chatId]/searchMembers/route.ts packages/web/src/app/api/(server)/chat/[chatId]/searchMembers/route.ts renamed to packages/web/src/app/api/(server)/ee/chat/[chatId]/searchMembers/route.ts

@@ -1,8 +1,11 @@
 import { apiHandler } from "@/lib/apiHandler";
 import { SOURCEBOT_GUEST_USER_ID } from "@/lib/constants";
+import { ErrorCode } from "@/lib/errorCodes";
 import { notFound, queryParamsSchemaValidationError, serviceErrorResponse } from "@/lib/serviceError";
 import { isServiceError } from "@/lib/utils";
 import { withAuthV2 } from "@/withAuthV2";
+import { env, hasEntitlement } from "@sourcebot/shared";
+import { StatusCodes } from "http-status-codes";
 import { NextRequest } from "next/server";
 import { z } from "zod";
 
@@ -28,6 +31,24 @@ export const GET = apiHandler(async (
     request: NextRequest,
     { params }: { params: Promise<{ chatId: string }> }
 ) => {
+    // Disable this API for the askgh experiment since we
+    // don't want to allow users to search other members.
+    if (env.EXPERIMENT_ASK_GH_ENABLED === 'true') {
+        return serviceErrorResponse({
+            statusCode: StatusCodes.FORBIDDEN,
+            errorCode: ErrorCode.NOT_FOUND,
+            message: "This API is not enabled with this experiment.",
+        })
+    }
+
+    if (!hasEntitlement('chat-sharing')) {
+        return serviceErrorResponse({
+            statusCode: StatusCodes.FORBIDDEN,
+            errorCode: ErrorCode.NOT_FOUND,
+            message: "Chat sharing is not enabled for your license",
+        })
+    }
+
     const { chatId } = await params;
 
     const rawParams = Object.fromEntries(
@@ -46,6 +67,7 @@ export const GET = apiHandler(async (
 
     const { query } = parsed.data;
 
+
     const result = await withAuthV2(async ({ org, user, prisma }) => {
         const chat = await prisma.chat.findUnique({
             where: {