@superblocksteam/shared 0.9561.4 → 0.9562.0

package/src/plugins/templates/databricks.ts

@@ -11,7 +11,7 @@ import {
 import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
 import { makeDropdownItem } from './shared/db.js';
 
-export const DatabricksPluginVersions = {
+const DatabricksPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V3: '0.0.3',
@@ -26,9 +26,9 @@ enum ConnectionType {
 }
 
 const CONNECTION_METHODS_AND_DISPLAY_NAMES_V2 = {
-  [ConnectionType.PERSONAL_ACCESS_TOKEN]: 'Personal access token',
   [ConnectionType.MACHINE_TO_MACHINE]: 'Machine-to-machine',
-  [ConnectionType.OAUTH_TOKEN_FEDERATION]: 'OAuth token federation'
+  [ConnectionType.OAUTH_TOKEN_FEDERATION]: 'OAuth token federation',
+  [ConnectionType.PERSONAL_ACCESS_TOKEN]: 'Personal access token'
 };
 
 const BASE_HOST = {
@@ -68,7 +68,7 @@ const BASE_DEFAULT_CATALOG = {
   label: 'Default catalog',
   name: 'connection.defaultCatalog',
   componentType: FormComponentType.INPUT_TEXT,
-  placeholder: 'hive_datastore'
+  placeholder: 'catalog'
 };
 
 const BASE_DEFAULT_SCHEMA = {
@@ -137,7 +137,7 @@ const BASE_SCOPE = {
 
 export const DatabricksPlugin: Plugin = {
   id: 'databricks',
-  name: 'Databricks',
+  name: 'Databricks SQL Warehouse',
   moduleName: 'DatabricksPlugin',
   modulePath: 'plugins/databricks/DatabricksPlugin',
   iconLocation: `${PUBLIC_INTEGRATIONS_LOGO_URL}/databricks.png`,
@@ -315,7 +315,7 @@ export const DatabricksPlugin: Plugin = {
             startVersion: DatabricksPluginVersions.V2,
             endVersion: DatabricksPluginVersions.V2,
             componentType: FormComponentType.DROPDOWN,
-            initialValue: ConnectionType.PERSONAL_ACCESS_TOKEN,
+            initialValue: ConnectionType.MACHINE_TO_MACHINE,
             rules: [{ required: true }],
             options: Object.entries(CONNECTION_METHODS_AND_DISPLAY_NAMES_V2).map(([value, displayName]) =>
               makeDropdownItem(value, displayName)
@@ -326,7 +326,7 @@ export const DatabricksPlugin: Plugin = {
             name: 'connection.connectionType',
             startVersion: DatabricksPluginVersions.V3,
             componentType: FormComponentType.DROPDOWN,
-            initialValue: ConnectionType.PERSONAL_ACCESS_TOKEN,
+            initialValue: ConnectionType.MACHINE_TO_MACHINE,
             rules: [{ required: true }],
             options: Object.entries(CONNECTION_METHODS_AND_DISPLAY_NAMES_V2).map(([value, displayName]) =>
               makeDropdownItem(value, displayName)

package/src/plugins/templates/dynamodb.ts

@@ -12,7 +12,7 @@ import { camelCaseToDisplay } from '../../utils/index.js';
 import { PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
 import { getAWSAuthSection } from './shared/index.js';
 
-export const DynamoDBActions = [
+const DynamoDBActions = [
   'getItem',
   'updateItem',
   'putItem',
@@ -78,7 +78,7 @@ function getDynamoDBActionDropdownOptions(): DropdownOption[] {
   });
 }
 
-export const DynamoDBPluginVersions = {
+const DynamoDBPluginVersions = {
   V1: '0.0.1',
   V4: '0.0.4',
   V5: '0.0.5'

package/src/plugins/templates/email.ts

@@ -4,14 +4,13 @@ import { PUBLIC_INTEGRATIONS_LOGO_URL } from './constants.js';
 // superblocksmail is set up as a verified domain in Sendgrid under a different account than superblockshq.com.
 // The current setup allows us to use any address with that domain.
 export const EMAIL_INTEGRATION_SENDER_ADDRESS_DEFAULT = 'app@superblocksmail.com';
-export const EMAIL_INTEGRATION_SENDER_NAME_DEFAULT = 'Superblocks';
-export const EMAIL_INTEGRATION_TEMPLATE_ID_DEFAULT = 'd-1cf30b5bba6e4dccbd14dc67e7b8f354';
+const EMAIL_INTEGRATION_SENDER_NAME_DEFAULT = 'Superblocks';
 
-export const EmailPluginVersions = {
+const EmailPluginVersions = {
   V1: '0.0.1'
 };
 
-export enum EmailActionFieldNames {
+enum EmailActionFieldNames {
   FROM = 'emailFrom',
   TO = 'emailTo',
   CC = 'emailCc',
@@ -21,7 +20,7 @@ export enum EmailActionFieldNames {
   ATTACHMENTS = 'emailAttachments'
 }
 
-export const EmailActionFieldsMap: Record<EmailActionFieldNames, FormItem> = {
+const EmailActionFieldsMap: Record<EmailActionFieldNames, FormItem> = {
   [EmailActionFieldNames.FROM]: {
     name: EmailActionFieldNames.FROM,
     label: 'From',

package/src/plugins/templates/gcs.ts

@@ -10,11 +10,11 @@ import {
 import { SERVICE_ACCOUNT_GHOST_TEXT, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
 import { S3ActionType } from './s3.js';
 
-export const GCSPluginVersions = {
+const GCSPluginVersions = {
   V1: '0.0.1'
 };
 
-export enum GCSActionType {
+enum GCSActionType {
   LIST_OBJECTS = 'LIST_OBJECTS',
   GET_OBJECT = 'GET_OBJECT',
   DELETE_OBJECT = 'DELETE_OBJECT',
@@ -25,7 +25,7 @@ export enum GCSActionType {
   GENERATE_PRESIGNED_URL = 'GENERATE_PRESIGNED_URL'
 }
 
-export const GCS_ACTION_DISPLAY_NAMES: Record<GCSActionType, string> = {
+const GCS_ACTION_DISPLAY_NAMES: Record<GCSActionType, string> = {
   [GCSActionType.LIST_OBJECTS]: 'List files',
   [GCSActionType.GET_OBJECT]: 'Read file',
   [GCSActionType.DELETE_OBJECT]: 'Delete file',
@@ -36,7 +36,7 @@ export const GCS_ACTION_DISPLAY_NAMES: Record<GCSActionType, string> = {
   [GCSActionType.GENERATE_PRESIGNED_URL]: 'Generate presigned URL'
 };
 
-export const DEFAULT_GCS_PRESIGNED_URL_EXPIRATION_SECONDS = 60 * 10;
+const DEFAULT_GCS_PRESIGNED_URL_EXPIRATION_SECONDS = 60 * 10;
 
 export const GCSPlugin: Plugin = {
   id: 'gcs',

package/src/plugins/templates/graphqlintegration.ts

@@ -11,7 +11,7 @@ import { PUBLIC_INTEGRATIONS_LOGO_URL } from './constants.js';
 import { GraphQLPluginVersions } from './graphql.js';
 import { authSections } from './shared/index.js';
 
-export const GraphQLIntegrationPluginVersions = GraphQLPluginVersions;
+const GraphQLIntegrationPluginVersions = GraphQLPluginVersions;
 
 export enum GraphQLIntegrationPluginMergedFieldNames {
   HEADERS = 'headers'

package/src/plugins/templates/gsheets.ts

@@ -22,7 +22,7 @@ const makeDropdownItem = (value: string, displayName?: string, subText?: string)
   };
 };
 
-export const GoogleSheetsPluginVersions = {
+const GoogleSheetsPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V3: '0.0.3',
@@ -51,7 +51,7 @@ export enum GoogleSheetsActionType {
   CLEAR_SPREADSHEET = 'CLEAR_SPREADSHEET'
 }
 
-export const GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V15: Record<string, string> = {
+const GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V15: Record<string, string> = {
   [GoogleSheetsActionType.READ_SPREADSHEET]: 'Read the whole spreadsheet',
   [GoogleSheetsActionType.READ_SPREADSHEET_RANGE]: 'Read from a range (e.g. A1:D100)',
   [GoogleSheetsActionType.APPEND_SPREADSHEET]: markAsDeprecated('Append rows to the spreadsheet'),
@@ -59,25 +59,19 @@ export const GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V15: Record<string, string> = {
   [GoogleSheetsActionType.CLEAR_SPREADSHEET]: 'Clear the spreadsheet'
 };
 
-export const GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V19: Record<string, string> = {
+const GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V19: Record<string, string> = {
   [GoogleSheetsActionType.CREATE_WORKSHEET]: 'Add a worksheet to a spreadsheet',
   ...GOOGLE_SHEETS_ACTION_DISPLAY_NAMES_V15
 };
 
 export const GOOGLE_SHEETS_PLUGIN_ID = 'gsheets';
 
-export const GOOGLE_SHEETS_FORMAT_DISPLAY_NAMES: Record<GoogleSheetsFormatType, string> = {
-  [GoogleSheetsFormatType.EFFECTIVE_VALUE]: 'Effective value',
-  [GoogleSheetsFormatType.FORMATTED_VALUE]: 'Formatted value',
-  [GoogleSheetsFormatType.USER_ENTERED_VALUE]: 'User entered value'
-};
-
-export enum GOOGLE_SHEETS_SCOPE {
+enum GOOGLE_SHEETS_SCOPE {
   READ = 'email https://www.googleapis.com/auth/drive.metadata.readonly https://www.googleapis.com/auth/spreadsheets.readonly https://www.googleapis.com/auth/userinfo.email openid',
   WRITE = 'email https://www.googleapis.com/auth/drive.metadata.readonly https://www.googleapis.com/auth/spreadsheets https://www.googleapis.com/auth/userinfo.email openid'
 }
 
-export const getGsheetsAuthDisplayName = (authType: GoogleSheetsAuthType): string => {
+const getGsheetsAuthDisplayName = (authType: GoogleSheetsAuthType): string => {
   switch (authType) {
     case GoogleSheetsAuthType.OAUTH2_CODE:
       return 'Connect using Google Account';

package/src/plugins/templates/index.ts

@@ -36,6 +36,7 @@ export * from './javascript.js';
 export * from './jira.js';
 export * from './kafka.js';
 export * from './kinesis.js';
+export * from './lakebase.js';
 export * from './launchdarkly.js';
 export * from './mariadb.js';
 export * from './mistral.js';

package/src/plugins/templates/javascript.ts

@@ -1,7 +1,7 @@
 import { EditorLanguage, FormComponentType, LanguagePluginID, Plugin, PluginResponseType, PluginType } from '../../types/index.js';
 import { PUBLIC_INTEGRATIONS_LOGO_URL } from './constants.js';
 
-export const JavascriptPluginVersions = {
+const JavascriptPluginVersions = {
   V1: '0.0.1'
 };
 

package/src/plugins/templates/kafka.ts

@@ -1,7 +1,8 @@
 import { KafkaV1 as Kafka } from '@superblocksteam/types';
 import { EditorLanguage, FormComponentType, InputDataType, Plugin, PluginResponseType, PluginType } from '../../types/index.js';
 import { PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
-export const KafkaPluginVersions = {
+
+const KafkaPluginVersions = {
   V1: '0.0.1'
 };
 
@@ -85,22 +86,6 @@ export const KafkaPlugin = generateKafkaPlugin(
   ALL_SASL_MECHANISMS_AND_DISPLAY_NAMES
 );
 
-export const AivenPlugin = generateKafkaPlugin(
-  'aivenkafka',
-  'Aiven for Apache Kafka',
-  `${PUBLIC_INTEGRATIONS_LOGO_URL}/aiven_kafka.png`,
-  `${DOCS_BASE_URL}/aivenkafka`,
-  EXCLUDE_AWS_SASL_MECHANISMS_AND_DISPLAY_NAMES
-);
-
-export const MskPlugin = generateKafkaPlugin(
-  'msk',
-  'MSK',
-  `${PUBLIC_INTEGRATIONS_LOGO_URL}/msk.png`,
-  `${DOCS_BASE_URL}/msk`,
-  ALL_SASL_MECHANISMS_AND_DISPLAY_NAMES
-);
-
 export const RedpandaPlugin = generateKafkaPlugin(
   'redpanda',
   'Redpanda',

package/src/plugins/templates/kinesis.ts

@@ -1,6 +1,7 @@
 import { FormItem, EditorLanguage, FormComponentType, InputDataType, Plugin, PluginResponseType, PluginType } from '../../types/index.js';
 import { PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
-export const KinesisPluginVersions = {
+
+const KinesisPluginVersions = {
   V1: '0.0.1'
 };
 

package/src/plugins/templates/lakebase.ts

@@ -0,0 +1,365 @@
+import {
+  DB_SQL_INITIAL_TEXT,
+  EditorLanguage,
+  FormComponentType,
+  InputDataType,
+  Plugin,
+  PluginResponseType,
+  PluginType,
+  FormItem
+} from '../../types/index.js';
+import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
+import { makeDropdownItem } from './shared/db.js';
+
+export const LakebasePluginVersions = {
+  V1: '0.0.1'
+};
+
+// These match the proto enum values
+enum ConnectionType {
+  USERNAME_PASSWORD = '1',
+  OAUTH_M2M = '2',
+  OAUTH_TOKEN_FEDERATION = '3'
+}
+
+const CONNECTION_METHODS_AND_DISPLAY_NAMES = {
+  [ConnectionType.OAUTH_M2M]: 'OAuth Machine-to-Machine',
+  [ConnectionType.OAUTH_TOKEN_FEDERATION]: 'OAuth Token Federation',
+  [ConnectionType.USERNAME_PASSWORD]: 'Username/Password'
+};
+
+const BASE_HOST = {
+  label: 'Host',
+  name: 'connection.host',
+  componentType: FormComponentType.INPUT_TEXT,
+  placeholder: 'instance-abc123.database.cloud.databricks.com',
+  rules: [{ required: true, message: 'Host is required' }]
+};
+
+const BASE_PORT = {
+  label: 'Port',
+  name: 'connection.port',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.NUMBER,
+  initialValue: 5432,
+  rules: [{ required: true, message: 'Port is required' }]
+};
+
+const BASE_DATABASE_NAME = {
+  label: 'Database name',
+  name: 'connection.databaseName',
+  componentType: FormComponentType.INPUT_TEXT,
+  placeholder: 'databricks_postgres',
+  rules: [{ required: true, message: 'Database name is required' }]
+};
+
+// OAuth M2M fields
+const BASE_OAUTH_CLIENT_ID = {
+  label: 'Client ID',
+  name: 'connection.oauthClientId',
+  componentType: FormComponentType.INPUT_TEXT,
+  rules: [{ required: true, message: 'Client ID is required' }]
+};
+
+const BASE_OAUTH_CLIENT_SECRET = {
+  label: 'Client secret',
+  name: 'connection.oauthClientSecret',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.PASSWORD,
+  rules: [{ required: true, message: 'Client secret is required' }]
+};
+
+const BASE_OAUTH_WORKSPACE_URL = {
+  label: 'Workspace URL',
+  name: 'connection.oauthWorkspaceUrl',
+  componentType: FormComponentType.INPUT_TEXT,
+  placeholder: 'https://dbc-a1b2345c-d6e7.cloud.databricks.com',
+  tooltip: {
+    markdownText: 'Databricks workspace URL used for OAuth token endpoint.'
+  },
+  rules: [{ required: true, message: 'Workspace URL is required for OAuth' }]
+};
+
+// OAuth Token Federation fields
+const BASE_SUBJECT_TOKEN_SOURCE = {
+  label: 'Subject token source',
+  name: 'authConfig.subjectTokenSource',
+  componentType: FormComponentType.DROPDOWN,
+  initialValue: 'SUBJECT_TOKEN_SOURCE_LOGIN_IDENTITY_PROVIDER',
+  rules: [{ required: true, message: 'Subject token source is required' }],
+  options: [
+    makeDropdownItem('SUBJECT_TOKEN_SOURCE_LOGIN_IDENTITY_PROVIDER', 'Login Identity Provider'),
+    makeDropdownItem('SUBJECT_TOKEN_SOURCE_STATIC_TOKEN', 'Static Token')
+  ]
+};
+
+const BASE_TOKEN_URL = {
+  label: 'Token URL',
+  name: 'authConfig.tokenUrl',
+  componentType: FormComponentType.INPUT_TEXT,
+  placeholder: 'https://your-databricks-instance.cloud.databricks.com/oidc/v1/token',
+  rules: [{ required: true, message: 'Token URL is required' }]
+};
+
+const BASE_STATIC_TOKEN = {
+  label: 'Static Token',
+  name: 'authConfig.subjectTokenSourceStaticToken',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.PASSWORD,
+  rules: [{ required: true, message: 'Static token is required when using static token source' }]
+};
+
+// Username/Password fields
+const BASE_USERNAME = {
+  label: 'Username',
+  name: 'connection.username',
+  componentType: FormComponentType.INPUT_TEXT,
+  rules: [{ required: true, message: 'Username is required' }]
+};
+
+const BASE_PASSWORD = {
+  label: 'Password',
+  name: 'connection.password',
+  componentType: FormComponentType.INPUT_TEXT,
+  dataType: InputDataType.PASSWORD,
+  rules: [{ required: true, message: 'Password is required' }]
+};
+
+export const LakebasePlugin: Plugin = {
+  id: 'lakebase',
+  name: 'Databricks Lakebase',
+  moduleName: 'LakebasePlugin',
+  modulePath: 'plugins/lakebase/LakebasePlugin',
+  iconLocation: `${PUBLIC_INTEGRATIONS_LOGO_URL}/databricks.png`,
+  docsUrl: `${DOCS_BASE_URL}/lakebase`,
+  type: PluginType.DB,
+  responseType: PluginResponseType.TABLE,
+  hasRawRequest: true,
+  hasMetadata: true,
+  rawRequestName: 'Executed SQL',
+  datasourceTemplate: {
+    sections: [
+      {
+        name: 'main',
+        items: [
+          // Display name
+          {
+            label: 'Display name',
+            name: 'name',
+            startVersion: LakebasePluginVersions.V1,
+            componentType: FormComponentType.INPUT_TEXT,
+            placeholder: 'Lakebase Production',
+            rules: [{ required: true, message: 'Display name is required' }]
+          },
+          // Hidden fields for OAuth Token Exchange integration
+          {
+            label: 'Auth Type Field',
+            name: 'authTypeField',
+            hidden: true,
+            componentType: FormComponentType.INPUT_TEXT,
+            startVersion: LakebasePluginVersions.V1,
+            initialValue: 'oauthType',
+            rules: [{ required: false }],
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          },
+          {
+            label: 'OAuth Type',
+            name: 'oauthType',
+            hidden: true,
+            componentType: FormComponentType.INPUT_TEXT,
+            startVersion: LakebasePluginVersions.V1,
+            initialValue: 'oauth-token-exchange',
+            rules: [{ required: false }],
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          },
+          {
+            label: 'Subject Token Type',
+            name: 'authConfig.subjectTokenType',
+            hidden: true,
+            componentType: FormComponentType.INPUT_TEXT,
+            startVersion: LakebasePluginVersions.V1,
+            initialValue: 'urn:ietf:params:oauth:token-type:jwt',
+            rules: [{ required: false }],
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          },
+          {
+            label: 'Scope',
+            name: 'authConfig.scope',
+            hidden: true,
+            componentType: FormComponentType.INPUT_TEXT,
+            startVersion: LakebasePluginVersions.V1,
+            initialValue: 'all-apis',
+            rules: [{ required: false }],
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          },
+          // Connection fields - always shown
+          {
+            ...BASE_HOST,
+            startVersion: LakebasePluginVersions.V1
+          } as FormItem,
+          {
+            ...BASE_PORT,
+            startVersion: LakebasePluginVersions.V1
+          } as FormItem,
+          {
+            ...BASE_DATABASE_NAME,
+            startVersion: LakebasePluginVersions.V1
+          } as FormItem,
+          // Authentication dropdown
+          {
+            label: 'Authentication',
+            name: 'connection.connectionType',
+            startVersion: LakebasePluginVersions.V1,
+            componentType: FormComponentType.DROPDOWN,
+            initialValue: ConnectionType.OAUTH_M2M,
+            rules: [{ required: true }],
+            options: Object.entries(CONNECTION_METHODS_AND_DISPLAY_NAMES).map(([value, displayName]) =>
+              makeDropdownItem(value, displayName)
+            )
+          },
+          // OAuth M2M fields
+          {
+            ...BASE_OAUTH_WORKSPACE_URL,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_M2M]
+              }
+            }
+          } as FormItem,
+          {
+            ...BASE_OAUTH_CLIENT_ID,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_M2M]
+              }
+            }
+          } as FormItem,
+          {
+            ...BASE_OAUTH_CLIENT_SECRET,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_M2M]
+              }
+            }
+          } as FormItem,
+          // OAuth Token Federation fields
+          {
+            ...BASE_SUBJECT_TOKEN_SOURCE,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          } as FormItem,
+          {
+            ...BASE_STATIC_TOKEN,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION],
+                'authConfig.subjectTokenSource': ['SUBJECT_TOKEN_SOURCE_STATIC_TOKEN']
+              }
+            }
+          } as FormItem,
+          {
+            ...BASE_TOKEN_URL,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.OAUTH_TOKEN_FEDERATION]
+              }
+            }
+          } as FormItem,
+          // Username/Password fields
+          {
+            ...BASE_USERNAME,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.USERNAME_PASSWORD]
+              }
+            }
+          } as FormItem,
+          {
+            ...BASE_PASSWORD,
+            startVersion: LakebasePluginVersions.V1,
+            display: {
+              show: {
+                'connection.connectionType': [ConnectionType.USERNAME_PASSWORD]
+              }
+            }
+          } as FormItem
+        ]
+      }
+    ]
+  },
+  actionTemplate: {
+    sections: [
+      {
+        name: 'main',
+        items: [
+          {
+            label: 'Parameters ($1, $2, ...)',
+            name: 'parameters',
+            startVersion: LakebasePluginVersions.V1,
+            ldFlag: 'clark.parameterized-apis.v2.enabled',
+            componentType: FormComponentType.DYNAMIC_INPUT_TEXT,
+            placeholder: '[param1, param2, ...]',
+            tooltip: {
+              markdownText:
+                'JavaScript expression that evaluates to an array of parameter values. Use `$1`, `$2`, etc. in your SQL query to reference these parameters.'
+            }
+          },
+          {
+            label: '', // Query
+            name: 'body',
+            startVersion: LakebasePluginVersions.V1,
+            componentType: FormComponentType.CODE_EDITOR,
+            language: EditorLanguage.SQL,
+            initialValue: DB_SQL_INITIAL_TEXT
+          }
+        ]
+      },
+      {
+        name: 'advanced:main',
+        items: [
+          {
+            label: 'Use parameterized SQL',
+            name: 'usePreparedSql',
+            startVersion: LakebasePluginVersions.V1,
+            componentType: FormComponentType.SWITCH,
+            initialValue: true,
+            tooltip: {
+              markdownText: PARAMETERIZED_SQL_DESCRIPTION
+            },
+            display: {
+              show: {
+                parameters: ['undefined']
+              }
+            }
+          }
+        ]
+      }
+    ]
+  }
+};

package/src/plugins/templates/mariadb.ts

@@ -12,7 +12,7 @@ import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_
 import { SqlOperations } from './shared/index.js';
 import { CONNECTION_METHODS_AND_DISPLAY_NAMES, makeDropdownItem } from './shared/db.js';
 
-export const MariaDBPluginVersions = {
+const MariaDBPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V6: '0.0.6',

package/src/plugins/templates/mongodb.ts

@@ -1,12 +1,12 @@
 import { EditorLanguage, FormComponentType, Plugin, PluginResponseType, PluginType } from '../../types/index.js';
 import { PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_URL } from './constants.js';
 
-export const MongoDBPluginVersions = {
+const MongoDBPluginVersions = {
   V1: '0.0.1',
   V7: '0.0.7'
 };
 
-export enum MongoDBOperationType {
+enum MongoDBOperationType {
   aggregate = 'aggregate',
   count = 'count',
   deleteOne = 'deleteOne',

package/src/plugins/templates/mssql.ts

@@ -12,7 +12,7 @@ import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_
 import { SqlOperations } from './shared/index.js';
 import { CONNECTION_METHODS_AND_DISPLAY_NAMES, makeDropdownItem } from './shared/db.js';
 
-export const MicrosoftSQLPluginVersions = {
+const MicrosoftSQLPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V6: '0.0.6',

package/src/plugins/templates/mysql.ts

@@ -12,7 +12,7 @@ import { PARAMETERIZED_SQL_DESCRIPTION, PUBLIC_INTEGRATIONS_LOGO_URL, DOCS_BASE_
 import { SqlOperations, sshTunnelSections } from './shared/index.js';
 import { CONNECTION_METHODS_AND_DISPLAY_NAMES, makeDropdownItem } from './shared/db.js';
 
-export const MySQLPluginVersions = {
+const MySQLPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V6: '0.0.6',

package/src/plugins/templates/openai.ts

@@ -25,13 +25,13 @@ import {
   DOCS_BASE_URL
 } from './constants.js';
 
-export const OpenAiPluginVersions = {
+const OpenAiPluginVersions = {
   V1: '0.0.1',
   V2: '0.0.2',
   V3: '0.0.3'
 };
 
-export const OPEN_AI_ACTION_AND_DESCRIPTION = {
+const OPEN_AI_ACTION_AND_DESCRIPTION = {
   [OPEN_AI_ACTION.GENERATE_CHATGPT_RESPONSE]: 'Given a prompt, the model will return a chat completion response',
   [OPEN_AI_ACTION.GENERATE_TEXT]: 'Given a prompt, the model will return a text completion response',
   [OPEN_AI_ACTION.GENERATE_CODE]: 'Given a prompt or code, the model will return a code completion response',