@budibase/server 2.6.27 → 2.6.28-alpha.0

package/dist/index.js

@@ -5397,11 +5397,11 @@ function makeCacheItem(doc, lastWrite = null) {
   return { doc, lastWrite: lastWrite || Date.now() };
 }
 async function put(db2, doc, writeRateMs = DEFAULT_WRITE_RATE_MS) {
-  const cache2 = await getCache();
+  const cache3 = await getCache();
   const key = doc._id;
   let cacheItem;
   if (key) {
-    cacheItem = await cache2.get(makeCacheKey(db2, key));
+    cacheItem = await cache3.get(makeCacheKey(db2, key));
   }
   const updateDb = !cacheItem || cacheItem.lastWrite < Date.now() - writeRateMs;
   let output = doc;
@@ -5439,30 +5439,30 @@ async function put(db2, doc, writeRateMs = DEFAULT_WRITE_RATE_MS) {
   }
   cacheItem = makeCacheItem(output, updateDb ? null : cacheItem == null ? void 0 : cacheItem.lastWrite);
   if (output._id) {
-    await cache2.store(makeCacheKey(db2, output._id), cacheItem);
+    await cache3.store(makeCacheKey(db2, output._id), cacheItem);
   }
   return { ok: true, id: output._id, rev: output._rev };
 }
 async function get2(db2, id) {
-  const cache2 = await getCache();
+  const cache3 = await getCache();
   const cacheKey = makeCacheKey(db2, id);
-  let cacheItem = await cache2.get(cacheKey);
+  let cacheItem = await cache3.get(cacheKey);
   if (!cacheItem) {
     const doc = await db2.get(id);
     cacheItem = makeCacheItem(doc);
-    await cache2.store(cacheKey, cacheItem);
+    await cache3.store(cacheKey, cacheItem);
   }
   return cacheItem.doc;
 }
 async function remove(db2, docOrId, rev) {
-  const cache2 = await getCache();
+  const cache3 = await getCache();
   if (!docOrId) {
     throw new Error("No ID/Rev provided.");
   }
   const id = typeof docOrId === "string" ? docOrId : docOrId._id;
   rev = typeof docOrId === "string" ? rev : docOrId._rev;
   try {
-    await cache2.delete(makeCacheKey(db2, id));
+    await cache3.delete(makeCacheKey(db2, id));
   } finally {
     await db2.remove(id, rev);
   }
@@ -10060,8 +10060,8 @@ async function preAuth(passport2, ctx, next) {
     callbackUrl,
     ssoSaveUserNoOp
   );
-  if (!ctx.query.appId || !ctx.query.datasourceId) {
-    ctx.throw(400, "appId and datasourceId query params not present.");
+  if (!ctx.query.appId) {
+    ctx.throw(400, "appId query param not present.");
   }
   return passport2.authenticate(strategy, {
     scope: ["profile", "email", "https://www.googleapis.com/auth/spreadsheets"],
@@ -10081,7 +10081,7 @@ async function postAuth(passport2, ctx, next) {
         clientSecret: config.clientSecret,
         callbackURL: callbackUrl
       },
-      (accessToken, refreshToken, profile, done) => {
+      (accessToken, refreshToken, _profile, done) => {
         clearCookie(ctx, "budibase:datasourceauth" /* DatasourceAuth */);
         done(null, { accessToken, refreshToken });
       }
@@ -10089,22 +10089,14 @@ async function postAuth(passport2, ctx, next) {
     { successRedirect: "/", failureRedirect: "/error" },
     async (err, tokens) => {
       const baseUrl = `/builder/app/${authStateCookie.appId}/data`;
-      await doWithDB(authStateCookie.appId, async (db2) => {
-        let datasource2;
-        try {
-          datasource2 = await db2.get(authStateCookie.datasourceId);
-        } catch (err2) {
-          if (err2.status === 404) {
-            ctx.redirect(baseUrl);
-          }
-        }
-        if (!datasource2.config) {
-          datasource2.config = {};
+      const id = utils_exports2.newid();
+      await cache_exports.store(
+        `datasource:creation:${authStateCookie.appId}:google:${id}`,
+        {
+          tokens
         }
-        datasource2.config.auth = { type: "google", ...tokens };
-        await db2.put(datasource2);
-        ctx.redirect(`${baseUrl}/datasource/${authStateCookie.datasourceId}`);
-      });
+      );
+      ctx.redirect(`${baseUrl}/new?continue_google_setup=${id}`);
     }
   )(ctx, next);
 }
@@ -10114,9 +10106,9 @@ var init_google2 = __esm({
     init_google();
     init_constants2();
     init_utils5();
-    init_db5();
     init_configs3();
     init_sso2();
+    init_src2();
     GoogleStrategy2 = require("passport-google-oauth").OAuth2Strategy;
   }
 });
@@ -24847,22 +24839,22 @@ function threadSetup() {
   init10();
 }
 async function checkCacheForDynamicVariable(queryId, variable) {
-  const cache2 = await getClient3();
-  return cache2.get(makeVariableKey(queryId, variable));
+  const cache3 = await getClient3();
+  return cache3.get(makeVariableKey(queryId, variable));
 }
 async function invalidateDynamicVariables(cachedVars) {
-  const cache2 = await getClient3();
+  const cache3 = await getClient3();
   let promises = [];
   for (let variable of cachedVars) {
     promises.push(
-      cache2.delete(makeVariableKey(variable.queryId, variable.name))
+      cache3.delete(makeVariableKey(variable.queryId, variable.name))
     );
   }
   await Promise.all(promises);
 }
 async function storeDynamicVariable(queryId, variable, value) {
-  const cache2 = await getClient3();
-  await cache2.store(
+  const cache3 = await getClient3();
+  await cache3.store(
     makeVariableKey(queryId, variable),
     value,
     VARIABLE_TTL_SECONDS
@@ -29038,9 +29030,20 @@ var init_rest = __esm({
 });
 
 // src/integrations/googlesheets.ts
+async function setupCreationAuth(datasouce) {
+  if (datasouce.continueSetupId) {
+    const appId = context_exports.getAppId();
+    const tokens = await cache_exports.get(
+      `datasource:creation:${appId}:google:${datasouce.continueSetupId}`
+    );
+    datasouce.auth = tokens.tokens;
+    delete datasouce.continueSetupId;
+  }
+}
 var import_google_auth_library, import_google_spreadsheet, import_node_fetch10, ALLOWED_TYPES, SCHEMA12, GoogleSheetsIntegration, googlesheets_default;
 var init_googlesheets = __esm({
   "src/integrations/googlesheets.ts"() {
+    "use strict";
     init_src();
     import_google_auth_library = require("google-auth-library");
     init_utils19();
@@ -29075,7 +29078,7 @@ var init_googlesheets = __esm({
       },
       datasource: {
         spreadsheetId: {
-          display: "Google Sheet URL",
+          display: "Spreadsheet URL",
           type: "string" /* STRING */,
           required: true
         }
@@ -29145,6 +29148,7 @@ var init_googlesheets = __esm({
       }
       async testConnection() {
         try {
+          await setupCreationAuth(this.config);
           await this.connect();
           return { connected: true };
         } catch (e) {
@@ -34793,6 +34797,9 @@ async function save13(ctx) {
     datasource2.entities = tables;
     setDefaultDisplayColumns(datasource2);
   }
+  if (preSaveAction[datasource2.source]) {
+    await preSaveAction[datasource2.source](datasource2);
+  }
   const dbResp = await db2.put(datasource2);
   await events_exports.datasource.created(datasource2);
   datasource2._rev = dbResp.rev;
@@ -34871,6 +34878,7 @@ async function query(ctx) {
     ctx.throw(400, err);
   }
 }
+var preSaveAction;
 var init_datasource5 = __esm({
   "src/api/controllers/datasource.ts"() {
     init_utils9();
@@ -34880,8 +34888,15 @@ var init_datasource5 = __esm({
     init_utils24();
     init_utils18();
     init_src2();
+    init_src();
     init_sdk3();
     init_websockets();
+    init_googlesheets();
+    preSaveAction = {
+      ["GOOGLE_SHEETS" /* GOOGLE_SHEETS */]: async (datasource2) => {
+        await setupCreationAuth(datasource2.config);
+      }
+    };
   }
 });