@deepnote/database-integrations 1.3.2 → 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);
@@ -211,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(),
@@ -225,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(),
@@ -695,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: {
@@ -877,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;