@mongoosejs/studio 0.2.11 → 0.2.13

package/backend/actions/Model/getEstimatedDocumentCounts.js

@@ -0,0 +1,38 @@
+'use strict';
+
+const Archetype = require('archetype');
+const authorize = require('../../authorize');
+
+const GetEstimatedDocumentCountsParams = new Archetype({
+  roles: {
+    $type: ['string']
+  }
+}).compile('GetEstimatedDocumentCountsParams');
+
+module.exports = ({ db }) => async function getEstimatedDocumentCounts(params) {
+  const { roles } = new GetEstimatedDocumentCountsParams(params);
+  await authorize('Model.getEstimatedDocumentCounts', roles);
+
+  const modelNames = Object.keys(db.models)
+    .filter(key => !key.startsWith('__Studio_'))
+    .sort();
+
+  const results = await Promise.allSettled(
+    modelNames.map(name => {
+      const Model = db.models[name];
+      return Model.estimatedDocumentCount().exec();
+    })
+  );
+
+  const counts = {};
+  results.forEach((result, index) => {
+    const name = modelNames[index];
+    if (result.status === 'fulfilled' && typeof result.value === 'number') {
+      counts[name] = result.value;
+    } else {
+      counts[name] = null;
+    }
+  });
+
+  return { counts };
+};

package/backend/actions/Model/index.js

@@ -13,6 +13,7 @@ exports.getDocuments = require('./getDocuments');
 exports.getDocumentsStream = require('./getDocumentsStream');
 exports.getCollectionInfo = require('./getCollectionInfo');
 exports.getIndexes = require('./getIndexes');
+exports.getEstimatedDocumentCounts = require('./getEstimatedDocumentCounts');
 exports.listModels = require('./listModels');
 exports.streamDocumentChanges = require('./streamDocumentChanges');
 exports.streamChatMessage = require('./streamChatMessage');

package/backend/actions/Model/streamDocumentChanges.js

@@ -27,7 +27,8 @@ module.exports = ({ db, changeStream }) => async function* streamDocumentChanges
     throw new Error(`Model ${model} not found`);
   }
 
-  if (!changeStream) {
+  const stream = changeStream();
+  if (!stream) {
     throw new Error('Change streams are not enabled');
   }
 
@@ -79,9 +80,9 @@ module.exports = ({ db, changeStream }) => async function* streamDocumentChanges
     enqueue({ type: 'end' });
   }
 
-  changeStream.on('change', handleChange);
-  changeStream.on('error', handleError);
-  changeStream.on('end', handleEnd);
+  stream.on('change', handleChange);
+  stream.on('error', handleError);
+  stream.on('end', handleEnd);
 
   try {
     while (true) {
@@ -112,9 +113,9 @@ module.exports = ({ db, changeStream }) => async function* streamDocumentChanges
       }
     }
   } finally {
-    changeStream.off('change', handleChange);
-    changeStream.off('error', handleError);
-    changeStream.off('end', handleEnd);
+    stream.off('change', handleChange);
+    stream.off('error', handleError);
+    stream.off('end', handleEnd);
     if (resolveQueue) {
       resolveQueue();
       resolveQueue = null;

package/backend/actions/Task/getTasks.js

@@ -4,7 +4,8 @@ const Archetype = require('archetype');
 
 const GetTasksParams = new Archetype({
   start: {
-    $type: Date
+    $type: Date,
+    $required: true
   },
   end: {
     $type: Date
@@ -31,22 +32,24 @@ module.exports = ({ db }) => async function getTasks(params) {
   }
   if (status) {
     filter.status = status;
+  } else {
+    filter.status = { $in: ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'] };
   }
   if (name) {
     filter.name = { $regex: name, $options: 'i' };
   }
 
   const tasks = await Task.find(filter);
-  
+
   // Define all possible statuses
   const allStatuses = ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'];
-  
+
   // Initialize groupedTasks with all statuses
   const groupedTasks = allStatuses.reduce((groups, status) => {
     groups[status] = [];
     return groups;
   }, {});
-  
+
   // Group tasks by status
   tasks.forEach(task => {
     const taskStatus = task.status || 'unknown';
@@ -54,9 +57,9 @@ module.exports = ({ db }) => async function getTasks(params) {
       groupedTasks[taskStatus].push(task);
     }
   });
- 
+
   return {
     tasks,
     groupedTasks
   };
-};
+};

package/backend/authorize.js

@@ -22,6 +22,7 @@ const actionsToRequiredRoles = {
   'Model.getDocument': ['owner', 'admin', 'member', 'readonly'],
   'Model.getDocuments': ['owner', 'admin', 'member', 'readonly'],
   'Model.getDocumentsStream': ['owner', 'admin', 'member', 'readonly'],
+  'Model.getEstimatedDocumentCounts': ['owner', 'admin', 'member', 'readonly'],
   'Model.getIndexes': ['owner', 'admin', 'member', 'readonly'],
   'Model.listModels': ['owner', 'admin', 'member', 'readonly'],
   'Model.streamDocumentChanges': ['owner', 'admin', 'member', 'readonly'],

package/backend/index.js

@@ -18,10 +18,18 @@ module.exports = function backend(db, studioConnection, options) {
 
   let changeStream = null;
   if (options?.changeStream) {
-    changeStream = db.watch();
+    const conn = db instanceof mongoose.Mongoose ? db.connection : db;
+    if (conn.readyState !== mongoose.Connection.STATES.connected) {
+      conn._waitForConnect().then(() => {
+        changeStream = conn.watch();
+      });
+    } else {
+      changeStream = conn.watch();
+    }
+
   }
 
-  const actions = applySpec(Actions, { db, studioConnection, options, changeStream });
-  actions.services = { changeStream };
+  const actions = applySpec(Actions, { db, studioConnection, options, changeStream: () => changeStream });
+  actions.services = { changeStream: () => changeStream };
   return actions;
 };

package/eslint.config.js

@@ -42,7 +42,8 @@ module.exports = defineConfig([
         navigator: true,
         TextDecoder: true,
         AbortController: true,
-        clearTimeout: true
+        clearTimeout: true,
+        requestAnimationFrame: true
       },
       sourceType: 'commonjs'
     },

package/express.js

@@ -33,6 +33,7 @@ module.exports = async function mongooseStudioExpressApp(apiUrl, conn, options)
 
   apiUrl = apiUrl || 'api';
   const backend = Backend(conn, options.studioConnection, options);
+  delete backend.services;
 
   router.use(
     '/api',