@mondaydotcomorg/atp-server 0.20.5 → 0.20.7

package/dist/index.js

@@ -6955,6 +6955,10 @@ function getRequestToolRules() {
   return asyncLocalStorage.getStore()?.toolRules;
 }
 __name(getRequestToolRules, "getRequestToolRules");
+function getRequestScope() {
+  return asyncLocalStorage.getStore();
+}
+__name(getRequestScope, "getRequestScope");
 function isGroupAllowed(groupName, rules) {
   if (!rules) return true;
   if (rules.allowOnlyApiGroups && rules.allowOnlyApiGroups.length > 0) {
@@ -10357,8 +10361,13 @@ var ExplorerService = class {
   }
   root;
   apiGroups;
+  apiGroupMap;
   constructor(apiGroups) {
     this.apiGroups = apiGroups;
+    this.apiGroupMap = new Map(apiGroups.map((group) => [
+      group.name,
+      group
+    ]));
     this.root = {
       type: "directory",
       name: "/",
@@ -10418,7 +10427,13 @@ var ExplorerService = class {
   buildTree(apiGroups) {
     for (const group of apiGroups) {
       if (!group.functions || group.functions.length === 0) continue;
-      const groupFolder = group.type === "graphql" ? this.ensureDirectory(this.root, group.name, group.description) : this.ensureDirectory(this.ensureDirectory(this.root, group.type), group.name, group.description);
+      let groupFolder;
+      if (group.type === "graphql") {
+        groupFolder = this.ensureDirectory(this.root, group.name, group.description);
+      } else {
+        groupFolder = this.ensureDirectory(this.ensureDirectory(this.root, group.type), group.name, group.description);
+      }
+      groupFolder.apiGroupName = group.name;
       for (const func of group.functions) {
         const segments = this.extractSegments(func, group);
         if (segments.length > 1) {
@@ -10580,11 +10595,18 @@ var ExplorerService = class {
         }
         return a.type === "directory" ? -1 : 1;
       });
-      return {
+      const result = {
         type: "directory",
         path: currentPath,
         items
       };
+      if (current.apiGroupName) {
+        const apiGroup = this.apiGroupMap.get(current.apiGroupName);
+        if (apiGroup?.documentation) {
+          result.documentation = apiGroup.documentation;
+        }
+      }
+      return result;
     } else {
       if (!current.functionDef) {
         return null;
@@ -11028,10 +11050,19 @@ __name(handleInit, "handleInit");
 // src/handlers/search.handler.ts
 async function handleSearch(ctx, searchEngine, config) {
   const searchOptions = ctx.body;
-  const results = await searchEngine.search(searchOptions, ctx.userId, ctx.auth, config.discovery.scopeFiltering);
-  return {
-    results
-  };
+  const { toolRules, ...cleanOptions } = searchOptions;
+  const executeSearch = /* @__PURE__ */ __name(async () => {
+    const results = await searchEngine.search(cleanOptions, ctx.userId, ctx.auth, config.discovery.scopeFiltering);
+    return {
+      results
+    };
+  }, "executeSearch");
+  if (toolRules && !getRequestScope()?.toolRules) {
+    return runInRequestScope({
+      toolRules
+    }, executeSearch);
+  }
+  return executeSearch();
 }
 __name(handleSearch, "handleSearch");
 async function handleSearchQuery(ctx, searchEngine, config) {
@@ -11049,11 +11080,20 @@ __name(handleSearchQuery, "handleSearchQuery");
 async function handleExplore(ctx, explorerService) {
   const body = ctx.body;
   const path = body.path || "/";
-  const result = explorerService.explore(path);
-  if (!result) {
-    ctx.throw(404, `Path not found: ${path}`);
+  const { toolRules } = body;
+  const executeExplore = /* @__PURE__ */ __name(() => {
+    const result = explorerService.explore(path);
+    if (!result) {
+      ctx.throw(404, `Path not found: ${path}`);
+    }
+    return result;
+  }, "executeExplore");
+  if (toolRules && !getRequestScope()?.toolRules) {
+    return runInRequestScope({
+      toolRules
+    }, executeExplore);
   }
-  return result;
+  return executeExplore();
 }
 __name(handleExplore, "handleExplore");
 async function emitProvenanceTokens(result, clientId, executionId, provenanceMode, cacheProvider, logger, maxTokens = 5e3, tokenTTL = 3600, provenanceSnapshot) {
@@ -11293,7 +11333,8 @@ async function handleExecute(ctx, executor, stateManager, config, auditSink, ses
       method: ctx.method
     },
     onToolCall,
-    eventCallback: requestConfig.eventCallback
+    eventCallback: requestConfig.eventCallback,
+    toolRules: requestConfig.toolRules
   };
   let hintMap;
   const prelimExecutionId = crypto.randomUUID();