@deepnote/database-integrations 1.3.4 → 1.4.0

package/dist/index.cjs

@@ -43,7 +43,7 @@ const BigQueryAuthMethods = {
 };
 const TrinoAuthMethods = {
 	Password: "password",
-	Oauth: "oauth"
+	Oauth: "trino-oauth"
 };
 const DatabaseAuthMethods = {
 	UsernameAndPassword: "username-and-password",
@@ -174,11 +174,21 @@ const commonSnowflakeMetadataSchema = zod.z.object({
 	accountName: zod.z.string(),
 	warehouse: zod.z.string().optional(),
 	database: zod.z.string().optional(),
-	role: zod.z.string().optional()
+	role: zod.z.string().optional(),
+	dbt: zod.z.boolean().optional(),
+	dbtServiceToken: zod.z.string().optional(),
+	dbtPrimaryJobId: zod.z.string().optional(),
+	dbtProxyServerUrl: zod.z.string().optional()
 });
-const snowflakeMetadataSchema = zod.z.discriminatedUnion("authMethod", [
+const snowflakeMetadataSchema = zod.z.preprocess((data) => {
+	if (typeof data === "object" && data !== null && (!("authMethod" in data) || !data.authMethod)) return {
+		...data,
+		authMethod: SnowflakeAuthMethods.Password
+	};
+	return data;
+}, zod.z.discriminatedUnion("authMethod", [
 	commonSnowflakeMetadataSchema.extend({
-		authMethod: zod.z.literal(SnowflakeAuthMethods.Password),
+		authMethod: zod.z.literal(SnowflakeAuthMethods.Password).optional(),
 		username: zod.z.string(),
 		password: zod.z.string()
 	}),
@@ -209,7 +219,7 @@ const snowflakeMetadataSchema = zod.z.discriminatedUnion("authMethod", [
 		privateKey: zod.z.string(),
 		privateKeyPassphrase: zod.z.string().optional()
 	})
-]);
+]));
 const spannerMetadataSchema = zod.z.object({
 	service_account: zod.z.string(),
 	dataBoostEnabled: zod.z.boolean(),
@@ -229,6 +239,7 @@ const trinoBaseMetadataSchema = zod.z.object({
 	caCertificateText: zod.z.string().optional()
 });
 const trinoUsernamePasswordMetadataSchema = trinoBaseMetadataSchema.extend({
+	authMethod: zod.z.literal(TrinoAuthMethods.Password).nullish(),
 	user: zod.z.string(),
 	password: zod.z.string()
 });
@@ -239,11 +250,7 @@ const trinoOAuthMetadataSchema = trinoBaseMetadataSchema.extend({
 	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 trinoMetadataSchema = zod.z.discriminatedUnion("authMethod", [trinoUsernamePasswordMetadataSchema, trinoOAuthMetadataSchema]);
 const commonDatabaseSchema = zod.z.object({
 	host: zod.z.string(),
 	user: zod.z.string(),