Feature flag conversation ingestion

Author: spastorelli

integrations/slack/gitbook-manifest.yaml

@@ -78,4 +78,5 @@ secrets:
     CLIENT_ID: ${{ env.SLACK_CLIENT_ID }}
     CLIENT_SECRET: ${{ env.SLACK_CLIENT_SECRET }}
     SIGNING_SECRET: ${{ env.SLACK_SIGNING_SECRET }}
+    REFLAG_SECRET_KEY: ${{ env.REFLAG_SECRET_KEY }}
 target: space

integrations/slack/src/actions/ingestConversation.ts

@@ -1,7 +1,11 @@
 import { ConversationInput, ConversationPart, IntegrationInstallation } from '@gitbook/api';
 import { SlackInstallationConfiguration, SlackRuntimeContext } from '../configuration';
 import { getSlackThread, slackAPI, SlackConversationThread } from '../slack';
-import { getInstallationApiClient, getIntegrationInstallationForTeam } from '../utils';
+import {
+    getInstallationApiClient,
+    getIntegrationInstallationForTeam,
+    isDocsAgentsConversationsEnabled,
+} from '../utils';
 import { IngestSlackConversationActionParams } from './types';
 import { Logger } from '@gitbook/runtime';
 
@@ -21,6 +25,11 @@ export async function ingestSlackConversation(params: IngestSlackConversationAct
     const accessToken = (installation.configuration as SlackInstallationConfiguration)
         .oauth_credentials?.access_token;
 
+    const isDocsAgentsEnabled = await isDocsAgentsConversationsEnabled({
+        organizationId: installation.target.organization,
+        context,
+    });
+
     await Promise.all([
         slackAPI(
             context,
@@ -29,24 +38,28 @@ export async function ingestSlackConversation(params: IngestSlackConversationAct
                 path: 'chat.postMessage',
                 payload: {
                     channel: channelId,
-                    text: `🚀 Sharing this conversation with Docs Agent to improve your docs...`,
+                    text: isDocsAgentsEnabled
+                        ? `🚀 Sharing this conversation with Docs Agent to improve your docs...`
+                        : `✨ Docs Agent is currently in private alpha.\n\nRequest early access for your organization: https://app.gitbook.com/o/${installation.target.organization}/agents`,
                     thread_ts: threadId,
                 },
             },
             {
                 accessToken,
             },
         ),
-        handleIngestSlackConversationAction(
-            {
-                channelId,
-                threadId,
-                installation,
-                accessToken,
-                conversationToIngest,
-            },
-            context,
-        ),
+        isDocsAgentsEnabled
+            ? handleIngestSlackConversationAction(
+                  {
+                      channelId,
+                      threadId,
+                      installation,
+                      accessToken,
+                      conversationToIngest,
+                  },
+                  context,
+              )
+            : Promise.resolve(),
     ]);
 }
 

integrations/slack/src/utils.ts

@@ -148,3 +148,36 @@ export function isAllowedToRespond(eventPayload: any) {
 
     return !isFromBot && !isExternalChannel;
 }
+
+/**
+ * Check if Docs agents conversation is enabled for the organization.
+ *
+ * TODO: Remove when Docs agents reached general availability.
+ */
+export async function isDocsAgentsConversationsEnabled(params: {
+    organizationId: string;
+    context: SlackRuntimeContext;
+}): Promise<boolean> {
+    const { organizationId, context } = params;
+    try {
+        const response = await fetch(
+            `https://front.reflag.com/features/enabled?context.company.id=${organizationId}&key=GIT_SYNC_STATIC_IP`,
+            {
+                method: 'GET',
+                headers: {
+                    Authorization: `Bearer ${context.environment.secrets.REFLAG_SECRET_KEY}`,
+                    'Content-Type': 'application/json',
+                },
+            },
+        );
+
+        const json = (await response.json()) as {
+            features: { DOCS_AGENTS_CONVERSATIONS: { isEnabled: boolean } };
+        };
+        const flag = json.features.DOCS_AGENTS_CONVERSATIONS;
+
+        return flag.isEnabled;
+    } catch (e) {
+        return false;
+    }
+}