@malloydata/malloy-explorer 0.0.295-dev250710213205 → 0.0.299-dev250804170242

package/dist/esm/index.js

@@ -15,9 +15,8 @@ var __superGet = (cls, obj, key2) => __reflectGet(__getProtoOf(cls), key2, obj);
 var _a, _b, _captureMap, _compiled, _pattern, _nameMap, _strategy, __EmulatedRegExp_instances, execCore_fn, _c;
 import { jsx, Fragment, jsxs } from "react/jsx-runtime";
 import * as React from "react";
-import React__default, { useLayoutEffect, useEffect, useMemo, useState, useRef, createElement, useContext, useCallback, createContext, memo, useReducer } from "react";
-import * as QB from "@malloydata/malloy-query-builder";
-import { ASTArrowQueryDefinition, ASTArrowViewDefinition, ASTSegmentViewDefinition, ASTNestViewOperation, ASTRefinementViewDefinition, ASTFilterWithLiteralEquality, ASTAnnotation, ASTOrderByViewOperation, ASTGroupByViewOperation, ASTAggregateViewOperation, ASTLimitViewOperation, ASTTimeTruncationExpression, ASTFilterWithFilterString, ASTReferenceExpression, ASTWhereViewOperation, ASTHavingViewOperation, ASTDrillViewOperation, ASTQuery } from "@malloydata/malloy-query-builder";
+import React__default, { useLayoutEffect, useEffect, useMemo, createContext, useContext, useCallback, useState, useRef, createElement, memo, useReducer } from "react";
+import { ASTQuery, ASTArrowQueryDefinition, ASTArrowViewDefinition, ASTSegmentViewDefinition, ASTNestViewOperation, ASTRefinementViewDefinition, ASTFilterWithLiteralEquality, ASTAnnotation, ASTOrderByViewOperation, ASTGroupByViewOperation, ASTAggregateViewOperation, ASTLimitViewOperation, ASTCalculateViewOperation, ASTTimeTruncationExpression, ASTFilterWithFilterString, ASTReferenceExpression, ASTWhereViewOperation, ASTHavingViewOperation, ASTDrillViewOperation, ASTReferenceQueryArrowSource } from "@malloydata/malloy-query-builder";
 import { Tag } from "@malloydata/malloy-tag";
 import { MalloyRenderer, isCoreVizPluginInstance } from "@malloydata/render";
 import { TemporalFilterExpression, BooleanFilterExpression, NumberFilterExpression, StringFilterExpression } from "@malloydata/malloy-filter";
@@ -3311,11 +3310,14 @@ var Root3$1 = Tooltip;
 var Trigger$4 = TooltipTrigger;
 var Portal$4 = TooltipPortal;
 var Content2$4 = TooltipContent;
-const QueryEditorContext = /* @__PURE__ */ React.createContext({});
+const QueryEditorContext = /* @__PURE__ */ React.createContext({
+  setQuery: () => {
+  }
+});
 function useQueryBuilder(source, query) {
   return useMemo(() => {
     if (source) {
-      return new QB.ASTQuery({
+      return new ASTQuery({
         query,
         source
       });
@@ -3399,10 +3401,29 @@ const findNestView = (currentView, remainingPath) => {
   }
   return null;
 };
+const UpdateQueryContext = /* @__PURE__ */ createContext({
+  updateQuery: () => {
+    console.warn("Missing <MalloyExplorerProvider>");
+  }
+});
+function useUpdateQuery() {
+  const {
+    updateQuery
+  } = useContext(UpdateQueryContext);
+  return updateQuery;
+}
+const TopValuesContext = /* @__PURE__ */ createContext({});
+function useTopValues() {
+  const {
+    topValues
+  } = useContext(TopValuesContext);
+  return topValues;
+}
 function MalloyExplorerProvider({
   source,
   query,
-  onQueryChange,
+  onQueryChange = () => {
+  },
   focusedNestViewPath,
   onFocusedNestViewPathChange,
   children,
@@ -3410,20 +3431,32 @@ function MalloyExplorerProvider({
   onDrill
 }) {
   const rootQuery = useQueryBuilder(source, query);
+  const updateQuery = useCallback(() => {
+    onQueryChange == null ? void 0 : onQueryChange(rootQuery == null ? void 0 : rootQuery.build());
+  }, [onQueryChange, rootQuery]);
   return /* @__PURE__ */ jsx(TooltipProvider, {
     children: /* @__PURE__ */ jsx(MalloyQueryFocusProvider, {
       rootQuery,
       focusedNestViewPath,
       onFocusedNestViewPathChange,
-      children: /* @__PURE__ */ jsx(QueryEditorContext.Provider, {
+      children: /* @__PURE__ */ jsx(UpdateQueryContext.Provider, {
         value: {
-          source,
-          rootQuery,
-          setQuery: onQueryChange,
-          topValues,
-          onDrill
+          updateQuery
         },
-        children
+        children: /* @__PURE__ */ jsx(TopValuesContext.Provider, {
+          value: {
+            topValues
+          },
+          children: /* @__PURE__ */ jsx(QueryEditorContext.Provider, {
+            value: {
+              source,
+              rootQuery,
+              setQuery: onQueryChange,
+              onDrill
+            },
+            children
+          })
+        })
       })
     })
   });
@@ -25986,9 +26019,9 @@ function QueryActionBar({
       children: [/* @__PURE__ */ jsx(Button, {
         onClick: () => {
           focusMainView();
-          setQuery == null ? void 0 : setQuery(void 0);
+          setQuery(void 0);
         },
-        isDisabled: !rootQuery || (rootQuery == null ? void 0 : rootQuery.isEmpty()),
+        isDisabled: !rootQuery || rootQuery.isEmpty(),
         label: "Clear",
         variant: "flat",
         size: "compact"
@@ -26198,6 +26231,7 @@ function fieldToIcon(field) {
       break;
     case "dimension":
     case "measure":
+    case "calculate":
       icon = atomicTypeToIcon(field.type.kind);
       break;
     case "join":
@@ -26209,8 +26243,7 @@ function fieldToIcon(field) {
 function relationshipToIcon(relationship) {
   return relationshipMap[relationship];
 }
-function viewToVisualizationIcon(view) {
-  const currentTag = view.getTag();
+function tagToVisualization(currentTag) {
   const currentRenderer = tagToRenderer(currentTag) ?? "table";
   return `viz_${currentRenderer}`;
 }
@@ -26230,6 +26263,7 @@ const fieldKindMap = {
   view: "purple",
   dimension: "cyan",
   measure: "green",
+  calculate: "green",
   join: void 0
 };
 const relationshipMap = {
@@ -26239,6 +26273,7 @@ const relationshipMap = {
 };
 function FieldToken({
   field,
+  additionalSiblings,
   hoverActions,
   hoverActionsVisible,
   ...props2
@@ -26263,7 +26298,7 @@ function FieldToken({
       color: fieldKindToColor(field.kind),
       icon: fieldToIcon(field),
       ...props2
-    }), hoverActions && /* @__PURE__ */ jsx("div", {
+    }), additionalSiblings, hoverActions && /* @__PURE__ */ jsx("div", {
       ...{
         className: "mly1vsucm1 mly2lah0s"
       },
@@ -30599,6 +30634,12 @@ const styles$s = {
     justifyContent: "mly1rfj78v",
     $$css: true
   },
+  nest: {
+    gridColumn: "mly623te4",
+    gridColumnStart: null,
+    gridColumnEnd: null,
+    $$css: true
+  },
   heading: {
     gridColumn: "mly623te4",
     gridColumnStart: null,
@@ -31111,6 +31152,20 @@ const colorVariants = {
     $$css: true
   }
 };
+function setAtPath(current, path, value) {
+  const newCurrent = {
+    ...current
+  };
+  let target = newCurrent;
+  for (let i2 = 0; i2 < path.length - 1; i2++) {
+    if (typeof target[path[i2]] !== "object") {
+      target[path[i2]] = {};
+    }
+    target = target[path[i2]];
+  }
+  target[path[path.length - 1]] = value;
+  return newCurrent;
+}
 function OneOfEditor({
   view,
   name,
@@ -31156,6 +31211,15 @@ function OneOfEditor({
             current,
             option: subOption
           }, key2);
+        case "object":
+          return /* @__PURE__ */ jsx(OneOfObjectEditor, {
+            view,
+            name,
+            path,
+            updateCurrent,
+            current,
+            option: subOption
+          }, key2);
       }
       return null;
     })]
@@ -31265,7 +31329,7 @@ function OneOfNumberEditor({
           valueAsNumber
         }
       }) => {
-        setValue(value);
+        setValue(valueAsNumber);
         updateCurrent(path, valueAsNumber);
       },
       ..._stylex.props(styles$s.input)
@@ -31304,6 +31368,48 @@ function OneOfArrayEditor({
     }), /* @__PURE__ */ jsx("div", {})]
   });
 }
+function OneOfObjectEditor({
+  path,
+  current,
+  option: option2,
+  updateCurrent,
+  view
+}) {
+  const isObject2 = current != null && typeof current === "object" && !Array.isArray(current);
+  const [objectValue, setObjectValue] = useState(isObject2 ? current : {});
+  return /* @__PURE__ */ jsxs(Fragment, {
+    children: [/* @__PURE__ */ jsx("div", {
+      ..._stylex.props(styles$s.left),
+      children: /* @__PURE__ */ jsx("input", {
+        type: "radio",
+        checked: isObject2,
+        onChange: ({
+          target: {
+            checked
+          }
+        }) => {
+          if (checked) {
+            updateCurrent(path, objectValue);
+          }
+        }
+      })
+    }), /* @__PURE__ */ jsx("div", {
+      ..._stylex.props(styles$s.nest, styles$s.editorGrid),
+      children: /* @__PURE__ */ jsx(ObjectEditor, {
+        current: objectValue,
+        updateCurrent: (path2, value) => {
+          const [_2, ...subPath] = path2;
+          setObjectValue(setAtPath(objectValue, subPath, value));
+          updateCurrent(path2, value);
+        },
+        view,
+        name: "",
+        path,
+        option: option2
+      })
+    })]
+  });
+}
 function ArrayEditor({
   name,
   path,
@@ -31334,6 +31440,7 @@ function NumberEditor$1({
   option: option2,
   updateCurrent
 }) {
+  const [value, setValue] = React.useState(current ?? option2.minimum);
   return /* @__PURE__ */ jsxs(Fragment, {
     children: [/* @__PURE__ */ jsx("div", {
       ..._stylex.props(styles$s.left, styles$s.label),
@@ -31341,7 +31448,7 @@ function NumberEditor$1({
         children: [option2.title ?? name, ":"]
       })
     }), /* @__PURE__ */ jsx("input", {
-      value: current,
+      value,
       type: "number",
       min: option2.minimum,
       max: option2.maximum,
@@ -31350,6 +31457,7 @@ function NumberEditor$1({
           valueAsNumber
         }
       }) => {
+        setValue(valueAsNumber);
         updateCurrent(path, valueAsNumber);
       },
       ..._stylex.props(styles$s.input)
@@ -31548,11 +31656,12 @@ function ObjectEditor({
   option: option2,
   updateCurrent
 }) {
-  const [isExpanded, setIsExpanded] = useState(path.length === 0);
+  const [isExpanded, setIsExpanded] = useState(path.length === 0 || !name);
   return /* @__PURE__ */ jsxs(Fragment, {
     children: [name ? /* @__PURE__ */ jsxs(Fragment, {
       children: [/* @__PURE__ */ jsxs("div", {
         ..._stylex.props(styles$s.label, styles$s.heading),
+        onClick: () => setIsExpanded(!isExpanded),
         children: [/* @__PURE__ */ jsx("label", {
           children: option2.title ?? name
         }), option2.description ? /* @__PURE__ */ jsx(InfoHover, {
@@ -31644,31 +31753,20 @@ function ObjectEditor({
   });
 }
 function VizEditor({
-  rootQuery,
   view,
   plugin,
   setOpen
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const [current, setCurrent] = useState(plugin.getSettings());
-  console.info("RendererEditor current settings:", current);
   const schema = plugin.getSchema();
   const updateCurrent = (path, value) => {
-    const newCurrent = {
-      ...current
-    };
-    let target = newCurrent;
-    for (let i2 = 0; i2 < path.length - 1; i2++) {
-      if (!(path[i2] in target)) {
-        target[path[i2]] = {};
-      }
-      target = target[path[i2]];
-    }
-    target[path[path.length - 1]] = value;
+    const newCurrent = setAtPath(current, path, value);
     setCurrent(newCurrent);
   };
+  React.useEffect(() => {
+    console.info("RendererEditor current settings:", current);
+  }, [current]);
   return /* @__PURE__ */ jsxs("div", {
     ..._stylex.props(styles$s.editor),
     children: [/* @__PURE__ */ jsx("div", {
@@ -31703,7 +31801,7 @@ function VizEditor({
           view.getOrAddAnnotations().add(new ASTAnnotation({
             value
           }));
-          setQuery == null ? void 0 : setQuery(rootQuery.build());
+          updateQuery();
           setOpen(false);
         },
         customStyle: styles$s.editorCell
@@ -31916,7 +32014,6 @@ const dialogStyles = {
   }
 };
 function VizEditorDialog({
-  rootQuery,
   plugin,
   view,
   customStyle,
@@ -31946,7 +32043,6 @@ function VizEditorDialog({
             ..._stylex.props(dialogStyles.displayNone),
             children: "Edit visualization parameters"
           }), /* @__PURE__ */ jsx(VizEditor, {
-            rootQuery,
             view,
             plugin,
             setOpen
@@ -31965,12 +32061,9 @@ const styles$p = {
   }
 };
 function Visualization$1({
-  rootQuery,
   view
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const renderer = useMemo(() => new MalloyRenderer(), []);
   const [currentRenderer, setCurrentRenderer] = useState("table");
   const [plugin, setPlugin] = useState();
@@ -32015,7 +32108,7 @@ function Visualization$1({
   }, [renderer, view, tag]);
   const updateViz = (renderer2) => {
     view.setTagProperty(["viz"], renderer2, RENDERER_PREFIX);
-    setQuery == null ? void 0 : setQuery(rootQuery.build());
+    updateQuery();
   };
   const items = VIZ_RENDERERS.map((viz) => ({
     icon: /* @__PURE__ */ jsx(Icon, {
@@ -32054,7 +32147,6 @@ function Visualization$1({
     }), /* @__PURE__ */ jsx(VizEditorDialog, {
       open,
       setOpen,
-      rootQuery,
       plugin,
       view
     })]
@@ -32197,9 +32289,7 @@ function TopValuesTable({
   field,
   path
 }) {
-  const {
-    topValues
-  } = useContext(QueryEditorContext);
+  const topValues = useTopValues();
   const fieldPath = [...path, field.name].join(".");
   const fieldTopValues = topValues == null ? void 0 : topValues.find((entry) => entry.fieldName === fieldPath);
   if (!fieldTopValues) {
@@ -32572,9 +32662,7 @@ function BooleanFilterCore({
   });
 }
 function useSearch(searchTerm, fieldPath) {
-  const {
-    topValues
-  } = useContext(QueryEditorContext);
+  const topValues = useTopValues();
   if (topValues && searchTerm) {
     let searchValues = topValues;
     if (fieldPath) {
@@ -33976,6 +34064,42 @@ const styles$g = {
     $$css: true
   }
 };
+const getPrimaryAxis = (view) => {
+  const schema = view.getOutputSchema();
+  return getPrimaryAxisFromSchema(schema);
+};
+const getPrimaryAxisFromSchema = (schema) => {
+  let primaryAxisField = void 0;
+  primaryAxisField = findAxisFieldByAnnotation(schema.fields) || findAxisFieldByType(schema.fields) || findDefaultAxisField(schema.fields);
+  return primaryAxisField;
+};
+const findAxisFieldByAnnotation = (fields) => {
+  return fields.find((field) => {
+    if (field.kind === "dimension" && field.annotations && field.annotations.find((annotation) => annotation.value === "# x")) {
+      return true;
+    }
+    return false;
+  });
+};
+const findAxisFieldByType = (fields) => {
+  return fields.find((field) => {
+    if (field.kind === "dimension") {
+      const dimension = field;
+      if (dimension.type.kind === "date_type" || dimension.type.kind === "timestamp_type") {
+        return true;
+      }
+    }
+    return false;
+  });
+};
+const findDefaultAxisField = (fields) => {
+  return fields.find((field) => {
+    if (field.kind === "dimension") {
+      return true;
+    }
+    return false;
+  });
+};
 function toFullName(path, name) {
   return [...path || [], name].join(".");
 }
@@ -34054,6 +34178,34 @@ function addGroupBy(view, field, path) {
     segment.addTimestampGroupBy(field.name, path, "second");
   } else {
     segment.addGroupBy(field.name, path, rename);
+    recomputePartitionByAndPrimaryAxis(segment);
+  }
+}
+function recomputePartitionByAndPrimaryAxis(segment) {
+  var _a2;
+  const primaryAxis = getPrimaryAxisFromSchema(segment.getOutputSchema());
+  const groupBys = segment.operations.items.filter((op) => op.kind === "group_by");
+  const movingAverages = segment.operations.items.filter((op) => op.kind === "calculate" && op.expression.kind === "moving_average");
+  const canSmooth = primaryAxis !== null && (primaryAxis == null ? void 0 : primaryAxis.kind) === "dimension" && (primaryAxis.type.kind === "timestamp_type" || primaryAxis.type.kind === "date_type");
+  if (!canSmooth) {
+    movingAverages.forEach((operation) => {
+      operation.delete();
+      segment.addAggregate(operation.getFieldInfo().name);
+    });
+  } else {
+    const nonPrimaryGroupByNames = groupBys.filter((gb) => gb.name !== primaryAxis.name);
+    movingAverages.forEach((operation) => {
+      operation.expression.edit();
+      operation.expression.setPartitionFields(nonPrimaryGroupByNames.map((gb) => gb.field.getReference()).filter((ref) => !!ref));
+    });
+    const primaryTimeAxisOperation = groupBys.find((gb) => gb.name === primaryAxis.name);
+    if (primaryTimeAxisOperation) {
+      const expression = (_a2 = primaryTimeAxisOperation.field) == null ? void 0 : _a2.expression;
+      if (expression && expression.kind === "time_truncation") {
+        const truncationExpression = primaryTimeAxisOperation.field.expression;
+        truncationExpression.truncation = "day";
+      }
+    }
   }
 }
 function addAggregate(view, field, path) {
@@ -34165,16 +34317,13 @@ function OperationActionTitle({
   });
 }
 function RenameDialog({
-  rootQuery,
   view,
   target,
   open,
   setOpen
 }) {
   const [name, setName] = useState("");
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   useEffect(() => {
     if (target) {
       setName(target.name);
@@ -34184,8 +34333,9 @@ function RenameDialog({
     return null;
   }
   const onRename = () => {
+    target.edit();
     target.name = name;
-    setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+    updateQuery();
     setOpen(false);
   };
   const {
@@ -34248,15 +34398,12 @@ function RenameDialog({
   });
 }
 function SortableOperations({
-  rootQuery,
   segment,
   view,
   operations,
   kind
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const sensors = useSensors(useSensor(PointerSensor));
   const items = useMemo(() => {
     return operations.map((operation) => ({
@@ -34278,7 +34425,7 @@ function SortableOperations({
       const oldIndex = names.indexOf(active.id);
       const newIndex = names.indexOf(over.id);
       segment.reorderFields(arrayMove(names, oldIndex, newIndex));
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     }
   }
   const {
@@ -34290,7 +34437,7 @@ function SortableOperations({
     types: ["dimension"],
     onClick: (field, path) => {
       addGroupBy(view, field, path);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     }
   } : {
     title: "aggregate",
@@ -34298,12 +34445,11 @@ function SortableOperations({
     types: ["measure"],
     onClick: (field, path) => {
       addAggregate(view, field, path);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     }
   };
   return /* @__PURE__ */ jsxs("div", {
     children: [/* @__PURE__ */ jsx(OperationActionTitle, {
-      rootQuery,
       view,
       fields,
       ...props2
@@ -34316,7 +34462,6 @@ function SortableOperations({
         children: /* @__PURE__ */ jsx(SortableContext, {
           items,
           children: items.map((item) => /* @__PURE__ */ jsx(SortableOperation, {
-            rootQuery,
             id: item.id,
             color: kind === "group_by" ? "cyan" : "green",
             view,
@@ -34329,18 +34474,16 @@ function SortableOperations({
 }
 const NULL_PATH = [];
 function SortableOperation({
-  rootQuery,
   id,
   view,
   operation,
   color
 }) {
   var _a2;
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const fieldInfo = operation.getFieldInfo();
-  const path = ((_a2 = operation.field.getReference()) == null ? void 0 : _a2.path) ?? NULL_PATH;
+  const field = operation instanceof ASTCalculateViewOperation ? null : operation.field;
+  const path = operation instanceof ASTCalculateViewOperation ? operation.expression.node.field_reference.path ?? NULL_PATH : ((_a2 = operation.field.getReference()) == null ? void 0 : _a2.path) ?? NULL_PATH;
   const {
     attributes,
     listeners,
@@ -34360,6 +34503,18 @@ function SortableOperation({
     transform: CSS$1.Transform.toString(transform2),
     transition
   };
+  const primaryAxis = useMemo(() => {
+    return getPrimaryAxis(view);
+  }, [view]);
+  const canSmooth = operation.kind === "aggregate" && primaryAxis !== null && (primaryAxis == null ? void 0 : primaryAxis.kind) === "dimension" && (primaryAxis.type.kind === "timestamp_type" || primaryAxis.type.kind === "date_type");
+  const enableSmoothing = useCallback((operation2) => {
+    if (!canSmooth) {
+      return;
+    }
+    operation2.convertToCalculateMovingAverage(operation2.name + "_smoothed", 7);
+    recomputePartitionByAndPrimaryAxis(view.getOrAddDefaultSegment());
+    updateQuery();
+  }, [canSmooth, updateQuery, view]);
   const hoverActions = useMemo(() => {
     return /* @__PURE__ */ jsxs(Fragment, {
       children: [/* @__PURE__ */ jsx(DropdownMenu, {
@@ -34370,7 +34525,14 @@ function SortableOperation({
           tooltip: "More Actions"
         }),
         onOpenChange: setHoverActionsVisible,
-        children: [/* @__PURE__ */ jsx(DropdownMenuItem, {
+        children: [canSmooth ? /* @__PURE__ */ jsx(DropdownMenuItem, {
+          label: "Add smoothing",
+          onClick: () => {
+            if (operation instanceof ASTAggregateViewOperation) {
+              enableSmoothing(operation);
+            }
+          }
+        }, "smoothing") : null, /* @__PURE__ */ jsx(DropdownMenuItem, {
           label: "Rename",
           onClick: () => {
             setRenameTarget(operation);
@@ -34380,12 +34542,21 @@ function SortableOperation({
       }, [...path, fieldInfo.name].join(".")), /* @__PURE__ */ jsx(ClearButton, {
         onClick: () => {
           operation.delete();
-          setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+          recomputePartitionByAndPrimaryAxis(view.getOrAddDefaultSegment());
+          updateQuery();
         }
       })]
     });
-  }, [fieldInfo, operation, path, rootQuery, setQuery]);
-  const granular = granularityMenuItems(fieldInfo, operation.field);
+  }, [canSmooth, enableSmoothing, fieldInfo.name, operation, path, updateQuery, view]);
+  const hasSmoothedField = useMemo(() => {
+    return view.getOrAddDefaultSegment().operations.items.some((operation2) => {
+      return operation2.kind === "calculate";
+    });
+  }, [view]);
+  const granular = granularityMenuItems(fieldInfo, field);
+  if (hasSmoothedField && granular && operation.name === (primaryAxis == null ? void 0 : primaryAxis.name)) {
+    granular.options = granular.options.filter((option2) => option2.value === "day");
+  }
   let icon = "orderBy";
   if (fieldInfo.kind === "dimension" || fieldInfo.kind === "measure") {
     icon = atomicTypeToIcon(fieldInfo.type.kind);
@@ -34394,7 +34565,7 @@ function SortableOperation({
     id,
     ref: setNodeRef,
     style,
-    children: [granular ? /* @__PURE__ */ jsxs("div", {
+    children: [field && granular ? /* @__PURE__ */ jsxs("div", {
       ...{
         0: {
           className: "mly78zum5 mly6s0dn4 mly1nhvcw1 mly1jnr06f mly11ogjb6 mly1g9of4s"
@@ -34417,8 +34588,8 @@ function SortableOperation({
           color,
           value: granular.value,
           onChange: (granulation) => {
-            if (operation.field.expression instanceof ASTTimeTruncationExpression) operation.field.expression.truncation = granulation;
-            setQuery == null ? void 0 : setQuery(rootQuery.build());
+            if (field.expression instanceof ASTTimeTruncationExpression) field.expression.truncation = granulation;
+            updateQuery();
           },
           items: granular.options
         })]
@@ -34428,7 +34599,7 @@ function SortableOperation({
         },
         children: hoverActions
       })]
-    }) : /* @__PURE__ */ jsx(FieldToken, {
+    }) : operation instanceof ASTCalculateViewOperation ? /* @__PURE__ */ jsx(FieldToken, {
       field: fieldInfo,
       color,
       hoverActionsVisible,
@@ -34445,9 +34616,67 @@ function SortableOperation({
       dragProps: {
         attributes,
         listeners
-      }
+      },
+      additionalSiblings: /* @__PURE__ */ jsx(SelectorToken, {
+        color,
+        value: "" + (operation.expression.node.rows_preceding ?? 7),
+        onChange: (value) => {
+          if (value === "1") {
+            view.getOrAddDefaultSegment().operations.add(new ASTAggregateViewOperation({
+              kind: "aggregate",
+              field: {
+                expression: {
+                  kind: "field_reference",
+                  name: operation.expression.node.field_reference.name
+                }
+              }
+            }));
+            operation.delete();
+            updateQuery();
+          } else {
+            operation.expression.edit();
+            operation.expression.rowsPreceding = parseInt(value, 10);
+            updateQuery();
+          }
+        },
+        items: [{
+          label: "7d",
+          value: "7"
+        }, {
+          label: "14d",
+          value: "14"
+        }, {
+          label: "28d",
+          value: "28"
+        }, {
+          label: "30d",
+          value: "30"
+        }, {
+          label: "Remove smoothing",
+          value: "1"
+        }]
+      })
+    }) : /* @__PURE__ */ jsx(Fragment, {
+      children: /* @__PURE__ */ jsx(FieldToken, {
+        field: fieldInfo,
+        color,
+        hoverActionsVisible,
+        hoverActions,
+        tooltip: /* @__PURE__ */ jsx(FieldHoverCard, {
+          field: fieldInfo,
+          path
+        }),
+        tooltipProps: {
+          side: "right",
+          align: "start",
+          alignOffset: 28
+        },
+        dragProps: {
+          attributes,
+          listeners
+        }
+      })
     }), /* @__PURE__ */ jsx(RenameDialog, {
-      rootQuery,
       view,
       target: renameTarget,
       open: renameOpen,
@@ -34458,7 +34687,7 @@ function SortableOperation({
 const DateGranulation = ["day", "week", "month", "quarter", "year"];
 const TimestampGranulation = ["second", "minute", "hour", "day", "week", "month", "quarter", "year"];
 function granularityMenuItems(fieldInfo, field) {
-  if (fieldInfo.kind !== "dimension" || !(field.expression instanceof ASTTimeTruncationExpression)) {
+  if (!field || fieldInfo.kind !== "dimension" || !(field.expression instanceof ASTTimeTruncationExpression)) {
     return null;
   }
   if (fieldInfo.type.kind === "timestamp_type") {
@@ -34488,13 +34717,11 @@ const customStyles = {
   }
 };
 function GroupByOperations({
-  rootQuery,
   segment,
   view,
   groupBys
 }) {
   return /* @__PURE__ */ jsx(SortableOperations, {
-    rootQuery,
     segment,
     view,
     operations: groupBys,
@@ -34737,12 +34964,9 @@ function displayTimeFromMoment(momentObj) {
   return momentObj.moment;
 }
 function FilterOperations({
-  rootQuery,
   filters
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   if (filters.length === 0) {
     return null;
   }
@@ -34758,13 +34982,12 @@ function FilterOperations({
             children: ["Invalid filter", /* @__PURE__ */ jsx(ClearButton, {
               onClick: () => {
                 filterOperation.delete();
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               }
             })]
           }),
           children: /* @__PURE__ */ jsx(SingleFilterOperation, {
-            filterOperation,
-            rootQuery
+            filterOperation
           })
         }, key2);
       })
@@ -34772,18 +34995,15 @@ function FilterOperations({
   });
 }
 function SingleFilterOperation({
-  rootQuery,
   filterOperation
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const setFilter = useCallback((filter2) => {
     if (filterOperation.filter instanceof ASTFilterWithFilterString) {
       filterOperation.filter.setFilter(filter2);
     }
-    setQuery == null ? void 0 : setQuery(rootQuery.build());
-  }, [filterOperation.filter, rootQuery, setQuery]);
+    updateQuery();
+  }, [filterOperation.filter, updateQuery]);
   const {
     filter
   } = filterOperation;
@@ -34827,19 +35047,16 @@ function SingleFilterOperation({
       children: /* @__PURE__ */ jsx(ClearButton, {
         onClick: () => {
           filterOperation.delete();
-          setQuery == null ? void 0 : setQuery(rootQuery.build());
+          updateQuery();
         }
       })
     })]
   });
 }
 function LimitOperation({
-  rootQuery,
   limit
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   if (!limit) {
     return null;
   }
@@ -34854,7 +35071,7 @@ function LimitOperation({
         value: limit.limit,
         onChange: (value) => {
           limit.limit = value;
-          setQuery == null ? void 0 : setQuery(rootQuery.build());
+          updateQuery();
         },
         errorMessage: limit.limit < 0 ? "Limit must be positive" : void 0
       }), /* @__PURE__ */ jsx("div", {
@@ -34862,7 +35079,7 @@ function LimitOperation({
         children: /* @__PURE__ */ jsx(ClearButton, {
           onClick: () => {
             limit.delete();
-            setQuery == null ? void 0 : setQuery(rootQuery.build());
+            updateQuery();
           }
         })
       })]
@@ -34870,13 +35087,11 @@ function LimitOperation({
   });
 }
 function AggregateOperations({
-  rootQuery,
   segment,
   view,
   aggregates
 }) {
   return /* @__PURE__ */ jsx(SortableOperations, {
-    rootQuery,
     segment,
     view,
     operations: aggregates,
@@ -34884,12 +35099,9 @@ function AggregateOperations({
   });
 }
 function OrderByOperations({
-  rootQuery,
   orderBys
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   if (orderBys.length === 0) {
     return null;
   }
@@ -34929,7 +35141,7 @@ function OrderByOperations({
               value: orderBy.direction ?? "asc",
               onChange: (direction) => {
                 orderBy.direction = direction;
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               }
             })]
           }), /* @__PURE__ */ jsx("div", {
@@ -34937,7 +35149,7 @@ function OrderByOperations({
             children: /* @__PURE__ */ jsx(ClearButton, {
               onClick: () => {
                 orderBy.delete();
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               }
             })
           })]
@@ -34947,7 +35159,6 @@ function OrderByOperations({
   });
 }
 function View({
-  rootQuery,
   view
 }) {
   return /* @__PURE__ */ jsxs("div", {
@@ -34957,10 +35168,8 @@ function View({
       gap: 8
     },
     children: [/* @__PURE__ */ jsx(Visualization$1, {
-      rootQuery,
       view
     }), /* @__PURE__ */ jsx(ViewDefinition, {
-      rootQuery,
       view,
       viewDef: view.definition
     })]
@@ -35025,12 +35234,9 @@ const styles$f = {
   }
 };
 function AddLimit({
-  rootQuery,
   view
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const segment = getSegmentIfPresent(view);
   const hasLimit = segment ? segmentHasLimit(segment) : false;
   return /* @__PURE__ */ jsx(AddItem, {
@@ -35042,17 +35248,14 @@ function AddLimit({
     onClick: () => {
       const segment2 = view.getOrAddDefaultSegment();
       segment2.setLimit(10);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     }
   });
 }
 function AddEmptyNest({
-  rootQuery,
   view
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   return /* @__PURE__ */ jsx(AddItem, {
     icon: /* @__PURE__ */ jsx(Icon, {
       name: "nest"
@@ -35062,7 +35265,7 @@ function AddEmptyNest({
       const segment = view.getOrAddDefaultSegment();
       const nestNo = segmentNestNo(segment, `Nest`);
       segment.addEmptyNest(nestNo > 1 ? `Nest ${nestNo}` : `Nest`);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     }
   });
 }
@@ -35094,13 +35297,10 @@ const styles$e = {
   }
 };
 function AddOrderBy({
-  rootQuery,
   view,
   search
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const outputSchemaFields = view.getOutputSchema().fields;
   const segment = getSegmentIfPresent(view);
   const fields = useMemo(() => outputSchemaFields.filter((field) => field.kind === "dimension").filter((field) => ORDERABLE_TYPES$1.includes(field.type.kind)).filter((field) => segment ? !segmentHasOrderBy(segment, field.name) : true), [outputSchemaFields, segment]);
@@ -35112,7 +35312,7 @@ function AddOrderBy({
     types: ["dimension"],
     onAddOperation: (field) => {
       addOrderBy(view, field);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     },
     disabledMessage: "There must be at least one field in the output to order by.",
     search
@@ -35120,13 +35320,10 @@ function AddOrderBy({
 }
 const ORDERABLE_TYPES$1 = ["string_type", "number_type", "boolean_type", "date_type", "timestamp_type"];
 function AddGroupBy({
-  rootQuery,
   view,
   search
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const {
     fields
   } = getInputSchemaFromViewParent(view);
@@ -35143,19 +35340,16 @@ function AddGroupBy({
     filter,
     onAddOperation: (field, path) => {
       addGroupBy(view, field, path);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     },
     search
   });
 }
 function AddAggregate({
-  rootQuery,
   view,
   search
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const {
     fields
   } = getInputSchemaFromViewParent(view);
@@ -35172,7 +35366,7 @@ function AddAggregate({
     filter,
     onAddOperation: (field, path) => {
       addAggregate(view, field, path);
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     },
     search
   });
@@ -35181,10 +35375,7 @@ function AddWhere({
   view,
   search
 }) {
-  const {
-    rootQuery,
-    setQuery
-  } = React.useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const {
     fields
   } = getInputSchemaFromViewParent(view);
@@ -35198,7 +35389,7 @@ function AddWhere({
     onAddOperation: (field, path, filter) => {
       if (filter && (field.kind === "dimension" || field.kind === "measure")) {
         addFilter(view, field, path, filter);
-        setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+        updateQuery();
       }
     },
     isFilterOperation: true,
@@ -35207,15 +35398,18 @@ function AddWhere({
 }
 const FILTERABLE_TYPES$1 = /* @__PURE__ */ new Set(["string_type", "boolean_type", "number_type", "date_type", "timestamp_type"]);
 function AddView({
-  rootQuery,
   view,
   search
 }) {
   const {
-    setQuery
+    rootQuery
   } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const allFields = getInputSchemaFromViewParent(view).fields;
   const fields = useMemo(() => allFields.filter((field) => field.kind === "view" && !isIndexView(field)), [allFields]);
+  if (!rootQuery) {
+    return null;
+  }
   return /* @__PURE__ */ jsx(AddFieldItem, {
     label: "Add view",
     icon: "view",
@@ -35229,20 +35423,17 @@ function AddView({
       } else {
         addNest(view, field);
       }
-      setQuery == null ? void 0 : setQuery(rootQuery.build());
+      updateQuery();
     },
     search
   });
 }
 function AddMenu({
-  rootQuery,
   view
 }) {
   const [open, setOpen] = useState(false);
   const [active, setActive] = useState("");
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const [search, setSearch] = useState("");
   const segment = getSegmentIfPresent(view);
   const toggleActive = (toggle) => {
@@ -35306,7 +35497,7 @@ function AddMenu({
                 } else {
                   addNest(view, field);
                 }
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               },
               search
             }), /* @__PURE__ */ jsx(Divider, {}), /* @__PURE__ */ jsx(ValueList, {
@@ -35322,7 +35513,7 @@ function AddMenu({
                     values: [value.fieldValue ?? "∅"]
                   }
                 });
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               }
             })]
           }) : /* @__PURE__ */ jsxs("div", {
@@ -35359,7 +35550,6 @@ function AddMenu({
               onClick: () => toggleActive("view"),
               open: active === "view"
             }), /* @__PURE__ */ jsx(Divider, {}), /* @__PURE__ */ jsx(AddLimit, {
-              rootQuery,
               view
             }), /* @__PURE__ */ jsx(AddItem, {
               icon: /* @__PURE__ */ jsx(Icon, {
@@ -35369,7 +35559,6 @@ function AddMenu({
               onClick: () => toggleActive("order_by"),
               open: active === "order_by"
             }), /* @__PURE__ */ jsx(AddEmptyNest, {
-              rootQuery,
               view
             })]
           }), /* @__PURE__ */ jsxs("div", {
@@ -35378,23 +35567,18 @@ function AddMenu({
               overflowY: "auto"
             },
             children: [active === "group_by" && /* @__PURE__ */ jsx(AddGroupBy, {
-              rootQuery,
               view,
               search
             }), active === "aggregate" && /* @__PURE__ */ jsx(AddAggregate, {
-              rootQuery,
               view,
               search
             }), active === "where" && /* @__PURE__ */ jsx(AddWhere, {
-              rootQuery,
               view,
               search
             }), active === "view" && /* @__PURE__ */ jsx(AddView, {
-              rootQuery,
               view,
               search
             }), active === "order_by" && /* @__PURE__ */ jsx(AddOrderBy, {
-              rootQuery,
               view,
               search
             })]
@@ -35440,7 +35624,6 @@ function FocusableView({
   });
 }
 function NestOperations({
-  rootQuery,
   view,
   nests
 }) {
@@ -35450,20 +35633,16 @@ function NestOperations({
   return /* @__PURE__ */ jsx("div", {
     ..._stylex.props(styles$u.tokenContainer),
     children: nests.map((nest) => /* @__PURE__ */ jsx(NestOperation, {
-      rootQuery,
       view,
       nest
     }, nest.name))
   });
 }
 function NestOperation({
-  rootQuery,
   view,
   nest
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const [renameOpen, setRenameOpen] = useState(false);
   const parentNestViewPath = useContext(NestViewPathContext);
   const {
@@ -35484,7 +35663,7 @@ function NestOperation({
         onClick: () => {
           focusNestView([...parentNestViewPath]);
           nest2.delete();
-          setQuery == null ? void 0 : setQuery(rootQuery.build());
+          updateQuery();
         }
       }), /* @__PURE__ */ jsx(DropdownMenuItem, {
         label: "Rename",
@@ -35493,7 +35672,6 @@ function NestOperation({
         }
       })]
     }), /* @__PURE__ */ jsx(AddMenu, {
-      rootQuery,
       view: nest2.view
     })]
   });
@@ -35505,17 +35683,15 @@ function NestOperation({
       },
       children: [/* @__PURE__ */ jsx(CollapsiblePanel, {
         title: nest.name,
-        icon: viewToVisualizationIcon(nest.view),
+        icon: tagToVisualization(nest.view.getTag()),
         defaultOpen: true,
         controls: getControls(nest),
         collapsedControls: getControls(nest),
         isFocused: isNestViewFocused([...parentNestViewPath, nest.name]),
         children: /* @__PURE__ */ jsx(View, {
-          rootQuery,
           view: nest.view
         })
       }), /* @__PURE__ */ jsx(RenameDialog, {
-        rootQuery,
         view,
         target: nest,
         open: renameOpen,
@@ -35525,7 +35701,6 @@ function NestOperation({
   });
 }
 function Operations({
-  rootQuery,
   view,
   viewDef
 }) {
@@ -35543,7 +35718,7 @@ function Operations({
   segment.operations.items.forEach((operation) => {
     if (operation instanceof ASTGroupByViewOperation) {
       groupBys.push(operation);
-    } else if (operation instanceof ASTAggregateViewOperation) {
+    } else if (operation instanceof ASTAggregateViewOperation || operation instanceof ASTCalculateViewOperation) {
       aggregates.push(operation);
     } else if (operation instanceof ASTWhereViewOperation) {
       filters.push(operation);
@@ -35564,66 +35739,53 @@ function Operations({
       className: "mly1tpqehw mly10lvyaf"
     },
     children: [/* @__PURE__ */ jsx(GroupByOperations, {
-      rootQuery,
       segment,
       view,
       groupBys
     }), /* @__PURE__ */ jsx(AggregateOperations, {
-      rootQuery,
       segment,
       view,
       aggregates
     }), /* @__PURE__ */ jsx(DrillOperations, {
-      rootQuery,
       drills
     }), /* @__PURE__ */ jsx(FilterOperations, {
-      rootQuery,
       filters
     }), /* @__PURE__ */ jsx(OrderByOperations, {
-      rootQuery,
       orderBys
     }), /* @__PURE__ */ jsx(NestOperations, {
-      rootQuery,
       view,
       nests
     }), /* @__PURE__ */ jsx(LimitOperation, {
-      rootQuery,
       limit
     })]
   });
 }
 function ViewDefinition({
-  rootQuery,
   view,
   viewDef
 }) {
   if (viewDef instanceof ASTArrowViewDefinition) {
     return /* @__PURE__ */ jsx(ViewDefinition, {
-      rootQuery,
       view,
       viewDef: viewDef.view
     });
   } else if (viewDef instanceof ASTRefinementViewDefinition) {
     return /* @__PURE__ */ jsxs("div", {
       children: [/* @__PURE__ */ jsx(ViewDefinition, {
-        rootQuery,
         view,
         viewDef: viewDef.base
       }), /* @__PURE__ */ jsx(ViewDefinition, {
-        rootQuery,
         view,
         viewDef: viewDef.refinement
       })]
     });
   } else if (viewDef instanceof ASTSegmentViewDefinition) {
     return /* @__PURE__ */ jsx(Operations, {
-      rootQuery,
       view,
       viewDef
     });
   } else {
     return /* @__PURE__ */ jsx(CollapsingView, {
-      rootQuery,
       viewDef
     });
   }
@@ -35692,14 +35854,21 @@ const styles$c = {
   }
 };
 function Query({
-  rootQuery,
-  query,
-  setQuery
+  definition
 }) {
   const {
     focusMainView,
     isMainViewFocused
   } = useQueryFocus();
+  const {
+    rootQuery,
+    setQuery
+  } = useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
+  if (!rootQuery) {
+    return null;
+  }
+  const isEmpty = rootQuery.isEmpty();
   return /* @__PURE__ */ jsx(FocusableView, {
     children: /* @__PURE__ */ jsxs(CollapsiblePanel, {
       title: "Main query",
@@ -35718,44 +35887,41 @@ function Query({
               label: "Clear query",
               onClick: () => {
                 focusMainView();
-                setQuery == null ? void 0 : setQuery(void 0);
+                setQuery(void 0);
               },
-              disabled: rootQuery.isEmpty()
+              disabled: isEmpty
             }), /* @__PURE__ */ jsx(DropdownMenuItem, {
               icon: "nest",
               label: "Nest query",
               onClick: () => {
-                if (rootQuery.definition instanceof ASTArrowQueryDefinition) {
-                  rootQuery.definition.view.convertToNest("Nest");
+                if (definition instanceof ASTArrowQueryDefinition) {
+                  definition.view.convertToNest("Nest");
                 }
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               },
-              disabled: rootQuery.isEmpty() || !(rootQuery.definition instanceof ASTArrowQueryDefinition)
+              disabled: isEmpty || !(definition instanceof ASTArrowQueryDefinition)
             })]
           }) : /* @__PURE__ */ jsx(Fragment, {})
-        }), query.definition instanceof ASTArrowQueryDefinition ? /* @__PURE__ */ jsx(AddMenu, {
-          rootQuery,
-          view: query.definition
+        }), definition instanceof ASTArrowQueryDefinition ? /* @__PURE__ */ jsx(AddMenu, {
+          view: definition
         }) : null]
       }),
       collapsedControls: /* @__PURE__ */ jsx(Icon, {
-        name: viewToVisualizationIcon(query)
+        name: tagToVisualization(rootQuery.getTag())
       }),
-      children: [query.definition instanceof ASTArrowQueryDefinition ? /* @__PURE__ */ jsxs("div", {
+      children: [definition instanceof ASTArrowQueryDefinition ? /* @__PURE__ */ jsxs("div", {
         style: {
           display: "flex",
           flexDirection: "column",
           gap: 8
         },
-        children: [!query.isEmpty() && /* @__PURE__ */ jsx(Visualization$1, {
-          rootQuery,
-          view: query
+        children: [!isEmpty && /* @__PURE__ */ jsx(Visualization$1, {
+          view: rootQuery
         }), /* @__PURE__ */ jsx(ViewDefinition, {
-          rootQuery,
-          view: query.definition,
-          viewDef: query.definition.view
+          view: definition,
+          viewDef: definition.view
         })]
-      }) : null, query.isEmpty() ? /* @__PURE__ */ jsx("div", {
+      }) : null, isEmpty ? /* @__PURE__ */ jsx("div", {
         ...{
           className: "mly78zum5 mlyl56j7k mly6s0dn4 mlyng8ra"
         },
@@ -35777,9 +35943,9 @@ function Query({
   });
 }
 function Source({
-  rootQuery
+  definition
 }) {
-  if (rootQuery.definition instanceof ASTArrowQueryDefinition) {
+  if (definition instanceof ASTArrowQueryDefinition) {
     return /* @__PURE__ */ jsx("div", {
       ..._stylex.props(styles$u.queryCard, styles$b.content),
       children: /* @__PURE__ */ jsxs("div", {
@@ -35790,7 +35956,7 @@ function Source({
           ...{
             className: "mlyb3r6kr mlylyipyv mly117nqv4"
           },
-          children: rootQuery.definition.as.ArrowQueryDefinition().source.as.ReferenceQueryArrowSource().name
+          children: definition.source.getSourceInfo().name
         })]
       })
     });
@@ -36127,13 +36293,13 @@ function LiteralValueEditor({
   }
 }
 function Parameters({
-  rootQuery
+  definition
 }) {
-  const {
-    setQuery
-  } = useContext(QueryEditorContext);
-  if (rootQuery.definition instanceof ASTArrowQueryDefinition) {
-    const source = rootQuery.definition.as.ArrowQueryDefinition().source.as.ReferenceQueryArrowSource();
+  const updateQuery = useUpdateQuery();
+  if (definition instanceof ASTArrowQueryDefinition && definition.source instanceof ASTReferenceQueryArrowSource) {
+    const {
+      source
+    } = definition;
     const sourceParameters = source.getSourceParameters();
     if (!sourceParameters || sourceParameters.length === 0) {
       return null;
@@ -36156,7 +36322,7 @@ function Parameters({
               value: ((_a2 = source.tryGetParameter(parameter.name)) == null ? void 0 : _a2.parameter.value) ?? parameter.default_value,
               setValue: (value) => {
                 source.setParameter(parameter.name, value);
-                setQuery == null ? void 0 : setQuery(rootQuery.build());
+                updateQuery();
               }
             })]
           }, parameter.name);
@@ -36174,23 +36340,23 @@ const styles$9 = {
 };
 function QueryEditor() {
   const {
-    rootQuery,
-    setQuery
+    rootQuery
   } = useContext(QueryEditorContext);
   if (!rootQuery) {
     console.error("Missing <MalloyExplorerProvider>");
     return null;
   }
+  const {
+    definition
+  } = rootQuery;
   return /* @__PURE__ */ jsxs("div", {
     ..._stylex.props(fontStyles.body, styles$8.main),
     children: [/* @__PURE__ */ jsx(Source, {
-      rootQuery
+      definition
     }), /* @__PURE__ */ jsx(Parameters, {
-      rootQuery
+      definition
     }), /* @__PURE__ */ jsx(Query, {
-      rootQuery,
-      query: rootQuery,
-      setQuery
+      definition
     })]
   });
 }
@@ -36733,7 +36899,7 @@ function RenderedResult({
           query: stableQuery,
           source
         });
-        setQuery == null ? void 0 : setQuery(rootQuery.build());
+        setQuery(rootQuery.build());
       },
       tableConfig: {
         enableDrill: true
@@ -37314,27 +37480,21 @@ function useOperations(view, field, path) {
     return "";
   }, [matchingFieldItem, view, path, field.name, fullName]);
   return {
-    isGroupByAllowed: !groupByDisabledReason,
     groupByDisabledReason,
-    isAggregateAllowed: !aggregateDisabledReason,
     aggregateDisabledReason,
-    isFilterAllowed: !filterDisabledReason,
     filterDisabledReason,
-    isOrderByAllowed: !orderByDisabledReason,
     orderByDisabledReason
   };
 }
 const FILTERABLE_TYPES = ["string_type", "boolean_type", "number_type", "date_type", "timestamp_type"];
 const ORDERABLE_TYPES = ["string_type", "number_type", "boolean_type", "date_type", "timestamp_type"];
 function FieldTokenWithActions({
+  rootQuery,
   field,
   path,
   viewDef
 }) {
-  const {
-    rootQuery,
-    setQuery
-  } = React__default.useContext(QueryEditorContext);
+  const updateQuery = useUpdateQuery();
   const {
     focusedNestView
   } = useQueryFocus();
@@ -37358,19 +37518,19 @@ function FieldTokenWithActions({
       } else if (operation === "filter" && !filterDisabledReason && filter) {
         addFilter(view, field, path, filter);
       }
-      setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+      updateQuery();
     }
   };
   const handleSetView = () => {
-    if (field.kind === "view" && (rootQuery == null ? void 0 : rootQuery.isEmpty())) {
-      rootQuery == null ? void 0 : rootQuery.setView(field.name);
-      setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+    if (field.kind === "view" && rootQuery.isEmpty()) {
+      rootQuery.setView(field.name);
+      updateQuery();
     }
   };
   const handleAddView = () => {
     if (field.kind === "view") {
       addNest(view, field);
-      setQuery == null ? void 0 : setQuery(rootQuery == null ? void 0 : rootQuery.build());
+      updateQuery();
     }
   };
   return /* @__PURE__ */ jsx(FieldToken, {
@@ -37378,9 +37538,9 @@ function FieldTokenWithActions({
     hoverActions: field.kind === "view" ? /* @__PURE__ */ jsxs(Fragment, {
       children: [/* @__PURE__ */ jsx(ActionButton, {
         icon: "insert",
-        disabled: !(rootQuery == null ? void 0 : rootQuery.isEmpty()),
+        disabled: !rootQuery.isEmpty(),
         onClick: handleSetView,
-        tooltip: !(rootQuery == null ? void 0 : rootQuery.isEmpty()) ? "Can only add a view to an empty query." : "Add view",
+        tooltip: !rootQuery.isEmpty() ? "Can only add a view to an empty query." : "Add view",
         onTooltipOpenChange: setIsTooltipOpen
       }), /* @__PURE__ */ jsx(ActionButton, {
         icon: "nest",
@@ -37446,7 +37606,7 @@ function FieldTokenWithActions({
       })]
     }) : null,
     onClick: field.kind === "dimension" && !groupByDisabledReason ? () => handleAddOperationAction("groupBy") : field.kind === "measure" && !aggregateDisabledReason ? () => handleAddOperationAction("aggregate") : field.kind === "view" ? () => {
-      if (rootQuery == null ? void 0 : rootQuery.isEmpty()) {
+      if (rootQuery.isEmpty()) {
         handleSetView();
       } else {
         handleAddView();
@@ -37490,6 +37650,7 @@ function ActionButton({
 }
 const FIELD_KIND_ORDER = ["dimension", "measure", "view"];
 function SearchResultList({
+  rootQuery,
   source,
   items
 }) {
@@ -37499,10 +37660,7 @@ function SearchResultList({
       items: groupFieldItemsByPath(source, group.items)
     }));
   }, [source, items]);
-  const {
-    rootQuery
-  } = React.useContext(QueryEditorContext);
-  const viewDef = rootQuery == null ? void 0 : rootQuery.definition;
+  const viewDef = rootQuery.definition;
   if (!(viewDef instanceof ASTArrowQueryDefinition)) {
     return null;
   }
@@ -37539,6 +37697,7 @@ function SearchResultList({
           field,
           path
         }) => /* @__PURE__ */ jsx(FieldTokenWithActions, {
+          rootQuery,
           field,
           path,
           viewDef
@@ -37565,6 +37724,7 @@ const getSublabelFromPath = (source, path) => {
   return path.length > 0 ? `joined to ${[...path.slice(0, -1), source.name].join(" > ")}` : void 0;
 };
 function FieldGroupList({
+  rootQuery,
   source,
   fieldItems,
   fieldGroupType
@@ -37584,10 +37744,7 @@ function FieldGroupList({
       group
     }) => group === fieldGroupType)) == null ? void 0 : _a2.items) ?? [];
   }, [fieldGroupsByKindByPath, fieldGroupType]);
-  const {
-    rootQuery
-  } = React.useContext(QueryEditorContext);
-  const viewDef = rootQuery == null ? void 0 : rootQuery.definition;
+  const viewDef = rootQuery.definition;
   if (!(viewDef instanceof ASTArrowQueryDefinition)) {
     return null;
   }
@@ -37604,6 +37761,7 @@ function FieldGroupList({
           field,
           path
         }) => /* @__PURE__ */ jsx(FieldTokenWithActions, {
+          rootQuery,
           field,
           path,
           viewDef
@@ -37616,7 +37774,8 @@ function SourcePanel({
   onRefresh
 }) {
   const {
-    source
+    source,
+    rootQuery
   } = React.useContext(QueryEditorContext);
   const [searchQuery, setSearchQuery] = React.useState("");
   const {
@@ -37639,7 +37798,7 @@ function SourcePanel({
     return [];
   }, [fieldItems, searchQuery]);
   const isSearchActive = !!searchQuery;
-  if (!source) {
+  if (!source || !rootQuery) {
     return null;
   }
   return /* @__PURE__ */ jsxs("div", {
@@ -37694,6 +37853,7 @@ function SourcePanel({
         className: "mly78zum5 mlydt5ytf mlyysyzu8 mly1iyjqo2"
       },
       children: isSearchActive ? /* @__PURE__ */ jsx(SearchResultList, {
+        rootQuery,
         source,
         items: searchResultItems
       }) : /* @__PURE__ */ jsxs(AccordionList, {
@@ -37710,6 +37870,7 @@ function SourcePanel({
             color: "cyan"
           }),
           children: /* @__PURE__ */ jsx(FieldGroupList, {
+            rootQuery,
             source,
             fieldItems,
             fieldGroupType: "dimension"
@@ -37726,6 +37887,7 @@ function SourcePanel({
             color: "green"
           }),
           children: /* @__PURE__ */ jsx(FieldGroupList, {
+            rootQuery,
             source,
             fieldItems,
             fieldGroupType: "measure"
@@ -37742,6 +37904,7 @@ function SourcePanel({
             color: "purple"
           }),
           children: /* @__PURE__ */ jsx(FieldGroupList, {
+            rootQuery,
             source,
             fieldItems,
             fieldGroupType: "view"