@budibase/server 1.0.212-alpha.14 → 1.0.212-alpha.15

package/coverage/lcov.info

@@ -7091,19 +7091,20 @@ BRH:20
 end_of_record
 TN:
 SF:src/threads/query.js
-FN:14,(anonymous_0)
-FN:33,(anonymous_1)
-FN:42,(anonymous_2)
-FN:124,(anonymous_3)
-FN:125,(anonymous_4)
-FN:138,(anonymous_5)
-FN:154,(anonymous_6)
-FN:172,(anonymous_7)
-FN:187,(anonymous_8)
-FN:196,(anonymous_9)
-FN:218,(anonymous_10)
-FN:219,(anonymous_11)
-FNF:12
+FN:20,(anonymous_0)
+FN:40,(anonymous_1)
+FN:50,(anonymous_2)
+FN:142,(anonymous_3)
+FN:143,(anonymous_4)
+FN:156,(anonymous_5)
+FN:172,(anonymous_6)
+FN:197,(anonymous_7)
+FN:215,(anonymous_8)
+FN:230,(anonymous_9)
+FN:239,(anonymous_10)
+FN:261,(anonymous_11)
+FN:262,(anonymous_12)
+FNF:13
 FNH:10
 FNDA:22,(anonymous_0)
 FNDA:23,(anonymous_1)
@@ -7111,171 +7112,204 @@ FNDA:0,(anonymous_2)
 FNDA:20,(anonymous_3)
 FNDA:0,(anonymous_4)
 FNDA:5,(anonymous_5)
-FNDA:6,(anonymous_6)
-FNDA:21,(anonymous_7)
-FNDA:6,(anonymous_8)
+FNDA:0,(anonymous_6)
+FNDA:6,(anonymous_7)
+FNDA:21,(anonymous_8)
 FNDA:6,(anonymous_9)
-FNDA:17,(anonymous_10)
+FNDA:6,(anonymous_10)
 FNDA:17,(anonymous_11)
+FNDA:17,(anonymous_12)
 DA:1,2
 DA:2,2
 DA:3,2
 DA:4,2
 DA:5,2
 DA:6,2
-DA:7,2
+DA:10,2
 DA:11,2
-DA:15,22
-DA:16,22
-DA:17,22
-DA:18,22
-DA:19,22
-DA:20,22
+DA:13,2
+DA:17,2
 DA:21,22
 DA:22,22
 DA:23,22
+DA:24,22
 DA:25,22
+DA:26,22
+DA:27,22
+DA:28,22
 DA:29,22
-DA:30,22
-DA:34,23
-DA:35,23
-DA:36,23
-DA:37,2
-DA:40,21
-DA:41,0
-DA:43,0
+DA:31,22
+DA:35,22
+DA:36,22
+DA:37,22
+DA:41,23
+DA:43,23
+DA:44,23
+DA:45,2
 DA:48,21
-DA:51,21
-DA:55,21
+DA:49,0
+DA:51,0
 DA:56,21
 DA:59,21
-DA:60,1
-DA:68,21
-DA:69,2
-DA:71,19
-DA:75,21
-DA:76,0
-DA:79,21
-DA:80,21
-DA:81,21
-DA:82,21
+DA:63,21
+DA:64,21
+DA:67,21
+DA:68,1
+DA:76,21
+DA:77,2
+DA:79,19
 DA:83,21
-DA:84,21
-DA:85,19
-DA:86,19
-DA:87,19
-DA:88,19
+DA:84,0
+DA:87,21
+DA:88,21
+DA:89,21
+DA:90,21
+DA:91,21
 DA:92,21
-DA:93,0
-DA:97,0
-DA:101,21
-DA:107,1
-DA:109,1
+DA:93,19
+DA:94,19
+DA:95,19
+DA:96,19
+DA:100,21
+DA:101,0
+DA:105,0
+DA:109,21
 DA:110,1
-DA:114,20
-DA:115,0
-DA:119,20
-DA:120,18
-DA:124,20
-DA:125,0
-DA:129,20
-DA:131,20
-DA:132,0
-DA:135,20
-DA:139,5
-DA:140,5
-DA:141,5
-DA:142,5
-DA:155,6
-DA:156,6
-DA:157,6
-DA:158,6
-DA:159,6
+DA:116,0
+DA:117,0
+DA:119,0
+DA:122,0
+DA:126,1
+DA:127,1
+DA:128,1
+DA:132,20
+DA:133,0
+DA:137,20
+DA:138,18
+DA:142,20
+DA:143,0
+DA:147,20
+DA:149,20
+DA:150,0
+DA:153,20
+DA:157,5
+DA:158,5
+DA:159,5
 DA:160,5
-DA:164,5
-DA:165,5
-DA:167,1
-DA:169,6
-DA:173,21
-DA:174,21
-DA:175,0
-DA:177,21
-DA:178,21
-DA:179,21
-DA:180,2
-DA:181,2
-DA:184,21
-DA:186,16
-DA:187,16
-DA:189,6
+DA:173,0
+DA:174,0
+DA:176,0
+DA:177,0
+DA:178,0
+DA:181,0
+DA:189,0
 DA:190,0
-DA:193,6
-DA:194,6
-DA:196,16
-DA:197,16
-DA:198,16
+DA:191,0
+DA:194,0
+DA:198,6
 DA:199,6
 DA:200,6
-DA:211,6
-DA:214,21
-DA:218,2
-DA:219,17
-DA:220,17
-DA:221,17
-DA:222,17
-DA:223,15
-DA:225,2
-LF:108
-LH:98
-BRDA:14,0,0,17
-BRDA:36,1,0,2
-BRDA:36,1,1,21
-BRDA:40,2,0,0
-BRDA:40,2,1,21
-BRDA:59,3,0,1
-BRDA:59,3,1,20
-BRDA:68,4,0,2
-BRDA:68,4,1,19
-BRDA:75,5,0,0
-BRDA:75,5,1,21
-BRDA:84,6,0,19
-BRDA:84,6,1,2
-BRDA:92,7,0,0
-BRDA:92,7,1,21
-BRDA:101,8,0,1
-BRDA:101,8,1,20
-BRDA:102,9,0,21
-BRDA:102,9,1,19
-BRDA:102,9,2,1
-BRDA:102,9,3,1
-BRDA:114,10,0,0
-BRDA:114,10,1,20
-BRDA:119,11,0,18
-BRDA:119,11,1,2
-BRDA:124,12,0,0
-BRDA:124,12,1,20
-BRDA:131,13,0,0
-BRDA:131,13,1,20
-BRDA:159,14,0,5
-BRDA:159,14,1,1
-BRDA:160,15,0,0
-BRDA:160,15,1,5
-BRDA:174,16,0,0
-BRDA:174,16,1,21
-BRDA:174,17,0,21
-BRDA:174,17,1,21
-BRDA:177,18,0,21
-BRDA:177,18,1,20
-BRDA:178,19,0,21
-BRDA:178,19,1,10
-BRDA:180,20,0,2
-BRDA:180,20,1,0
-BRDA:184,21,0,16
-BRDA:184,21,1,5
-BRDA:189,22,0,0
-BRDA:189,22,1,6
-BRF:47
-BRH:37
+DA:201,6
+DA:202,6
+DA:203,5
+DA:207,5
+DA:208,5
+DA:210,1
+DA:212,6
+DA:216,21
+DA:217,21
+DA:218,0
+DA:220,21
+DA:221,21
+DA:222,21
+DA:223,2
+DA:224,2
+DA:227,21
+DA:229,16
+DA:230,16
+DA:232,6
+DA:233,0
+DA:236,6
+DA:237,6
+DA:239,16
+DA:240,16
+DA:241,16
+DA:242,6
+DA:243,6
+DA:254,6
+DA:257,21
+DA:261,2
+DA:262,17
+DA:263,17
+DA:264,17
+DA:265,17
+DA:266,15
+DA:268,2
+LF:126
+LH:102
+BRDA:20,0,0,17
+BRDA:44,1,0,2
+BRDA:44,1,1,21
+BRDA:48,2,0,0
+BRDA:48,2,1,21
+BRDA:67,3,0,1
+BRDA:67,3,1,20
+BRDA:76,4,0,2
+BRDA:76,4,1,19
+BRDA:83,5,0,0
+BRDA:83,5,1,21
+BRDA:92,6,0,19
+BRDA:92,6,1,2
+BRDA:100,7,0,0
+BRDA:100,7,1,21
+BRDA:109,8,0,1
+BRDA:109,8,1,20
+BRDA:109,9,0,21
+BRDA:109,9,1,19
+BRDA:109,9,2,1
+BRDA:110,10,0,0
+BRDA:110,10,1,1
+BRDA:111,11,0,1
+BRDA:111,11,1,0
+BRDA:111,11,2,0
+BRDA:119,12,0,0
+BRDA:119,12,1,0
+BRDA:119,13,0,0
+BRDA:119,13,1,0
+BRDA:132,14,0,0
+BRDA:132,14,1,20
+BRDA:137,15,0,18
+BRDA:137,15,1,2
+BRDA:142,16,0,0
+BRDA:142,16,1,20
+BRDA:149,17,0,0
+BRDA:149,17,1,20
+BRDA:176,18,0,0
+BRDA:176,18,1,0
+BRDA:176,19,0,0
+BRDA:176,19,1,0
+BRDA:189,20,0,0
+BRDA:189,20,1,0
+BRDA:202,21,0,5
+BRDA:202,21,1,1
+BRDA:203,22,0,0
+BRDA:203,22,1,5
+BRDA:217,23,0,0
+BRDA:217,23,1,21
+BRDA:217,24,0,21
+BRDA:217,24,1,21
+BRDA:220,25,0,21
+BRDA:220,25,1,20
+BRDA:221,26,0,21
+BRDA:221,26,1,10
+BRDA:223,27,0,2
+BRDA:223,27,1,0
+BRDA:227,28,0,16
+BRDA:227,28,1,5
+BRDA:232,29,0,0
+BRDA:232,29,1,6
+BRF:61
+BRH:38
 end_of_record
 TN:
 SF:src/threads/utils.js

package/dist/api/controllers/query/index.js

@@ -20,6 +20,8 @@ const environment_1 = require("../../../environment");
 const context_1 = require("@budibase/backend-core/context");
 const pro_1 = require("@budibase/pro");
 const backend_core_1 = require("@budibase/backend-core");
+const utils_3 = require("@budibase/backend-core/utils");
+const constants_2 = require("@budibase/backend-core/constants");
 const Runner = new threads_1.Thread(threads_1.ThreadType.QUERY, {
     timeoutMs: environment_1.QUERY_THREAD_TIMEOUT || 10000,
 });
@@ -111,6 +113,19 @@ function find(ctx) {
     });
 }
 exports.find = find;
+//Required to discern between OIDC OAuth config entries
+function getOAuthConfigCookieId(ctx) {
+    if (ctx.user.providerType === constants_2.Configs.OIDC) {
+        return (0, utils_3.getCookie)(ctx, constants_2.Cookies.OIDC_CONFIG);
+    }
+}
+function getAuthConfig(ctx) {
+    const authCookie = (0, utils_3.getCookie)(ctx, constants_2.Cookies.Auth);
+    let authConfigCtx = {};
+    authConfigCtx["configId"] = getOAuthConfigCookieId(ctx);
+    authConfigCtx["sessionId"] = authCookie ? authCookie.sessionId : null;
+    return authConfigCtx;
+}
 function preview(ctx) {
     return __awaiter(this, void 0, void 0, function* () {
         const db = (0, context_1.getAppDB)();
@@ -119,6 +134,7 @@ function preview(ctx) {
         // preview may not have a queryId as it hasn't been saved, but if it does
         // this stops dynamic variables from calling the same query
         const { fields, parameters, queryVerb, transformer, queryId } = query;
+        const authConfigCtx = getAuthConfig(ctx);
         try {
             const runFn = () => Runner.run({
                 appId: ctx.appId,
@@ -130,6 +146,7 @@ function preview(ctx) {
                 queryId,
                 ctx: {
                     user: ctx.user,
+                    auth: Object.assign({}, authConfigCtx),
                 },
             });
             const { rows, keys, info, extra } = yield pro_1.quotas.addQuery(runFn);
@@ -152,6 +169,7 @@ function execute(ctx, opts = { rowsOnly: false }) {
         const db = (0, context_1.getAppDB)();
         const query = yield db.get(ctx.params.queryId);
         const datasource = yield db.get(query.datasourceId);
+        const authConfigCtx = getAuthConfig(ctx);
         const enrichedParameters = ctx.request.body.parameters || {};
         // make sure parameters are fully enriched with defaults
         if (query && query.parameters) {
@@ -174,6 +192,7 @@ function execute(ctx, opts = { rowsOnly: false }) {
                 queryId: ctx.params.queryId,
                 ctx: {
                     user: ctx.user,
+                    auth: Object.assign({}, authConfigCtx),
                 },
             });
             const { rows, pagination, extra } = yield pro_1.quotas.addQuery(runFn);

package/dist/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@budibase/server",
     "email": "hi@budibase.com",
-    "version": "1.0.212-alpha.13",
+    "version": "1.0.212-alpha.14",
     "description": "Budibase Web Server",
     "main": "src/index.ts",
     "repository": {
@@ -77,11 +77,11 @@
     "license": "GPL-3.0",
     "dependencies": {
         "@apidevtools/swagger-parser": "10.0.3",
-        "@budibase/backend-core": "^1.0.212-alpha.13",
-        "@budibase/client": "^1.0.212-alpha.13",
-        "@budibase/pro": "1.0.212-alpha.13",
-        "@budibase/string-templates": "^1.0.212-alpha.13",
-        "@budibase/types": "^1.0.212-alpha.13",
+        "@budibase/backend-core": "^1.0.212-alpha.14",
+        "@budibase/client": "^1.0.212-alpha.14",
+        "@budibase/pro": "1.0.212-alpha.14",
+        "@budibase/string-templates": "^1.0.212-alpha.14",
+        "@budibase/types": "^1.0.212-alpha.14",
         "@bull-board/api": "3.7.0",
         "@bull-board/koa": "3.9.4",
         "@elastic/elasticsearch": "7.10.0",

package/dist/threads/query.js

@@ -14,6 +14,8 @@ const ScriptRunner = require("../utilities/scriptRunner");
 const { integrations } = require("../integrations");
 const { processStringSync } = require("@budibase/string-templates");
 const { doInAppContext, getAppDB } = require("@budibase/backend-core/context");
+const { refreshOAuthToken, updateUserOAuth, } = require("@budibase/backend-core/auth");
+const { getGlobalIDFromUserMetadataID } = require("../db/utils");
 const { isSQL } = require("../integrations/utils");
 const { enrichQueryFields, interpolateSQL, } = require("../integrations/queries/sql");
 class QueryRunner {
@@ -34,8 +36,10 @@ class QueryRunner {
         // it can be
         this.queryResponse = {};
         this.hasRerun = false;
+        this.hasRefreshedOAuth = false;
     }
     execute() {
+        var _a;
         return __awaiter(this, void 0, void 0, function* () {
             let { datasource, fields, queryVerb, transformer } = this;
             const Integration = integrations[datasource.source];
@@ -87,12 +91,20 @@ class QueryRunner {
                 rows = runner.execute();
             }
             // if the request fails we retry once, invalidating the cached value
-            if (info &&
-                info.code >= 400 &&
-                this.cachedVariables.length > 0 &&
-                !this.hasRerun) {
+            if (info && info.code >= 400 && !this.hasRerun) {
+                if (((_a = this.ctx.user) === null || _a === void 0 ? void 0 : _a.provider) &&
+                    info.code === 401 &&
+                    !this.hasRefreshedOAuth) {
+                    // Attempt to refresh the access token from the provider
+                    this.hasRefreshedOAuth = true;
+                    const authResponse = yield this.refreshOAuth2(this.ctx);
+                    if (!authResponse || authResponse.err) {
+                        // In this event the user may have oAuth issues that
+                        // could require re-authenticating with their provider.
+                        throw new Error("OAuth2 access token could not be refreshed");
+                    }
+                }
                 this.hasRerun = true;
-                // invalidate the cache value
                 yield threadUtils.invalidateDynamicVariables(this.cachedVariables);
                 return this.execute();
             }
@@ -130,6 +142,24 @@ class QueryRunner {
             }, { noRecursiveQuery: true }).execute();
         });
     }
+    refreshOAuth2(ctx) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { oauth2, providerType, _id } = ctx.user;
+            const { configId } = ctx.auth;
+            if (!providerType || !(oauth2 === null || oauth2 === void 0 ? void 0 : oauth2.refreshToken)) {
+                console.error("No refresh token found for authenticated user");
+                return;
+            }
+            const resp = yield refreshOAuthToken(oauth2.refreshToken, providerType, configId);
+            // Refresh session flow. Should be in same location as refreshOAuthToken
+            // There are several other properties available in 'resp'
+            if (!resp.error) {
+                const globalUserId = getGlobalIDFromUserMetadataID(_id);
+                yield updateUserOAuth(globalUserId, resp);
+            }
+            return resp;
+        });
+    }
     getDynamicVariable(variable) {
         return __awaiter(this, void 0, void 0, function* () {
             let { parameters } = this;