@deepnote/database-integrations 1.3.1 → 1.3.3

package/dist/index.cjs

@@ -41,6 +41,10 @@ const BigQueryAuthMethods = {
 	ServiceAccount: "service-account",
 	GoogleOauth: "google-oauth"
 };
+const TrinoAuthMethods = {
+	Password: "password",
+	Oauth: "oauth"
+};
 const DatabaseAuthMethods = {
 	UsernameAndPassword: "username-and-password",
 	IndividualCredentials: "individual-credentials"
@@ -51,7 +55,8 @@ const federatedAuthMethods = [
 	SnowflakeAuthMethods.AzureAd,
 	SnowflakeAuthMethods.KeyPair,
 	BigQueryAuthMethods.GoogleOauth,
-	DatabaseAuthMethods.IndividualCredentials
+	DatabaseAuthMethods.IndividualCredentials,
+	TrinoAuthMethods.Oauth
 ];
 function isFederatedAuthMethod(authMethod) {
 	return federatedAuthMethods.includes(authMethod);
@@ -145,9 +150,15 @@ const commonRedshiftMetadataSchema = zod.z.object({
 	caCertificateName: zod.z.string().optional(),
 	caCertificateText: zod.z.string().optional()
 });
-const redshiftMetadataSchema = zod.z.discriminatedUnion("authMethod", [
+const redshiftMetadataSchema = zod.z.preprocess((data) => {
+	if (typeof data === "object" && data !== null && (!("authMethod" in data) || !data.authMethod)) return {
+		...data,
+		authMethod: DatabaseAuthMethods.UsernameAndPassword
+	};
+	return data;
+}, zod.z.union([
 	commonRedshiftMetadataSchema.extend({
-		authMethod: zod.z.literal(DatabaseAuthMethods.UsernameAndPassword),
+		authMethod: zod.z.literal(DatabaseAuthMethods.UsernameAndPassword).optional(),
 		user: zod.z.string(),
 		password: zod.z.string()
 	}),
@@ -158,7 +169,7 @@ const redshiftMetadataSchema = zod.z.discriminatedUnion("authMethod", [
 		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(),
@@ -205,12 +216,10 @@ const spannerMetadataSchema = zod.z.object({
 	instance: zod.z.string(),
 	database: zod.z.string()
 });
-const trinoMetadataSchema = zod.z.object({
+const trinoBaseMetadataSchema = zod.z.object({
 	host: zod.z.string(),
-	user: zod.z.string(),
-	password: zod.z.string(),
+	port: zod.z.string().optional(),
 	database: zod.z.string(),
-	port: zod.z.string(),
 	sshEnabled: zod.z.boolean().optional(),
 	sshHost: zod.z.string().optional(),
 	sshPort: zod.z.string().optional(),
@@ -219,6 +228,22 @@ const trinoMetadataSchema = zod.z.object({
 	caCertificateName: zod.z.string().optional(),
 	caCertificateText: zod.z.string().optional()
 });
+const trinoUsernamePasswordMetadataSchema = trinoBaseMetadataSchema.extend({
+	user: zod.z.string(),
+	password: zod.z.string()
+});
+const trinoOAuthMetadataSchema = trinoBaseMetadataSchema.extend({
+	authMethod: zod.z.literal(TrinoAuthMethods.Oauth),
+	clientId: zod.z.string(),
+	clientSecret: zod.z.string(),
+	authUrl: zod.z.string().url(),
+	tokenUrl: zod.z.string().url()
+});
+const trinoMetadataSchema = zod.z.discriminatedUnion("authMethod", [
+	trinoUsernamePasswordMetadataSchema.extend({ authMethod: zod.z.literal(null) }),
+	trinoUsernamePasswordMetadataSchema.extend({ authMethod: zod.z.literal(TrinoAuthMethods.Password) }),
+	trinoOAuthMetadataSchema
+]);
 const commonDatabaseSchema = zod.z.object({
 	host: zod.z.string(),
 	user: zod.z.string(),
@@ -628,6 +653,7 @@ const getPostgresSqlAlchemyInput = (integrationId, projectRootDirectory, { user,
 	};
 };
 const getRedshiftSqlAlchemyInput = (integrationId, projectRootDirectory, metadata) => {
+	if (!metadata.authMethod) metadata.authMethod = DatabaseAuthMethods.UsernameAndPassword;
 	if (isFederatedAuthMethod(metadata.authMethod)) return null;
 	const portSuffix = metadata.port ? `:${metadata.port}` : "";
 	const mode = getPostgresStyleMode(metadata.caCertificateName, metadata.sslEnabled);
@@ -688,6 +714,7 @@ const getMaterializePostgresSqlAlchemyInput = (integrationId, projectRootDirecto
 	};
 };
 const getTrinoEnvVars = (integrationId, metadata, projectRootDirectory) => {
+	if (metadata.authMethod === TrinoAuthMethods.Oauth) return null;
 	return {
 		url: `trino://${encodeURIComponent(metadata.user)}:${encodeURIComponent(metadata.password)}@${metadata.host}:${metadata.port}/${encodeURIComponent(metadata.database)}`,
 		params: { connect_args: {
@@ -870,6 +897,7 @@ exports.BigQueryServiceAccountParseError = BigQueryServiceAccountParseError;
 exports.DatabaseAuthMethods = DatabaseAuthMethods;
 exports.SnowflakeAuthMethods = SnowflakeAuthMethods;
 exports.SpannerServiceAccountParseError = SpannerServiceAccountParseError;
+exports.TrinoAuthMethods = TrinoAuthMethods;
 exports.databaseIntegrationConfigSchema = databaseIntegrationConfigSchema;
 exports.databaseIntegrationTypes = databaseIntegrationTypes;
 exports.databaseIntegrationTypesWithSslSupport = databaseIntegrationTypesWithSslSupport;