@mongoosejs/studio 0.2.12 → 0.3.0

package/backend/actions/ChatMessage/executeScript.js

@@ -43,7 +43,11 @@ module.exports = ({ db, studioConnection }) => async function executeScript(para
   if (!db.Types) {
     db.Types = mongoose.Types;
   }
-  const sandbox = { db, mongoose, console: {}, ObjectId: mongoose.Types.ObjectId };
+  const MongooseStudioChartColors = [
+    '#4e79a7', '#e15759', '#59a14f', '#9c755f',
+    '#f28e2b', '#b07aa1', '#76b7b2', '#edc948'
+  ];
+  const sandbox = { db, mongoose, console: {}, ObjectId: mongoose.Types.ObjectId, MongooseStudioChartColors };
 
   // Capture console logs
   sandbox.console.log = function() {

package/backend/actions/ChatThread/createChatMessage.js

@@ -108,6 +108,7 @@ The following globals are available. Assume no other globals exist.
 - mongoose: the output of require('mongoose').
 - ObjectId: MongoDB ObjectId class from mongoose.Types.ObjectId
 - console: has a stubbed log() function that logs to the console and is accessible in the output.
+- MongooseStudioChartColors: an array of 8 hex color strings for use as chart dataset colors.
 
 Keep scripts concise. Avoid unnecessary comments, error handling, and temporary variables.
 
@@ -129,7 +130,7 @@ Format output as Markdown, including code fences for any scripts the user reques
 
 Add a brief text description of what the script does.
 
-If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it. Set responsive: true, maintainAspectRatio: false options unless the user explicitly asks.
+If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it. Set responsive: true, maintainAspectRatio: false options unless the user explicitly asks. Use MongooseStudioChartColors for dataset backgroundColor and borderColor by default. For line/bar charts, use MongooseStudioChartColors[i] as borderColor and MongooseStudioChartColors[i] + '33' as backgroundColor for each dataset. For pie/doughnut charts, use MongooseStudioChartColors.slice(0, data.length) as backgroundColor. Only use custom colors if the user explicitly requests specific colors.
 
 If the user\'s query is best answered by a map, return an object { $featureCollection } which contains a GeoJSON FeatureCollection
 

package/backend/actions/ChatThread/streamChatMessage.js

@@ -123,7 +123,7 @@ const systemPrompt = `
 
   Do not write any imports or require() statements, that will cause the script to break.
 
-  If the user approves the script, the script will run in the Node.js server in a sandboxed vm.createContext() call with only 1 global variable: db, which contains the Mongoose connection. The script return value will then send the response via JSON to the client. Be aware that the result of the query will be serialized to JSON before being displayed to the user. MAKE SURE TO RETURN A VALUE FROM THE SCRIPT.
+  If the user approves the script, the script will run in the Node.js server in a sandboxed vm.createContext() call with the following globals: db (the Mongoose connection), mongoose, ObjectId (mongoose.Types.ObjectId), console, and MongooseStudioChartColors (an array of 8 hex color strings for chart dataset colors). The script return value will then send the response via JSON to the client. Be aware that the result of the query will be serialized to JSON before being displayed to the user. MAKE SURE TO RETURN A VALUE FROM THE SCRIPT.
 
   Optimize scripts for readability first, followed by reliability, followed by performance. Avoid using the aggregation framework unless explicitly requested by the user. Use indexed fields in queries where possible.
 
@@ -139,7 +139,7 @@ const systemPrompt = `
 
   Add a brief text description of what the script does.
 
-  If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it. Set responsive: true, maintainAspectRatio: false options unless the user explicitly asks.
+  If the user's query is best answered with a chart, return a Chart.js 4 configuration as \`return { $chart: chartJSConfig };\`. Disable ChartJS animation by default unless user asks for it. Set responsive: true, maintainAspectRatio: false options unless the user explicitly asks. Use MongooseStudioChartColors for dataset backgroundColor and borderColor by default. For line/bar charts, use MongooseStudioChartColors[i] as borderColor and MongooseStudioChartColors[i] + '33' as backgroundColor for each dataset. For pie/doughnut charts, use MongooseStudioChartColors.slice(0, data.length) as backgroundColor. Only use custom colors if the user explicitly requests specific colors.
 
   If the user\'s query is best answered by a map, return an object { $featureCollection } which contains a GeoJSON FeatureCollection
 

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 instanceof mongoose.Mongoose ? db.connection.watch() : 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,11 @@ module.exports = defineConfig([
         navigator: true,
         TextDecoder: true,
         AbortController: true,
-        clearTimeout: true
+        clearTimeout: true,
+        requestAnimationFrame: true,
+        localStorage: true,
+        getComputedStyle: true,
+        CustomEvent: 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',