@airmcp-dev/core 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +17 -0
- package/dist/config/defaults.d.ts +3 -0
- package/dist/config/defaults.d.ts.map +1 -0
- package/dist/config/defaults.js +11 -0
- package/dist/config/defaults.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +5 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/loader.d.ts +4 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +59 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/context/index.d.ts +3 -0
- package/dist/context/index.d.ts.map +1 -0
- package/dist/context/index.js +5 -0
- package/dist/context/index.js.map +1 -0
- package/dist/context/request-context.d.ts +4 -0
- package/dist/context/request-context.d.ts.map +1 -0
- package/dist/context/request-context.js +14 -0
- package/dist/context/request-context.js.map +1 -0
- package/dist/context/server-context.d.ts +12 -0
- package/dist/context/server-context.d.ts.map +1 -0
- package/dist/context/server-context.js +25 -0
- package/dist/context/server-context.js.map +1 -0
- package/dist/index.d.ts +47 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +62 -0
- package/dist/index.js.map +1 -0
- package/dist/middleware/after-hook.d.ts +4 -0
- package/dist/middleware/after-hook.d.ts.map +1 -0
- package/dist/middleware/after-hook.js +18 -0
- package/dist/middleware/after-hook.js.map +1 -0
- package/dist/middleware/before-hook.d.ts +4 -0
- package/dist/middleware/before-hook.d.ts.map +1 -0
- package/dist/middleware/before-hook.js +40 -0
- package/dist/middleware/before-hook.js.map +1 -0
- package/dist/middleware/chain.d.ts +11 -0
- package/dist/middleware/chain.d.ts.map +1 -0
- package/dist/middleware/chain.js +84 -0
- package/dist/middleware/chain.js.map +1 -0
- package/dist/middleware/error-handler.d.ts +27 -0
- package/dist/middleware/error-handler.d.ts.map +1 -0
- package/dist/middleware/error-handler.js +63 -0
- package/dist/middleware/error-handler.js.map +1 -0
- package/dist/middleware/index.d.ts +7 -0
- package/dist/middleware/index.d.ts.map +1 -0
- package/dist/middleware/index.js +9 -0
- package/dist/middleware/index.js.map +1 -0
- package/dist/middleware/meter-middleware.d.ts +19 -0
- package/dist/middleware/meter-middleware.d.ts.map +1 -0
- package/dist/middleware/meter-middleware.js +139 -0
- package/dist/middleware/meter-middleware.js.map +1 -0
- package/dist/middleware/shield-middleware.d.ts +16 -0
- package/dist/middleware/shield-middleware.d.ts.map +1 -0
- package/dist/middleware/shield-middleware.js +163 -0
- package/dist/middleware/shield-middleware.js.map +1 -0
- package/dist/plugin/builtin/auth.d.ts +16 -0
- package/dist/plugin/builtin/auth.d.ts.map +1 -0
- package/dist/plugin/builtin/auth.js +60 -0
- package/dist/plugin/builtin/auth.js.map +1 -0
- package/dist/plugin/builtin/cache.d.ts +12 -0
- package/dist/plugin/builtin/cache.d.ts.map +1 -0
- package/dist/plugin/builtin/cache.js +65 -0
- package/dist/plugin/builtin/cache.js.map +1 -0
- package/dist/plugin/builtin/circuit-breaker.d.ts +12 -0
- package/dist/plugin/builtin/circuit-breaker.d.ts.map +1 -0
- package/dist/plugin/builtin/circuit-breaker.js +76 -0
- package/dist/plugin/builtin/circuit-breaker.js.map +1 -0
- package/dist/plugin/builtin/cors.d.ts +12 -0
- package/dist/plugin/builtin/cors.d.ts.map +1 -0
- package/dist/plugin/builtin/cors.js +29 -0
- package/dist/plugin/builtin/cors.js.map +1 -0
- package/dist/plugin/builtin/dedup.d.ts +8 -0
- package/dist/plugin/builtin/dedup.d.ts.map +1 -0
- package/dist/plugin/builtin/dedup.js +64 -0
- package/dist/plugin/builtin/dedup.js.map +1 -0
- package/dist/plugin/builtin/dryrun.d.ts +12 -0
- package/dist/plugin/builtin/dryrun.d.ts.map +1 -0
- package/dist/plugin/builtin/dryrun.js +61 -0
- package/dist/plugin/builtin/dryrun.js.map +1 -0
- package/dist/plugin/builtin/fallback.d.ts +5 -0
- package/dist/plugin/builtin/fallback.d.ts.map +1 -0
- package/dist/plugin/builtin/fallback.js +42 -0
- package/dist/plugin/builtin/fallback.js.map +1 -0
- package/dist/plugin/builtin/i18n.d.ts +12 -0
- package/dist/plugin/builtin/i18n.d.ts.map +1 -0
- package/dist/plugin/builtin/i18n.js +51 -0
- package/dist/plugin/builtin/i18n.js.map +1 -0
- package/dist/plugin/builtin/logger-json.d.ts +12 -0
- package/dist/plugin/builtin/logger-json.d.ts.map +1 -0
- package/dist/plugin/builtin/logger-json.js +52 -0
- package/dist/plugin/builtin/logger-json.js.map +1 -0
- package/dist/plugin/builtin/logger.d.ts +3 -0
- package/dist/plugin/builtin/logger.d.ts.map +1 -0
- package/dist/plugin/builtin/logger.js +22 -0
- package/dist/plugin/builtin/logger.js.map +1 -0
- package/dist/plugin/builtin/metrics.d.ts +14 -0
- package/dist/plugin/builtin/metrics.d.ts.map +1 -0
- package/dist/plugin/builtin/metrics.js +56 -0
- package/dist/plugin/builtin/metrics.js.map +1 -0
- package/dist/plugin/builtin/queue.d.ts +12 -0
- package/dist/plugin/builtin/queue.d.ts.map +1 -0
- package/dist/plugin/builtin/queue.js +98 -0
- package/dist/plugin/builtin/queue.js.map +1 -0
- package/dist/plugin/builtin/ratelimit-per-user.d.ts +12 -0
- package/dist/plugin/builtin/ratelimit-per-user.d.ts.map +1 -0
- package/dist/plugin/builtin/ratelimit-per-user.js +45 -0
- package/dist/plugin/builtin/ratelimit-per-user.js.map +1 -0
- package/dist/plugin/builtin/retry.d.ts +12 -0
- package/dist/plugin/builtin/retry.d.ts.map +1 -0
- package/dist/plugin/builtin/retry.js +51 -0
- package/dist/plugin/builtin/retry.js.map +1 -0
- package/dist/plugin/builtin/sanitizer.d.ts +14 -0
- package/dist/plugin/builtin/sanitizer.d.ts.map +1 -0
- package/dist/plugin/builtin/sanitizer.js +55 -0
- package/dist/plugin/builtin/sanitizer.js.map +1 -0
- package/dist/plugin/builtin/timeout.d.ts +3 -0
- package/dist/plugin/builtin/timeout.d.ts.map +1 -0
- package/dist/plugin/builtin/timeout.js +30 -0
- package/dist/plugin/builtin/timeout.js.map +1 -0
- package/dist/plugin/builtin/transform.d.ts +11 -0
- package/dist/plugin/builtin/transform.d.ts.map +1 -0
- package/dist/plugin/builtin/transform.js +51 -0
- package/dist/plugin/builtin/transform.js.map +1 -0
- package/dist/plugin/builtin/validator.d.ts +9 -0
- package/dist/plugin/builtin/validator.d.ts.map +1 -0
- package/dist/plugin/builtin/validator.js +43 -0
- package/dist/plugin/builtin/validator.js.map +1 -0
- package/dist/plugin/builtin/webhook.d.ts +16 -0
- package/dist/plugin/builtin/webhook.d.ts.map +1 -0
- package/dist/plugin/builtin/webhook.js +78 -0
- package/dist/plugin/builtin/webhook.js.map +1 -0
- package/dist/plugin/index.d.ts +23 -0
- package/dist/plugin/index.d.ts.map +1 -0
- package/dist/plugin/index.js +34 -0
- package/dist/plugin/index.js.map +1 -0
- package/dist/plugin/plugin-hook.d.ts +11 -0
- package/dist/plugin/plugin-hook.d.ts.map +1 -0
- package/dist/plugin/plugin-hook.js +39 -0
- package/dist/plugin/plugin-hook.js.map +1 -0
- package/dist/plugin/plugin-manager.d.ts +22 -0
- package/dist/plugin/plugin-manager.d.ts.map +1 -0
- package/dist/plugin/plugin-manager.js +71 -0
- package/dist/plugin/plugin-manager.js.map +1 -0
- package/dist/plugin/plugin-schema.d.ts +6 -0
- package/dist/plugin/plugin-schema.d.ts.map +1 -0
- package/dist/plugin/plugin-schema.js +18 -0
- package/dist/plugin/plugin-schema.js.map +1 -0
- package/dist/prompt/define-prompt.d.ts +3 -0
- package/dist/prompt/define-prompt.d.ts.map +1 -0
- package/dist/prompt/define-prompt.js +17 -0
- package/dist/prompt/define-prompt.js.map +1 -0
- package/dist/prompt/index.d.ts +2 -0
- package/dist/prompt/index.d.ts.map +1 -0
- package/dist/prompt/index.js +4 -0
- package/dist/prompt/index.js.map +1 -0
- package/dist/resource/define-resource.d.ts +3 -0
- package/dist/resource/define-resource.d.ts.map +1 -0
- package/dist/resource/define-resource.js +19 -0
- package/dist/resource/define-resource.js.map +1 -0
- package/dist/resource/index.d.ts +3 -0
- package/dist/resource/index.d.ts.map +1 -0
- package/dist/resource/index.js +5 -0
- package/dist/resource/index.js.map +1 -0
- package/dist/resource/resource-template.d.ts +5 -0
- package/dist/resource/resource-template.d.ts.map +1 -0
- package/dist/resource/resource-template.js +17 -0
- package/dist/resource/resource-template.js.map +1 -0
- package/dist/server/define-server.d.ts +14 -0
- package/dist/server/define-server.d.ts.map +1 -0
- package/dist/server/define-server.js +175 -0
- package/dist/server/define-server.js.map +1 -0
- package/dist/server/index.d.ts +4 -0
- package/dist/server/index.d.ts.map +1 -0
- package/dist/server/index.js +6 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/lifecycle.d.ts +3 -0
- package/dist/server/lifecycle.d.ts.map +1 -0
- package/dist/server/lifecycle.js +32 -0
- package/dist/server/lifecycle.js.map +1 -0
- package/dist/server/server-runner.d.ts +42 -0
- package/dist/server/server-runner.d.ts.map +1 -0
- package/dist/server/server-runner.js +255 -0
- package/dist/server/server-runner.js.map +1 -0
- package/dist/storage/file-store.d.ts +25 -0
- package/dist/storage/file-store.d.ts.map +1 -0
- package/dist/storage/file-store.js +177 -0
- package/dist/storage/file-store.js.map +1 -0
- package/dist/storage/index.d.ts +4 -0
- package/dist/storage/index.d.ts.map +1 -0
- package/dist/storage/index.js +6 -0
- package/dist/storage/index.js.map +1 -0
- package/dist/storage/memory-store.d.ts +18 -0
- package/dist/storage/memory-store.d.ts.map +1 -0
- package/dist/storage/memory-store.js +80 -0
- package/dist/storage/memory-store.js.map +1 -0
- package/dist/storage/storage-adapter.d.ts +4 -0
- package/dist/storage/storage-adapter.d.ts.map +1 -0
- package/dist/storage/storage-adapter.js +32 -0
- package/dist/storage/storage-adapter.js.map +1 -0
- package/dist/telemetry/index.d.ts +3 -0
- package/dist/telemetry/index.d.ts.map +1 -0
- package/dist/telemetry/index.js +4 -0
- package/dist/telemetry/index.js.map +1 -0
- package/dist/telemetry/telemetry.d.ts +56 -0
- package/dist/telemetry/telemetry.d.ts.map +1 -0
- package/dist/telemetry/telemetry.js +95 -0
- package/dist/telemetry/telemetry.js.map +1 -0
- package/dist/tool/define-tool.d.ts +23 -0
- package/dist/tool/define-tool.d.ts.map +1 -0
- package/dist/tool/define-tool.js +28 -0
- package/dist/tool/define-tool.js.map +1 -0
- package/dist/tool/index.d.ts +5 -0
- package/dist/tool/index.d.ts.map +1 -0
- package/dist/tool/index.js +6 -0
- package/dist/tool/index.js.map +1 -0
- package/dist/tool/tool-result.d.ts +10 -0
- package/dist/tool/tool-result.d.ts.map +1 -0
- package/dist/tool/tool-result.js +34 -0
- package/dist/tool/tool-result.js.map +1 -0
- package/dist/tool/tool-schema.d.ts +7 -0
- package/dist/tool/tool-schema.d.ts.map +1 -0
- package/dist/tool/tool-schema.js +97 -0
- package/dist/tool/tool-schema.js.map +1 -0
- package/dist/transport/auto-detect.d.ts +4 -0
- package/dist/transport/auto-detect.d.ts.map +1 -0
- package/dist/transport/auto-detect.js +21 -0
- package/dist/transport/auto-detect.js.map +1 -0
- package/dist/transport/http-adapter.d.ts +4 -0
- package/dist/transport/http-adapter.d.ts.map +1 -0
- package/dist/transport/http-adapter.js +10 -0
- package/dist/transport/http-adapter.js.map +1 -0
- package/dist/transport/index.d.ts +5 -0
- package/dist/transport/index.d.ts.map +1 -0
- package/dist/transport/index.js +7 -0
- package/dist/transport/index.js.map +1 -0
- package/dist/transport/sse-adapter.d.ts +3 -0
- package/dist/transport/sse-adapter.d.ts.map +1 -0
- package/dist/transport/sse-adapter.js +8 -0
- package/dist/transport/sse-adapter.js.map +1 -0
- package/dist/transport/stdio-adapter.d.ts +3 -0
- package/dist/transport/stdio-adapter.d.ts.map +1 -0
- package/dist/transport/stdio-adapter.js +8 -0
- package/dist/transport/stdio-adapter.js.map +1 -0
- package/dist/types/config.d.ts +101 -0
- package/dist/types/config.d.ts.map +1 -0
- package/dist/types/config.js +5 -0
- package/dist/types/config.js.map +1 -0
- package/dist/types/index.d.ts +12 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +4 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/middleware.d.ts +48 -0
- package/dist/types/middleware.d.ts.map +1 -0
- package/dist/types/middleware.js +4 -0
- package/dist/types/middleware.js.map +1 -0
- package/dist/types/plugin-manifest.d.ts +59 -0
- package/dist/types/plugin-manifest.d.ts.map +1 -0
- package/dist/types/plugin-manifest.js +9 -0
- package/dist/types/plugin-manifest.js.map +1 -0
- package/dist/types/plugin.d.ts +50 -0
- package/dist/types/plugin.d.ts.map +1 -0
- package/dist/types/plugin.js +4 -0
- package/dist/types/plugin.js.map +1 -0
- package/dist/types/prompt.d.ts +21 -0
- package/dist/types/prompt.d.ts.map +1 -0
- package/dist/types/prompt.js +4 -0
- package/dist/types/prompt.js.map +1 -0
- package/dist/types/resource.d.ts +25 -0
- package/dist/types/resource.d.ts.map +1 -0
- package/dist/types/resource.js +4 -0
- package/dist/types/resource.js.map +1 -0
- package/dist/types/server.d.ts +54 -0
- package/dist/types/server.d.ts.map +1 -0
- package/dist/types/server.js +4 -0
- package/dist/types/server.js.map +1 -0
- package/dist/types/storage.d.ts +40 -0
- package/dist/types/storage.d.ts.map +1 -0
- package/dist/types/storage.js +4 -0
- package/dist/types/storage.js.map +1 -0
- package/dist/types/tool.d.ts +54 -0
- package/dist/types/tool.d.ts.map +1 -0
- package/dist/types/tool.js +4 -0
- package/dist/types/tool.js.map +1 -0
- package/dist/types/transport.d.ts +12 -0
- package/dist/types/transport.d.ts.map +1 -0
- package/dist/types/transport.js +4 -0
- package/dist/types/transport.js.map +1 -0
- package/package.json +23 -0
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/metrics.ts
|
|
3
|
+
// 내장 메트릭 플러그인 — 호출 수, 에러율, 지연시간 자동 수집
|
|
4
|
+
const metrics = new Map();
|
|
5
|
+
export function getMetrics() {
|
|
6
|
+
const result = {};
|
|
7
|
+
for (const [name, m] of metrics) {
|
|
8
|
+
result[name] = {
|
|
9
|
+
...m,
|
|
10
|
+
avgDuration: m.calls > 0 ? Math.round(m.totalDuration / m.calls) : 0,
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
return result;
|
|
14
|
+
}
|
|
15
|
+
export function resetMetrics() {
|
|
16
|
+
metrics.clear();
|
|
17
|
+
}
|
|
18
|
+
export function builtinMetricsPlugin() {
|
|
19
|
+
const mw = {
|
|
20
|
+
name: 'air:builtin-metrics',
|
|
21
|
+
after: async (ctx) => {
|
|
22
|
+
const m = metrics.get(ctx.tool.name) || {
|
|
23
|
+
calls: 0,
|
|
24
|
+
errors: 0,
|
|
25
|
+
totalDuration: 0,
|
|
26
|
+
lastCalledAt: 0,
|
|
27
|
+
};
|
|
28
|
+
m.calls++;
|
|
29
|
+
m.totalDuration += ctx.duration;
|
|
30
|
+
m.lastCalledAt = Date.now();
|
|
31
|
+
metrics.set(ctx.tool.name, m);
|
|
32
|
+
},
|
|
33
|
+
onError: async (ctx, _error) => {
|
|
34
|
+
const m = metrics.get(ctx.tool.name) || {
|
|
35
|
+
calls: 0,
|
|
36
|
+
errors: 0,
|
|
37
|
+
totalDuration: 0,
|
|
38
|
+
lastCalledAt: 0,
|
|
39
|
+
};
|
|
40
|
+
m.calls++;
|
|
41
|
+
m.errors++;
|
|
42
|
+
m.lastCalledAt = Date.now();
|
|
43
|
+
metrics.set(ctx.tool.name, m);
|
|
44
|
+
return undefined; // 에러를 먹지 않음
|
|
45
|
+
},
|
|
46
|
+
};
|
|
47
|
+
return {
|
|
48
|
+
meta: {
|
|
49
|
+
name: 'air:builtin-metrics',
|
|
50
|
+
version: '0.1.0',
|
|
51
|
+
description: 'Built-in metrics collection',
|
|
52
|
+
},
|
|
53
|
+
middleware: [mw],
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=metrics.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../../../src/plugin/builtin/metrics.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,+CAA+C;AAC/C,sCAAsC;AAYtC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAC;AAE9C,MAAM,UAAU,UAAU;IACxB,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,OAAO,EAAE,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,GAAG;YACb,GAAG,CAAC;YACJ,WAAW,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;SACrE,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,OAAO,CAAC,KAAK,EAAE,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,oBAAoB;IAClC,MAAM,EAAE,GAAkB;QACxB,IAAI,EAAE,qBAAqB;QAC3B,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;gBACtC,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,CAAC;aAChB,CAAC;YACF,CAAC,CAAC,KAAK,EAAE,CAAC;YACV,CAAC,CAAC,aAAa,IAAI,GAAG,CAAC,QAAQ,CAAC;YAChC,CAAC,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChC,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE;YAC7B,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;gBACtC,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,CAAC;aAChB,CAAC;YACF,CAAC,CAAC,KAAK,EAAE,CAAC;YACV,CAAC,CAAC,MAAM,EAAE,CAAC;YACX,CAAC,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC9B,OAAO,SAAS,CAAC,CAAC,YAAY;QAChC,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE;YACJ,IAAI,EAAE,qBAAqB;YAC3B,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,6BAA6B;SAC3C;QACD,UAAU,EAAE,CAAC,EAAE,CAAC;KACjB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
interface QueueOptions {
|
|
3
|
+
/** 도구별 동시 실행 수 (기본: '*': 10) */
|
|
4
|
+
concurrency?: Record<string, number>;
|
|
5
|
+
/** 대기열 최대 크기 (기본: 100) */
|
|
6
|
+
maxQueueSize?: number;
|
|
7
|
+
/** 대기 타임아웃 ms (기본: 30000) */
|
|
8
|
+
queueTimeoutMs?: number;
|
|
9
|
+
}
|
|
10
|
+
export declare function queuePlugin(options?: QueueOptions): AirPlugin;
|
|
11
|
+
export {};
|
|
12
|
+
//# sourceMappingURL=queue.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"queue.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/queue.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,UAAU,YAAY;IACpB,gCAAgC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,0BAA0B;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,6BAA6B;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAQD,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,YAAY,GAAG,SAAS,CA0F7D"}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/queue.ts
|
|
3
|
+
//
|
|
4
|
+
// 동시성 제어 큐 플러그인.
|
|
5
|
+
// 도구별로 동시 실행 수를 제한. 초과 시 대기열에 넣고 순차 처리.
|
|
6
|
+
// DB 커넥션 풀, 외부 API 호출 제한 등에 사용.
|
|
7
|
+
//
|
|
8
|
+
// @example
|
|
9
|
+
// defineServer({
|
|
10
|
+
// use: [queuePlugin({
|
|
11
|
+
// concurrency: { 'db_query': 3, '*': 10 },
|
|
12
|
+
// })],
|
|
13
|
+
// });
|
|
14
|
+
export function queuePlugin(options) {
|
|
15
|
+
const concurrencyMap = options?.concurrency ?? { '*': 10 };
|
|
16
|
+
const maxQueueSize = options?.maxQueueSize ?? 100;
|
|
17
|
+
const queueTimeoutMs = options?.queueTimeoutMs ?? 30_000;
|
|
18
|
+
const active = new Map();
|
|
19
|
+
const queues = new Map();
|
|
20
|
+
function getLimit(toolName) {
|
|
21
|
+
return concurrencyMap[toolName] ?? concurrencyMap['*'] ?? 10;
|
|
22
|
+
}
|
|
23
|
+
function getActive(toolName) {
|
|
24
|
+
return active.get(toolName) ?? 0;
|
|
25
|
+
}
|
|
26
|
+
function acquire(toolName) {
|
|
27
|
+
const limit = getLimit(toolName);
|
|
28
|
+
const current = getActive(toolName);
|
|
29
|
+
if (current < limit) {
|
|
30
|
+
active.set(toolName, current + 1);
|
|
31
|
+
return Promise.resolve();
|
|
32
|
+
}
|
|
33
|
+
// 대기열에 추가
|
|
34
|
+
const queue = queues.get(toolName) ?? [];
|
|
35
|
+
if (queue.length >= maxQueueSize) {
|
|
36
|
+
return Promise.reject(new Error(`[Queue] Queue full for "${toolName}" (max: ${maxQueueSize})`));
|
|
37
|
+
}
|
|
38
|
+
return new Promise((resolve, reject) => {
|
|
39
|
+
const entry = { resolve, reject, addedAt: Date.now() };
|
|
40
|
+
queue.push(entry);
|
|
41
|
+
queues.set(toolName, queue);
|
|
42
|
+
// 타임아웃
|
|
43
|
+
setTimeout(() => {
|
|
44
|
+
const idx = queue.indexOf(entry);
|
|
45
|
+
if (idx >= 0) {
|
|
46
|
+
queue.splice(idx, 1);
|
|
47
|
+
reject(new Error(`[Queue] Timeout waiting for "${toolName}" (${queueTimeoutMs}ms)`));
|
|
48
|
+
}
|
|
49
|
+
}, queueTimeoutMs);
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
function release(toolName) {
|
|
53
|
+
const current = getActive(toolName);
|
|
54
|
+
if (current <= 0)
|
|
55
|
+
return;
|
|
56
|
+
const queue = queues.get(toolName);
|
|
57
|
+
if (queue && queue.length > 0) {
|
|
58
|
+
// 대기 중인 요청을 깨움
|
|
59
|
+
const next = queue.shift();
|
|
60
|
+
next.resolve();
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
active.set(toolName, current - 1);
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
const middleware = {
|
|
67
|
+
name: 'air:queue',
|
|
68
|
+
before: async (ctx) => {
|
|
69
|
+
try {
|
|
70
|
+
await acquire(ctx.tool.name);
|
|
71
|
+
ctx.meta._queued = true;
|
|
72
|
+
}
|
|
73
|
+
catch (err) {
|
|
74
|
+
return {
|
|
75
|
+
abort: true,
|
|
76
|
+
abortResponse: {
|
|
77
|
+
content: [{ type: 'text', text: err.message }],
|
|
78
|
+
isError: true,
|
|
79
|
+
},
|
|
80
|
+
};
|
|
81
|
+
}
|
|
82
|
+
},
|
|
83
|
+
after: async (ctx) => {
|
|
84
|
+
if (ctx.meta._queued)
|
|
85
|
+
release(ctx.tool.name);
|
|
86
|
+
},
|
|
87
|
+
onError: async (ctx) => {
|
|
88
|
+
if (ctx.meta._queued)
|
|
89
|
+
release(ctx.tool.name);
|
|
90
|
+
return undefined;
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
return {
|
|
94
|
+
meta: { name: 'air:queue', version: '1.0.0' },
|
|
95
|
+
middleware: [middleware],
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
//# sourceMappingURL=queue.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"queue.js","sourceRoot":"","sources":["../../../src/plugin/builtin/queue.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,6CAA6C;AAC7C,EAAE;AACF,iBAAiB;AACjB,wCAAwC;AACxC,gCAAgC;AAChC,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,0BAA0B;AAC1B,iDAAiD;AACjD,WAAW;AACX,QAAQ;AAoBR,MAAM,UAAU,WAAW,CAAC,OAAsB;IAChD,MAAM,cAAc,GAAG,OAAO,EAAE,WAAW,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;IAC3D,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,GAAG,CAAC;IAClD,MAAM,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,MAAM,CAAC;IAEzD,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IAE/C,SAAS,QAAQ,CAAC,QAAgB;QAChC,OAAO,cAAc,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;IAC/D,CAAC;IAED,SAAS,SAAS,CAAC,QAAgB;QACjC,OAAO,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,SAAS,OAAO,CAAC,QAAgB;QAC/B,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;QAEpC,IAAI,OAAO,GAAG,KAAK,EAAE,CAAC;YACpB,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;YAClC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QAED,UAAU;QACV,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACzC,IAAI,KAAK,CAAC,MAAM,IAAI,YAAY,EAAE,CAAC;YACjC,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,2BAA2B,QAAQ,WAAW,YAAY,GAAG,CAAC,CAAC,CAAC;QAClG,CAAC;QAED,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,MAAM,KAAK,GAAe,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YACnE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClB,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAE5B,OAAO;YACP,UAAU,CAAC,GAAG,EAAE;gBACd,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjC,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC;oBACb,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;oBACrB,MAAM,CAAC,IAAI,KAAK,CAAC,gCAAgC,QAAQ,MAAM,cAAc,KAAK,CAAC,CAAC,CAAC;gBACvF,CAAC;YACH,CAAC,EAAE,cAAc,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,OAAO,CAAC,QAAgB;QAC/B,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,OAAO,IAAI,CAAC;YAAE,OAAO;QAEzB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,eAAe;YACf,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,WAAW;QACjB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7B,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,OAAO;oBACL,KAAK,EAAE,IAAI;oBACX,aAAa,EAAE;wBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;wBAC9C,OAAO,EAAE,IAAI;qBACd;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO;gBAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACrB,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO;gBAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7C,OAAO,SAAS,CAAC;QACnB,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;QAC7C,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
interface PerUserRateLimitOptions {
|
|
3
|
+
/** 윈도우 내 최대 호출 수 (기본: 10) */
|
|
4
|
+
maxCalls?: number;
|
|
5
|
+
/** 윈도우 크기 ms (기본: 60000) */
|
|
6
|
+
windowMs?: number;
|
|
7
|
+
/** 사용자 식별 파라미터 이름 (기본: '_userId') */
|
|
8
|
+
identifyBy?: string;
|
|
9
|
+
}
|
|
10
|
+
export declare function perUserRateLimitPlugin(options?: PerUserRateLimitOptions): AirPlugin;
|
|
11
|
+
export {};
|
|
12
|
+
//# sourceMappingURL=ratelimit-per-user.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ratelimit-per-user.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/ratelimit-per-user.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,UAAU,uBAAuB;IAC/B,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,4BAA4B;IAC5B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,qCAAqC;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,wBAAgB,sBAAsB,CAAC,OAAO,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAiCnF"}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/ratelimit-per-user.ts
|
|
3
|
+
//
|
|
4
|
+
// 사용자별 레이트 리밋 플러그인.
|
|
5
|
+
// shield의 글로벌 레이트 리밋과 달리, 사용자/세션별 개별 추적.
|
|
6
|
+
//
|
|
7
|
+
// @example
|
|
8
|
+
// defineServer({
|
|
9
|
+
// use: [perUserRateLimitPlugin({
|
|
10
|
+
// maxCalls: 10,
|
|
11
|
+
// windowMs: 60_000,
|
|
12
|
+
// identifyBy: '_userId',
|
|
13
|
+
// })],
|
|
14
|
+
// });
|
|
15
|
+
export function perUserRateLimitPlugin(options) {
|
|
16
|
+
const maxCalls = options?.maxCalls ?? 10;
|
|
17
|
+
const windowMs = options?.windowMs ?? 60_000;
|
|
18
|
+
const identifyBy = options?.identifyBy ?? '_userId';
|
|
19
|
+
const windows = new Map();
|
|
20
|
+
const middleware = {
|
|
21
|
+
name: 'air:per-user-ratelimit',
|
|
22
|
+
before: async (ctx) => {
|
|
23
|
+
const userId = ctx.params[identifyBy] || 'anonymous';
|
|
24
|
+
const now = Date.now();
|
|
25
|
+
const key = `${userId}:${ctx.tool.name}`;
|
|
26
|
+
const timestamps = (windows.get(key) || []).filter((t) => t > now - windowMs);
|
|
27
|
+
if (timestamps.length >= maxCalls) {
|
|
28
|
+
return {
|
|
29
|
+
abort: true,
|
|
30
|
+
abortResponse: {
|
|
31
|
+
content: [{ type: 'text', text: `[RateLimit] User "${userId}" exceeded ${maxCalls} calls per ${windowMs / 1000}s for "${ctx.tool.name}".` }],
|
|
32
|
+
isError: true,
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
timestamps.push(now);
|
|
37
|
+
windows.set(key, timestamps);
|
|
38
|
+
},
|
|
39
|
+
};
|
|
40
|
+
return {
|
|
41
|
+
meta: { name: 'air:per-user-ratelimit', version: '1.0.0' },
|
|
42
|
+
middleware: [middleware],
|
|
43
|
+
};
|
|
44
|
+
}
|
|
45
|
+
//# sourceMappingURL=ratelimit-per-user.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ratelimit-per-user.js","sourceRoot":"","sources":["../../../src/plugin/builtin/ratelimit-per-user.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,0DAA0D;AAC1D,EAAE;AACF,oBAAoB;AACpB,yCAAyC;AACzC,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,qCAAqC;AACrC,sBAAsB;AACtB,0BAA0B;AAC1B,+BAA+B;AAC/B,WAAW;AACX,QAAQ;AAcR,MAAM,UAAU,sBAAsB,CAAC,OAAiC;IACtE,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,MAAM,CAAC;IAC7C,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,SAAS,CAAC;IACpD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAoB,CAAC;IAE5C,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,wBAAwB;QAC9B,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,MAAM,GAAI,GAAG,CAAC,MAAM,CAAC,UAAU,CAAY,IAAI,WAAW,CAAC;YACjE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,GAAG,GAAG,GAAG,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACzC,MAAM,UAAU,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,GAAG,GAAG,QAAQ,CAAC,CAAC;YAE9E,IAAI,UAAU,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAClC,OAAO;oBACL,KAAK,EAAE,IAAI;oBACX,aAAa,EAAE;wBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,MAAM,cAAc,QAAQ,cAAc,QAAQ,GAAG,IAAI,UAAU,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;wBAC5I,OAAO,EAAE,IAAI;qBACd;iBACF,CAAC;YACJ,CAAC;YAED,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAC/B,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,OAAO,EAAE,OAAO,EAAE;QAC1D,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
interface RetryOptions {
|
|
3
|
+
/** 최대 재시도 횟수 (기본: 3) */
|
|
4
|
+
maxRetries?: number;
|
|
5
|
+
/** 초기 대기 시간 ms (기본: 200) */
|
|
6
|
+
delayMs?: number;
|
|
7
|
+
/** 재시도할 에러 패턴 (기본: 전부) */
|
|
8
|
+
retryOn?: (error: Error) => boolean;
|
|
9
|
+
}
|
|
10
|
+
export declare function retryPlugin(options?: RetryOptions): AirPlugin;
|
|
11
|
+
export {};
|
|
12
|
+
//# sourceMappingURL=retry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retry.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/retry.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,UAAU,YAAY;IACpB,wBAAwB;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4BAA4B;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,0BAA0B;IAC1B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC;CACrC;AAED,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,YAAY,GAAG,SAAS,CA4C7D"}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/retry.ts
|
|
3
|
+
//
|
|
4
|
+
// 도구 실행 재시도 플러그인.
|
|
5
|
+
// 핸들러에서 에러가 발생하면 지정 횟수만큼 재시도.
|
|
6
|
+
// 지수 백오프(exponential backoff) 적용.
|
|
7
|
+
//
|
|
8
|
+
// @example
|
|
9
|
+
// defineServer({
|
|
10
|
+
// use: [retryPlugin({ maxRetries: 3, delayMs: 200 })],
|
|
11
|
+
// });
|
|
12
|
+
export function retryPlugin(options) {
|
|
13
|
+
const maxRetries = options?.maxRetries ?? 3;
|
|
14
|
+
const baseDelay = options?.delayMs ?? 200;
|
|
15
|
+
const retryOn = options?.retryOn ?? (() => true);
|
|
16
|
+
const middleware = {
|
|
17
|
+
name: 'air:retry',
|
|
18
|
+
onError: async (ctx, error) => {
|
|
19
|
+
if (!retryOn(error))
|
|
20
|
+
return undefined;
|
|
21
|
+
const retryCount = (ctx.meta._retryCount || 0);
|
|
22
|
+
if (retryCount >= maxRetries)
|
|
23
|
+
return undefined; // 재시도 초과
|
|
24
|
+
ctx.meta._retryCount = retryCount + 1;
|
|
25
|
+
const delay = baseDelay * Math.pow(2, retryCount);
|
|
26
|
+
console.warn(`[air:retry] ${ctx.tool.name} failed (attempt ${retryCount + 1}/${maxRetries}), retrying in ${delay}ms...`);
|
|
27
|
+
await new Promise((r) => setTimeout(r, delay));
|
|
28
|
+
// 핸들러 재실행
|
|
29
|
+
try {
|
|
30
|
+
const result = await ctx.tool.handler(ctx.params, {
|
|
31
|
+
requestId: ctx.requestId,
|
|
32
|
+
serverName: ctx.serverName,
|
|
33
|
+
startedAt: ctx.startedAt,
|
|
34
|
+
state: {},
|
|
35
|
+
});
|
|
36
|
+
return result;
|
|
37
|
+
}
|
|
38
|
+
catch (retryError) {
|
|
39
|
+
if (ctx.meta._retryCount < maxRetries) {
|
|
40
|
+
return undefined; // 다음 onError 미들웨어에서 다시 시도
|
|
41
|
+
}
|
|
42
|
+
return undefined;
|
|
43
|
+
}
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
return {
|
|
47
|
+
meta: { name: 'air:retry', version: '1.0.0' },
|
|
48
|
+
middleware: [middleware],
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=retry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../../src/plugin/builtin/retry.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,6CAA6C;AAC7C,EAAE;AACF,kBAAkB;AAClB,8BAA8B;AAC9B,kCAAkC;AAClC,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,2DAA2D;AAC3D,QAAQ;AAcR,MAAM,UAAU,WAAW,CAAC,OAAsB;IAChD,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,CAAC;IAC5C,MAAM,SAAS,GAAG,OAAO,EAAE,OAAO,IAAI,GAAG,CAAC;IAC1C,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAEjD,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;YAC5B,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAC;YAEtC,MAAM,UAAU,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,CAAW,CAAC;YACzD,IAAI,UAAU,IAAI,UAAU;gBAAE,OAAO,SAAS,CAAC,CAAC,SAAS;YAEzD,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,UAAU,GAAG,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;YAElD,OAAO,CAAC,IAAI,CACV,eAAe,GAAG,CAAC,IAAI,CAAC,IAAI,oBAAoB,UAAU,GAAG,CAAC,IAAI,UAAU,kBAAkB,KAAK,OAAO,CAC3G,CAAC;YAEF,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;YAE/C,UAAU;YACV,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE;oBAChD,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,UAAU,EAAE,GAAG,CAAC,UAAU;oBAC1B,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,KAAK,EAAE,EAAE;iBACV,CAAC,CAAC;gBACH,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,GAAG,UAAU,EAAE,CAAC;oBACtC,OAAO,SAAS,CAAC,CAAC,0BAA0B;gBAC9C,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;QAC7C,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
interface SanitizerOptions {
|
|
3
|
+
/** HTML 태그 제거 (기본: true) */
|
|
4
|
+
stripHtml?: boolean;
|
|
5
|
+
/** 제어 문자 제거 (기본: true) */
|
|
6
|
+
stripControl?: boolean;
|
|
7
|
+
/** 최대 문자열 길이 (기본: 10000) */
|
|
8
|
+
maxStringLength?: number;
|
|
9
|
+
/** 적용 제외 도구 */
|
|
10
|
+
exclude?: string[];
|
|
11
|
+
}
|
|
12
|
+
export declare function sanitizerPlugin(options?: SanitizerOptions): AirPlugin;
|
|
13
|
+
export {};
|
|
14
|
+
//# sourceMappingURL=sanitizer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sanitizer.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/sanitizer.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,UAAU,gBAAgB;IACxB,4BAA4B;IAC5B,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,0BAA0B;IAC1B,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,4BAA4B;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,eAAe;IACf,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB;AAqBD,wBAAgB,eAAe,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAuBrE"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/sanitizer.ts
|
|
3
|
+
//
|
|
4
|
+
// 입력 데이터 정제 플러그인.
|
|
5
|
+
// XSS, HTML 태그, 제어 문자를 자동으로 제거.
|
|
6
|
+
//
|
|
7
|
+
// @example
|
|
8
|
+
// defineServer({
|
|
9
|
+
// use: [sanitizerPlugin()],
|
|
10
|
+
// });
|
|
11
|
+
function sanitizeValue(value, opts) {
|
|
12
|
+
if (typeof value === 'string') {
|
|
13
|
+
let s = value;
|
|
14
|
+
if (opts.stripHtml)
|
|
15
|
+
s = s.replace(/<[^>]*>/g, '');
|
|
16
|
+
if (opts.stripControl)
|
|
17
|
+
s = s.replace(/[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]/g, '');
|
|
18
|
+
if (s.length > opts.maxStringLength)
|
|
19
|
+
s = s.slice(0, opts.maxStringLength);
|
|
20
|
+
return s;
|
|
21
|
+
}
|
|
22
|
+
if (Array.isArray(value))
|
|
23
|
+
return value.map((v) => sanitizeValue(v, opts));
|
|
24
|
+
if (value && typeof value === 'object') {
|
|
25
|
+
const cleaned = {};
|
|
26
|
+
for (const [k, v] of Object.entries(value)) {
|
|
27
|
+
cleaned[k] = sanitizeValue(v, opts);
|
|
28
|
+
}
|
|
29
|
+
return cleaned;
|
|
30
|
+
}
|
|
31
|
+
return value;
|
|
32
|
+
}
|
|
33
|
+
export function sanitizerPlugin(options) {
|
|
34
|
+
const opts = {
|
|
35
|
+
stripHtml: options?.stripHtml !== false,
|
|
36
|
+
stripControl: options?.stripControl !== false,
|
|
37
|
+
maxStringLength: options?.maxStringLength ?? 10_000,
|
|
38
|
+
exclude: options?.exclude || [],
|
|
39
|
+
};
|
|
40
|
+
const excludeSet = new Set(opts.exclude);
|
|
41
|
+
const middleware = {
|
|
42
|
+
name: 'air:sanitizer',
|
|
43
|
+
before: async (ctx) => {
|
|
44
|
+
if (excludeSet.has(ctx.tool.name))
|
|
45
|
+
return;
|
|
46
|
+
const cleaned = sanitizeValue(ctx.params, opts);
|
|
47
|
+
return { params: cleaned };
|
|
48
|
+
},
|
|
49
|
+
};
|
|
50
|
+
return {
|
|
51
|
+
meta: { name: 'air:sanitizer', version: '1.0.0' },
|
|
52
|
+
middleware: [middleware],
|
|
53
|
+
};
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=sanitizer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sanitizer.js","sourceRoot":"","sources":["../../../src/plugin/builtin/sanitizer.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,iDAAiD;AACjD,EAAE;AACF,kBAAkB;AAClB,gCAAgC;AAChC,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,gCAAgC;AAChC,QAAQ;AAgBR,SAAS,aAAa,CAAC,KAAU,EAAE,IAAgC;IACjE,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,IAAI,CAAC,GAAG,KAAK,CAAC;QACd,IAAI,IAAI,CAAC,SAAS;YAAE,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QAClD,IAAI,IAAI,CAAC,YAAY;YAAE,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,mCAAmC,EAAE,EAAE,CAAC,CAAC;QAC9E,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe;YAAE,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1E,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IAC1E,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvC,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,OAA0B;IACxD,MAAM,IAAI,GAAG;QACX,SAAS,EAAE,OAAO,EAAE,SAAS,KAAK,KAAK;QACvC,YAAY,EAAE,OAAO,EAAE,YAAY,KAAK,KAAK;QAC7C,eAAe,EAAE,OAAO,EAAE,eAAe,IAAI,MAAM;QACnD,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;KAChC,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAEzC,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,eAAe;QACrB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAAE,OAAO;YAC1C,MAAM,OAAO,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAChD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE;QACjD,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"timeout.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/timeout.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,wBAAgB,aAAa,CAAC,SAAS,GAAE,MAAe,GAAG,SAAS,CAqBnE"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/timeout.ts
|
|
3
|
+
//
|
|
4
|
+
// 도구 실행 타임아웃 플러그인.
|
|
5
|
+
// 지정 시간 내에 핸들러가 완료되지 않으면 자동으로 에러를 반환.
|
|
6
|
+
//
|
|
7
|
+
// @example
|
|
8
|
+
// defineServer({
|
|
9
|
+
// use: [timeoutPlugin(5000)], // 5초 타임아웃
|
|
10
|
+
// });
|
|
11
|
+
export function timeoutPlugin(timeoutMs = 30_000) {
|
|
12
|
+
const middleware = {
|
|
13
|
+
name: 'air:timeout',
|
|
14
|
+
before: async (ctx) => {
|
|
15
|
+
ctx.meta._timeoutMs = timeoutMs;
|
|
16
|
+
ctx.meta._timeoutStart = Date.now();
|
|
17
|
+
},
|
|
18
|
+
after: async (ctx) => {
|
|
19
|
+
const elapsed = Date.now() - (ctx.meta._timeoutStart || ctx.startedAt);
|
|
20
|
+
if (elapsed > timeoutMs) {
|
|
21
|
+
console.warn(`[air:timeout] ${ctx.tool.name} took ${elapsed}ms (limit: ${timeoutMs}ms)`);
|
|
22
|
+
}
|
|
23
|
+
},
|
|
24
|
+
};
|
|
25
|
+
return {
|
|
26
|
+
meta: { name: 'air:timeout', version: '1.0.0' },
|
|
27
|
+
middleware: [middleware],
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=timeout.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"timeout.js","sourceRoot":"","sources":["../../../src/plugin/builtin/timeout.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,+CAA+C;AAC/C,EAAE;AACF,mBAAmB;AACnB,sCAAsC;AACtC,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,8CAA8C;AAC9C,QAAQ;AAKR,MAAM,UAAU,aAAa,CAAC,YAAoB,MAAM;IACtD,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,aAAa;QACnB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,GAAG,CAAC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtC,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;YACvE,IAAI,OAAO,GAAG,SAAS,EAAE,CAAC;gBACxB,OAAO,CAAC,IAAI,CACV,iBAAiB,GAAG,CAAC,IAAI,CAAC,IAAI,SAAS,OAAO,cAAc,SAAS,KAAK,CAC3E,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE;QAC/C,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
type TransformFn = (data: any) => any;
|
|
3
|
+
interface TransformOptions {
|
|
4
|
+
/** 입력 파라미터 변환 (도구명 → 변환 함수, '*'는 전체 적용) */
|
|
5
|
+
before?: Record<string, TransformFn>;
|
|
6
|
+
/** 출력 결과 변환 */
|
|
7
|
+
after?: Record<string, TransformFn>;
|
|
8
|
+
}
|
|
9
|
+
export declare function transformPlugin(options: TransformOptions): AirPlugin;
|
|
10
|
+
export {};
|
|
11
|
+
//# sourceMappingURL=transform.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transform.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/transform.ts"],"names":[],"mappings":"AAmBA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,KAAK,WAAW,GAAG,CAAC,IAAI,EAAE,GAAG,KAAK,GAAG,CAAC;AAEtC,UAAU,gBAAgB;IACxB,2CAA2C;IAC3C,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACrC,eAAe;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;CACrC;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,gBAAgB,GAAG,SAAS,CAoCpE"}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/transform.ts
|
|
3
|
+
//
|
|
4
|
+
// 도구 입출력 변환 플러그인.
|
|
5
|
+
// 파라미터 전처리, 응답 후처리를 선언적으로 등록.
|
|
6
|
+
//
|
|
7
|
+
// @example
|
|
8
|
+
// defineServer({
|
|
9
|
+
// use: [transformPlugin({
|
|
10
|
+
// before: {
|
|
11
|
+
// '*': (params) => ({ ...params, _caller: 'air' }),
|
|
12
|
+
// 'search': (params) => ({ ...params, query: params.query.trim() }),
|
|
13
|
+
// },
|
|
14
|
+
// after: {
|
|
15
|
+
// '*': (result) => ({ ...result, _processedAt: new Date().toISOString() }),
|
|
16
|
+
// },
|
|
17
|
+
// })],
|
|
18
|
+
// });
|
|
19
|
+
export function transformPlugin(options) {
|
|
20
|
+
const beforeMap = options.before || {};
|
|
21
|
+
const afterMap = options.after || {};
|
|
22
|
+
const middleware = {
|
|
23
|
+
name: 'air:transform',
|
|
24
|
+
before: async (ctx) => {
|
|
25
|
+
const toolName = ctx.tool.name;
|
|
26
|
+
// 글로벌 변환
|
|
27
|
+
if (beforeMap['*']) {
|
|
28
|
+
ctx.params = beforeMap['*'](ctx.params);
|
|
29
|
+
}
|
|
30
|
+
// 도구별 변환
|
|
31
|
+
if (beforeMap[toolName]) {
|
|
32
|
+
ctx.params = beforeMap[toolName](ctx.params);
|
|
33
|
+
}
|
|
34
|
+
return { params: ctx.params };
|
|
35
|
+
},
|
|
36
|
+
after: async (ctx) => {
|
|
37
|
+
const toolName = ctx.tool.name;
|
|
38
|
+
if (afterMap['*']) {
|
|
39
|
+
ctx.result = afterMap['*'](ctx.result);
|
|
40
|
+
}
|
|
41
|
+
if (afterMap[toolName]) {
|
|
42
|
+
ctx.result = afterMap[toolName](ctx.result);
|
|
43
|
+
}
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
return {
|
|
47
|
+
meta: { name: 'air:transform', version: '1.0.0' },
|
|
48
|
+
middleware: [middleware],
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=transform.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../../src/plugin/builtin/transform.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,iDAAiD;AACjD,EAAE;AACF,kBAAkB;AAClB,8BAA8B;AAC9B,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,8BAA8B;AAC9B,kBAAkB;AAClB,4DAA4D;AAC5D,6EAA6E;AAC7E,WAAW;AACX,iBAAiB;AACjB,oFAAoF;AACpF,WAAW;AACX,WAAW;AACX,QAAQ;AAcR,MAAM,UAAU,eAAe,CAAC,OAAyB;IACvD,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAErC,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,eAAe;QACrB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,QAAQ,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YAE/B,SAAS;YACT,IAAI,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnB,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC1C,CAAC;YACD,SAAS;YACT,IAAI,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACxB,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/C,CAAC;YAED,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;QAChC,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACnB,MAAM,QAAQ,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YAE/B,IAAI,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClB,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACzC,CAAC;YACD,IAAI,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvB,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE;QACjD,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
type ValidationRule = (params: Record<string, any>) => string | null;
|
|
3
|
+
interface ValidatorOptions {
|
|
4
|
+
/** 도구별 검증 규칙 */
|
|
5
|
+
rules: Record<string, ValidationRule>;
|
|
6
|
+
}
|
|
7
|
+
export declare function validatorPlugin(options: ValidatorOptions): AirPlugin;
|
|
8
|
+
export {};
|
|
9
|
+
//# sourceMappingURL=validator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"validator.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/validator.ts"],"names":[],"mappings":"AAmBA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,KAAK,cAAc,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,MAAM,GAAG,IAAI,CAAC;AAErE,UAAU,gBAAgB;IACxB,gBAAgB;IAChB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CACvC;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,gBAAgB,GAAG,SAAS,CAwBpE"}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
// Copyright 2026 CodePedia Labs. Licensed under Apache-2.0.
|
|
2
|
+
// @airmcp-dev/core — plugin/builtin/validator.ts
|
|
3
|
+
//
|
|
4
|
+
// 커스텀 검증 플러그인.
|
|
5
|
+
// zod 스키마 외에 비즈니스 로직 검증을 추가.
|
|
6
|
+
// 도구별로 검증 규칙을 등록할 수 있다.
|
|
7
|
+
//
|
|
8
|
+
// @example
|
|
9
|
+
// defineServer({
|
|
10
|
+
// use: [validatorPlugin({
|
|
11
|
+
// rules: {
|
|
12
|
+
// 'transfer': (params) => {
|
|
13
|
+
// if (params.amount > 10000) return 'Amount exceeds limit';
|
|
14
|
+
// return null; // OK
|
|
15
|
+
// },
|
|
16
|
+
// },
|
|
17
|
+
// })],
|
|
18
|
+
// });
|
|
19
|
+
export function validatorPlugin(options) {
|
|
20
|
+
const middleware = {
|
|
21
|
+
name: 'air:validator',
|
|
22
|
+
before: async (ctx) => {
|
|
23
|
+
const rule = options.rules[ctx.tool.name];
|
|
24
|
+
if (!rule)
|
|
25
|
+
return;
|
|
26
|
+
const error = rule(ctx.params);
|
|
27
|
+
if (error) {
|
|
28
|
+
return {
|
|
29
|
+
abort: true,
|
|
30
|
+
abortResponse: {
|
|
31
|
+
content: [{ type: 'text', text: `[Validator] ${ctx.tool.name}: ${error}` }],
|
|
32
|
+
isError: true,
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
},
|
|
37
|
+
};
|
|
38
|
+
return {
|
|
39
|
+
meta: { name: 'air:validator', version: '1.0.0' },
|
|
40
|
+
middleware: [middleware],
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
//# sourceMappingURL=validator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"validator.js","sourceRoot":"","sources":["../../../src/plugin/builtin/validator.ts"],"names":[],"mappings":"AAAA,4DAA4D;AAC5D,iDAAiD;AACjD,EAAE;AACF,eAAe;AACf,6BAA6B;AAC7B,wBAAwB;AACxB,EAAE;AACF,WAAW;AACX,mBAAmB;AACnB,8BAA8B;AAC9B,iBAAiB;AACjB,oCAAoC;AACpC,sEAAsE;AACtE,+BAA+B;AAC/B,aAAa;AACb,WAAW;AACX,WAAW;AACX,QAAQ;AAYR,MAAM,UAAU,eAAe,CAAC,OAAyB;IACvD,MAAM,UAAU,GAAkB;QAChC,IAAI,EAAE,eAAe;QACrB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACpB,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI;gBAAE,OAAO;YAElB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/B,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO;oBACL,KAAK,EAAE,IAAI;oBACX,aAAa,EAAE;wBACb,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK,EAAE,EAAE,CAAC;wBAC3E,OAAO,EAAE,IAAI;qBACd;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO;QACL,IAAI,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE;QACjD,UAAU,EAAE,CAAC,UAAU,CAAC;KACzB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { AirPlugin } from '../../types/plugin.js';
|
|
2
|
+
interface WebhookOptions {
|
|
3
|
+
/** 웹훅 URL */
|
|
4
|
+
url: string;
|
|
5
|
+
/** 전송할 이벤트 타입 (기본: ['tool.call']) */
|
|
6
|
+
events?: Array<'tool.call' | 'tool.error' | 'tool.slow'>;
|
|
7
|
+
/** 느린 호출 기준 ms (기본: 5000) */
|
|
8
|
+
slowThresholdMs?: number;
|
|
9
|
+
/** 요청 헤더 */
|
|
10
|
+
headers?: Record<string, string>;
|
|
11
|
+
/** 배치 크기 (기본: 1, 즉시 전송) */
|
|
12
|
+
batchSize?: number;
|
|
13
|
+
}
|
|
14
|
+
export declare function webhookPlugin(options: WebhookOptions): AirPlugin;
|
|
15
|
+
export {};
|
|
16
|
+
//# sourceMappingURL=webhook.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"webhook.d.ts","sourceRoot":"","sources":["../../../src/plugin/builtin/webhook.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGvD,UAAU,cAAc;IACtB,aAAa;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,qCAAqC;IACrC,MAAM,CAAC,EAAE,KAAK,CAAC,WAAW,GAAG,YAAY,GAAG,WAAW,CAAC,CAAC;IACzD,6BAA6B;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY;IACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,2BAA2B;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,wBAAgB,aAAa,CAAC,OAAO,EAAE,cAAc,GAAG,SAAS,CAiEhE"}
|