@mongoosejs/studio 0.3.6 → 0.3.8

package/backend/actions/Dashboard/getDashboard.js

@@ -3,6 +3,7 @@
 const Archetype = require('archetype');
 const vm = require('vm');
 const authorize = require('../../authorize');
+const { defaultMothershipURL } = require('../../../constants');
 
 const GetDashboardParams = new Archetype({
   dashboardId: {
@@ -26,7 +27,7 @@ const GetDashboardParams = new Archetype({
 module.exports = ({ db, options }) => async function getDashboard(params) {
   const { $workspaceId, authorization, dashboardId, evaluate, roles } = new GetDashboardParams(params);
   const Dashboard = db.model('__Studio_Dashboard');
-  const mothershipUrl = options?._mothershipUrl ?? 'https://mongoose-js.netlify.app/.netlify/functions';
+  const mothershipUrl = options?._mothershipUrl ?? defaultMothershipURL;
 
   await authorize('Dashboard.getDashboard', roles);
 

package/backend/actions/Task/getTasksOverTime.js

@@ -0,0 +1,66 @@
+'use strict';
+
+const Archetype = require('archetype');
+
+const GetTasksOverTimeParams = new Archetype({
+  start: { $type: Date },
+  end: { $type: Date },
+  bucketSizeMs: { $type: 'number' }
+}).compile('GetTasksOverTimeParams');
+
+const TRACKED_STATUSES = ['succeeded', 'failed', 'cancelled'];
+
+module.exports = ({ db }) => async function getTasksOverTime(params) {
+  params = new GetTasksOverTimeParams(params);
+  const { Task } = db.models;
+  const { start, end, bucketSizeMs } = params;
+
+  const bucketMs = (bucketSizeMs != null && bucketSizeMs > 0) ? bucketSizeMs : 5 * 60 * 1000;
+
+  const match = { status: { $in: TRACKED_STATUSES } };
+  if (start != null && end != null) {
+    match.scheduledAt = { $gte: start, $lt: end };
+  } else if (start != null) {
+    match.scheduledAt = { $gte: start };
+  }
+
+  const pipeline = [
+    { $match: match },
+    {
+      $project: {
+        status: 1,
+        bucket: {
+          $toDate: {
+            $multiply: [
+              { $floor: { $divide: [{ $toLong: '$scheduledAt' }, bucketMs] } },
+              bucketMs
+            ]
+          }
+        }
+      }
+    },
+    {
+      $group: {
+        _id: { bucket: '$bucket', status: '$status' },
+        count: { $sum: 1 }
+      }
+    },
+    {
+      $group: {
+        _id: '$_id.bucket',
+        counts: { $push: { status: '$_id.status', count: '$count' } }
+      }
+    },
+    { $sort: { _id: 1 } }
+  ];
+
+  const results = await Task.aggregate(pipeline);
+
+  return results.map(r => {
+    const bucket = { timestamp: r._id, succeeded: 0, failed: 0, cancelled: 0 };
+    for (const { status, count } of r.counts) {
+      if (status in bucket) bucket[status] = count;
+    }
+    return bucket;
+  });
+};

package/backend/actions/Task/index.js

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

package/backend/integrations/callLLM.js

@@ -4,6 +4,7 @@ const { createAnthropic } = require('@ai-sdk/anthropic');
 const { createGoogleGenerativeAI } = require('@ai-sdk/google');
 const { createOpenAI } = require('@ai-sdk/openai');
 const { generateText } = require('ai');
+const { defaultMothershipURL } = require('../../constants');
 
 module.exports = async function callLLM(messages, system, options) {
   let provider = null;
@@ -50,7 +51,7 @@ module.exports = async function callLLM(messages, system, options) {
   }
 
   const headers = { 'Content-Type': 'application/json' };
-  const response = await fetch('https://mongoose-js.netlify.app/.netlify/functions/getChatCompletion', {
+  const response = await fetch(`${defaultMothershipURL}/getChatCompletion`, {
     method: 'POST',
     headers,
     body: JSON.stringify({

package/backend/integrations/streamLLM.js

@@ -4,6 +4,7 @@ const { createAnthropic } = require('@ai-sdk/anthropic');
 const { createGoogleGenerativeAI } = require('@ai-sdk/google');
 const { createOpenAI } = require('@ai-sdk/openai');
 const { streamText } = require('ai');
+const { defaultMothershipURL } = require('../../constants');
 
 module.exports = async function* streamLLM(messages, system, options) {
   let provider = null;
@@ -62,7 +63,7 @@ module.exports = async function* streamLLM(messages, system, options) {
 
   // If not using OpenAI, Anthropic, or Google Gemini, fallback to Mongoose (no streaming)
   const headers = { 'Content-Type': 'application/json' };
-  const response = await fetch('https://mongoose-js.netlify.app/.netlify/functions/getChatCompletion', {
+  const response = await fetch(`${defaultMothershipURL}/getChatCompletion`, {
     method: 'POST',
     headers,
     body: JSON.stringify({

package/backend/netlify.js

@@ -2,10 +2,11 @@
 
 const Backend = require('./');
 const { toNetlifyFunction } = require('extrovert');
+const { defaultMothershipURL } = require('../constants');
 
 module.exports = function netlify(conn, options) {
   const backend = Backend(conn, options?.studioConnection, options);
-  const mothershipUrl = options?._mothershipUrl || 'https://mongoose-js.netlify.app/.netlify/functions';
+  const mothershipUrl = options?._mothershipUrl || defaultMothershipURL;
 
   let workspace = null;
 

package/backend/next.js

@@ -1,11 +1,12 @@
 'use strict';
 
 const Backend = require('./');
+const { defaultMothershipURL } = require('../constants');
 
 module.exports = function next(conn, options) {
   const backend = Backend(conn, options?.studioConnection, options);
 
-  const mothershipUrl = options?._mothershipUrl || 'https://mongoose-js.netlify.app/.netlify/functions';
+  const mothershipUrl = options?._mothershipUrl || defaultMothershipURL;
   let workspace = null;
 
   return async function wrappedNextJSFunction(req, res) {

package/constants.js

@@ -0,0 +1,7 @@
+'use strict';
+
+const defaultMothershipURL = 'https://mothership.mongoosestudio.app/.netlify/functions';
+
+module.exports = {
+  defaultMothershipURL
+};

package/express.js

@@ -4,12 +4,13 @@ const Backend = require('./backend');
 const express = require('express');
 const frontend = require('./frontend');
 const { toRoute, objectRouter } = require('extrovert');
+const { defaultMothershipURL } = require('./constants');
 
 module.exports = async function mongooseStudioExpressApp(apiUrl, conn, options) {
   const router = express.Router();
   options = options ? { changeStream: true, ...options } : { changeStream: true };
 
-  const mothershipUrl = options._mothershipUrl || 'https://mongoose-js.netlify.app/.netlify/functions';
+  const mothershipUrl = options._mothershipUrl || defaultMothershipURL;
   let workspace = null;
   if (options?.apiKey) {
     ({ workspace } = await fetch(`${mothershipUrl}/getWorkspace`, {

package/frontend/index.js

@@ -4,10 +4,11 @@ const { execSync, exec } = require('child_process');
 const fs = require('fs');
 const path = require('path');
 const webpack = require('webpack');
+const { defaultMothershipURL } = require('../constants');
 const webpackConfig = require('./webpack.config');
 
 module.exports = async function frontend(apiUrl, isLambda, options, workspace) {
-  const mothershipUrl = options?._mothershipUrl || 'https://mongoose-js.netlify.app/.netlify/functions';
+  const mothershipUrl = options?._mothershipUrl || defaultMothershipURL;
 
   if (workspace == null && options?.apiKey) {
     ({ workspace } = await fetch(`${mothershipUrl}/getWorkspace`, {