@deepnote/database-integrations 1.1.0 → 1.2.0

package/dist/index.cjs

@@ -21,8 +21,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 }) : target, mod));
 
 //#endregion
-let build_url_ts = require("build-url-ts");
-build_url_ts = __toESM(build_url_ts);
 let zod = require("zod");
 zod = __toESM(zod);
 
@@ -62,12 +60,349 @@ 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
+/**
+* Build a URL with path and query parameters using native URL API
+*/
+function buildUrl(baseUrl, options) {
+	const url = new URL(baseUrl.replace(/^(\w+):\/\//, "http://"));
+	if (options.path) url.pathname = options.path;
+	if (options.queryParams) {
+		for (const [key, value] of Object.entries(options.queryParams)) if (value !== void 0) url.searchParams.set(key, value);
+	}
+	return url.href.replace(/^http:\/\//, baseUrl.match(/^(\w+):\/\//)?.[0] || "http://");
+}
 function getSnowflakeSqlAlchemyInput(metadata, params) {
 	if (isFederatedAuthMetadata(metadata)) return null;
 	const { username, accountName, database, authMethod } = metadata;
-	if (authMethod === SnowflakeAuthMethods.ServiceAccountKeyPair) return createSqlAlchemyInputFromUrl((0, build_url_ts.default)(`snowflake://${encodeURIComponent(username)}@${encodeURIComponent(accountName)}`, {
+	if (authMethod === SnowflakeAuthMethods.ServiceAccountKeyPair) return createSqlAlchemyInputFromUrl(buildUrl(`snowflake://${encodeURIComponent(username)}@${encodeURIComponent(accountName)}`, {
 		path: database,
 		queryParams: {
 			...createBaseQueryParams(metadata, params.snowflakePartnerIdentifier),
@@ -78,11 +413,27 @@ function getSnowflakeSqlAlchemyInput(metadata, params) {
 		password: metadata.privateKeyPassphrase,
 		username: metadata.username
 	});
-	return createSqlAlchemyInputFromUrl((0, build_url_ts.default)(`snowflake://${encodeURIComponent(username)}:${encodeURIComponent(metadata.password)}@${encodeURIComponent(accountName)}`, {
+	return createSqlAlchemyInputFromUrl(buildUrl(`snowflake://${encodeURIComponent(username)}:${encodeURIComponent(metadata.password)}@${encodeURIComponent(accountName)}`, {
 		path: database,
 		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,
@@ -108,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 {
@@ -460,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 = [
@@ -699,6 +836,7 @@ const sqlIntegrationTypes = [
 ];
 const databaseIntegrationTypes = [...sqlIntegrationTypes, "mongodb"];
 const databaseIntegrationTypesWithSslSupport = [
+	"alloydb",
 	"clickhouse",
 	"dremio",
 	"mariadb",
@@ -709,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;
@@ -717,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;