agent-swarm-kit 1.0.124 → 1.0.126

package/build/index.cjs

@@ -4387,6 +4387,7 @@ var ClientSession = /** @class */ (function () {
 
 var METHOD_NAME_BAN_CLIENT = "PolicyUtils.banClient";
 var METHOD_NAME_UNBAN_CLIENT = "PolicyUtils.unbanClient";
+var METHOD_NAME_HAS_BAN = "PolicyUtils.hasBan";
 /**
  * NoopPolicy class implements the IPolicy interface with no-op methods.
  */
@@ -4400,6 +4401,15 @@ var NoopPolicy = /** @class */ (function () {
         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
             swarm$1.loggerService.debug("NoopPolicy CTOR swarmName=".concat(swarmName));
     }
+    /**
+     * Check if has ban in any policy
+     * @returns {Promise<boolean>}
+     */
+    NoopPolicy.prototype.hasBan = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy hasBan swarmName=".concat(this.swarmName));
+        return Promise.resolve(false);
+    };
     /**
      * Gets the ban message.
      * @returns {Promise<string>} The ban message.
@@ -4465,6 +4475,56 @@ var MergePolicy = /** @class */ (function () {
                 policies: policies,
             });
     }
+    /**
+     * Check if has ban in any policy
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<boolean>}
+     */
+    MergePolicy.prototype.hasBan = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy, e_1_1;
+            var e_1, _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy hasBan swarmName=".concat(swarmName), {
+                                clientId: clientId,
+                            });
+                        _d.label = 1;
+                    case 1:
+                        _d.trys.push([1, 6, 7, 8]);
+                        _a = __values(this.policies), _b = _a.next();
+                        _d.label = 2;
+                    case 2:
+                        if (!!_b.done) return [3 /*break*/, 5];
+                        policy = _b.value;
+                        return [4 /*yield*/, policy.hasBan(clientId, swarmName)];
+                    case 3:
+                        if (_d.sent()) {
+                            return [2 /*return*/, true];
+                        }
+                        _d.label = 4;
+                    case 4:
+                        _b = _a.next();
+                        return [3 /*break*/, 2];
+                    case 5: return [3 /*break*/, 8];
+                    case 6:
+                        e_1_1 = _d.sent();
+                        e_1 = { error: e_1_1 };
+                        return [3 /*break*/, 8];
+                    case 7:
+                        try {
+                            if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                        }
+                        finally { if (e_1) throw e_1.error; }
+                        return [7 /*endfinally*/];
+                    case 8: return [2 /*return*/, false];
+                }
+            });
+        });
+    };
     /**
      * Gets the ban message.
      * @param {SessionId} clientId - The client ID.
@@ -4500,8 +4560,8 @@ var MergePolicy = /** @class */ (function () {
      */
     MergePolicy.prototype.validateInput = function (incoming, clientId, swarmName) {
         return __awaiter(this, void 0, void 0, function () {
-            var _a, _b, policy, e_1_1;
-            var e_1, _c;
+            var _a, _b, policy, e_2_1;
+            var e_2, _c;
             return __generator(this, function (_d) {
                 switch (_d.label) {
                     case 0:
@@ -4529,14 +4589,14 @@ var MergePolicy = /** @class */ (function () {
                         return [3 /*break*/, 2];
                     case 5: return [3 /*break*/, 8];
                     case 6:
-                        e_1_1 = _d.sent();
-                        e_1 = { error: e_1_1 };
+                        e_2_1 = _d.sent();
+                        e_2 = { error: e_2_1 };
                         return [3 /*break*/, 8];
                     case 7:
                         try {
                             if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
                         }
-                        finally { if (e_1) throw e_1.error; }
+                        finally { if (e_2) throw e_2.error; }
                         return [7 /*endfinally*/];
                     case 8:
                         this._targetPolicy = null;
@@ -4554,8 +4614,8 @@ var MergePolicy = /** @class */ (function () {
      */
     MergePolicy.prototype.validateOutput = function (outgoing, clientId, swarmName) {
         return __awaiter(this, void 0, void 0, function () {
-            var _a, _b, policy, e_2_1;
-            var e_2, _c;
+            var _a, _b, policy, e_3_1;
+            var e_3, _c;
             return __generator(this, function (_d) {
                 switch (_d.label) {
                     case 0:
@@ -4583,14 +4643,14 @@ var MergePolicy = /** @class */ (function () {
                         return [3 /*break*/, 2];
                     case 5: return [3 /*break*/, 8];
                     case 6:
-                        e_2_1 = _d.sent();
-                        e_2 = { error: e_2_1 };
+                        e_3_1 = _d.sent();
+                        e_3 = { error: e_3_1 };
                         return [3 /*break*/, 8];
                     case 7:
                         try {
                             if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
                         }
-                        finally { if (e_2) throw e_2.error; }
+                        finally { if (e_3) throw e_3.error; }
                         return [7 /*endfinally*/];
                     case 8:
                         this._targetPolicy = null;
@@ -4717,6 +4777,28 @@ var PolicyUtils = /** @class */ (function () {
                 }
             });
         }); });
+        /**
+         * Check if client is banned
+         * @param {Object} payload - The payload containing clientId, swarmName, and policyName.
+         * @param {string} payload.clientId - The client ID.
+         * @param {SwarmName} payload.swarmName - The name of the swarm.
+         * @param {PolicyName} payload.policyName - The name of the policy.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_HAS_BAN, payload);
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_HAS_BAN);
+                        swarm$1.swarmValidationService.validate(payload.swarmName, METHOD_NAME_HAS_BAN);
+                        swarm$1.policyValidationService.validate(payload.policyName, METHOD_NAME_HAS_BAN);
+                        return [4 /*yield*/, swarm$1.policyPublicService.hasBan(payload.swarmName, METHOD_NAME_HAS_BAN, payload.clientId, payload.policyName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); });
     }
     return PolicyUtils;
 }());
@@ -6952,6 +7034,24 @@ var SwarmValidationService = /** @class */ (function () {
             }
             return swarm.agentList;
         };
+        /**
+         * Retrieves the list of ban policies for a given swarm.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {string[]} The list of policy names.
+         * @throws Will throw an error if the swarm is not found.
+         */
+        this.getPolicyList = function (swarmName) {
+            var _a;
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("swarmValidationService getAgentList", {
+                    swarmName: swarmName,
+                });
+            var swarm = _this._swarmMap.get(swarmName);
+            if (!swarm) {
+                throw new Error("agent-swarm swarm ".concat(swarmName, " not found"));
+            }
+            return (_a = swarm.policies) !== null && _a !== void 0 ? _a : [];
+        };
         /**
          * Retrieves the list of swarms
          * @returns {string[]} The list of swarm names
@@ -10368,6 +10468,7 @@ var PerfService = /** @class */ (function () {
         this.agentValidationService = inject(TYPES.agentValidationService);
         this.statePublicService = inject(TYPES.statePublicService);
         this.swarmPublicService = inject(TYPES.swarmPublicService);
+        this.policyPublicService = inject(TYPES.policyPublicService);
         this.stateConnectionService = inject(TYPES.stateConnectionService);
         this.executionScheduleMap = new Map();
         this.executionOutputLenMap = new Map();
@@ -10382,7 +10483,7 @@ var PerfService = /** @class */ (function () {
          * @returns {Promise<Record<string, unknown>>} A promise that resolves to an object containing the aggregated state of the client.
          */
         this.computeClientState = function (clientId) { return __awaiter(_this, void 0, void 0, function () {
-            var swarmName, agentName, result, stateFetchSet_1;
+            var swarmName, agentName, policyBans, result, stateFetchSet_1;
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -10395,11 +10496,29 @@ var PerfService = /** @class */ (function () {
                         return [4 /*yield*/, this.swarmPublicService.getAgentName(METHOD_NAME_COMPUTE_STATE, clientId, swarmName)];
                     case 1:
                         agentName = _a.sent();
+                        return [4 /*yield*/, Promise.all(this.swarmValidationService
+                                .getPolicyList(swarmName)
+                                .map(function (policyName) { return __awaiter(_this, void 0, void 0, function () {
+                                var _a;
+                                return __generator(this, function (_b) {
+                                    switch (_b.label) {
+                                        case 0:
+                                            _a = [policyName];
+                                            return [4 /*yield*/, this.policyPublicService.hasBan(swarmName, METHOD_NAME_COMPUTE_STATE, clientId, policyName)];
+                                        case 1: return [2 /*return*/, _a.concat([
+                                                _b.sent()
+                                            ])];
+                                    }
+                                });
+                            }); }))];
+                    case 2:
+                        policyBans = _a.sent();
                         result = {
                             swarmStatus: {
                                 swarmName: swarmName,
                                 agentName: agentName,
-                            }
+                            },
+                            policyBans: Object.fromEntries(policyBans),
                         };
                         stateFetchSet_1 = new Set();
                         return [4 /*yield*/, Promise.all(this.swarmValidationService
@@ -10432,7 +10551,7 @@ var PerfService = /** @class */ (function () {
                                     }
                                 });
                             }); }))];
-                    case 2:
+                    case 3:
                         _a.sent();
                         return [2 /*return*/, result];
                 }
@@ -10855,6 +10974,46 @@ var PolicyPublicService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.policyConnectionService = inject(TYPES.policyConnectionService);
+        /**
+         * Check if has ban message
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = function (swarmName, methodName, clientId, policyName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("policyPublicService hasBan", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        return [4 /*yield*/, MethodContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.policyConnectionService.hasBan(clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Retrieves the ban message for a client in a specific swarm.
          * @param {SwarmName} swarmName - The name of the swarm.
@@ -11085,6 +11244,35 @@ var ClientPolicy = /** @class */ (function () {
             this.params.callbacks.onInit(params.policyName);
         }
     }
+    /**
+     * Check if client is banned
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<boolean>}
+     */
+    ClientPolicy.prototype.hasBan = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b;
+            return __generator(this, function (_c) {
+                switch (_c.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " hasBan"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 2];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 1:
+                        _a._banSet = new (_b.apply(Set, [void 0, _c.sent()]))();
+                        _c.label = 2;
+                    case 2: return [2 /*return*/, this._banSet.has(clientId)];
+                }
+            });
+        });
+    };
     /**
      * Gets the ban message for a client.
      * @param {SessionId} clientId - The client ID.
@@ -11388,6 +11576,26 @@ var PolicyConnectionService = /** @class */ (function () {
             var schema = _this.policySchemaService.get(policyName);
             return new ClientPolicy(__assign({ policyName: policyName, bus: _this.busService, logger: _this.loggerService }, schema));
         });
+        /**
+         * Check if got ban flag
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("policyConnectionService hasBan", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).hasBan(clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Retrieves the ban message for a client in a swarm.
          * @param {SessionId} clientId - The ID of the client.

package/build/index.mjs

@@ -4385,6 +4385,7 @@ var ClientSession = /** @class */ (function () {
 
 var METHOD_NAME_BAN_CLIENT = "PolicyUtils.banClient";
 var METHOD_NAME_UNBAN_CLIENT = "PolicyUtils.unbanClient";
+var METHOD_NAME_HAS_BAN = "PolicyUtils.hasBan";
 /**
  * NoopPolicy class implements the IPolicy interface with no-op methods.
  */
@@ -4398,6 +4399,15 @@ var NoopPolicy = /** @class */ (function () {
         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
             swarm$1.loggerService.debug("NoopPolicy CTOR swarmName=".concat(swarmName));
     }
+    /**
+     * Check if has ban in any policy
+     * @returns {Promise<boolean>}
+     */
+    NoopPolicy.prototype.hasBan = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy hasBan swarmName=".concat(this.swarmName));
+        return Promise.resolve(false);
+    };
     /**
      * Gets the ban message.
      * @returns {Promise<string>} The ban message.
@@ -4463,6 +4473,56 @@ var MergePolicy = /** @class */ (function () {
                 policies: policies,
             });
     }
+    /**
+     * Check if has ban in any policy
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<boolean>}
+     */
+    MergePolicy.prototype.hasBan = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy, e_1_1;
+            var e_1, _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy hasBan swarmName=".concat(swarmName), {
+                                clientId: clientId,
+                            });
+                        _d.label = 1;
+                    case 1:
+                        _d.trys.push([1, 6, 7, 8]);
+                        _a = __values(this.policies), _b = _a.next();
+                        _d.label = 2;
+                    case 2:
+                        if (!!_b.done) return [3 /*break*/, 5];
+                        policy = _b.value;
+                        return [4 /*yield*/, policy.hasBan(clientId, swarmName)];
+                    case 3:
+                        if (_d.sent()) {
+                            return [2 /*return*/, true];
+                        }
+                        _d.label = 4;
+                    case 4:
+                        _b = _a.next();
+                        return [3 /*break*/, 2];
+                    case 5: return [3 /*break*/, 8];
+                    case 6:
+                        e_1_1 = _d.sent();
+                        e_1 = { error: e_1_1 };
+                        return [3 /*break*/, 8];
+                    case 7:
+                        try {
+                            if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                        }
+                        finally { if (e_1) throw e_1.error; }
+                        return [7 /*endfinally*/];
+                    case 8: return [2 /*return*/, false];
+                }
+            });
+        });
+    };
     /**
      * Gets the ban message.
      * @param {SessionId} clientId - The client ID.
@@ -4498,8 +4558,8 @@ var MergePolicy = /** @class */ (function () {
      */
     MergePolicy.prototype.validateInput = function (incoming, clientId, swarmName) {
         return __awaiter(this, void 0, void 0, function () {
-            var _a, _b, policy, e_1_1;
-            var e_1, _c;
+            var _a, _b, policy, e_2_1;
+            var e_2, _c;
             return __generator(this, function (_d) {
                 switch (_d.label) {
                     case 0:
@@ -4527,14 +4587,14 @@ var MergePolicy = /** @class */ (function () {
                         return [3 /*break*/, 2];
                     case 5: return [3 /*break*/, 8];
                     case 6:
-                        e_1_1 = _d.sent();
-                        e_1 = { error: e_1_1 };
+                        e_2_1 = _d.sent();
+                        e_2 = { error: e_2_1 };
                         return [3 /*break*/, 8];
                     case 7:
                         try {
                             if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
                         }
-                        finally { if (e_1) throw e_1.error; }
+                        finally { if (e_2) throw e_2.error; }
                         return [7 /*endfinally*/];
                     case 8:
                         this._targetPolicy = null;
@@ -4552,8 +4612,8 @@ var MergePolicy = /** @class */ (function () {
      */
     MergePolicy.prototype.validateOutput = function (outgoing, clientId, swarmName) {
         return __awaiter(this, void 0, void 0, function () {
-            var _a, _b, policy, e_2_1;
-            var e_2, _c;
+            var _a, _b, policy, e_3_1;
+            var e_3, _c;
             return __generator(this, function (_d) {
                 switch (_d.label) {
                     case 0:
@@ -4581,14 +4641,14 @@ var MergePolicy = /** @class */ (function () {
                         return [3 /*break*/, 2];
                     case 5: return [3 /*break*/, 8];
                     case 6:
-                        e_2_1 = _d.sent();
-                        e_2 = { error: e_2_1 };
+                        e_3_1 = _d.sent();
+                        e_3 = { error: e_3_1 };
                         return [3 /*break*/, 8];
                     case 7:
                         try {
                             if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
                         }
-                        finally { if (e_2) throw e_2.error; }
+                        finally { if (e_3) throw e_3.error; }
                         return [7 /*endfinally*/];
                     case 8:
                         this._targetPolicy = null;
@@ -4715,6 +4775,28 @@ var PolicyUtils = /** @class */ (function () {
                 }
             });
         }); });
+        /**
+         * Check if client is banned
+         * @param {Object} payload - The payload containing clientId, swarmName, and policyName.
+         * @param {string} payload.clientId - The client ID.
+         * @param {SwarmName} payload.swarmName - The name of the swarm.
+         * @param {PolicyName} payload.policyName - The name of the policy.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_HAS_BAN, payload);
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_HAS_BAN);
+                        swarm$1.swarmValidationService.validate(payload.swarmName, METHOD_NAME_HAS_BAN);
+                        swarm$1.policyValidationService.validate(payload.policyName, METHOD_NAME_HAS_BAN);
+                        return [4 /*yield*/, swarm$1.policyPublicService.hasBan(payload.swarmName, METHOD_NAME_HAS_BAN, payload.clientId, payload.policyName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); });
     }
     return PolicyUtils;
 }());
@@ -6950,6 +7032,24 @@ var SwarmValidationService = /** @class */ (function () {
             }
             return swarm.agentList;
         };
+        /**
+         * Retrieves the list of ban policies for a given swarm.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {string[]} The list of policy names.
+         * @throws Will throw an error if the swarm is not found.
+         */
+        this.getPolicyList = function (swarmName) {
+            var _a;
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("swarmValidationService getAgentList", {
+                    swarmName: swarmName,
+                });
+            var swarm = _this._swarmMap.get(swarmName);
+            if (!swarm) {
+                throw new Error("agent-swarm swarm ".concat(swarmName, " not found"));
+            }
+            return (_a = swarm.policies) !== null && _a !== void 0 ? _a : [];
+        };
         /**
          * Retrieves the list of swarms
          * @returns {string[]} The list of swarm names
@@ -10366,6 +10466,7 @@ var PerfService = /** @class */ (function () {
         this.agentValidationService = inject(TYPES.agentValidationService);
         this.statePublicService = inject(TYPES.statePublicService);
         this.swarmPublicService = inject(TYPES.swarmPublicService);
+        this.policyPublicService = inject(TYPES.policyPublicService);
         this.stateConnectionService = inject(TYPES.stateConnectionService);
         this.executionScheduleMap = new Map();
         this.executionOutputLenMap = new Map();
@@ -10380,7 +10481,7 @@ var PerfService = /** @class */ (function () {
          * @returns {Promise<Record<string, unknown>>} A promise that resolves to an object containing the aggregated state of the client.
          */
         this.computeClientState = function (clientId) { return __awaiter(_this, void 0, void 0, function () {
-            var swarmName, agentName, result, stateFetchSet_1;
+            var swarmName, agentName, policyBans, result, stateFetchSet_1;
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -10393,11 +10494,29 @@ var PerfService = /** @class */ (function () {
                         return [4 /*yield*/, this.swarmPublicService.getAgentName(METHOD_NAME_COMPUTE_STATE, clientId, swarmName)];
                     case 1:
                         agentName = _a.sent();
+                        return [4 /*yield*/, Promise.all(this.swarmValidationService
+                                .getPolicyList(swarmName)
+                                .map(function (policyName) { return __awaiter(_this, void 0, void 0, function () {
+                                var _a;
+                                return __generator(this, function (_b) {
+                                    switch (_b.label) {
+                                        case 0:
+                                            _a = [policyName];
+                                            return [4 /*yield*/, this.policyPublicService.hasBan(swarmName, METHOD_NAME_COMPUTE_STATE, clientId, policyName)];
+                                        case 1: return [2 /*return*/, _a.concat([
+                                                _b.sent()
+                                            ])];
+                                    }
+                                });
+                            }); }))];
+                    case 2:
+                        policyBans = _a.sent();
                         result = {
                             swarmStatus: {
                                 swarmName: swarmName,
                                 agentName: agentName,
-                            }
+                            },
+                            policyBans: Object.fromEntries(policyBans),
                         };
                         stateFetchSet_1 = new Set();
                         return [4 /*yield*/, Promise.all(this.swarmValidationService
@@ -10430,7 +10549,7 @@ var PerfService = /** @class */ (function () {
                                     }
                                 });
                             }); }))];
-                    case 2:
+                    case 3:
                         _a.sent();
                         return [2 /*return*/, result];
                 }
@@ -10853,6 +10972,46 @@ var PolicyPublicService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.policyConnectionService = inject(TYPES.policyConnectionService);
+        /**
+         * Check if has ban message
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = function (swarmName, methodName, clientId, policyName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("policyPublicService hasBan", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        return [4 /*yield*/, MethodContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.policyConnectionService.hasBan(clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Retrieves the ban message for a client in a specific swarm.
          * @param {SwarmName} swarmName - The name of the swarm.
@@ -11083,6 +11242,35 @@ var ClientPolicy = /** @class */ (function () {
             this.params.callbacks.onInit(params.policyName);
         }
     }
+    /**
+     * Check if client is banned
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<boolean>}
+     */
+    ClientPolicy.prototype.hasBan = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b;
+            return __generator(this, function (_c) {
+                switch (_c.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " hasBan"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 2];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 1:
+                        _a._banSet = new (_b.apply(Set, [void 0, _c.sent()]))();
+                        _c.label = 2;
+                    case 2: return [2 /*return*/, this._banSet.has(clientId)];
+                }
+            });
+        });
+    };
     /**
      * Gets the ban message for a client.
      * @param {SessionId} clientId - The client ID.
@@ -11386,6 +11574,26 @@ var PolicyConnectionService = /** @class */ (function () {
             var schema = _this.policySchemaService.get(policyName);
             return new ClientPolicy(__assign({ policyName: policyName, bus: _this.busService, logger: _this.loggerService }, schema));
         });
+        /**
+         * Check if got ban flag
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<boolean>}
+         */
+        this.hasBan = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("policyConnectionService hasBan", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).hasBan(clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Retrieves the ban message for a client in a swarm.
          * @param {SessionId} clientId - The ID of the client.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.0.124",
+  "version": "1.0.126",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",
@@ -12,7 +12,7 @@
     "url": "http://paypal.me/tripolskypetr"
   },
   "license": "MIT",
-  "homepage": "https://react-declarative-playground.github.io",
+  "homepage": "https://agent-swarm.github.io",
   "keywords": [
     "NVIDIA NIM",
     "OpenAI",
@@ -43,6 +43,7 @@
     "build:docs": "rimraf docs && mkdir docs && node ./scripts/dts-docs.cjs ./types.d.ts ./docs",
     "docs:gpt": "npm run build && node ./scripts/gpt-docs.mjs",
     "docs:uml": "npm run build && node ./scripts/uml.mjs",
+    "docs:www": "typedoc",
     "repl": "dotenv -e .env -- npm run build && node -e \"import('./scripts/repl.mjs')\" --interactive"
   },
   "main": "build/index.cjs",
@@ -67,7 +68,8 @@
     "rollup-plugin-dts": "6.1.1",
     "rollup-plugin-peer-deps-external": "2.2.4",
     "tslib": "2.7.0",
-    "worker-testbed": "1.0.10"
+    "worker-testbed": "1.0.10",
+    "typedoc": "0.27.9"
   },
   "peerDependencies": {
     "typescript": "^5.0.0"

package/types.d.ts

@@ -463,6 +463,12 @@ interface IPolicyCallbacks {
  * Interface for a policy.
  */
 interface IPolicy {
+    /**
+     * Check if got banhammer flag
+     * @param clientId - The session ID of the client.
+     * @param swarmName - The name of the swarm.
+     */
+    hasBan(clientId: SessionId, swarmName: SwarmName): Promise<boolean>;
     /**
      * Gets the ban message for a client.
      * @param clientId - The session ID of the client.
@@ -2958,6 +2964,13 @@ declare class SwarmValidationService {
      * @throws Will throw an error if the swarm is not found.
      */
     getAgentList: (swarmName: SwarmName) => string[];
+    /**
+     * Retrieves the list of ban policies for a given swarm.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {string[]} The list of policy names.
+     * @throws Will throw an error if the swarm is not found.
+     */
+    getPolicyList: (swarmName: SwarmName) => string[];
     /**
      * Retrieves the list of swarms
      * @returns {string[]} The list of swarm names
@@ -3882,6 +3895,7 @@ declare class PerfService {
     private readonly agentValidationService;
     private readonly statePublicService;
     private readonly swarmPublicService;
+    private readonly policyPublicService;
     private readonly stateConnectionService;
     private executionScheduleMap;
     private executionOutputLenMap;
@@ -4050,6 +4064,13 @@ declare class ClientPolicy implements IPolicy {
      * @param {IPolicyParams} params - The policy parameters.
      */
     constructor(params: IPolicyParams);
+    /**
+     * Check if client is banned
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<boolean>}
+     */
+    hasBan(clientId: SessionId, swarmName: SwarmName): Promise<boolean>;
     /**
      * Gets the ban message for a client.
      * @param {SessionId} clientId - The client ID.
@@ -4104,6 +4125,13 @@ declare class PolicyConnectionService implements IPolicy {
      * @returns {ClientPolicy} The client policy.
      */
     getPolicy: ((policyName: PolicyName) => ClientPolicy) & functools_kit.IClearableMemoize<string> & functools_kit.IControlMemoize<string, ClientPolicy>;
+    /**
+     * Check if got ban flag
+     * @param {SessionId} clientId - The ID of the client.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<boolean>}
+     */
+    hasBan: (clientId: SessionId, swarmName: SwarmName) => Promise<boolean>;
     /**
      * Retrieves the ban message for a client in a swarm.
      * @param {SessionId} clientId - The ID of the client.
@@ -4157,6 +4185,15 @@ type TPolicyConnectionService = {
 declare class PolicyPublicService implements TPolicyConnectionService {
     private readonly loggerService;
     private readonly policyConnectionService;
+    /**
+     * Check if has ban message
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @param {string} methodName - The name of the method.
+     * @param {string} clientId - The ID of the client.
+     * @param {PolicyName} policyName - The name of the policy.
+     * @returns {Promise<boolean>}
+     */
+    hasBan: (swarmName: SwarmName, methodName: string, clientId: string, policyName: PolicyName) => Promise<boolean>;
     /**
      * Retrieves the ban message for a client in a specific swarm.
      * @param {SwarmName} swarmName - The name of the swarm.
@@ -5297,6 +5334,19 @@ declare class PolicyUtils {
         swarmName: SwarmName;
         policyName: PolicyName;
     }) => Promise<void>;
+    /**
+     * Check if client is banned
+     * @param {Object} payload - The payload containing clientId, swarmName, and policyName.
+     * @param {string} payload.clientId - The client ID.
+     * @param {SwarmName} payload.swarmName - The name of the swarm.
+     * @param {PolicyName} payload.policyName - The name of the policy.
+     * @returns {Promise<boolean>}
+     */
+    hasBan: (payload: {
+        clientId: string;
+        swarmName: SwarmName;
+        policyName: PolicyName;
+    }) => Promise<boolean>;
 }
 /**
  * An instance of PolicyUtils.
@@ -5614,4 +5664,4 @@ declare const Schema: SchemaUtils;
  */
 declare const beginContext: <T extends (...args: any[]) => any>(run: T) => ((...args: Parameters<T>) => ReturnType<T>);
 
-export { type EventSource, ExecutionContextService, History, HistoryAdapter, HistoryInstance, type IAgentSchema, type IAgentTool, type IBaseEvent, type IBusEvent, type IBusEventContext, type ICompletionArgs, type ICompletionSchema, type ICustomEvent, type IEmbeddingSchema, type IHistoryAdapter, type IHistoryInstance, type IHistoryInstanceCallbacks, type IIncomingMessage, type ILoggerAdapter, type ILoggerInstance, type ILoggerInstanceCallbacks, type IMakeConnectionConfig, type IMakeDisposeParams, type IModelMessage, type IOutgoingMessage, type ISessionConfig, type IStateSchema, type IStorageSchema, type ISwarmSchema, type ITool, type IToolCall, Logger, LoggerAdapter, LoggerInstance, MethodContextService, Policy, type ReceiveMessageFn, Schema, type SendMessageFn$1 as SendMessageFn, SharedState, SharedStorage, State, Storage, addAgent, addCompletion, addEmbedding, addPolicy, addState, addStorage, addSwarm, addTool, beginContext, cancelOutput, cancelOutputForce, changeToAgent, changeToDefaultAgent, changeToPrevAgent, commitAssistantMessage, commitAssistantMessageForce, commitFlush, commitFlushForce, commitStopTools, commitStopToolsForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, dumpAgent, dumpClientPerformance, dumpDocs, dumpPerfomance, dumpSwarm, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionContext, getSessionMode, getUserHistory, listenAgentEvent, listenAgentEventOnce, listenEvent, listenEventOnce, listenExecutionEvent, listenExecutionEventOnce, listenHistoryEvent, listenHistoryEventOnce, listenPolicyEvent, listenPolicyEventOnce, listenSessionEvent, listenSessionEventOnce, listenStateEvent, listenStateEventOnce, listenStorageEvent, listenStorageEventOnce, listenSwarmEvent, listenSwarmEventOnce, makeAutoDispose, makeConnection, runStateless, runStatelessForce, session, setConfig, swarm };
+export { type EventSource, ExecutionContextService, History, HistoryAdapter, HistoryInstance, type IAgentSchema, type IAgentTool, type IBaseEvent, type IBusEvent, type IBusEventContext, type ICompletionArgs, type ICompletionSchema, type ICustomEvent, type IEmbeddingSchema, type IHistoryAdapter, type IHistoryInstance, type IHistoryInstanceCallbacks, type IIncomingMessage, type ILoggerAdapter, type ILoggerInstance, type ILoggerInstanceCallbacks, type IMakeConnectionConfig, type IMakeDisposeParams, type IModelMessage, type IOutgoingMessage, type IPolicySchema, type ISessionConfig, type IStateSchema, type IStorageSchema, type ISwarmSchema, type ITool, type IToolCall, Logger, LoggerAdapter, LoggerInstance, MethodContextService, Policy, type ReceiveMessageFn, Schema, type SendMessageFn$1 as SendMessageFn, SharedState, SharedStorage, State, Storage, addAgent, addCompletion, addEmbedding, addPolicy, addState, addStorage, addSwarm, addTool, beginContext, cancelOutput, cancelOutputForce, changeToAgent, changeToDefaultAgent, changeToPrevAgent, commitAssistantMessage, commitAssistantMessageForce, commitFlush, commitFlushForce, commitStopTools, commitStopToolsForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, dumpAgent, dumpClientPerformance, dumpDocs, dumpPerfomance, dumpSwarm, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionContext, getSessionMode, getUserHistory, listenAgentEvent, listenAgentEventOnce, listenEvent, listenEventOnce, listenExecutionEvent, listenExecutionEventOnce, listenHistoryEvent, listenHistoryEventOnce, listenPolicyEvent, listenPolicyEventOnce, listenSessionEvent, listenSessionEventOnce, listenStateEvent, listenStateEventOnce, listenStorageEvent, listenStorageEventOnce, listenSwarmEvent, listenSwarmEventOnce, makeAutoDispose, makeConnection, runStateless, runStatelessForce, session, setConfig, swarm };