@mongoosejs/studio 0.2.13 → 0.3.1

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,10 +130,12 @@ 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
 
+If the user's query is best answered by a table, return an object { $table: { columns: string[], rows: any[][] } }
+
 Example output:
 
 The following script counts the number of users which are not deleted.

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,10 +139,12 @@ 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
 
+  If the user's query is best answered by a table, return an object { $table: { columns: string[], rows: any[][] } }
+
   Example output:
 
   The following script counts the number of users which are not deleted.

package/backend/actions/Dashboard/updateDashboard.js

@@ -32,11 +32,11 @@ module.exports = ({ db }) => async function updateDashboard(params) {
 
   const updateObj = { code };
 
-  if (title) {
+  if (title != null) {
     updateObj.title = title;
   }
 
-  if (description) {
+  if (description != null) {
     updateObj.description = description;
   }
 

package/backend/actions/Task/getTaskOverview.js

@@ -0,0 +1,102 @@
+'use strict';
+
+const Archetype = require('archetype');
+const escape = require('regexp.escape');
+
+const GetTaskOverviewParams = new Archetype({
+  start: { $type: Date },
+  end: { $type: Date },
+  status: { $type: 'string' },
+  name: { $type: 'string' }
+}).compile('GetTaskOverviewParams');
+
+/** Statuses shown on the Task overview page. */
+const OVERVIEW_STATUSES = ['pending', 'succeeded', 'failed', 'cancelled'];
+
+function buildMatch(params) {
+  const { start, end, status, name } = params;
+  const match = {};
+  if (start != null && end != null) {
+    match.scheduledAt = { $gte: start, $lt: end };
+  } else if (start != null) {
+    match.scheduledAt = { $gte: start };
+  }
+  const statusVal = typeof status === 'string' ? status.trim() : status;
+  if (statusVal != null && statusVal !== '') {
+    match.status = statusVal;
+  } else {
+    match.status = { $in: ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'] };
+  }
+  if (name != null && name !== '') {
+    const nameStr = typeof name === 'string' ? name.trim() : String(name);
+    match.name = { $regex: escape(nameStr), $options: 'i' };
+  }
+  return match;
+}
+
+module.exports = ({ db }) => async function getTaskOverview(params) {
+  params = new GetTaskOverviewParams(params);
+  if (typeof params.status === 'string') params.status = params.status.trim();
+  if (typeof params.name === 'string') params.name = params.name.trim();
+  const { Task } = db.models;
+  const match = buildMatch(params);
+
+  const defaultCounts = OVERVIEW_STATUSES.map(s => ({ k: s, v: 0 }));
+
+  const pipeline = [
+    { $match: match },
+    {
+      $facet: {
+        statusCounts: [
+          { $group: { _id: { $ifNull: ['$status', 'unknown'] }, count: { $sum: 1 } } },
+          { $group: { _id: null, counts: { $push: { k: '$_id', v: '$count' } } } },
+          {
+            $project: {
+              statusCounts: {
+                $arrayToObject: {
+                  $concatArrays: [{ $literal: defaultCounts }, '$counts']
+                }
+              }
+            }
+          },
+          { $replaceRoot: { newRoot: '$statusCounts' } }
+        ],
+        tasksByName: [
+          {
+            $group: {
+              _id: '$name',
+              totalCount: { $sum: 1 },
+              lastRun: { $max: '$scheduledAt' },
+              pending: { $sum: { $cond: [{ $eq: ['$status', 'pending'] }, 1, 0] } },
+              succeeded: { $sum: { $cond: [{ $eq: ['$status', 'succeeded'] }, 1, 0] } },
+              failed: { $sum: { $cond: [{ $eq: ['$status', 'failed'] }, 1, 0] } },
+              cancelled: { $sum: { $cond: [{ $eq: ['$status', 'cancelled'] }, 1, 0] } }
+            }
+          },
+          {
+            $project: {
+              _id: 0,
+              name: '$_id',
+              totalCount: 1,
+              lastRun: 1,
+              statusCounts: {
+                pending: '$pending',
+                succeeded: '$succeeded',
+                failed: '$failed',
+                cancelled: '$cancelled'
+              }
+            }
+          },
+          { $sort: { name: 1 } }
+        ]
+      }
+    }
+  ];
+
+  const [result] = await Task.aggregate(pipeline);
+
+  return {
+    statusCounts: result.statusCounts?.[0] ?? {},
+    tasksByName: result.tasksByName || []
+  };
+};

package/backend/actions/Task/getTasks.js

@@ -1,65 +1,105 @@
 'use strict';
 
 const Archetype = require('archetype');
+const escape = require('regexp.escape');
 
 const GetTasksParams = new Archetype({
-  start: {
-    $type: Date,
-    $required: true
-  },
-  end: {
-    $type: Date
-  },
-  status: {
-    $type: 'string'
-  },
-  name: {
-    $type: 'string'
-  }
+  start: { $type: Date },
+  end: { $type: Date },
+  status: { $type: 'string' },
+  name: { $type: 'string' },
+  skip: { $type: 'number', $default: 0 },
+  limit: { $type: 'number', $default: 100 }
 }).compile('GetTasksParams');
 
-module.exports = ({ db }) => async function getTasks(params) {
-  params = new GetTasksParams(params);
-  const { start, end, status, name } = params;
-  const { Task } = db.models;
+const ALL_STATUSES = ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'];
+
+/** Status keys for statusCounts (same shape as getTaskOverview). */
+const STATUS_COUNT_KEYS = ['pending', 'succeeded', 'failed', 'cancelled'];
 
-  const filter = {};
+/** Max documents per request to avoid excessive memory and response size. */
+const MAX_LIMIT = 2000;
 
-  if (start && end) {
-    filter.scheduledAt = { $gte: start, $lt: end };
-  } else if (start) {
-    filter.scheduledAt = { $gte: start };
+function buildMatch(params) {
+  const { start, end, status, name } = params;
+  const match = {};
+  if (start != null && end != null) {
+    match.scheduledAt = { $gte: start, $lt: end };
+  } else if (start != null) {
+    match.scheduledAt = { $gte: start };
   }
-  if (status) {
-    filter.status = status;
+  const statusVal = typeof status === 'string' ? status.trim() : status;
+  if (statusVal != null && statusVal !== '') {
+    match.status = statusVal;
   } else {
-    filter.status = { $in: ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'] };
+    match.status = { $in: ALL_STATUSES };
   }
-  if (name) {
-    filter.name = { $regex: name, $options: 'i' };
+  if (name != null && name !== '') {
+    const nameStr = typeof name === 'string' ? name.trim() : String(name);
+    match.name = { $regex: escape(nameStr), $options: 'i' };
   }
+  return match;
+}
 
-  const tasks = await Task.find(filter);
+/** Projection done in aggregation: only fields needed by frontend, payload → parameters, _id → id. */
+const TASK_PROJECT_STAGE = {
+  _id: 1,
+  id: '$_id',
+  name: 1,
+  status: 1,
+  scheduledAt: 1,
+  createdAt: 1,
+  startedAt: 1,
+  completedAt: 1,
+  error: 1,
+  parameters: '$payload'
+};
+
+module.exports = ({ db }) => async function getTasks(params) {
+  params = new GetTasksParams(params);
+  if (typeof params.status === 'string') params.status = params.status.trim();
+  if (typeof params.name === 'string') params.name = params.name.trim();
 
-  // Define all possible statuses
-  const allStatuses = ['pending', 'in_progress', 'succeeded', 'failed', 'cancelled', 'unknown'];
+  const skip = Math.max(0, Number(params.skip) || 0);
+  const limit = Math.min(MAX_LIMIT, Math.max(1, Number(params.limit) || 100));
+  const { Task } = db.models;
+  const match = buildMatch(params);
 
-  // Initialize groupedTasks with all statuses
-  const groupedTasks = allStatuses.reduce((groups, status) => {
-    groups[status] = [];
-    return groups;
-  }, {});
+  const defaultCounts = STATUS_COUNT_KEYS.map(s => ({ k: s, v: 0 }));
 
-  // Group tasks by status
-  tasks.forEach(task => {
-    const taskStatus = task.status || 'unknown';
-    if (groupedTasks.hasOwnProperty(taskStatus)) {
-      groupedTasks[taskStatus].push(task);
+  const pipeline = [
+    { $match: match },
+    {
+      $facet: {
+        tasks: [
+          { $sort: { scheduledAt: -1 } },
+          { $skip: skip },
+          { $limit: limit },
+          { $project: TASK_PROJECT_STAGE }
+        ],
+        count: [{ $count: 'total' }],
+        statusCounts: [
+          { $group: { _id: { $ifNull: ['$status', 'unknown'] }, count: { $sum: 1 } } },
+          { $group: { _id: null, counts: { $push: { k: '$_id', v: '$count' } } } },
+          {
+            $project: {
+              statusCounts: {
+                $arrayToObject: {
+                  $concatArrays: [{ $literal: defaultCounts }, '$counts']
+                }
+              }
+            }
+          },
+          { $replaceRoot: { newRoot: '$statusCounts' } }
+        ]
+      }
     }
-  });
+  ];
+
+  const [result] = await Task.aggregate(pipeline);
+  const tasks = result.tasks || [];
+  const numDocs = (result.count && result.count[0] && result.count[0].total) || 0;
+  const statusCounts = result.statusCounts?.[0] ?? {};
 
-  return {
-    tasks,
-    groupedTasks
-  };
+  return { tasks, numDocs, statusCounts };
 };

package/backend/actions/Task/index.js

@@ -3,5 +3,6 @@
 exports.cancelTask = require('./cancelTask');
 exports.createTask = require('./createTask');
 exports.getTasks = require('./getTasks');
+exports.getTaskOverview = require('./getTaskOverview');
 exports.rescheduleTask = require('./rescheduleTask');
 exports.runTask = require('./runTask');

package/eslint.config.js

@@ -43,7 +43,10 @@ module.exports = defineConfig([
         TextDecoder: true,
         AbortController: true,
         clearTimeout: true,
-        requestAnimationFrame: true
+        requestAnimationFrame: true,
+        localStorage: true,
+        getComputedStyle: true,
+        CustomEvent: true
       },
       sourceType: 'commonjs'
     },