@malloy-publisher/server 0.2.7 → 0.3.1

package/dist/package_load_worker.mjs

@@ -14820,6 +14820,9 @@ var init_motly = __esm(() => {
 });
 
 // src/service/given.ts
+import {
+  isSourceDef as isSourceDef5
+} from "@malloydata/malloy";
 function presentText(tag, ...path) {
   const raw = tagText(tag, ...path);
   return raw !== undefined && raw.trim() !== "" ? raw : undefined;
@@ -14875,8 +14878,78 @@ function malloyGivenToApi(given) {
     default: given._internal?.defaultText
   };
 }
+function collectGivenRefs(value, into) {
+  if (Array.isArray(value)) {
+    for (const item of value)
+      collectGivenRefs(item, into);
+    return;
+  }
+  if (value === null || typeof value !== "object")
+    return;
+  const node = value;
+  if (node.node === "given" && typeof node.refName === "string") {
+    into.add(node.refName);
+  }
+  for (const child of Object.values(node))
+    collectGivenRefs(child, into);
+}
+function suggestGivenLookup(modelDef, authorizeBySource, surfaced) {
+  const registry = modelDef.givens ?? {};
+  const bySource = new Map;
+  const byQuery = new Map;
+  for (const obj of Object.values(modelDef.contents)) {
+    if (isSourceDef5(obj)) {
+      const name = obj.as || obj.name;
+      const refs = new Set;
+      collectGivenRefs(obj.filterList, refs);
+      for (const expr of authorizeBySource(name) ?? []) {
+        for (const given of referencedGivenNames(expr))
+          refs.add(given);
+      }
+      bySource.set(name, Array.from(refs));
+    } else if (obj.type === "query") {
+      const query = obj;
+      byQuery.set(query.as || query.name, {
+        own: (query.givenUsage ?? []).map((usage) => registry[usage.id]?.name).filter((n) => n !== undefined),
+        source: typeof query.structRef === "string" ? query.structRef : undefined
+      });
+    }
+  }
+  const narrow = (names) => Array.from(new Set(names)).filter((name) => surfaced === undefined || surfaced.has(name));
+  return {
+    forSource: (name) => {
+      const found = bySource.get(name);
+      return found && narrow(found);
+    },
+    forQuery: (name) => {
+      const found = byQuery.get(name);
+      if (!found)
+        return;
+      return narrow([
+        ...found.own,
+        ...found.source ? bySource.get(found.source) ?? [] : []
+      ]);
+    }
+  };
+}
+function suggestGivenNames(suggest, lookup) {
+  const names = suggest.query !== undefined ? lookup.forQuery(suggest.query) : suggest.source !== undefined ? lookup.forSource(suggest.source) : undefined;
+  return names && names.length > 0 ? names : undefined;
+}
+function attachSuggestGivenNames(givens, lookup) {
+  for (const given of givens ?? []) {
+    if (!given.suggest)
+      continue;
+    const names = suggestGivenNames(given.suggest, lookup);
+    if (names)
+      given.suggest.givenNames = names;
+    else
+      delete given.suggest.givenNames;
+  }
+}
 var init_given = __esm(() => {
   init_annotations();
+  init_authorize();
   init_motly();
 });
 
@@ -14914,7 +14987,7 @@ init_gate_dimension();
 var import_recursive_readdir = __toESM(require_recursive_readdir(), 1);
 import {
   contextOverlay,
-  isSourceDef as isSourceDef5,
+  isSourceDef as isSourceDef6,
   MalloyConfig,
   MalloyError as MalloyError2,
   modelDefToModelInfo,
@@ -15415,6 +15488,7 @@ async function compileMalloyModel(job, malloyConfig, modelPath) {
     authorizeOwnNotes,
     attributedAuthorizeOwnNotes
   } = extractSources(modelDef, givens);
+  attachSuggestGivenNames(givens, suggestGivenLookup(modelDef, (name) => sources.find((source) => source.name === name)?.authorize, new Set((givens ?? []).map((given) => given.name))));
   const queryResult = extractQueries(modelDef);
   const queries = queryResult.queries;
   assertPartitionAnnotationsValid(modelDef);
@@ -15434,7 +15508,7 @@ async function compileMalloyModel(job, malloyConfig, modelPath) {
     onRowLevelGateUnexpressible: authorizeWarningCollection.onRowLevelGateUnexpressible,
     onOwnRowLevelConditionCompiled: (sourceName, condition) => {
       const struct = modelDef.contents[sourceName];
-      if (!struct || !isSourceDef5(struct))
+      if (!struct || !isSourceDef6(struct))
         return;
       validateSourceLineGateGivenUsage(sourceName, struct, condition.refSummary, condition.e, modelDef, (cause, detail) => {
         recordRowLevelGateRejected(cause);
@@ -15553,6 +15627,7 @@ async function compileNotebookModel(job, malloyConfig, modelPath) {
     finalSourceInfos = collected.sourceInfos;
     const extracted = extractSources(finalModelDef, finalGivens);
     finalSources = extracted.sources;
+    attachSuggestGivenNames(finalGivens, suggestGivenLookup(finalModelDef, (name) => extracted.sources.find((source) => source.name === name)?.authorize, new Set((finalGivens ?? []).map((given) => given.name))));
     finalFilterMap = extracted.filterMap;
     const finalQueryResult = extractQueries(finalModelDef);
     finalQueries = finalQueryResult.queries;
@@ -15573,7 +15648,7 @@ async function compileNotebookModel(job, malloyConfig, modelPath) {
       onRowLevelGateUnexpressible: authorizeWarningCollection.onRowLevelGateUnexpressible,
       onOwnRowLevelConditionCompiled: (sourceName, condition) => {
         const struct = finalCompiledModelDef.contents[sourceName];
-        if (!struct || !isSourceDef5(struct))
+        if (!struct || !isSourceDef6(struct))
           return;
         validateSourceLineGateGivenUsage(sourceName, struct, condition.refSummary, condition.e, finalCompiledModelDef, (cause, detail) => {
           recordRowLevelGateRejected(cause);