feat(data-warehouse): add salesforce integration

frontend/src/scenes/data-warehouse/external/forms/SourceForm.tsx

@@ -1,4 +1,5 @@
 import { LemonInput, LemonSelect, LemonSwitch, LemonTextArea } from '@posthog/lemon-ui'
+import { useValues } from 'kea'
 import { Form, Group } from 'kea-forms'
 import { LemonField } from 'lib/lemon-ui/LemonField'
 
@@ -8,6 +9,8 @@ import { SOURCE_DETAILS, sourceWizardLogic } from '../../new/sourceWizardLogic'
 
 interface SourceFormProps {
     sourceConfig: SourceConfig
+    showPrefix?: boolean
+    showSourceFields?: boolean
 }
 
 const sourceFieldToElement = (field: SourceFieldConfig): JSX.Element => {
@@ -73,14 +76,26 @@ const sourceFieldToElement = (field: SourceFieldConfig): JSX.Element => {
 }
 
 export default function SourceForm({ sourceConfig }: SourceFormProps): JSX.Element {
+    const { source } = useValues(sourceWizardLogic)
+    const showSourceFields = SOURCE_DETAILS[sourceConfig.name].showSourceForm
+        ? SOURCE_DETAILS[sourceConfig.name].showSourceForm?.(source.payload)
+        : true
+    const showPrefix = SOURCE_DETAILS[sourceConfig.name].showPrefix
+        ? SOURCE_DETAILS[sourceConfig.name].showPrefix?.(source.payload)
+        : true
+
     return (
         <Form logic={sourceWizardLogic} formKey="sourceConnectionDetails" className="space-y-4" enableFormOnSubmit>
-            <Group name="payload">
-                {SOURCE_DETAILS[sourceConfig.name].fields.map((field) => sourceFieldToElement(field))}
-            </Group>
-            <LemonField name="prefix" label="Table Prefix (optional)">
-                <LemonInput className="ph-ignore-input" data-attr="prefix" placeholder="internal_" />
-            </LemonField>
+            {showSourceFields && (
+                <Group name="payload">
+                    {SOURCE_DETAILS[sourceConfig.name].fields.map((field) => sourceFieldToElement(field))}
+                </Group>
+            )}
+            {showPrefix && (
+                <LemonField name="prefix" label="Table Prefix (optional)">
+                    <LemonInput className="ph-ignore-input" data-attr="prefix" placeholder="internal_" />
+                </LemonField>
+            )}
         </Form>
     )
 }

frontend/src/scenes/data-warehouse/new/sourceWizardLogic.tsx

@@ -25,6 +25,7 @@ import { dataWarehouseTableLogic } from './dataWarehouseTableLogic'
 import type { sourceWizardLogicType } from './sourceWizardLogicType'
 
 export const getHubspotRedirectUri = (): string => `${window.location.origin}/data-warehouse/hubspot/redirect`
+export const getSalesforceRedirectUri = (): string => `${window.location.origin}/data-warehouse/salesforce/redirect`
 
 export const SOURCE_DETAILS: Record<ExternalDataSourceType, SourceConfig> = {
     Stripe: {
@@ -65,6 +66,7 @@ export const SOURCE_DETAILS: Record<ExternalDataSourceType, SourceConfig> = {
         name: 'Hubspot',
         fields: [],
         caption: 'Succesfully authenticated with Hubspot. Please continue here to complete the source setup',
+        oauthPayload: ['code'],
     },
     Postgres: {
         name: 'Postgres',
@@ -420,6 +422,22 @@ export const SOURCE_DETAILS: Record<ExternalDataSourceType, SourceConfig> = {
             },
         ],
     },
+    Salesforce: {
+        name: 'Salesforce',
+        fields: [
+            {
+                name: 'subdomain',
+                label: 'Salesforce subdomain',
+                type: 'text',
+                required: true,
+                placeholder: '',
+            },
+        ],
+        caption: 'Succesfully authenticated with Salesforce. Please continue here to complete the source setup',
+        showPrefix: (payload) => !!payload.code,
+        showSourceForm: (payload) => !payload.code,
+        oauthPayload: ['code', 'subdomain'],
+    },
 }
 
 export const buildKeaFormDefaultFromSourceDetails = (
@@ -746,6 +764,27 @@ export const sourceWizardLogic = kea<sourceWizardLogicType>([
                 }
             },
         ],
+        addToSalesforceButtonUrl: [
+            (s) => [s.preflight],
+            (preflight) => {
+                return (subdomain: string) => {
+                    const clientId = preflight?.data_warehouse_integrations?.salesforce.client_id
+
+                    if (!clientId) {
+                        return null
+                    }
+
+                    const params = new URLSearchParams()
+                    params.set('client_id', clientId)
+                    params.set('redirect_uri', `${window.location.origin}/data-warehouse/salesforce/redirect`)
+                    params.set('response_type', 'code')
+                    params.set('scope', 'refresh_token api')
+                    params.set('state', subdomain)
+
+                    return `https://${subdomain}.my.salesforce.com/services/oauth2/authorize?${params.toString()}`
+                }
+            },
+        ],
         modalTitle: [
             (s) => [s.currentStep],
             (currentStep) => {
@@ -883,6 +922,17 @@ export const sourceWizardLogic = kea<sourceWizardLogicType>([
                     })
                     return
                 }
+                case 'salesforce': {
+                    actions.updateSource({
+                        source_type: 'Salesforce',
+                        payload: {
+                            code: searchParams.code,
+                            subdomain: searchParams.subdomain,
+                            redirect_uri: getSalesforceRedirectUri(),
+                        },
+                    })
+                    break
+                }
                 default:
                     lemonToast.error(`Something went wrong.`)
             }
@@ -923,6 +973,13 @@ export const sourceWizardLogic = kea<sourceWizardLogicType>([
             if (kind === 'hubspot') {
                 router.actions.push(urls.dataWarehouseTable(), { kind, code: searchParams.code })
             }
+            if (kind === 'salesforce') {
+                router.actions.push(urls.dataWarehouseTable(), {
+                    kind,
+                    code: searchParams.code,
+                    subdomain: searchParams.state,
+                })
+            }
         },
         '/data-warehouse/new': (_, searchParams) => {
             if (searchParams.kind == 'hubspot' && searchParams.code) {
@@ -932,16 +989,41 @@ export const sourceWizardLogic = kea<sourceWizardLogicType>([
                 })
                 actions.setStep(2)
             }
+            if (searchParams.kind == 'salesforce' && searchParams.code) {
+                actions.selectConnector(SOURCE_DETAILS['Salesforce'])
+                actions.handleRedirect(searchParams.kind, {
+                    code: searchParams.code,
+                    subdomain: searchParams.subdomain,
+                })
+                actions.setStep(2)
+            }
         },
     })),
     forms(({ actions, values }) => ({
         sourceConnectionDetails: {
             defaults: buildKeaFormDefaultFromSourceDetails(SOURCE_DETAILS),
             errors: (sourceValues) => {
+                if (
+                    values.selectedConnector &&
+                    SOURCE_DETAILS[values.selectedConnector?.name].oauthPayload &&
+                    SOURCE_DETAILS[values.selectedConnector.name].oauthPayload?.every(
+                        (element) => values.source.payload[element]
+                    )
+                ) {
+                    return {}
+                }
                 return getErrorsForFields(values.selectedConnector?.fields ?? [], sourceValues as any)
             },
             submit: async (sourceValues) => {
                 if (values.selectedConnector) {
+                    if (
+                        values.selectedConnector.name === 'Salesforce' &&
+                        (!values.source.payload.code || !values.source.payload.subdomain)
+                    ) {
+                        window.open(values.addToSalesforceButtonUrl(sourceValues.payload.subdomain) as string)
+                        return
+                    }
+
                     const payload = {
                         ...sourceValues,
                         source_type: values.selectedConnector.name,
@@ -951,19 +1033,28 @@ export const sourceWizardLogic = kea<sourceWizardLogicType>([
                     try {
                         await api.externalDataSources.source_prefix(payload.source_type, sourceValues.prefix)
 
-                        const payloadKeys = (values.selectedConnector?.fields ?? []).map((n) => n.name)
+                        // salesforce doesn't need to store the payload. The relevant fielsd will already be handled from the URL
+                        // only update the prefix
+                        if (values.selectedConnector.name === 'Salesforce') {
+                            actions.updateSource({
+                                ...values.source,
+                                prefix: sourceValues.prefix,
+                            })
+                        } else {
+                            const payloadKeys = (values.selectedConnector?.fields ?? []).map((n) => n.name)
 
-                        // Only store the keys of the source type we're using
-                        actions.updateSource({
-                            ...payload,
-                            payload: {
-                                source_type: values.selectedConnector.name,
-                                ...payloadKeys.reduce((acc, cur) => {
-                                    acc[cur] = payload['payload'][cur]
-                                    return acc
-                                }, {} as Record<string, any>),
-                            },
-                        })
+                            // Only store the keys of the source type we're using
+                            actions.updateSource({
+                                ...payload,
+                                payload: {
+                                    source_type: values.selectedConnector.name,
+                                    ...payloadKeys.reduce((acc, cur) => {
+                                        acc[cur] = payload['payload'][cur]
+                                        return acc
+                                    }, {} as Record<string, any>),
+                                },
+                            })
+                        }
 
                         actions.setIsLoading(false)
                     } catch (e: any) {

frontend/src/scenes/data-warehouse/settings/DataWarehouseManagedSourcesTable.tsx

@@ -10,6 +10,7 @@ import IconGoogleCloudStorage from 'public/services/google-cloud-storage.png'
 import IconHubspot from 'public/services/hubspot.png'
 import IconMySQL from 'public/services/mysql.png'
 import IconPostgres from 'public/services/postgres.png'
+import IconSalesforce from 'public/services/salesforce.png'
 import IconSnowflake from 'public/services/snowflake.png'
 import IconStripe from 'public/services/stripe.png'
 import IconZendesk from 'public/services/zendesk.png'
@@ -183,6 +184,7 @@ export function RenderDataWarehouseSourceIcon({
         'google-cloud': IconGoogleCloudStorage,
         'cloudflare-r2': IconCloudflare,
         azure: Iconazure,
+        Salesforce: IconSalesforce,
     }[type]
 
     return (

frontend/src/types.ts

@@ -2961,6 +2961,9 @@ export interface PreflightStatus {
         hubspot: {
             client_id?: string
         }
+        salesforce: {
+            client_id?: string
+        }
     }
     /** Whether PostHog is running in DEBUG mode. */
     is_debug?: boolean
@@ -3852,7 +3855,15 @@ export enum DataWarehouseSettingsTab {
     SelfManaged = 'self-managed',
 }
 
-export const externalDataSources = ['Stripe', 'Hubspot', 'Postgres', 'MySQL', 'Zendesk', 'Snowflake'] as const
+export const externalDataSources = [
+    'Stripe',
+    'Hubspot',
+    'Postgres',
+    'MySQL',
+    'Zendesk',
+    'Snowflake',
+    'Salesforce',
+] as const
 
 export type ExternalDataSourceType = (typeof externalDataSources)[number]
 
@@ -4229,6 +4240,9 @@ export interface SourceConfig {
     caption: string | React.ReactNode
     fields: SourceFieldConfig[]
     disabledReason?: string | null
+    showPrefix?: (payload: Record<string, any>) => boolean
+    showSourceForm?: (payload: Record<string, any>) => boolean
+    oauthPayload?: string[]
 }
 
 export interface ProductPricingTierSubrows {