@deepnote/database-integrations 1.1.1 → 1.2.1

package/dist/index.js

@@ -1,4 +1,4 @@
-import { z } from "zod";
+import z$1, { z } from "zod";
 
 //#region src/sql-integration-auth-methods.ts
 const AwsAuthMethods = {
@@ -36,6 +36,332 @@ function isFederatedAuthMetadata(metadata) {
 	return isFederatedAuthMethod(metadata.authMethod);
 }
 
+//#endregion
+//#region src/database-integration-metadata-schemas.ts
+const athenaMetadataSchema = z.object({
+	access_key_id: z.string(),
+	region: z.string(),
+	s3_output_path: z.string(),
+	secret_access_key: z.string(),
+	workgroup: z.string().optional()
+});
+const bigqueryMetadataSchema = z.discriminatedUnion("authMethod", [z.object({
+	authMethod: z.literal(BigQueryAuthMethods.ServiceAccount),
+	service_account: z.string()
+}), z.object({
+	authMethod: z.literal(BigQueryAuthMethods.GoogleOauth),
+	project: z.string(),
+	clientId: z.string(),
+	clientSecret: z.string()
+})]);
+const clickhouseMetadataSchema = z.object({
+	host: z.string(),
+	user: z.string(),
+	password: z.string().optional(),
+	database: z.string(),
+	port: z.string().optional(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const databricksMetadataSchema = z.object({
+	host: z.string(),
+	httpPath: z.string(),
+	token: z.string(),
+	port: z.string(),
+	schema: z.string().optional(),
+	catalog: z.string().optional(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional()
+});
+const dremioMetadataSchema = z.object({
+	schema: z.string(),
+	host: z.string(),
+	port: z.string(),
+	token: z.string(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional()
+});
+const mongodbMetadataSchema = z.object({
+	connection_string: z.string(),
+	rawConnectionString: z.string().optional(),
+	prefix: z.string().optional(),
+	host: z.string().optional(),
+	port: z.string().optional(),
+	user: z.string().optional(),
+	password: z.string().optional(),
+	database: z.string().optional(),
+	options: z.string().optional(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const pandasDataframeMetadataSchema = z.object({});
+const commonRedshiftMetadataSchema = z.object({
+	database: z.string(),
+	host: z.string(),
+	port: z.string().optional(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const redshiftMetadataSchema = z.discriminatedUnion("authMethod", [
+	commonRedshiftMetadataSchema.extend({
+		authMethod: z.literal(DatabaseAuthMethods.UsernameAndPassword),
+		user: z.string(),
+		password: z.string()
+	}),
+	commonRedshiftMetadataSchema.extend({
+		authMethod: z.literal(AwsAuthMethods.IamRole),
+		roleArn: z.string(),
+		roleExternalId: z.string(),
+		roleNonce: z.string()
+	}),
+	commonRedshiftMetadataSchema.extend({ authMethod: z.literal(DatabaseAuthMethods.IndividualCredentials) })
+]);
+const commonSnowflakeMetadataSchema = z.object({
+	accountName: z.string(),
+	warehouse: z.string().optional(),
+	database: z.string().optional(),
+	role: z.string().optional()
+});
+const snowflakeMetadataSchema = z.discriminatedUnion("authMethod", [
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: z.literal(SnowflakeAuthMethods.Password),
+		username: z.string(),
+		password: z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: z.literal(SnowflakeAuthMethods.Okta),
+		clientId: z.string(),
+		clientSecret: z.string(),
+		oktaSubdomain: z.string(),
+		identityProvider: z.string(),
+		authorizationServer: z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: z.literal(SnowflakeAuthMethods.NativeSnowflake),
+		clientId: z.string(),
+		clientSecret: z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: z.literal(SnowflakeAuthMethods.AzureAd),
+		clientId: z.string(),
+		clientSecret: z.string(),
+		resource: z.string(),
+		tenant: z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({ authMethod: z.literal(SnowflakeAuthMethods.KeyPair) }),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: z.literal(SnowflakeAuthMethods.ServiceAccountKeyPair),
+		username: z.string(),
+		privateKey: z.string(),
+		privateKeyPassphrase: z.string().optional()
+	})
+]);
+const spannerMetadataSchema = z.object({
+	service_account: z.string(),
+	dataBoostEnabled: z.boolean(),
+	instance: z.string(),
+	database: z.string()
+});
+const trinoMetadataSchema = z.object({
+	host: z.string(),
+	user: z.string(),
+	password: z.string(),
+	database: z.string(),
+	port: z.string(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const commonDatabaseSchema = z.object({
+	host: z.string(),
+	user: z.string(),
+	password: z.string(),
+	database: z.string(),
+	sshEnabled: z.boolean().optional(),
+	sshHost: z.string().optional(),
+	sshPort: z.string().optional(),
+	sshUser: z.string().optional()
+});
+const alloydbMetadataSchema = commonDatabaseSchema.extend({
+	port: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const mariadbMetadataSchema = commonDatabaseSchema.extend({
+	port: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const mindsdbMetadataSchema = commonDatabaseSchema.extend({
+	port: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const mysqlMetadataSchema = commonDatabaseSchema.extend({
+	port: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const pgsqlMetadataSchema = commonDatabaseSchema.extend({
+	port: z.string().optional(),
+	sslEnabled: z.boolean().optional(),
+	caCertificateName: z.string().optional(),
+	caCertificateText: z.string().optional()
+});
+const materializeMetadataSchema = pgsqlMetadataSchema.extend({ cluster: z.string() });
+const sqlServerMetadataSchema = commonDatabaseSchema.extend({ port: z.string() });
+const databaseMetadataSchemasByType = {
+	"alloydb": alloydbMetadataSchema,
+	"athena": athenaMetadataSchema,
+	"big-query": bigqueryMetadataSchema,
+	"clickhouse": clickhouseMetadataSchema,
+	"databricks": databricksMetadataSchema,
+	"dremio": dremioMetadataSchema,
+	"mariadb": mariadbMetadataSchema,
+	"materialize": materializeMetadataSchema,
+	"mindsdb": mindsdbMetadataSchema,
+	"mongodb": mongodbMetadataSchema,
+	"mysql": mysqlMetadataSchema,
+	"pandas-dataframe": pandasDataframeMetadataSchema,
+	"pgsql": pgsqlMetadataSchema,
+	"redshift": redshiftMetadataSchema,
+	"snowflake": snowflakeMetadataSchema,
+	"spanner": spannerMetadataSchema,
+	"sql-server": sqlServerMetadataSchema,
+	"trino": trinoMetadataSchema
+};
+
+//#endregion
+//#region src/database-integration-config.ts
+const commonIntegrationConfig = z$1.object({
+	id: z$1.string(),
+	name: z$1.string()
+});
+const databaseIntegrationConfigSchema = z$1.discriminatedUnion("type", [
+	commonIntegrationConfig.extend({
+		type: z$1.literal("alloydb"),
+		metadata: databaseMetadataSchemasByType["alloydb"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("athena"),
+		metadata: databaseMetadataSchemasByType["athena"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("big-query"),
+		metadata: databaseMetadataSchemasByType["big-query"],
+		federated_auth_method: z$1.enum(["google-oauth"]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("clickhouse"),
+		metadata: databaseMetadataSchemasByType["clickhouse"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("databricks"),
+		metadata: databaseMetadataSchemasByType["databricks"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("dremio"),
+		metadata: databaseMetadataSchemasByType["dremio"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("mariadb"),
+		metadata: databaseMetadataSchemasByType["mariadb"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("materialize"),
+		metadata: databaseMetadataSchemasByType["materialize"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("mindsdb"),
+		metadata: databaseMetadataSchemasByType["mindsdb"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("mongodb"),
+		metadata: databaseMetadataSchemasByType["mongodb"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("mysql"),
+		metadata: databaseMetadataSchemasByType["mysql"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("pandas-dataframe"),
+		metadata: databaseMetadataSchemasByType["pandas-dataframe"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("pgsql"),
+		metadata: databaseMetadataSchemasByType["pgsql"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("redshift"),
+		metadata: databaseMetadataSchemasByType["redshift"],
+		federated_auth_method: z$1.enum(["individual-credentials"]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("snowflake"),
+		metadata: databaseMetadataSchemasByType["snowflake"],
+		federated_auth_method: z$1.enum([
+			"okta",
+			"snowflake",
+			"azure",
+			"key-pair"
+		]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("spanner"),
+		metadata: databaseMetadataSchemasByType["spanner"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("sql-server"),
+		metadata: databaseMetadataSchemasByType["sql-server"],
+		federated_auth_method: z$1.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: z$1.literal("trino"),
+		metadata: databaseMetadataSchemasByType["trino"],
+		federated_auth_method: z$1.null().optional()
+	})
+]);
+
 //#endregion
 //#region src/snowflake-integration-env-vars.ts
 /**
@@ -68,6 +394,22 @@ function getSnowflakeSqlAlchemyInput(metadata, params) {
 		queryParams: { ...createBaseQueryParams(metadata, params.snowflakePartnerIdentifier) }
 	}));
 }
+function getSnowflakeFederatedAuthSqlAlchemyInput(metadata, options = {}) {
+	return createSqlAlchemyInputFromUrl(buildSnowflakeUrlForFederatedAuth(metadata, options), options.keyPair);
+}
+function buildSnowflakeUrlForFederatedAuth(metadata, options) {
+	const queryParams = { ...createBaseQueryParams(metadata, options.snowflakePartnerIdentifier) };
+	if (options.role) queryParams.role = options.role;
+	if (options.accessToken) {
+		queryParams.token = options.accessToken;
+		queryParams.authenticator = "oauth";
+	}
+	const accountIdentifier = metadata.accountName.replace(".privatelink", "");
+	return buildUrl(options.keyPair?.username ? `snowflake://${encodeURIComponent(options.keyPair.username)}@${encodeURIComponent(accountIdentifier)}` : `snowflake://:@${encodeURIComponent(accountIdentifier)}`, {
+		path: metadata.database,
+		queryParams
+	});
+}
 function createBaseQueryParams(metadata, snowflakePartnerIdentifier) {
 	return {
 		warehouse: metadata.warehouse,
@@ -93,7 +435,9 @@ function getEnvironmentVariablesForIntegrations(integrations, params) {
 	const errors = [];
 	integrations.forEach((integration) => {
 		const namePrefix = convertToEnvironmentVariableName(integration.name);
-		const envVarsForThisIntegration = Object.entries(integration.metadata).map(([key, rawValue]) => {
+		const envVarsForThisIntegration = Object.entries(integration.metadata).filter(([key]) => {
+			return key !== "caCertificateText";
+		}).map(([key, rawValue]) => {
 			const name = `${namePrefix}_${key.toUpperCase()}`;
 			const value = String(rawValue);
 			if (integration.type === "mongodb" && integration.metadata.sslEnabled && key === "connection_string") return {
@@ -189,12 +533,12 @@ const getDatabricksSqlAlchemyInput = ({ httpPath, host, token, port, schema, cat
 			catalog
 		} },
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 const getDremioSqlAlchemyInput = ({ port, host, token, schema, sshEnabled, sshHost, sshPort, sshUser }) => {
@@ -205,12 +549,12 @@ const getDremioSqlAlchemyInput = ({ port, host, token, schema, sshEnabled, sshHo
 		url: url.href,
 		params: {},
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 const getSQLServerVar = ({ user, password, host, port, database, sshEnabled, sshHost, sshPort, sshUser }) => {
@@ -219,12 +563,12 @@ const getSQLServerVar = ({ user, password, host, port, database, sshEnabled, ssh
 		url: `mssql+pymssql://${encodeURIComponent(user)}:${encodeURIComponent(password)}@${host}${portSuffix}/${database}`,
 		params: {},
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 const getPostgresSqlAlchemyInput = (integrationId, projectRootDirectory, { user, password, host, port, database, sshEnabled, sshHost, sshPort, sshUser, sslEnabled, caCertificateName }) => {
@@ -245,12 +589,12 @@ const getPostgresSqlAlchemyInput = (integrationId, projectRootDirectory, { user,
 			}) : void 0
 		} },
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 const getRedshiftSqlAlchemyInput = (integrationId, projectRootDirectory, metadata) => {
@@ -273,12 +617,12 @@ const getRedshiftSqlAlchemyInput = (integrationId, projectRootDirectory, metadat
 			}) : void 0
 		} },
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: metadata.sshEnabled == null ? void 0 : String(metadata.sshEnabled),
+		ssh_options: metadata.sshEnabled ? {
+			enabled: true,
 			host: metadata.sshHost,
 			port: metadata.sshPort,
 			user: metadata.sshUser
-		}
+		} : {}
 	};
 	if (metadata.authMethod === AwsAuthMethods.IamRole && metadata.roleArn && metadata.roleExternalId) vars.iamParams = {
 		integrationId,
@@ -305,12 +649,12 @@ const getMaterializePostgresSqlAlchemyInput = (integrationId, projectRootDirecto
 			}) : void 0
 		} },
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 const getTrinoEnvVars = (integrationId, metadata, projectRootDirectory) => {
@@ -358,12 +702,12 @@ const getClickHouseSqlAlchemyInput = (integrationId, projectRootDirectory, { use
 		url: `clickhouse://${password ? `${encodeURIComponent(user)}:${encodeURIComponent(password)}` : encodeURIComponent(user)}@${host}${portSuffix}/${database}?${params.toString()}`,
 		params: {},
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 var BigQueryServiceAccountParseError = class extends Error {
@@ -424,12 +768,12 @@ const getMySqlSqlAlchemyInput = (integrationId, projectRootDirectory, { user, pa
 		url: `mysql+pymysql://${encodeURIComponent(user)}:${encodeURIComponent(password)}@${host}${portSuffix}/${database}`,
 		params: { connect_args: { ssl } },
 		param_style: "pyformat",
-		ssh_options: {
-			enabled: sshEnabled == null ? void 0 : String(sshEnabled),
+		ssh_options: sshEnabled ? {
+			enabled: true,
 			host: sshHost,
 			port: sshPort,
 			user: sshUser
-		}
+		} : {}
 	};
 };
 function getCACertificatePath({ projectRoot, integrationId, caCertificateName }) {
@@ -445,222 +789,6 @@ function assertNever(value) {
 	throw new Error(`Unexpected value: ${value}`);
 }
 
-//#endregion
-//#region src/database-integration-metadata-schemas.ts
-const athenaMetadataSchema = z.object({
-	access_key_id: z.string(),
-	region: z.string(),
-	s3_output_path: z.string(),
-	secret_access_key: z.string(),
-	workgroup: z.string().optional()
-});
-const bigqueryMetadataSchema = z.discriminatedUnion("authMethod", [z.object({
-	authMethod: z.literal(BigQueryAuthMethods.ServiceAccount),
-	service_account: z.string()
-}), z.object({
-	authMethod: z.literal(BigQueryAuthMethods.GoogleOauth),
-	project: z.string(),
-	clientId: z.string(),
-	clientSecret: z.string()
-})]);
-const clickhouseMetadataSchema = z.object({
-	host: z.string(),
-	user: z.string(),
-	password: z.string().optional(),
-	database: z.string(),
-	port: z.string().optional(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const databricksMetadataSchema = z.object({
-	host: z.string(),
-	httpPath: z.string(),
-	token: z.string(),
-	port: z.string(),
-	schema: z.string().optional(),
-	catalog: z.string().optional(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional()
-});
-const dremioMetadataSchema = z.object({
-	schema: z.string(),
-	host: z.string(),
-	port: z.string(),
-	token: z.string(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional()
-});
-const mongodbMetadataSchema = z.object({
-	connection_string: z.string(),
-	rawConnectionString: z.string().optional(),
-	prefix: z.string().optional(),
-	host: z.string().optional(),
-	port: z.string().optional(),
-	user: z.string().optional(),
-	password: z.string().optional(),
-	database: z.string().optional(),
-	options: z.string().optional(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const pandasDataframeMetadataSchema = z.object({});
-const commonRedshiftMetadataSchema = z.object({
-	database: z.string(),
-	host: z.string(),
-	port: z.string().optional(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const redshiftMetadataSchema = z.discriminatedUnion("authMethod", [
-	commonRedshiftMetadataSchema.extend({
-		authMethod: z.literal(DatabaseAuthMethods.UsernameAndPassword),
-		user: z.string(),
-		password: z.string()
-	}),
-	commonRedshiftMetadataSchema.extend({
-		authMethod: z.literal(AwsAuthMethods.IamRole),
-		roleArn: z.string(),
-		roleExternalId: z.string(),
-		roleNonce: z.string()
-	}),
-	commonRedshiftMetadataSchema.extend({ authMethod: z.literal(DatabaseAuthMethods.IndividualCredentials) })
-]);
-const commonSnowflakeMetadataSchema = z.object({
-	accountName: z.string(),
-	warehouse: z.string().optional(),
-	database: z.string().optional(),
-	role: z.string().optional()
-});
-const snowflakeMetadataSchema = z.discriminatedUnion("authMethod", [
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: z.literal(SnowflakeAuthMethods.Password),
-		username: z.string(),
-		password: z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: z.literal(SnowflakeAuthMethods.Okta),
-		clientId: z.string(),
-		clientSecret: z.string(),
-		oktaSubdomain: z.string(),
-		identityProvider: z.string(),
-		authorizationServer: z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: z.literal(SnowflakeAuthMethods.NativeSnowflake),
-		clientId: z.string(),
-		clientSecret: z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: z.literal(SnowflakeAuthMethods.AzureAd),
-		clientId: z.string(),
-		clientSecret: z.string(),
-		resource: z.string(),
-		tenant: z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({ authMethod: z.literal(SnowflakeAuthMethods.KeyPair) }),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: z.literal(SnowflakeAuthMethods.ServiceAccountKeyPair),
-		username: z.string(),
-		privateKey: z.string(),
-		privateKeyPassphrase: z.string().optional()
-	})
-]);
-const spannerMetadataSchema = z.object({
-	service_account: z.string(),
-	dataBoostEnabled: z.boolean(),
-	instance: z.string(),
-	database: z.string()
-});
-const trinoMetadataSchema = z.object({
-	host: z.string(),
-	user: z.string(),
-	password: z.string(),
-	database: z.string(),
-	port: z.string(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const commonDatabaseSchema = z.object({
-	host: z.string(),
-	user: z.string(),
-	password: z.string(),
-	database: z.string(),
-	sshEnabled: z.boolean().optional(),
-	sshHost: z.string().optional(),
-	sshPort: z.string().optional(),
-	sshUser: z.string().optional()
-});
-const alloydbMetadataSchema = commonDatabaseSchema.extend({
-	port: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const mariadbMetadataSchema = commonDatabaseSchema.extend({
-	port: z.string().optional(),
-	caCertificateName: z.string().optional()
-});
-const mindsdbMetadataSchema = commonDatabaseSchema.extend({
-	port: z.string().optional(),
-	caCertificateName: z.string().optional()
-});
-const mysqlMetadataSchema = commonDatabaseSchema.extend({
-	port: z.string().optional(),
-	caCertificateName: z.string().optional()
-});
-const pgsqlMetadataSchema = commonDatabaseSchema.extend({
-	port: z.string().optional(),
-	sslEnabled: z.boolean().optional(),
-	caCertificateName: z.string().optional(),
-	caCertificateText: z.string().optional()
-});
-const materializeMetadataSchema = pgsqlMetadataSchema.extend({ cluster: z.string() });
-const sqlServerMetadataSchema = commonDatabaseSchema.extend({ port: z.string() });
-const databaseMetadataSchemasByType = {
-	"alloydb": alloydbMetadataSchema,
-	"athena": athenaMetadataSchema,
-	"big-query": bigqueryMetadataSchema,
-	"clickhouse": clickhouseMetadataSchema,
-	"databricks": databricksMetadataSchema,
-	"dremio": dremioMetadataSchema,
-	"mariadb": mariadbMetadataSchema,
-	"materialize": materializeMetadataSchema,
-	"mindsdb": mindsdbMetadataSchema,
-	"mongodb": mongodbMetadataSchema,
-	"mysql": mysqlMetadataSchema,
-	"pandas-dataframe": pandasDataframeMetadataSchema,
-	"pgsql": pgsqlMetadataSchema,
-	"redshift": redshiftMetadataSchema,
-	"snowflake": snowflakeMetadataSchema,
-	"spanner": spannerMetadataSchema,
-	"sql-server": sqlServerMetadataSchema,
-	"trino": trinoMetadataSchema
-};
-
 //#endregion
 //#region src/database-integration-types.ts
 const sqlIntegrationTypes = [
@@ -684,6 +812,7 @@ const sqlIntegrationTypes = [
 ];
 const databaseIntegrationTypes = [...sqlIntegrationTypes, "mongodb"];
 const databaseIntegrationTypesWithSslSupport = [
+	"alloydb",
 	"clickhouse",
 	"dremio",
 	"mariadb",
@@ -694,6 +823,15 @@ const databaseIntegrationTypesWithSslSupport = [
 	"redshift",
 	"trino"
 ];
+function isDatabaseIntegrationType(type) {
+	return databaseIntegrationTypes.includes(type);
+}
+function isSqlIntegrationType(type) {
+	return sqlIntegrationTypes.includes(type);
+}
+function isDatabaseIntegrationTypeWithSslSupport(type) {
+	return databaseIntegrationTypesWithSslSupport.includes(type);
+}
 
 //#endregion
-export { AwsAuthMethods, BigQueryAuthMethods, BigQueryServiceAccountParseError, DatabaseAuthMethods, SnowflakeAuthMethods, SpannerServiceAccountParseError, databaseIntegrationTypes, databaseIntegrationTypesWithSslSupport, databaseMetadataSchemasByType, getEnvironmentVariablesForIntegrations, getSqlAlchemyInput, isFederatedAuthMetadata, isFederatedAuthMethod, sqlIntegrationTypes };
+export { AwsAuthMethods, BigQueryAuthMethods, BigQueryServiceAccountParseError, DatabaseAuthMethods, SnowflakeAuthMethods, SpannerServiceAccountParseError, databaseIntegrationConfigSchema, databaseIntegrationTypes, databaseIntegrationTypesWithSslSupport, databaseMetadataSchemasByType, federatedAuthMethods, getEnvironmentVariablesForIntegrations, getSnowflakeFederatedAuthSqlAlchemyInput, getSqlAlchemyInput, isDatabaseIntegrationType, isDatabaseIntegrationTypeWithSslSupport, isFederatedAuthMetadata, isFederatedAuthMethod, isSqlIntegrationType, sqlIntegrationTypes };