Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 62 additions & 0 deletions backend/actions/Model/aggregate.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
'use strict';

const Archetype = require('archetype');
const { EJSON } = require('mongoose').mongo.BSON;
const authorize = require('../../authorize');

const AggregateParams = new Archetype({
model: {
$type: 'string',
$required: true
},
pipeline: {
$type: Archetype.Any,
$required: true
},
limit: {
$type: 'number',
$default: 20
},
roles: {
$type: ['string']
}
}).compile('AggregateParams');

module.exports = ({ db }) => async function aggregate(params) {
params = new AggregateParams(params);
const { model, roles } = params;
await authorize('Model.aggregate', roles);

const Model = db.models[model];
if (Model == null) {
throw new Error(`Model ${model} not found`);
}

if (!Array.isArray(params.pipeline)) {
throw new Error('`pipeline` must be an array');
}

let pipeline;
try {
pipeline = EJSON.deserialize(params.pipeline);
} catch (err) {
throw new Error(`Invalid pipeline (EJSON): ${err.message}`);
}

if (!Array.isArray(pipeline)) {
throw new Error('`pipeline` must be an array');
}

pipeline = pipeline.map((stage, index) => {
if (stage == null || Array.isArray(stage) || typeof stage !== 'object') {
throw new Error(`Invalid stage at index ${index}`);
}
return stage;
});

const limit = Math.max(1, Math.min(200, Math.floor(params.limit ?? 20)));
pipeline.push({ $limit: limit });

Comment thread
vkarpov15 marked this conversation as resolved.
const docs = await Model.aggregate(pipeline).exec();
return { docs };
};
Comment thread
IslandRhythms marked this conversation as resolved.
1 change: 1 addition & 0 deletions backend/actions/Model/index.js
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
'use strict';

exports.addField = require('./addField');
exports.aggregate = require('./aggregate');
exports.createChatMessage = require('./createChatMessage');
exports.createDocument = require('./createDocument');
exports.deleteDocument = require('./deleteDocument');
Expand Down
1 change: 1 addition & 0 deletions backend/authorize.js
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ const actionsToRequiredRoles = {
'Model.deleteDocuments': ['owner', 'admin', 'member'],
'Model.dropCollection': ['owner', 'admin'],
'Model.dropIndex': ['owner', 'admin'],
'Model.aggregate': ['owner', 'admin', 'member', 'readonly'],
'Model.executeDocumentScript': ['owner', 'admin', 'member'],
'Model.exportQueryResults': ['owner', 'admin', 'member', 'readonly'],
'Model.getDocument': ['owner', 'admin', 'member', 'readonly'],
Expand Down
247 changes: 247 additions & 0 deletions frontend/src/aggregation-builder/aggregation-builder.html
Original file line number Diff line number Diff line change
@@ -0,0 +1,247 @@
<div
class="aggregation-builder flex min-h-0 flex-col overflow-hidden bg-slate-50 p-4 md:p-6"
style="height: calc(100vh - 55px); height: calc(100dvh - 55px)">
<div class="relative mx-auto flex min-h-0 w-full max-w-[1600px] flex-1 flex-col gap-4">
<div class="relative flex min-h-0 flex-1 flex-col gap-3 overflow-hidden pr-1 md:gap-4">
<div
class="flex min-h-0 flex-1 flex-col gap-4 overflow-y-auto overscroll-y-contain pb-2 [-webkit-overflow-scrolling:touch] md:gap-5">
<section class="shrink-0 overflow-hidden rounded-lg border border-edge bg-page shadow-sm">
<header class="shrink-0 border-b border-edge bg-surface/60 px-4 py-2.5 md:px-5">
<h2 class="text-sm font-semibold tracking-tight text-content">Aggregation options</h2>
<p class="mt-1 text-xs leading-snug text-content-tertiary">Model, how many documents to return, and pipeline actions.</p>
</header>
<div class="flex flex-col gap-5 p-4 md:flex-row md:items-end md:justify-between md:gap-6 md:p-5">
<div class="flex min-w-0 flex-1 flex-col gap-4 sm:flex-row sm:flex-wrap sm:items-stretch sm:gap-x-5 sm:gap-y-4">
<div class="flex min-w-0 flex-1 flex-col justify-end gap-1.5 sm:max-w-md">
<label
for="aggregation-builder-model"
class="text-xs font-medium text-content-secondary">Model</label>
<select
id="aggregation-builder-model"
v-model="selectedModel"
class="box-border h-11 w-full rounded-lg border border-edge bg-surface px-3 text-sm leading-none text-content shadow-sm focus:border-edge-strong focus:outline-none focus:ring-2 focus:ring-gray-300/50">
<option v-for="modelName in models" :key="modelName" :value="modelName">{{modelName}}</option>
</select>
</div>
<div class="flex w-full flex-col justify-end gap-1.5 sm:w-[6.5rem] sm:flex-none">
<label
for="aggregation-builder-limit"
class="text-xs font-medium text-content-secondary">Limit</label>
<input
id="aggregation-builder-limit"
v-model.number="resultLimit"
type="number"
min="1"
max="200"
class="box-border h-11 w-full rounded-lg border border-edge bg-surface px-3 text-sm tabular-nums leading-none text-content shadow-sm focus:border-edge-strong focus:outline-none focus:ring-2 focus:ring-gray-300/50" />
</div>
</div>
<div class="flex w-full flex-wrap items-center gap-2 sm:w-auto sm:justify-end sm:gap-3">
<button
type="button"
@click="addStage"
class="min-h-[2.5rem] flex-1 rounded-lg border border-edge bg-surface px-4 py-2 text-sm font-medium text-content-secondary shadow-sm transition-colors hover:bg-muted sm:flex-none">
Add Stage
</button>
<button
type="button"
@click="runAggregation"
:disabled="isRunning || !selectedModel"
class="min-h-[2.5rem] flex-1 rounded-lg bg-primary px-4 py-2 text-sm font-semibold text-primary-text shadow-sm transition-colors hover:bg-primary-hover disabled:cursor-not-allowed disabled:bg-gray-500 disabled:opacity-90 sm:flex-none">
{{ isRunning ? 'Running...' : 'Run Now' }}
</button>
</div>
</div>
</section>

<section
v-for="(stage, index) in stages"
:key="stage.id + '-workflow-row'"
ref="workflowStageRows"
class="flex shrink-0 flex-col overflow-hidden rounded-lg border border-edge bg-page shadow-sm">
<header class="flex shrink-0 flex-wrap items-center justify-between gap-2 border-b border-edge bg-surface/60 px-4 py-2.5 md:px-5">
<h2 class="text-sm font-semibold tracking-tight text-content">Stage {{index + 1}}</h2>
<button
type="button"
@click="removeStage(index)"
class="rounded-md border border-edge bg-page px-3 py-1.5 text-xs font-semibold text-content-secondary shadow-sm hover:bg-muted">
Remove stage
</button>
</header>
<div class="p-4 md:p-5">
<div
class="grid min-h-0 grid-cols-1 gap-4 lg:min-h-[20rem] lg:grid-cols-[minmax(0,1.45fr)_minmax(0,0.95fr)_minmax(0,1.1fr)] lg:grid-rows-1 lg:items-stretch lg:gap-5">
<div class="flex min-h-0 flex-col gap-3 overflow-hidden rounded-lg border border-edge bg-surface p-4">
<div class="flex shrink-0 flex-wrap items-center gap-2 border-b border-edge pb-2">
<span class="text-xs font-semibold uppercase tracking-wide text-content-tertiary">Operator</span>
<select
v-model="stage.operator"
class="min-w-0 flex-1 rounded-md border border-edge bg-page px-2 py-1 text-sm text-content focus:border-edge-strong focus:outline-none sm:max-w-[11rem]">
<option v-for="operator in stageOperators" :key="operator" :value="operator">{{operator}}</option>
</select>
</div>
<textarea
v-model="stage.bodyText"
spellcheck="false"
@focus="onStageBodyFocus(stage)"
@blur="onStageBodyBlur(stage)"
class="min-h-[11rem] w-full min-w-0 flex-1 resize-y rounded-md border border-edge bg-page px-3 py-2 font-mono text-sm text-content focus:border-edge-strong focus:outline-none [tab-size:2] md:min-h-[12rem]"
placeholder="{ _id: ObjectId('...'), createdAt: new Date('2024-01-01') }"
></textarea>
<p class="shrink-0 text-xs leading-snug text-content-tertiary">
Same idea as the create document editor: JSON or JavaScript-style objects with
<span class="font-mono text-[0.7rem] text-content-secondary">ObjectId()</span>,
<span class="font-mono text-[0.7rem] text-content-secondary">new Date()</span>,
<span class="font-mono text-[0.7rem] text-content-secondary">RegExp</span>, etc. Types are sent with EJSON so they match on the server.
</p>
</div>

<div class="flex min-h-0 flex-col overflow-hidden rounded-lg border border-edge bg-surface">
<div class="shrink-0 border-b border-edge px-3 py-2.5 text-xs font-semibold text-content-tertiary">
Pipeline JSON (EJSON)
<span class="font-normal text-content-tertiary">· through stage {{index + 1}}</span>
</div>
<pre
class="min-h-0 flex-1 overflow-auto p-3 font-mono text-xs text-content [&_code]:text-inherit md:p-4"><code>{{ pipelinePreviewThrough(index) }}</code></pre>
</div>

<div class="flex min-h-0 flex-col overflow-hidden rounded-lg border border-edge bg-surface lg:h-full lg:max-h-full">
<div class="shrink-0 border-b border-edge px-3 py-2 text-xs font-semibold text-content-tertiary">Stage output preview</div>
<div class="flex min-h-0 flex-1 flex-col overflow-hidden p-2">
<div class="flex shrink-0 flex-col gap-2">
<button
type="button"
@click="runStagePreview(index)"
:disabled="!selectedModel || stage.previewLoading"
class="w-full rounded-md border border-edge bg-page px-3 py-1.5 text-xs font-semibold text-content-secondary hover:bg-muted disabled:cursor-not-allowed disabled:opacity-50 sm:w-auto">
Run preview
</button>
<button
type="button"
@click="toggleStagePreview(stage)"
class="flex w-full shrink-0 items-center justify-between rounded-md px-2 py-1.5 text-xs font-semibold text-content-tertiary hover:bg-muted">
<span>Preview</span>
<span>
<span v-if="stage.previewLoading">Loading...</span>
<span v-else-if="stage.previewError" class="text-red-600">Error</span>
<span v-else-if="!stage.previewLoaded" class="text-content-tertiary">Not previewed</span>
<span v-else-if="stage.previewDocs.length">{{stage.previewDocs.length}} doc(s)</span>
<span v-else class="text-content-tertiary">0 doc(s)</span>
<span class="ml-2">{{stage.previewExpanded ? 'Hide' : 'Show'}}</span>
</span>
</button>
</div>
<div
class="min-h-0 w-full min-w-0 flex-1 overflow-y-auto overscroll-y-contain border-t border-edge pt-2 [-webkit-overflow-scrolling:touch] max-h-[min(38vh,320px)] lg:max-h-[min(42vh,400px)]">
<template v-if="stage.previewExpanded">
<p v-if="stage.previewLoading" class="text-xs text-content-tertiary">Loading preview...</p>
<p v-else-if="stage.previewError" class="text-xs text-red-600">{{stage.previewError}}</p>
<p v-else-if="!stage.previewLoaded" class="text-xs text-content-tertiary">
Click Run preview to load sample documents through this stage.
</p>
<p v-else-if="stage.previewDocs.length === 0" class="text-xs text-content-tertiary">No documents returned.</p>
<div v-else class="flex flex-col gap-2">
<div
v-for="(doc, docIndex) in stage.previewDocs"
:key="stage.id + '-preview-' + docIndex"
class="shrink-0 overflow-hidden rounded-md border border-edge bg-page">
<div class="border-b border-edge px-[0.6rem] py-[0.4rem] text-xs font-semibold text-content-tertiary">
Document {{docIndex + 1}}
</div>
<pre
class="m-0 max-w-full overflow-x-auto whitespace-pre-wrap break-words border-0 bg-surface p-2 font-mono text-xs text-content [&_code]:text-inherit"><code>{{formatDoc(doc)}}</code></pre>
</div>
</div>
</template>
</div>
</div>
</div>
</div>
</div>
</section>
</div>

<div
v-if="hasVisibleErrors"
class="pointer-events-none absolute bottom-3 right-3 z-20 flex flex-col items-end gap-2 md:bottom-4 md:right-4">
<div
v-if="pipelineErrorsPanelOpen"
role="dialog"
aria-labelledby="aggregation-pipeline-errors-title"
class="pointer-events-auto flex max-h-[min(50vh,22rem)] w-[min(100vw-2rem,22rem)] flex-col overflow-hidden rounded-lg border border-red-200 bg-page shadow-lg">
<div class="flex shrink-0 items-center justify-between gap-2 border-b border-red-200 bg-red-50 px-3 py-2">
<h3 id="aggregation-pipeline-errors-title" class="text-sm font-semibold text-red-900">Pipeline errors</h3>
<button
type="button"
@click="closePipelineErrorsPanel"
class="rounded-md px-2 py-1 text-xs font-semibold text-red-800 hover:bg-red-100"
aria-label="Close errors">
Close
</button>
</div>
<div class="min-h-0 flex-1 overflow-y-auto overscroll-y-contain p-3 [-webkit-overflow-scrolling:touch]">
<p v-if="errorMessage" class="mb-3 text-sm leading-snug text-red-800">{{ errorMessage }}</p>
<ul v-if="visiblePipelineStageErrors.length" class="list-none space-y-2.5 text-sm text-red-800">
<li
v-for="(item, errIdx) in visiblePipelineStageErrors"
:key="'pipeline-err-' + errIdx"
class="rounded-md border border-red-100 bg-red-50/80 px-2.5 py-2">
<span class="font-semibold">Stage {{ item.stageNumber }}</span>
<p class="mt-0.5 font-mono text-xs leading-snug">{{ item.message }}</p>
</li>
</ul>
</div>
</div>
<button
type="button"
@click="togglePipelineErrorsPanel"
class="pointer-events-auto inline-flex items-center gap-2 rounded-lg border border-red-300 bg-red-50 px-3 py-2 text-sm font-semibold text-red-900 shadow-md transition-colors hover:bg-red-100"
:aria-expanded="pipelineErrorsPanelOpen"
aria-controls="aggregation-pipeline-errors-title">
<span>Errors</span>
<span class="inline-flex min-w-[1.25rem] items-center justify-center rounded-full bg-red-600 px-1.5 py-0.5 text-xs font-bold text-white">
{{ visibleErrorCount }}
</span>
</button>
</div>
</div>

<section
class="flex h-[min(34vh,440px)] min-h-[13rem] shrink-0 flex-col overflow-hidden rounded-lg border border-edge bg-page">
<div class="flex shrink-0 items-center justify-between gap-3 border-b border-edge px-4 py-2">
<span class="text-sm font-semibold text-content">Results</span>
<span v-if="isRunning" class="text-xs font-medium text-content-tertiary">Running…</span>
</div>
<div class="flex min-h-0 flex-1 flex-col overflow-hidden p-3">
<p v-if="results.length === 0 && !isRunning && !errorMessage" class="shrink-0 text-sm text-content-tertiary">
Click Run Now to execute the aggregation and show results here.
</p>
<p v-else-if="results.length === 0 && isRunning && !errorMessage" class="shrink-0 py-2 text-sm text-content-tertiary">Fetching results…</p>
<p v-else-if="errorMessage && results.length === 0 && !isRunning" class="shrink-0 text-sm text-content-tertiary">
Aggregation failed. Open <button type="button" @click="togglePipelineErrorsPanel" class="font-semibold text-red-700 underline hover:text-red-900">Errors</button> for details.
</p>
<div v-else class="flex min-h-0 flex-1 flex-col gap-3 overflow-hidden">
<p class="shrink-0 text-xs text-content-tertiary">
Showing {{ visibleResults.length }} of {{ results.length }} document(s).
</p>
<div
class="min-h-0 flex-1 overflow-y-auto overscroll-y-contain rounded-md border border-edge bg-surface p-2 [-webkit-overflow-scrolling:touch]"
:class="{ 'pointer-events-none opacity-60': isRunning }">
<list-json
:key="resultsRenderKey"
:value="visibleResults"
:expanded-fields="visibleResultsExpandedFields"
/>
</div>
<button
v-if="hasMoreResults"
type="button"
@click="loadMoreResults"
class="shrink-0 rounded-md border border-edge bg-surface px-3 py-1.5 text-xs font-semibold text-content-secondary hover:bg-muted">
Load {{ nextLoadMoreCount }} more
</button>
</div>
</div>
</section>
</div>
</div>
Loading
Loading