@deepnote/database-integrations 1.1.1 → 1.2.1

package/dist/index.cjs

@@ -60,6 +60,332 @@ function isFederatedAuthMetadata(metadata) {
 	return isFederatedAuthMethod(metadata.authMethod);
 }
 
+//#endregion
+//#region src/database-integration-metadata-schemas.ts
+const athenaMetadataSchema = zod.z.object({
+	access_key_id: zod.z.string(),
+	region: zod.z.string(),
+	s3_output_path: zod.z.string(),
+	secret_access_key: zod.z.string(),
+	workgroup: zod.z.string().optional()
+});
+const bigqueryMetadataSchema = zod.z.discriminatedUnion("authMethod", [zod.z.object({
+	authMethod: zod.z.literal(BigQueryAuthMethods.ServiceAccount),
+	service_account: zod.z.string()
+}), zod.z.object({
+	authMethod: zod.z.literal(BigQueryAuthMethods.GoogleOauth),
+	project: zod.z.string(),
+	clientId: zod.z.string(),
+	clientSecret: zod.z.string()
+})]);
+const clickhouseMetadataSchema = zod.z.object({
+	host: zod.z.string(),
+	user: zod.z.string(),
+	password: zod.z.string().optional(),
+	database: zod.z.string(),
+	port: zod.z.string().optional(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const databricksMetadataSchema = zod.z.object({
+	host: zod.z.string(),
+	httpPath: zod.z.string(),
+	token: zod.z.string(),
+	port: zod.z.string(),
+	schema: zod.z.string().optional(),
+	catalog: zod.z.string().optional(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional()
+});
+const dremioMetadataSchema = zod.z.object({
+	schema: zod.z.string(),
+	host: zod.z.string(),
+	port: zod.z.string(),
+	token: zod.z.string(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional()
+});
+const mongodbMetadataSchema = zod.z.object({
+	connection_string: zod.z.string(),
+	rawConnectionString: zod.z.string().optional(),
+	prefix: zod.z.string().optional(),
+	host: zod.z.string().optional(),
+	port: zod.z.string().optional(),
+	user: zod.z.string().optional(),
+	password: zod.z.string().optional(),
+	database: zod.z.string().optional(),
+	options: zod.z.string().optional(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const pandasDataframeMetadataSchema = zod.z.object({});
+const commonRedshiftMetadataSchema = zod.z.object({
+	database: zod.z.string(),
+	host: zod.z.string(),
+	port: zod.z.string().optional(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const redshiftMetadataSchema = zod.z.discriminatedUnion("authMethod", [
+	commonRedshiftMetadataSchema.extend({
+		authMethod: zod.z.literal(DatabaseAuthMethods.UsernameAndPassword),
+		user: zod.z.string(),
+		password: zod.z.string()
+	}),
+	commonRedshiftMetadataSchema.extend({
+		authMethod: zod.z.literal(AwsAuthMethods.IamRole),
+		roleArn: zod.z.string(),
+		roleExternalId: zod.z.string(),
+		roleNonce: zod.z.string()
+	}),
+	commonRedshiftMetadataSchema.extend({ authMethod: zod.z.literal(DatabaseAuthMethods.IndividualCredentials) })
+]);
+const commonSnowflakeMetadataSchema = zod.z.object({
+	accountName: zod.z.string(),
+	warehouse: zod.z.string().optional(),
+	database: zod.z.string().optional(),
+	role: zod.z.string().optional()
+});
+const snowflakeMetadataSchema = zod.z.discriminatedUnion("authMethod", [
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: zod.z.literal(SnowflakeAuthMethods.Password),
+		username: zod.z.string(),
+		password: zod.z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: zod.z.literal(SnowflakeAuthMethods.Okta),
+		clientId: zod.z.string(),
+		clientSecret: zod.z.string(),
+		oktaSubdomain: zod.z.string(),
+		identityProvider: zod.z.string(),
+		authorizationServer: zod.z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: zod.z.literal(SnowflakeAuthMethods.NativeSnowflake),
+		clientId: zod.z.string(),
+		clientSecret: zod.z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: zod.z.literal(SnowflakeAuthMethods.AzureAd),
+		clientId: zod.z.string(),
+		clientSecret: zod.z.string(),
+		resource: zod.z.string(),
+		tenant: zod.z.string()
+	}),
+	commonSnowflakeMetadataSchema.extend({ authMethod: zod.z.literal(SnowflakeAuthMethods.KeyPair) }),
+	commonSnowflakeMetadataSchema.extend({
+		authMethod: zod.z.literal(SnowflakeAuthMethods.ServiceAccountKeyPair),
+		username: zod.z.string(),
+		privateKey: zod.z.string(),
+		privateKeyPassphrase: zod.z.string().optional()
+	})
+]);
+const spannerMetadataSchema = zod.z.object({
+	service_account: zod.z.string(),
+	dataBoostEnabled: zod.z.boolean(),
+	instance: zod.z.string(),
+	database: zod.z.string()
+});
+const trinoMetadataSchema = zod.z.object({
+	host: zod.z.string(),
+	user: zod.z.string(),
+	password: zod.z.string(),
+	database: zod.z.string(),
+	port: zod.z.string(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const commonDatabaseSchema = zod.z.object({
+	host: zod.z.string(),
+	user: zod.z.string(),
+	password: zod.z.string(),
+	database: zod.z.string(),
+	sshEnabled: zod.z.boolean().optional(),
+	sshHost: zod.z.string().optional(),
+	sshPort: zod.z.string().optional(),
+	sshUser: zod.z.string().optional()
+});
+const alloydbMetadataSchema = commonDatabaseSchema.extend({
+	port: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const mariadbMetadataSchema = commonDatabaseSchema.extend({
+	port: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const mindsdbMetadataSchema = commonDatabaseSchema.extend({
+	port: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const mysqlMetadataSchema = commonDatabaseSchema.extend({
+	port: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const pgsqlMetadataSchema = commonDatabaseSchema.extend({
+	port: zod.z.string().optional(),
+	sslEnabled: zod.z.boolean().optional(),
+	caCertificateName: zod.z.string().optional(),
+	caCertificateText: zod.z.string().optional()
+});
+const materializeMetadataSchema = pgsqlMetadataSchema.extend({ cluster: zod.z.string() });
+const sqlServerMetadataSchema = commonDatabaseSchema.extend({ port: zod.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 = zod.default.object({
+	id: zod.default.string(),
+	name: zod.default.string()
+});
+const databaseIntegrationConfigSchema = zod.default.discriminatedUnion("type", [
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("alloydb"),
+		metadata: databaseMetadataSchemasByType["alloydb"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("athena"),
+		metadata: databaseMetadataSchemasByType["athena"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("big-query"),
+		metadata: databaseMetadataSchemasByType["big-query"],
+		federated_auth_method: zod.default.enum(["google-oauth"]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("clickhouse"),
+		metadata: databaseMetadataSchemasByType["clickhouse"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("databricks"),
+		metadata: databaseMetadataSchemasByType["databricks"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("dremio"),
+		metadata: databaseMetadataSchemasByType["dremio"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("mariadb"),
+		metadata: databaseMetadataSchemasByType["mariadb"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("materialize"),
+		metadata: databaseMetadataSchemasByType["materialize"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("mindsdb"),
+		metadata: databaseMetadataSchemasByType["mindsdb"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("mongodb"),
+		metadata: databaseMetadataSchemasByType["mongodb"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("mysql"),
+		metadata: databaseMetadataSchemasByType["mysql"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("pandas-dataframe"),
+		metadata: databaseMetadataSchemasByType["pandas-dataframe"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("pgsql"),
+		metadata: databaseMetadataSchemasByType["pgsql"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("redshift"),
+		metadata: databaseMetadataSchemasByType["redshift"],
+		federated_auth_method: zod.default.enum(["individual-credentials"]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("snowflake"),
+		metadata: databaseMetadataSchemasByType["snowflake"],
+		federated_auth_method: zod.default.enum([
+			"okta",
+			"snowflake",
+			"azure",
+			"key-pair"
+		]).optional().nullable()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("spanner"),
+		metadata: databaseMetadataSchemasByType["spanner"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("sql-server"),
+		metadata: databaseMetadataSchemasByType["sql-server"],
+		federated_auth_method: zod.default.null().optional()
+	}),
+	commonIntegrationConfig.extend({
+		type: zod.default.literal("trino"),
+		metadata: databaseMetadataSchemasByType["trino"],
+		federated_auth_method: zod.default.null().optional()
+	})
+]);
+
 //#endregion
 //#region src/snowflake-integration-env-vars.ts
 /**
@@ -92,6 +418,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,
@@ -117,7 +459,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 {
@@ -213,12 +557,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 }) => {
@@ -229,12 +573,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 }) => {
@@ -243,12 +587,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 }) => {
@@ -269,12 +613,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) => {
@@ -297,12 +641,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,
@@ -329,12 +673,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) => {
@@ -382,12 +726,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 {
@@ -448,12 +792,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 }) {
@@ -469,222 +813,6 @@ function assertNever(value) {
 	throw new Error(`Unexpected value: ${value}`);
 }
 
-//#endregion
-//#region src/database-integration-metadata-schemas.ts
-const athenaMetadataSchema = zod.z.object({
-	access_key_id: zod.z.string(),
-	region: zod.z.string(),
-	s3_output_path: zod.z.string(),
-	secret_access_key: zod.z.string(),
-	workgroup: zod.z.string().optional()
-});
-const bigqueryMetadataSchema = zod.z.discriminatedUnion("authMethod", [zod.z.object({
-	authMethod: zod.z.literal(BigQueryAuthMethods.ServiceAccount),
-	service_account: zod.z.string()
-}), zod.z.object({
-	authMethod: zod.z.literal(BigQueryAuthMethods.GoogleOauth),
-	project: zod.z.string(),
-	clientId: zod.z.string(),
-	clientSecret: zod.z.string()
-})]);
-const clickhouseMetadataSchema = zod.z.object({
-	host: zod.z.string(),
-	user: zod.z.string(),
-	password: zod.z.string().optional(),
-	database: zod.z.string(),
-	port: zod.z.string().optional(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const databricksMetadataSchema = zod.z.object({
-	host: zod.z.string(),
-	httpPath: zod.z.string(),
-	token: zod.z.string(),
-	port: zod.z.string(),
-	schema: zod.z.string().optional(),
-	catalog: zod.z.string().optional(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional()
-});
-const dremioMetadataSchema = zod.z.object({
-	schema: zod.z.string(),
-	host: zod.z.string(),
-	port: zod.z.string(),
-	token: zod.z.string(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional()
-});
-const mongodbMetadataSchema = zod.z.object({
-	connection_string: zod.z.string(),
-	rawConnectionString: zod.z.string().optional(),
-	prefix: zod.z.string().optional(),
-	host: zod.z.string().optional(),
-	port: zod.z.string().optional(),
-	user: zod.z.string().optional(),
-	password: zod.z.string().optional(),
-	database: zod.z.string().optional(),
-	options: zod.z.string().optional(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const pandasDataframeMetadataSchema = zod.z.object({});
-const commonRedshiftMetadataSchema = zod.z.object({
-	database: zod.z.string(),
-	host: zod.z.string(),
-	port: zod.z.string().optional(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const redshiftMetadataSchema = zod.z.discriminatedUnion("authMethod", [
-	commonRedshiftMetadataSchema.extend({
-		authMethod: zod.z.literal(DatabaseAuthMethods.UsernameAndPassword),
-		user: zod.z.string(),
-		password: zod.z.string()
-	}),
-	commonRedshiftMetadataSchema.extend({
-		authMethod: zod.z.literal(AwsAuthMethods.IamRole),
-		roleArn: zod.z.string(),
-		roleExternalId: zod.z.string(),
-		roleNonce: zod.z.string()
-	}),
-	commonRedshiftMetadataSchema.extend({ authMethod: zod.z.literal(DatabaseAuthMethods.IndividualCredentials) })
-]);
-const commonSnowflakeMetadataSchema = zod.z.object({
-	accountName: zod.z.string(),
-	warehouse: zod.z.string().optional(),
-	database: zod.z.string().optional(),
-	role: zod.z.string().optional()
-});
-const snowflakeMetadataSchema = zod.z.discriminatedUnion("authMethod", [
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.Password),
-		username: zod.z.string(),
-		password: zod.z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.Okta),
-		clientId: zod.z.string(),
-		clientSecret: zod.z.string(),
-		oktaSubdomain: zod.z.string(),
-		identityProvider: zod.z.string(),
-		authorizationServer: zod.z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.NativeSnowflake),
-		clientId: zod.z.string(),
-		clientSecret: zod.z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.AzureAd),
-		clientId: zod.z.string(),
-		clientSecret: zod.z.string(),
-		resource: zod.z.string(),
-		tenant: zod.z.string()
-	}),
-	commonSnowflakeMetadataSchema.extend({ authMethod: zod.z.literal(SnowflakeAuthMethods.KeyPair) }),
-	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.ServiceAccountKeyPair),
-		username: zod.z.string(),
-		privateKey: zod.z.string(),
-		privateKeyPassphrase: zod.z.string().optional()
-	})
-]);
-const spannerMetadataSchema = zod.z.object({
-	service_account: zod.z.string(),
-	dataBoostEnabled: zod.z.boolean(),
-	instance: zod.z.string(),
-	database: zod.z.string()
-});
-const trinoMetadataSchema = zod.z.object({
-	host: zod.z.string(),
-	user: zod.z.string(),
-	password: zod.z.string(),
-	database: zod.z.string(),
-	port: zod.z.string(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const commonDatabaseSchema = zod.z.object({
-	host: zod.z.string(),
-	user: zod.z.string(),
-	password: zod.z.string(),
-	database: zod.z.string(),
-	sshEnabled: zod.z.boolean().optional(),
-	sshHost: zod.z.string().optional(),
-	sshPort: zod.z.string().optional(),
-	sshUser: zod.z.string().optional()
-});
-const alloydbMetadataSchema = commonDatabaseSchema.extend({
-	port: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const mariadbMetadataSchema = commonDatabaseSchema.extend({
-	port: zod.z.string().optional(),
-	caCertificateName: zod.z.string().optional()
-});
-const mindsdbMetadataSchema = commonDatabaseSchema.extend({
-	port: zod.z.string().optional(),
-	caCertificateName: zod.z.string().optional()
-});
-const mysqlMetadataSchema = commonDatabaseSchema.extend({
-	port: zod.z.string().optional(),
-	caCertificateName: zod.z.string().optional()
-});
-const pgsqlMetadataSchema = commonDatabaseSchema.extend({
-	port: zod.z.string().optional(),
-	sslEnabled: zod.z.boolean().optional(),
-	caCertificateName: zod.z.string().optional(),
-	caCertificateText: zod.z.string().optional()
-});
-const materializeMetadataSchema = pgsqlMetadataSchema.extend({ cluster: zod.z.string() });
-const sqlServerMetadataSchema = commonDatabaseSchema.extend({ port: zod.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 = [
@@ -708,6 +836,7 @@ const sqlIntegrationTypes = [
 ];
 const databaseIntegrationTypes = [...sqlIntegrationTypes, "mongodb"];
 const databaseIntegrationTypesWithSslSupport = [
+	"alloydb",
 	"clickhouse",
 	"dremio",
 	"mariadb",
@@ -718,6 +847,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
 exports.AwsAuthMethods = AwsAuthMethods;
@@ -726,11 +864,17 @@ exports.BigQueryServiceAccountParseError = BigQueryServiceAccountParseError;
 exports.DatabaseAuthMethods = DatabaseAuthMethods;
 exports.SnowflakeAuthMethods = SnowflakeAuthMethods;
 exports.SpannerServiceAccountParseError = SpannerServiceAccountParseError;
+exports.databaseIntegrationConfigSchema = databaseIntegrationConfigSchema;
 exports.databaseIntegrationTypes = databaseIntegrationTypes;
 exports.databaseIntegrationTypesWithSslSupport = databaseIntegrationTypesWithSslSupport;
 exports.databaseMetadataSchemasByType = databaseMetadataSchemasByType;
+exports.federatedAuthMethods = federatedAuthMethods;
 exports.getEnvironmentVariablesForIntegrations = getEnvironmentVariablesForIntegrations;
+exports.getSnowflakeFederatedAuthSqlAlchemyInput = getSnowflakeFederatedAuthSqlAlchemyInput;
 exports.getSqlAlchemyInput = getSqlAlchemyInput;
+exports.isDatabaseIntegrationType = isDatabaseIntegrationType;
+exports.isDatabaseIntegrationTypeWithSslSupport = isDatabaseIntegrationTypeWithSslSupport;
 exports.isFederatedAuthMetadata = isFederatedAuthMetadata;
 exports.isFederatedAuthMethod = isFederatedAuthMethod;
+exports.isSqlIntegrationType = isSqlIntegrationType;
 exports.sqlIntegrationTypes = sqlIntegrationTypes;