feat: Allow integration to override connector verticals

openint-bak · Oct 31, 2024 · 21e388e · 21e388e
1 parent 3a1f35a
commit 21e388e
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/connectors/connector-google/server.ts b/connectors/connector-google/server.ts
@@ -50,7 +50,7 @@ export const googleServer = {
         },
       }
     }
-    if (context.integrationExternalId === 'drive') {
+    if (context.integrationExternalId === 'gmail') {
       return {
         authorization_params: {
           scope: 'https://www.googleapis.com/auth/gmail.readonly',
@@ -71,20 +71,23 @@ export const googleServer = {
           name: 'Google Drive',
           // TODO: Differ oauth scope use in Connect based on which integration
           raw_data: {} as any,
+          verticals: ['file-storage'],
           updated_at: new Date().toISOString(),
           logo_url: '/_assets/logo-google-drive.svg',
         },
         {
           id: 'gmail',
           name: 'Gmail',
           raw_data: {} as any,
+          verticals: ['email'],
           updated_at: new Date().toISOString(),
           logo_url: '/_assets/logo-google-gmail.svg',
         },
         {
           id: 'calendar',
           name: 'Google Calendar',
           raw_data: {} as any,
+          verticals: ['calendar'],
           updated_at: new Date().toISOString(),
           logo_url: '/_assets/logo-google-calendar.svg',
         },

diff --git a/kits/cdk/connector.types.ts b/kits/cdk/connector.types.ts
@@ -24,6 +24,7 @@ import type {
 import type {EndUserId, Id} from './id.types'
 import {makeId} from './id.types'
 import type {ZStandard} from './models'
+import {VerticalKey} from './verticals'
 
 /**
  * Equivalent to to airbyte's low code connector spec,
@@ -226,6 +227,7 @@ export interface ConnectorServer<
       name: string
       logo_url?: string | null
       updated_at: string
+      verticals?: VerticalKey[]
       raw_data: T['_intOpType']['data']
     }>
   }>

diff --git a/kits/cdk/verticals.ts b/kits/cdk/verticals.ts
@@ -40,6 +40,7 @@ const _VERTICAL_BY_KEY = {
                 one-way with no impact on original data.`,
     objects: ['job', 'offer', 'candidate', 'opening'],
   },
+  email: {},
 } satisfies Record<string, VerticalInfo>
 
 // MARK: -

diff --git a/packages/engine-frontend/components/IntegrationSearch.tsx b/packages/engine-frontend/components/IntegrationSearch.tsx
@@ -39,7 +39,7 @@ export function IntegrationSearch({
 
   const intsByCategory = ints?.reduce(
     (acc, int) => {
-      int.ccfg.verticals.forEach((vertical) => {
+      ;(int.verticals ?? int.ccfg.verticals).forEach((vertical) => {
         acc[vertical] = (acc[vertical] || []).concat(int)
       })
       return acc