feat: Add Dremio Cloud Support

- Adds support to cubejs-dremio-driver for Dremio Cloud
- Adds support for personal access tokens for Dremio Server

Author: maladroitthief

packages/cubejs-dremio-driver/README.md

@@ -9,11 +9,24 @@
 
 Pure Javascript Dremio driver.
 
+## Dremio Cloud
+
+To use this driver with [Dremio Cloud](https://docs.dremio.com/cloud/reference/api/), use the following setup:
+
+| Environment Variable | Value                                              |
+| -------------------- | -------------------------------------------------- |
+| CUBEJS_DB_TYPE       | dremio                                             |
+| CUBEJS_DB_URL        | https://api.dremio.cloud/v0/projects/${PROJECT_ID} |
+| CUBEJS_DB_NAME       | ${DB_NAME}                                         |
+| CUBEJS_DB_PASS       | Bearer ${PERSONAL_ACCESS_TOKEN}                    |
+
+> It's important to note that "Bearer" is required for using personal access tokens.
+
 ## Support
 
-This package is **community supported** and should be used at your own risk. 
+This package is **community supported** and should be used at your own risk.
 
-While the Cube Dev team is happy to review and accept future community contributions, we don't have active plans for further development. This includes bug fixes unless they affect different parts of Cube.js. **We're looking for maintainers for this package.** If you'd like to become a maintainer, please contact us in Cube.js Slack. 
+While the Cube Dev team is happy to review and accept future community contributions, we don't have active plans for further development. This includes bug fixes unless they affect different parts of Cube.js. **We're looking for maintainers for this package.** If you'd like to become a maintainer, please contact us in Cube.js Slack.
 
 ## License
 

packages/cubejs-dremio-driver/driver/DremioDriver.js

@@ -49,6 +49,10 @@ class DremioDriver extends BaseDriver {
       assertDataSource('default');
 
     this.config = {
+      dbUrl:
+        config.dbUrl ||
+        getEnv('dbUrl', { dataSource }) ||
+        '',
       host:
         config.host ||
         getEnv('dbHost', { dataSource }) ||
@@ -80,11 +84,24 @@ class DremioDriver extends BaseDriver {
         getEnv('dbPollMaxInterval', { dataSource })
       ) * 1000,
     };
-    const protocol = (this.config.ssl === true || this.config.ssl === 'true')
-      ? 'https'
-      : 'http';
-    this.config.url =
-      `${protocol}://${this.config.host}:${this.config.port}`;
+
+    if (this.config.dbUrl) {
+      this.config.url = this.config.dbUrl;
+      this.config.apiVersion = '';
+      this.config.auth = 'PAT';
+    } else {
+      const protocol = (this.config.ssl === true || this.config.ssl === 'true')
+        ? 'https'
+        : 'http';
+      this.config.url = `${protocol}://${this.config.host}:${this.config.port}`;
+      this.config.apiVersion = '/api/v3';
+    }
+
+    if (this.config.auth === 'PAT' || this.config.password.startsWith('Bearer ')) {
+      this.config.auth = 'PAT';
+    } else {
+      this.config.auth = 'BASIC';
+    }
   }
 
   /**
@@ -103,6 +120,19 @@ class DremioDriver extends BaseDriver {
    * @protected
    */
   async getToken() {
+    if (this.config.auth === 'PAT') {
+      await axios.get(
+        `${this.config.url}${this.config.apiVersion}/catalog`,
+        {
+          headers: {
+            Authorization: this.config.password
+          },
+        },
+      );
+
+      return this.config.password;
+    }
+
     if (this.authToken && this.authToken.expires > new Date().getTime()) {
       return `_dremio${this.authToken.token}`;
     }
@@ -129,7 +159,7 @@ class DremioDriver extends BaseDriver {
 
     return axios.request({
       method,
-      url: `${this.config.url}${url}`,
+      url: `${this.config.url}${this.config.apiVersion}${url}`,
       headers: {
         Authorization: token
       },
@@ -141,7 +171,7 @@ class DremioDriver extends BaseDriver {
    * @protected
    */
   async getJobStatus(jobId) {
-    const { data } = await this.restDremioQuery('get', `/api/v3/job/${jobId}`);
+    const { data } = await this.restDremioQuery('get', `/job/${jobId}`);
 
     if (data.jobState === 'FAILED') {
       throw new Error(data.errorMessage);
@@ -162,7 +192,7 @@ class DremioDriver extends BaseDriver {
    * @protected
    */
   async getJobResults(jobId, limit = 500, offset = 0) {
-    return this.restDremioQuery('get', `/api/v3/job/${jobId}/results?offset=${offset}&limit=${limit}`);
+    return this.restDremioQuery('get', `/job/${jobId}/results?offset=${offset}&limit=${limit}`);
   }
 
   /**
@@ -171,7 +201,7 @@ class DremioDriver extends BaseDriver {
    * @return {Promise<*>}
    */
   async executeQuery(sql) {
-    const { data } = await this.restDremioQuery('post', '/api/v3/sql', { sql });
+    const { data } = await this.restDremioQuery('post', '/sql', { sql });
     return data.id;
   }
 
@@ -216,7 +246,7 @@ class DremioDriver extends BaseDriver {
   }
 
   async refreshTablesSchema(path) {
-    const { data } = await this.restDremioQuery('get', `/api/v3/catalog/by-path/${path}`);
+    const { data } = await this.restDremioQuery('get', `/catalog/by-path/${path}`);
     if (!data || !data.children) {
       return true;
     }