@kya-os/mcp-i-core 1.2.3-canary.7 → 1.3.0-canary.clientinfo.20251126003544
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/.turbo/turbo-build.log +4 -0
- package/.turbo/turbo-test$colon$coverage.log +4239 -0
- package/.turbo/turbo-test.log +2973 -0
- package/COMPLIANCE_IMPROVEMENT_REPORT.md +483 -0
- package/Composer 3.md +615 -0
- package/GPT-5.md +1169 -0
- package/OPUS-plan.md +352 -0
- package/PHASE_3_AND_4.1_SUMMARY.md +585 -0
- package/PHASE_3_SUMMARY.md +317 -0
- package/PHASE_4.1.3_SUMMARY.md +428 -0
- package/PHASE_4.1_COMPLETE.md +525 -0
- package/PHASE_4_USER_DID_IDENTITY_LINKING_PLAN.md +1240 -0
- package/SCHEMA_COMPLIANCE_REPORT.md +275 -0
- package/TEST_PLAN.md +571 -0
- package/coverage/coverage-final.json +57 -0
- package/dist/__tests__/utils/mock-providers.d.ts +1 -2
- package/dist/__tests__/utils/mock-providers.d.ts.map +1 -1
- package/dist/__tests__/utils/mock-providers.js.map +1 -1
- package/dist/cache/oauth-config-cache.d.ts +69 -0
- package/dist/cache/oauth-config-cache.d.ts.map +1 -0
- package/dist/cache/oauth-config-cache.js +76 -0
- package/dist/cache/oauth-config-cache.js.map +1 -0
- package/dist/identity/idp-token-resolver.d.ts +53 -0
- package/dist/identity/idp-token-resolver.d.ts.map +1 -0
- package/dist/identity/idp-token-resolver.js +108 -0
- package/dist/identity/idp-token-resolver.js.map +1 -0
- package/dist/identity/idp-token-storage.interface.d.ts +42 -0
- package/dist/identity/idp-token-storage.interface.d.ts.map +1 -0
- package/dist/identity/idp-token-storage.interface.js +12 -0
- package/dist/identity/idp-token-storage.interface.js.map +1 -0
- package/dist/identity/user-did-manager.d.ts +39 -1
- package/dist/identity/user-did-manager.d.ts.map +1 -1
- package/dist/identity/user-did-manager.js +69 -3
- package/dist/identity/user-did-manager.js.map +1 -1
- package/dist/index.d.ts +24 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +43 -1
- package/dist/index.js.map +1 -1
- package/dist/runtime/audit-logger.d.ts +37 -0
- package/dist/runtime/audit-logger.d.ts.map +1 -0
- package/dist/runtime/audit-logger.js +9 -0
- package/dist/runtime/audit-logger.js.map +1 -0
- package/dist/runtime/base.d.ts +58 -2
- package/dist/runtime/base.d.ts.map +1 -1
- package/dist/runtime/base.js +266 -11
- package/dist/runtime/base.js.map +1 -1
- package/dist/services/access-control.service.d.ts.map +1 -1
- package/dist/services/access-control.service.js +200 -35
- package/dist/services/access-control.service.js.map +1 -1
- package/dist/services/authorization/authorization-registry.d.ts +29 -0
- package/dist/services/authorization/authorization-registry.d.ts.map +1 -0
- package/dist/services/authorization/authorization-registry.js +57 -0
- package/dist/services/authorization/authorization-registry.js.map +1 -0
- package/dist/services/authorization/types.d.ts +53 -0
- package/dist/services/authorization/types.d.ts.map +1 -0
- package/dist/services/authorization/types.js +10 -0
- package/dist/services/authorization/types.js.map +1 -0
- package/dist/services/batch-delegation.service.d.ts +53 -0
- package/dist/services/batch-delegation.service.d.ts.map +1 -0
- package/dist/services/batch-delegation.service.js +95 -0
- package/dist/services/batch-delegation.service.js.map +1 -0
- package/dist/services/index.d.ts +2 -0
- package/dist/services/index.d.ts.map +1 -1
- package/dist/services/index.js +4 -1
- package/dist/services/index.js.map +1 -1
- package/dist/services/oauth-config.service.d.ts +53 -0
- package/dist/services/oauth-config.service.d.ts.map +1 -0
- package/dist/services/oauth-config.service.js +117 -0
- package/dist/services/oauth-config.service.js.map +1 -0
- package/dist/services/oauth-provider-registry.d.ts +77 -0
- package/dist/services/oauth-provider-registry.d.ts.map +1 -0
- package/dist/services/oauth-provider-registry.js +112 -0
- package/dist/services/oauth-provider-registry.js.map +1 -0
- package/dist/services/oauth-service.d.ts +77 -0
- package/dist/services/oauth-service.d.ts.map +1 -0
- package/dist/services/oauth-service.js +348 -0
- package/dist/services/oauth-service.js.map +1 -0
- package/dist/services/oauth-token-retrieval.service.d.ts +49 -0
- package/dist/services/oauth-token-retrieval.service.d.ts.map +1 -0
- package/dist/services/oauth-token-retrieval.service.js +150 -0
- package/dist/services/oauth-token-retrieval.service.js.map +1 -0
- package/dist/services/provider-resolver.d.ts +48 -0
- package/dist/services/provider-resolver.d.ts.map +1 -0
- package/dist/services/provider-resolver.js +120 -0
- package/dist/services/provider-resolver.js.map +1 -0
- package/dist/services/provider-validator.d.ts +55 -0
- package/dist/services/provider-validator.d.ts.map +1 -0
- package/dist/services/provider-validator.js +135 -0
- package/dist/services/provider-validator.js.map +1 -0
- package/dist/services/session-registration.service.d.ts +80 -0
- package/dist/services/session-registration.service.d.ts.map +1 -0
- package/dist/services/session-registration.service.js +172 -0
- package/dist/services/session-registration.service.js.map +1 -0
- package/dist/services/tool-context-builder.d.ts +57 -0
- package/dist/services/tool-context-builder.d.ts.map +1 -0
- package/dist/services/tool-context-builder.js +125 -0
- package/dist/services/tool-context-builder.js.map +1 -0
- package/dist/services/tool-protection.service.d.ts +87 -10
- package/dist/services/tool-protection.service.d.ts.map +1 -1
- package/dist/services/tool-protection.service.js +282 -112
- package/dist/services/tool-protection.service.js.map +1 -1
- package/dist/types/oauth-required-error.d.ts +40 -0
- package/dist/types/oauth-required-error.d.ts.map +1 -0
- package/dist/types/oauth-required-error.js +40 -0
- package/dist/types/oauth-required-error.js.map +1 -0
- package/dist/utils/did-helpers.d.ts +33 -0
- package/dist/utils/did-helpers.d.ts.map +1 -1
- package/dist/utils/did-helpers.js +40 -0
- package/dist/utils/did-helpers.js.map +1 -1
- package/dist/utils/index.d.ts +1 -0
- package/dist/utils/index.d.ts.map +1 -1
- package/dist/utils/index.js +1 -0
- package/dist/utils/index.js.map +1 -1
- package/docs/API_REFERENCE.md +1362 -0
- package/docs/COMPLIANCE_MATRIX.md +691 -0
- package/docs/STATUSLIST2021_GUIDE.md +696 -0
- package/docs/W3C_VC_DELEGATION_GUIDE.md +710 -0
- package/package.json +24 -50
- package/scripts/audit-compliance.ts +724 -0
- package/src/__tests__/cache/tool-protection-cache.test.ts +640 -0
- package/src/__tests__/config/provider-runtime-config.test.ts +309 -0
- package/src/__tests__/delegation-e2e.test.ts +690 -0
- package/src/__tests__/identity/user-did-manager.test.ts +213 -0
- package/src/__tests__/index.test.ts +56 -0
- package/src/__tests__/integration/full-flow.test.ts +776 -0
- package/src/__tests__/integration.test.ts +281 -0
- package/src/__tests__/providers/base.test.ts +173 -0
- package/src/__tests__/providers/memory.test.ts +319 -0
- package/src/__tests__/regression/phase2-regression.test.ts +427 -0
- package/src/__tests__/runtime/audit-logger.test.ts +154 -0
- package/src/__tests__/runtime/base-extensions.test.ts +593 -0
- package/src/__tests__/runtime/base.test.ts +869 -0
- package/src/__tests__/runtime/delegation-flow.test.ts +164 -0
- package/src/__tests__/runtime/proof-client-did.test.ts +375 -0
- package/src/__tests__/runtime/route-interception.test.ts +686 -0
- package/src/__tests__/runtime/tool-protection-enforcement.test.ts +908 -0
- package/src/__tests__/services/agentshield-integration.test.ts +784 -0
- package/src/__tests__/services/provider-resolver-edge-cases.test.ts +487 -0
- package/src/__tests__/services/tool-protection-oauth-provider.test.ts +480 -0
- package/src/__tests__/services/tool-protection.service.test.ts +1366 -0
- package/src/__tests__/utils/mock-providers.ts +340 -0
- package/src/cache/oauth-config-cache.d.ts +69 -0
- package/src/cache/oauth-config-cache.d.ts.map +1 -0
- package/src/cache/oauth-config-cache.js +71 -0
- package/src/cache/oauth-config-cache.js.map +1 -0
- package/src/cache/oauth-config-cache.ts +123 -0
- package/src/cache/tool-protection-cache.ts +171 -0
- package/src/compliance/EXAMPLE.md +412 -0
- package/src/compliance/__tests__/schema-verifier.test.ts +797 -0
- package/src/compliance/index.ts +8 -0
- package/src/compliance/schema-registry.ts +460 -0
- package/src/compliance/schema-verifier.ts +708 -0
- package/src/config/__tests__/remote-config.spec.ts +268 -0
- package/src/config/remote-config.ts +174 -0
- package/src/config.ts +309 -0
- package/src/delegation/__tests__/audience-validator.test.ts +112 -0
- package/src/delegation/__tests__/bitstring.test.ts +346 -0
- package/src/delegation/__tests__/cascading-revocation.test.ts +628 -0
- package/src/delegation/__tests__/delegation-graph.test.ts +584 -0
- package/src/delegation/__tests__/utils.test.ts +152 -0
- package/src/delegation/__tests__/vc-issuer.test.ts +442 -0
- package/src/delegation/__tests__/vc-verifier.test.ts +922 -0
- package/src/delegation/audience-validator.ts +52 -0
- package/src/delegation/bitstring.ts +278 -0
- package/src/delegation/cascading-revocation.ts +370 -0
- package/src/delegation/delegation-graph.ts +299 -0
- package/src/delegation/index.ts +14 -0
- package/src/delegation/statuslist-manager.ts +353 -0
- package/src/delegation/storage/__tests__/memory-graph-storage.test.ts +366 -0
- package/src/delegation/storage/__tests__/memory-statuslist-storage.test.ts +228 -0
- package/src/delegation/storage/index.ts +9 -0
- package/src/delegation/storage/memory-graph-storage.ts +178 -0
- package/src/delegation/storage/memory-statuslist-storage.ts +77 -0
- package/src/delegation/utils.ts +42 -0
- package/src/delegation/vc-issuer.ts +232 -0
- package/src/delegation/vc-verifier.ts +568 -0
- package/src/identity/idp-token-resolver.ts +147 -0
- package/src/identity/idp-token-storage.interface.ts +59 -0
- package/src/identity/user-did-manager.ts +370 -0
- package/src/index.ts +271 -0
- package/src/providers/base.d.ts +91 -0
- package/src/providers/base.d.ts.map +1 -0
- package/src/providers/base.js +38 -0
- package/src/providers/base.js.map +1 -0
- package/src/providers/base.ts +96 -0
- package/src/providers/memory.ts +142 -0
- package/src/runtime/audit-logger.ts +39 -0
- package/src/runtime/base.ts +1329 -0
- package/src/services/__tests__/access-control.integration.test.ts +443 -0
- package/src/services/__tests__/access-control.proof-response-validation.test.ts +578 -0
- package/src/services/__tests__/access-control.service.test.ts +970 -0
- package/src/services/__tests__/batch-delegation.service.test.ts +351 -0
- package/src/services/__tests__/crypto.service.test.ts +531 -0
- package/src/services/__tests__/oauth-provider-registry.test.ts +142 -0
- package/src/services/__tests__/proof-verifier.integration.test.ts +485 -0
- package/src/services/__tests__/proof-verifier.test.ts +489 -0
- package/src/services/__tests__/provider-resolution.integration.test.ts +198 -0
- package/src/services/__tests__/provider-resolver.test.ts +217 -0
- package/src/services/__tests__/storage.service.test.ts +358 -0
- package/src/services/access-control.service.ts +990 -0
- package/src/services/authorization/authorization-registry.ts +66 -0
- package/src/services/authorization/types.ts +71 -0
- package/src/services/batch-delegation.service.ts +137 -0
- package/src/services/crypto.service.ts +302 -0
- package/src/services/errors.ts +76 -0
- package/src/services/index.ts +18 -0
- package/src/services/oauth-config.service.d.ts +53 -0
- package/src/services/oauth-config.service.d.ts.map +1 -0
- package/src/services/oauth-config.service.js +113 -0
- package/src/services/oauth-config.service.js.map +1 -0
- package/src/services/oauth-config.service.ts +166 -0
- package/src/services/oauth-provider-registry.d.ts +57 -0
- package/src/services/oauth-provider-registry.d.ts.map +1 -0
- package/src/services/oauth-provider-registry.js +73 -0
- package/src/services/oauth-provider-registry.js.map +1 -0
- package/src/services/oauth-provider-registry.ts +123 -0
- package/src/services/oauth-service.ts +510 -0
- package/src/services/oauth-token-retrieval.service.ts +245 -0
- package/src/services/proof-verifier.ts +478 -0
- package/src/services/provider-resolver.d.ts +48 -0
- package/src/services/provider-resolver.d.ts.map +1 -0
- package/src/services/provider-resolver.js +106 -0
- package/src/services/provider-resolver.js.map +1 -0
- package/src/services/provider-resolver.ts +144 -0
- package/src/services/provider-validator.ts +170 -0
- package/src/services/session-registration.service.ts +251 -0
- package/src/services/storage.service.ts +566 -0
- package/src/services/tool-context-builder.ts +172 -0
- package/src/services/tool-protection.service.ts +958 -0
- package/src/types/oauth-required-error.ts +63 -0
- package/src/types/tool-protection.ts +155 -0
- package/src/utils/__tests__/did-helpers.test.ts +101 -0
- package/src/utils/base64.ts +148 -0
- package/src/utils/cors.ts +83 -0
- package/src/utils/did-helpers.ts +150 -0
- package/src/utils/index.ts +8 -0
- package/src/utils/storage-keys.ts +278 -0
- package/tsconfig.json +21 -0
- package/vitest.config.ts +56 -0
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Provider Resolver
|
|
4
|
+
*
|
|
5
|
+
* Resolves OAuth provider for tools using priority-based resolution strategy.
|
|
6
|
+
* Supports Phase 2+ tool-specific providers with backward compatibility for Phase 1.
|
|
7
|
+
*
|
|
8
|
+
* @package @kya-os/mcp-i-core
|
|
9
|
+
*/
|
|
10
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
11
|
+
exports.ProviderResolver = void 0;
|
|
12
|
+
/**
|
|
13
|
+
* Resolves OAuth provider for tools with priority-based fallback strategy
|
|
14
|
+
*
|
|
15
|
+
* Priority order:
|
|
16
|
+
* 1. Tool-specific oauthProvider field (Phase 2+ preferred)
|
|
17
|
+
* 2. Scope prefix inference (fallback)
|
|
18
|
+
* 3. First configured provider (Phase 1 compatibility fallback)
|
|
19
|
+
* 4. Error if no provider can be resolved
|
|
20
|
+
*/
|
|
21
|
+
class ProviderResolver {
|
|
22
|
+
registry;
|
|
23
|
+
configService;
|
|
24
|
+
constructor(registry, configService) {
|
|
25
|
+
this.registry = registry;
|
|
26
|
+
this.configService = configService;
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Resolve OAuth provider for a tool
|
|
30
|
+
*
|
|
31
|
+
* @param toolProtection - Tool protection configuration
|
|
32
|
+
* @param projectId - Project ID for fetching provider config
|
|
33
|
+
* @returns Provider name (never null - throws if cannot resolve)
|
|
34
|
+
* @throws Error if provider cannot be resolved
|
|
35
|
+
*/
|
|
36
|
+
async resolveProvider(toolProtection, projectId) {
|
|
37
|
+
// Priority 1: Tool-specific provider (Phase 2+ preferred)
|
|
38
|
+
if (toolProtection.oauthProvider) {
|
|
39
|
+
// Ensure registry is loaded before checking
|
|
40
|
+
if (this.registry.getProviderNames().length === 0) {
|
|
41
|
+
await this.registry.loadFromAgentShield(projectId);
|
|
42
|
+
}
|
|
43
|
+
if (!this.registry.hasProvider(toolProtection.oauthProvider)) {
|
|
44
|
+
throw new Error(`Provider "${toolProtection.oauthProvider}" not configured for project "${projectId}". ` +
|
|
45
|
+
`Add provider in project settings.`);
|
|
46
|
+
}
|
|
47
|
+
return toolProtection.oauthProvider;
|
|
48
|
+
}
|
|
49
|
+
// Priority 2: Scope prefix inference (fallback)
|
|
50
|
+
const inferredProvider = this.inferProviderFromScopes(toolProtection.requiredScopes || []);
|
|
51
|
+
if (inferredProvider) {
|
|
52
|
+
// Ensure registry is loaded before checking
|
|
53
|
+
if (this.registry.getProviderNames().length === 0) {
|
|
54
|
+
await this.registry.loadFromAgentShield(projectId);
|
|
55
|
+
}
|
|
56
|
+
if (this.registry.hasProvider(inferredProvider)) {
|
|
57
|
+
console.log(`[ProviderResolver] Inferred provider "${inferredProvider}" from scopes`);
|
|
58
|
+
return inferredProvider;
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
// Priority 3: First configured provider (Phase 1 compatibility fallback)
|
|
62
|
+
// Ensure registry is loaded
|
|
63
|
+
await this.registry.loadFromAgentShield(projectId);
|
|
64
|
+
const providers = this.registry.getAllProviders();
|
|
65
|
+
if (providers.length > 0) {
|
|
66
|
+
// Log deprecation warning for Phase 1 tools
|
|
67
|
+
const firstProviderName = this.registry.getProviderNames()[0];
|
|
68
|
+
console.warn(`[ProviderResolver] Tool does not specify oauthProvider. ` +
|
|
69
|
+
`Using first configured provider "${firstProviderName}" as fallback. ` +
|
|
70
|
+
`This is deprecated - configure oauthProvider in AgentShield dashboard for Phase 2+.`);
|
|
71
|
+
return firstProviderName;
|
|
72
|
+
}
|
|
73
|
+
// Priority 4: Error if no provider can be resolved
|
|
74
|
+
throw new Error(`Tool requires OAuth but no provider could be resolved. ` +
|
|
75
|
+
`Either specify oauthProvider in tool protection config, or configure at least one provider for project "${projectId}".`);
|
|
76
|
+
}
|
|
77
|
+
/**
|
|
78
|
+
* Infer provider from scope prefixes
|
|
79
|
+
*
|
|
80
|
+
* Used as Priority 2 fallback when oauthProvider is not specified.
|
|
81
|
+
* Examples:
|
|
82
|
+
* - github:repo:read → github
|
|
83
|
+
* - gmail:read → google
|
|
84
|
+
* - microsoft:calendar:read → microsoft
|
|
85
|
+
*
|
|
86
|
+
* @param scopes - Required scopes for the tool
|
|
87
|
+
* @returns Provider name if uniquely inferred, null otherwise
|
|
88
|
+
*/
|
|
89
|
+
inferProviderFromScopes(scopes) {
|
|
90
|
+
if (!scopes || scopes.length === 0) {
|
|
91
|
+
return null;
|
|
92
|
+
}
|
|
93
|
+
// Extract first part of scope (before first colon)
|
|
94
|
+
const scopePrefixes = scopes.map((scope) => {
|
|
95
|
+
const parts = scope.split(":");
|
|
96
|
+
return parts[0].toLowerCase();
|
|
97
|
+
});
|
|
98
|
+
// Provider mapping
|
|
99
|
+
const providerMap = {
|
|
100
|
+
github: "github",
|
|
101
|
+
google: "google",
|
|
102
|
+
gmail: "google", // gmail:read → google
|
|
103
|
+
calendar: "google", // calendar:read → google (if ambiguous, use project default)
|
|
104
|
+
microsoft: "microsoft",
|
|
105
|
+
outlook: "microsoft",
|
|
106
|
+
slack: "slack",
|
|
107
|
+
auth0: "auth0",
|
|
108
|
+
okta: "okta",
|
|
109
|
+
};
|
|
110
|
+
// Find unique provider
|
|
111
|
+
const providers = new Set(scopePrefixes.map((prefix) => providerMap[prefix]).filter(Boolean));
|
|
112
|
+
if (providers.size === 1) {
|
|
113
|
+
return Array.from(providers)[0];
|
|
114
|
+
}
|
|
115
|
+
// Ambiguous or no prefix → return null (use project-level provider)
|
|
116
|
+
return null;
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
exports.ProviderResolver = ProviderResolver;
|
|
120
|
+
//# sourceMappingURL=provider-resolver.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-resolver.js","sourceRoot":"","sources":["../../src/services/provider-resolver.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAMH;;;;;;;;GAQG;AACH,MAAa,gBAAgB;IAEjB;IACA;IAFV,YACU,QAA+B,EAC/B,aAAiC;QADjC,aAAQ,GAAR,QAAQ,CAAuB;QAC/B,kBAAa,GAAb,aAAa,CAAoB;IACxC,CAAC;IAEJ;;;;;;;OAOG;IACH,KAAK,CAAC,eAAe,CACnB,cAA8B,EAC9B,SAAiB;QAEjB,0DAA0D;QAC1D,IAAI,cAAc,CAAC,aAAa,EAAE,CAAC;YACjC,4CAA4C;YAC5C,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClD,MAAM,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC7D,MAAM,IAAI,KAAK,CACb,aAAa,cAAc,CAAC,aAAa,iCAAiC,SAAS,KAAK;oBACtF,mCAAmC,CACtC,CAAC;YACJ,CAAC;YACD,OAAO,cAAc,CAAC,aAAa,CAAC;QACtC,CAAC;QAED,gDAAgD;QAChD,MAAM,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CACnD,cAAc,CAAC,cAAc,IAAI,EAAE,CACpC,CAAC;QACF,IAAI,gBAAgB,EAAE,CAAC;YACrB,4CAA4C;YAC5C,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClD,MAAM,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAChD,OAAO,CAAC,GAAG,CACT,yCAAyC,gBAAgB,eAAe,CACzE,CAAC;gBACF,OAAO,gBAAgB,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,yEAAyE;QACzE,4BAA4B;QAC5B,MAAM,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;QAClD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,4CAA4C;YAC5C,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9D,OAAO,CAAC,IAAI,CACV,0DAA0D;gBACxD,oCAAoC,iBAAiB,iBAAiB;gBACtE,qFAAqF,CACxF,CAAC;YACF,OAAO,iBAAiB,CAAC;QAC3B,CAAC;QAED,mDAAmD;QACnD,MAAM,IAAI,KAAK,CACb,yDAAyD;YACvD,2GAA2G,SAAS,IAAI,CAC3H,CAAC;IACJ,CAAC;IAED;;;;;;;;;;;OAWG;IACK,uBAAuB,CAAC,MAAgB;QAC9C,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,mDAAmD;QACnD,MAAM,aAAa,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACzC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC/B,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,mBAAmB;QACnB,MAAM,WAAW,GAA2B;YAC1C,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,QAAQ,EAAE,sBAAsB;YACvC,QAAQ,EAAE,QAAQ,EAAE,6DAA6D;YACjF,SAAS,EAAE,WAAW;YACtB,OAAO,EAAE,WAAW;YACpB,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,MAAM;SACb,CAAC;QAEF,uBAAuB;QACvB,MAAM,SAAS,GAAG,IAAI,GAAG,CACvB,aAAa,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CACnE,CAAC;QAEF,IAAI,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,oEAAoE;QACpE,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAxHD,4CAwHC"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Provider Validator
|
|
3
|
+
*
|
|
4
|
+
* Validates OAuth provider configurations for custom IDP support.
|
|
5
|
+
* Ensures provider configurations are valid before registration.
|
|
6
|
+
*
|
|
7
|
+
* @package @kya-os/mcp-i-core
|
|
8
|
+
*/
|
|
9
|
+
import type { OAuthProvider } from "@kya-os/contracts/config";
|
|
10
|
+
/**
|
|
11
|
+
* Validation error for provider configuration issues
|
|
12
|
+
*/
|
|
13
|
+
export declare class ProviderValidationError extends Error {
|
|
14
|
+
readonly field?: string | undefined;
|
|
15
|
+
constructor(message: string, field?: string | undefined);
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Service for validating OAuth provider configurations
|
|
19
|
+
*/
|
|
20
|
+
export declare class ProviderValidator {
|
|
21
|
+
/**
|
|
22
|
+
* Validate provider configuration
|
|
23
|
+
*
|
|
24
|
+
* @param provider - Provider configuration to validate
|
|
25
|
+
* @param name - Provider name (for error messages)
|
|
26
|
+
* @throws ProviderValidationError if validation fails
|
|
27
|
+
*/
|
|
28
|
+
validate(provider: OAuthProvider, name: string): void;
|
|
29
|
+
/**
|
|
30
|
+
* Validate URL format
|
|
31
|
+
*
|
|
32
|
+
* @param url - URL to validate
|
|
33
|
+
* @param providerName - Provider name (for error messages)
|
|
34
|
+
* @param fieldName - Field name (for error messages)
|
|
35
|
+
* @throws ProviderValidationError if URL is invalid
|
|
36
|
+
*/
|
|
37
|
+
private validateUrl;
|
|
38
|
+
/**
|
|
39
|
+
* Validate custom parameters don't override reserved OAuth parameters
|
|
40
|
+
*
|
|
41
|
+
* @param customParams - Custom parameters to validate
|
|
42
|
+
* @param providerName - Provider name (for error messages)
|
|
43
|
+
* @throws ProviderValidationError if reserved parameter is overridden
|
|
44
|
+
*/
|
|
45
|
+
private validateCustomParams;
|
|
46
|
+
/**
|
|
47
|
+
* Test provider endpoint reachability (optional)
|
|
48
|
+
*
|
|
49
|
+
* @param provider - Provider configuration
|
|
50
|
+
* @param fetchProvider - Fetch implementation
|
|
51
|
+
* @returns True if endpoint is reachable, false otherwise
|
|
52
|
+
*/
|
|
53
|
+
testProvider(provider: OAuthProvider, fetchProvider: typeof fetch): Promise<boolean>;
|
|
54
|
+
}
|
|
55
|
+
//# sourceMappingURL=provider-validator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-validator.d.ts","sourceRoot":"","sources":["../../src/services/provider-validator.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAe9D;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,KAAK;aACH,KAAK,CAAC,EAAE,MAAM;gBAA/C,OAAO,EAAE,MAAM,EAAkB,KAAK,CAAC,EAAE,MAAM,YAAA;CAI5D;AAED;;GAEG;AACH,qBAAa,iBAAiB;IAC5B;;;;;;OAMG;IACH,QAAQ,CAAC,QAAQ,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IA6CrD;;;;;;;OAOG;IACH,OAAO,CAAC,WAAW;IAoBnB;;;;;;OAMG;IACH,OAAO,CAAC,oBAAoB;IAqB5B;;;;;;OAMG;IACG,YAAY,CAChB,QAAQ,EAAE,aAAa,EACvB,aAAa,EAAE,OAAO,KAAK,GAC1B,OAAO,CAAC,OAAO,CAAC;CAYpB"}
|
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Provider Validator
|
|
4
|
+
*
|
|
5
|
+
* Validates OAuth provider configurations for custom IDP support.
|
|
6
|
+
* Ensures provider configurations are valid before registration.
|
|
7
|
+
*
|
|
8
|
+
* @package @kya-os/mcp-i-core
|
|
9
|
+
*/
|
|
10
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
11
|
+
exports.ProviderValidator = exports.ProviderValidationError = void 0;
|
|
12
|
+
/**
|
|
13
|
+
* Reserved OAuth parameters that cannot be overridden by custom parameters
|
|
14
|
+
*/
|
|
15
|
+
const RESERVED_PARAMETERS = [
|
|
16
|
+
"response_type",
|
|
17
|
+
"client_id",
|
|
18
|
+
"redirect_uri",
|
|
19
|
+
"scope",
|
|
20
|
+
"state",
|
|
21
|
+
"code_challenge",
|
|
22
|
+
"code_challenge_method",
|
|
23
|
+
];
|
|
24
|
+
/**
|
|
25
|
+
* Validation error for provider configuration issues
|
|
26
|
+
*/
|
|
27
|
+
class ProviderValidationError extends Error {
|
|
28
|
+
field;
|
|
29
|
+
constructor(message, field) {
|
|
30
|
+
super(message);
|
|
31
|
+
this.field = field;
|
|
32
|
+
this.name = "ProviderValidationError";
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
exports.ProviderValidationError = ProviderValidationError;
|
|
36
|
+
/**
|
|
37
|
+
* Service for validating OAuth provider configurations
|
|
38
|
+
*/
|
|
39
|
+
class ProviderValidator {
|
|
40
|
+
/**
|
|
41
|
+
* Validate provider configuration
|
|
42
|
+
*
|
|
43
|
+
* @param provider - Provider configuration to validate
|
|
44
|
+
* @param name - Provider name (for error messages)
|
|
45
|
+
* @throws ProviderValidationError if validation fails
|
|
46
|
+
*/
|
|
47
|
+
validate(provider, name) {
|
|
48
|
+
// Validate required fields
|
|
49
|
+
if (!provider.clientId || provider.clientId.trim().length === 0) {
|
|
50
|
+
throw new ProviderValidationError(`Provider "${name}" must have a clientId`, "clientId");
|
|
51
|
+
}
|
|
52
|
+
if (!provider.authorizationUrl || provider.authorizationUrl.trim().length === 0) {
|
|
53
|
+
throw new ProviderValidationError(`Provider "${name}" must have an authorizationUrl`, "authorizationUrl");
|
|
54
|
+
}
|
|
55
|
+
if (!provider.tokenUrl || provider.tokenUrl.trim().length === 0) {
|
|
56
|
+
throw new ProviderValidationError(`Provider "${name}" must have a tokenUrl`, "tokenUrl");
|
|
57
|
+
}
|
|
58
|
+
// Validate URL formats
|
|
59
|
+
this.validateUrl(provider.authorizationUrl, name, "authorizationUrl");
|
|
60
|
+
this.validateUrl(provider.tokenUrl, name, "tokenUrl");
|
|
61
|
+
if (provider.userInfoUrl) {
|
|
62
|
+
this.validateUrl(provider.userInfoUrl, name, "userInfoUrl");
|
|
63
|
+
}
|
|
64
|
+
// Validate proxy mode requirements
|
|
65
|
+
if (provider.proxyMode && !provider.requiresClientSecret) {
|
|
66
|
+
throw new ProviderValidationError(`Provider "${name}" with proxyMode=true must have requiresClientSecret=true`, "proxyMode");
|
|
67
|
+
}
|
|
68
|
+
// Validate custom parameters don't conflict with reserved parameters
|
|
69
|
+
if (provider.customParams) {
|
|
70
|
+
this.validateCustomParams(provider.customParams, name);
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
/**
|
|
74
|
+
* Validate URL format
|
|
75
|
+
*
|
|
76
|
+
* @param url - URL to validate
|
|
77
|
+
* @param providerName - Provider name (for error messages)
|
|
78
|
+
* @param fieldName - Field name (for error messages)
|
|
79
|
+
* @throws ProviderValidationError if URL is invalid
|
|
80
|
+
*/
|
|
81
|
+
validateUrl(url, providerName, fieldName) {
|
|
82
|
+
try {
|
|
83
|
+
const parsedUrl = new URL(url);
|
|
84
|
+
if (parsedUrl.protocol !== "http:" && parsedUrl.protocol !== "https:") {
|
|
85
|
+
throw new ProviderValidationError(`Provider "${providerName}" ${fieldName} must use HTTP or HTTPS protocol`, fieldName);
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
catch (error) {
|
|
89
|
+
if (error instanceof ProviderValidationError) {
|
|
90
|
+
throw error;
|
|
91
|
+
}
|
|
92
|
+
throw new ProviderValidationError(`Provider "${providerName}" ${fieldName} is not a valid URL: ${error instanceof Error ? error.message : String(error)}`, fieldName);
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
/**
|
|
96
|
+
* Validate custom parameters don't override reserved OAuth parameters
|
|
97
|
+
*
|
|
98
|
+
* @param customParams - Custom parameters to validate
|
|
99
|
+
* @param providerName - Provider name (for error messages)
|
|
100
|
+
* @throws ProviderValidationError if reserved parameter is overridden
|
|
101
|
+
*/
|
|
102
|
+
validateCustomParams(customParams, providerName) {
|
|
103
|
+
for (const [key, value] of Object.entries(customParams)) {
|
|
104
|
+
const normalizedKey = key.toLowerCase();
|
|
105
|
+
if (RESERVED_PARAMETERS.includes(normalizedKey)) {
|
|
106
|
+
throw new ProviderValidationError(`Provider "${providerName}" custom parameter "${key}" conflicts with reserved OAuth parameter. Reserved parameters: ${RESERVED_PARAMETERS.join(", ")}`, `customParams.${key}`);
|
|
107
|
+
}
|
|
108
|
+
if (!value || value.trim().length === 0) {
|
|
109
|
+
throw new ProviderValidationError(`Provider "${providerName}" custom parameter "${key}" has empty value`, `customParams.${key}`);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
/**
|
|
114
|
+
* Test provider endpoint reachability (optional)
|
|
115
|
+
*
|
|
116
|
+
* @param provider - Provider configuration
|
|
117
|
+
* @param fetchProvider - Fetch implementation
|
|
118
|
+
* @returns True if endpoint is reachable, false otherwise
|
|
119
|
+
*/
|
|
120
|
+
async testProvider(provider, fetchProvider) {
|
|
121
|
+
try {
|
|
122
|
+
// Test authorization URL (HEAD request to avoid triggering OAuth flow)
|
|
123
|
+
const authResponse = await fetchProvider(provider.authorizationUrl, {
|
|
124
|
+
method: "HEAD",
|
|
125
|
+
signal: AbortSignal.timeout(5000), // 5 second timeout
|
|
126
|
+
});
|
|
127
|
+
return authResponse.ok || authResponse.status === 405; // 405 Method Not Allowed is OK
|
|
128
|
+
}
|
|
129
|
+
catch (error) {
|
|
130
|
+
return false;
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
exports.ProviderValidator = ProviderValidator;
|
|
135
|
+
//# sourceMappingURL=provider-validator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-validator.js","sourceRoot":"","sources":["../../src/services/provider-validator.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAIH;;GAEG;AACH,MAAM,mBAAmB,GAAG;IAC1B,eAAe;IACf,WAAW;IACX,cAAc;IACd,OAAO;IACP,OAAO;IACP,gBAAgB;IAChB,uBAAuB;CACf,CAAC;AAEX;;GAEG;AACH,MAAa,uBAAwB,SAAQ,KAAK;IACH;IAA7C,YAAY,OAAe,EAAkB,KAAc;QACzD,KAAK,CAAC,OAAO,CAAC,CAAC;QAD4B,UAAK,GAAL,KAAK,CAAS;QAEzD,IAAI,CAAC,IAAI,GAAG,yBAAyB,CAAC;IACxC,CAAC;CACF;AALD,0DAKC;AAED;;GAEG;AACH,MAAa,iBAAiB;IAC5B;;;;;;OAMG;IACH,QAAQ,CAAC,QAAuB,EAAE,IAAY;QAC5C,2BAA2B;QAC3B,IAAI,CAAC,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,uBAAuB,CAC/B,aAAa,IAAI,wBAAwB,EACzC,UAAU,CACX,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,gBAAgB,IAAI,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChF,MAAM,IAAI,uBAAuB,CAC/B,aAAa,IAAI,iCAAiC,EAClD,kBAAkB,CACnB,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,uBAAuB,CAC/B,aAAa,IAAI,wBAAwB,EACzC,UAAU,CACX,CAAC;QACJ,CAAC;QAED,uBAAuB;QACvB,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB,EAAE,IAAI,EAAE,kBAAkB,CAAC,CAAC;QACtE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;QAEtD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;QAC9D,CAAC;QAED,mCAAmC;QACnC,IAAI,QAAQ,CAAC,SAAS,IAAI,CAAC,QAAQ,CAAC,oBAAoB,EAAE,CAAC;YACzD,MAAM,IAAI,uBAAuB,CAC/B,aAAa,IAAI,2DAA2D,EAC5E,WAAW,CACZ,CAAC;QACJ,CAAC;QAED,qEAAqE;QACrE,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC1B,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,WAAW,CAAC,GAAW,EAAE,YAAoB,EAAE,SAAiB;QACtE,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;YAC/B,IAAI,SAAS,CAAC,QAAQ,KAAK,OAAO,IAAI,SAAS,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACtE,MAAM,IAAI,uBAAuB,CAC/B,aAAa,YAAY,KAAK,SAAS,kCAAkC,EACzE,SAAS,CACV,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,uBAAuB,EAAE,CAAC;gBAC7C,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,uBAAuB,CAC/B,aAAa,YAAY,KAAK,SAAS,wBAAwB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACvH,SAAS,CACV,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,oBAAoB,CAC1B,YAAoC,EACpC,YAAoB;QAEpB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YACxD,MAAM,aAAa,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;YACxC,IAAI,mBAAmB,CAAC,QAAQ,CAAC,aAAoB,CAAC,EAAE,CAAC;gBACvD,MAAM,IAAI,uBAAuB,CAC/B,aAAa,YAAY,uBAAuB,GAAG,mEAAmE,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EACtJ,gBAAgB,GAAG,EAAE,CACtB,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxC,MAAM,IAAI,uBAAuB,CAC/B,aAAa,YAAY,uBAAuB,GAAG,mBAAmB,EACtE,gBAAgB,GAAG,EAAE,CACtB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,YAAY,CAChB,QAAuB,EACvB,aAA2B;QAE3B,IAAI,CAAC;YACH,uEAAuE;YACvE,MAAM,YAAY,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAClE,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,mBAAmB;aACvD,CAAC,CAAC;YACH,OAAO,YAAY,CAAC,EAAE,IAAI,YAAY,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,+BAA+B;QACxF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF;AAnID,8CAmIC"}
|
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Session Registration Service
|
|
3
|
+
*
|
|
4
|
+
* Registers MCP sessions with the AgentShield dashboard, enabling
|
|
5
|
+
* visibility into which MCP clients are connecting to agents.
|
|
6
|
+
*
|
|
7
|
+
* This is a fire-and-forget service - session registration should not
|
|
8
|
+
* block tool execution or affect the user experience.
|
|
9
|
+
*
|
|
10
|
+
* @package @kya-os/mcp-i-core
|
|
11
|
+
*/
|
|
12
|
+
import type { RegisterSessionRequest } from "@kya-os/contracts/agentshield-api";
|
|
13
|
+
import type { FetchProvider } from "../providers/base.js";
|
|
14
|
+
/**
|
|
15
|
+
* Configuration for the session registration service
|
|
16
|
+
*/
|
|
17
|
+
export interface SessionRegistrationServiceConfig {
|
|
18
|
+
/** Base URL for the AgentShield API (e.g., "https://kya.vouched.id") */
|
|
19
|
+
baseUrl: string;
|
|
20
|
+
/** API key for authentication */
|
|
21
|
+
apiKey: string;
|
|
22
|
+
/** Fetch provider for making HTTP requests (platform-agnostic) */
|
|
23
|
+
fetchProvider: FetchProvider;
|
|
24
|
+
/** Optional logger callback for diagnostics */
|
|
25
|
+
logger?: (message: string, data?: unknown) => void;
|
|
26
|
+
/** Timeout in milliseconds for the registration request (default: 5000) */
|
|
27
|
+
timeoutMs?: number;
|
|
28
|
+
}
|
|
29
|
+
/**
|
|
30
|
+
* Result of a session registration attempt
|
|
31
|
+
*/
|
|
32
|
+
export interface SessionRegistrationResult {
|
|
33
|
+
/** Whether registration was successful */
|
|
34
|
+
success: boolean;
|
|
35
|
+
/** Session ID that was registered */
|
|
36
|
+
sessionId: string;
|
|
37
|
+
/** Error message if registration failed */
|
|
38
|
+
error?: string;
|
|
39
|
+
}
|
|
40
|
+
/**
|
|
41
|
+
* Session Registration Service
|
|
42
|
+
*
|
|
43
|
+
* Registers MCP sessions with AgentShield for dashboard visibility.
|
|
44
|
+
* Designed to be non-blocking - failures are logged but don't throw.
|
|
45
|
+
*/
|
|
46
|
+
export declare class SessionRegistrationService {
|
|
47
|
+
private config;
|
|
48
|
+
constructor(config: SessionRegistrationServiceConfig);
|
|
49
|
+
/**
|
|
50
|
+
* Register a session with AgentShield
|
|
51
|
+
*
|
|
52
|
+
* This is a fire-and-forget operation. Failures are logged but don't throw.
|
|
53
|
+
* The method returns quickly and doesn't block the caller.
|
|
54
|
+
*
|
|
55
|
+
* @param request - Session registration request data
|
|
56
|
+
* @returns Result indicating success or failure
|
|
57
|
+
*/
|
|
58
|
+
registerSession(request: RegisterSessionRequest): Promise<SessionRegistrationResult>;
|
|
59
|
+
/**
|
|
60
|
+
* Fire-and-forget session registration
|
|
61
|
+
*
|
|
62
|
+
* Starts registration in the background without waiting for completion.
|
|
63
|
+
* Useful when you want to register a session but not delay the response.
|
|
64
|
+
*
|
|
65
|
+
* @param request - Session registration request data
|
|
66
|
+
*/
|
|
67
|
+
registerSessionAsync(request: RegisterSessionRequest): void;
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* Create a session registration service from common runtime config
|
|
71
|
+
*
|
|
72
|
+
* Helper function to create the service from typical environment config.
|
|
73
|
+
*/
|
|
74
|
+
export declare function createSessionRegistrationService(options: {
|
|
75
|
+
apiUrl: string;
|
|
76
|
+
apiKey: string;
|
|
77
|
+
fetchProvider: FetchProvider;
|
|
78
|
+
logger?: (message: string, data?: unknown) => void;
|
|
79
|
+
}): SessionRegistrationService | null;
|
|
80
|
+
//# sourceMappingURL=session-registration.service.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-registration.service.d.ts","sourceRoot":"","sources":["../../src/services/session-registration.service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EACV,sBAAsB,EAEvB,MAAM,mCAAmC,CAAC;AAM3C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAE1D;;GAEG;AACH,MAAM,WAAW,gCAAgC;IAC/C,wEAAwE;IACxE,OAAO,EAAE,MAAM,CAAC;IAEhB,iCAAiC;IACjC,MAAM,EAAE,MAAM,CAAC;IAEf,kEAAkE;IAClE,aAAa,EAAE,aAAa,CAAC;IAE7B,+CAA+C;IAC/C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IAEnD,2EAA2E;IAC3E,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC,0CAA0C;IAC1C,OAAO,EAAE,OAAO,CAAC;IACjB,qCAAqC;IACrC,SAAS,EAAE,MAAM,CAAC;IAClB,2CAA2C;IAC3C,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;;;GAKG;AACH,qBAAa,0BAA0B;IACrC,OAAO,CAAC,MAAM,CAKZ;gBAEU,MAAM,EAAE,gCAAgC;IAUpD;;;;;;;;OAQG;IACG,eAAe,CACnB,OAAO,EAAE,sBAAsB,GAC9B,OAAO,CAAC,yBAAyB,CAAC;IA6GrC;;;;;;;OAOG;IACH,oBAAoB,CAAC,OAAO,EAAE,sBAAsB,GAAG,IAAI;CAW5D;AAED;;;;GAIG;AACH,wBAAgB,gCAAgC,CAAC,OAAO,EAAE;IACxD,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,aAAa,EAAE,aAAa,CAAC;IAC7B,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD,GAAG,0BAA0B,GAAG,IAAI,CAmBpC"}
|
|
@@ -0,0 +1,172 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Session Registration Service
|
|
4
|
+
*
|
|
5
|
+
* Registers MCP sessions with the AgentShield dashboard, enabling
|
|
6
|
+
* visibility into which MCP clients are connecting to agents.
|
|
7
|
+
*
|
|
8
|
+
* This is a fire-and-forget service - session registration should not
|
|
9
|
+
* block tool execution or affect the user experience.
|
|
10
|
+
*
|
|
11
|
+
* @package @kya-os/mcp-i-core
|
|
12
|
+
*/
|
|
13
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
14
|
+
exports.SessionRegistrationService = void 0;
|
|
15
|
+
exports.createSessionRegistrationService = createSessionRegistrationService;
|
|
16
|
+
const agentshield_api_1 = require("@kya-os/contracts/agentshield-api");
|
|
17
|
+
/**
|
|
18
|
+
* Session Registration Service
|
|
19
|
+
*
|
|
20
|
+
* Registers MCP sessions with AgentShield for dashboard visibility.
|
|
21
|
+
* Designed to be non-blocking - failures are logged but don't throw.
|
|
22
|
+
*/
|
|
23
|
+
class SessionRegistrationService {
|
|
24
|
+
config;
|
|
25
|
+
constructor(config) {
|
|
26
|
+
this.config = {
|
|
27
|
+
baseUrl: config.baseUrl,
|
|
28
|
+
apiKey: config.apiKey,
|
|
29
|
+
fetchProvider: config.fetchProvider,
|
|
30
|
+
logger: config.logger || (() => { }),
|
|
31
|
+
timeoutMs: config.timeoutMs ?? 5000,
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Register a session with AgentShield
|
|
36
|
+
*
|
|
37
|
+
* This is a fire-and-forget operation. Failures are logged but don't throw.
|
|
38
|
+
* The method returns quickly and doesn't block the caller.
|
|
39
|
+
*
|
|
40
|
+
* @param request - Session registration request data
|
|
41
|
+
* @returns Result indicating success or failure
|
|
42
|
+
*/
|
|
43
|
+
async registerSession(request) {
|
|
44
|
+
const sessionId = request.session_id;
|
|
45
|
+
try {
|
|
46
|
+
// Validate request
|
|
47
|
+
const validationResult = agentshield_api_1.registerSessionRequestSchema.safeParse(request);
|
|
48
|
+
if (!validationResult.success) {
|
|
49
|
+
const errorMsg = `Invalid session registration request: ${validationResult.error.message}`;
|
|
50
|
+
this.config.logger("[SessionRegistration] Validation failed", {
|
|
51
|
+
sessionId,
|
|
52
|
+
error: errorMsg,
|
|
53
|
+
});
|
|
54
|
+
return { success: false, sessionId, error: errorMsg };
|
|
55
|
+
}
|
|
56
|
+
const url = `${this.config.baseUrl}${agentshield_api_1.AGENTSHIELD_ENDPOINTS.SESSIONS}`;
|
|
57
|
+
this.config.logger("[SessionRegistration] Registering session", {
|
|
58
|
+
sessionId,
|
|
59
|
+
agentDid: request.agent_did,
|
|
60
|
+
clientName: request.client_info.name,
|
|
61
|
+
url,
|
|
62
|
+
});
|
|
63
|
+
// Make the request with timeout
|
|
64
|
+
const controller = new AbortController();
|
|
65
|
+
const timeoutId = setTimeout(() => controller.abort(), this.config.timeoutMs);
|
|
66
|
+
try {
|
|
67
|
+
const response = await this.config.fetchProvider.fetch(url, {
|
|
68
|
+
method: "POST",
|
|
69
|
+
headers: {
|
|
70
|
+
"Content-Type": "application/json",
|
|
71
|
+
Authorization: `Bearer ${this.config.apiKey}`,
|
|
72
|
+
},
|
|
73
|
+
body: JSON.stringify(request),
|
|
74
|
+
signal: controller.signal,
|
|
75
|
+
});
|
|
76
|
+
clearTimeout(timeoutId);
|
|
77
|
+
if (!response.ok) {
|
|
78
|
+
// Log error but don't throw - this is fire-and-forget
|
|
79
|
+
const errorText = await response.text().catch(() => "Unknown error");
|
|
80
|
+
this.config.logger("[SessionRegistration] Registration failed", {
|
|
81
|
+
sessionId,
|
|
82
|
+
status: response.status,
|
|
83
|
+
error: errorText,
|
|
84
|
+
});
|
|
85
|
+
return {
|
|
86
|
+
success: false,
|
|
87
|
+
sessionId,
|
|
88
|
+
error: `HTTP ${response.status}: ${errorText}`,
|
|
89
|
+
};
|
|
90
|
+
}
|
|
91
|
+
// Parse response
|
|
92
|
+
const responseData = (await response.json());
|
|
93
|
+
const parseResult = agentshield_api_1.registerSessionResponseSchema.safeParse(responseData.data || responseData);
|
|
94
|
+
if (!parseResult.success) {
|
|
95
|
+
this.config.logger("[SessionRegistration] Invalid response format", {
|
|
96
|
+
sessionId,
|
|
97
|
+
response: responseData,
|
|
98
|
+
});
|
|
99
|
+
// Still consider it a success if we got a 200 OK
|
|
100
|
+
return { success: true, sessionId };
|
|
101
|
+
}
|
|
102
|
+
this.config.logger("[SessionRegistration] Session registered", {
|
|
103
|
+
sessionId,
|
|
104
|
+
registered: parseResult.data.registered,
|
|
105
|
+
});
|
|
106
|
+
return { success: true, sessionId };
|
|
107
|
+
}
|
|
108
|
+
finally {
|
|
109
|
+
clearTimeout(timeoutId);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
catch (error) {
|
|
113
|
+
// Handle abort/timeout
|
|
114
|
+
if (error instanceof Error && error.name === "AbortError") {
|
|
115
|
+
this.config.logger("[SessionRegistration] Request timed out", {
|
|
116
|
+
sessionId,
|
|
117
|
+
timeoutMs: this.config.timeoutMs,
|
|
118
|
+
});
|
|
119
|
+
return { success: false, sessionId, error: "Request timed out" };
|
|
120
|
+
}
|
|
121
|
+
// Log any other error
|
|
122
|
+
const errorMsg = error instanceof Error ? error.message : "Unknown error";
|
|
123
|
+
this.config.logger("[SessionRegistration] Unexpected error", {
|
|
124
|
+
sessionId,
|
|
125
|
+
error: errorMsg,
|
|
126
|
+
});
|
|
127
|
+
return { success: false, sessionId, error: errorMsg };
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
/**
|
|
131
|
+
* Fire-and-forget session registration
|
|
132
|
+
*
|
|
133
|
+
* Starts registration in the background without waiting for completion.
|
|
134
|
+
* Useful when you want to register a session but not delay the response.
|
|
135
|
+
*
|
|
136
|
+
* @param request - Session registration request data
|
|
137
|
+
*/
|
|
138
|
+
registerSessionAsync(request) {
|
|
139
|
+
// Start registration in background - don't await
|
|
140
|
+
this.registerSession(request).catch((error) => {
|
|
141
|
+
// This should never happen since registerSession catches all errors,
|
|
142
|
+
// but just in case
|
|
143
|
+
this.config.logger("[SessionRegistration] Background registration failed", {
|
|
144
|
+
sessionId: request.session_id,
|
|
145
|
+
error: error instanceof Error ? error.message : "Unknown error",
|
|
146
|
+
});
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
exports.SessionRegistrationService = SessionRegistrationService;
|
|
151
|
+
/**
|
|
152
|
+
* Create a session registration service from common runtime config
|
|
153
|
+
*
|
|
154
|
+
* Helper function to create the service from typical environment config.
|
|
155
|
+
*/
|
|
156
|
+
function createSessionRegistrationService(options) {
|
|
157
|
+
// Validate required config
|
|
158
|
+
if (!options.apiUrl || !options.apiKey) {
|
|
159
|
+
options.logger?.("[SessionRegistration] Missing required config - session registration disabled", {
|
|
160
|
+
hasApiUrl: !!options.apiUrl,
|
|
161
|
+
hasApiKey: !!options.apiKey,
|
|
162
|
+
});
|
|
163
|
+
return null;
|
|
164
|
+
}
|
|
165
|
+
return new SessionRegistrationService({
|
|
166
|
+
baseUrl: options.apiUrl,
|
|
167
|
+
apiKey: options.apiKey,
|
|
168
|
+
fetchProvider: options.fetchProvider,
|
|
169
|
+
logger: options.logger,
|
|
170
|
+
});
|
|
171
|
+
}
|
|
172
|
+
//# sourceMappingURL=session-registration.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-registration.service.js","sourceRoot":"","sources":["../../src/services/session-registration.service.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;GAUG;;;AAuNH,4EAwBC;AAzOD,uEAI2C;AAmC3C;;;;;GAKG;AACH,MAAa,0BAA0B;IAC7B,MAAM,CAKZ;IAEF,YAAY,MAAwC;QAClD,IAAI,CAAC,MAAM,GAAG;YACZ,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;YACnC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;SACpC,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,eAAe,CACnB,OAA+B;QAE/B,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;QAErC,IAAI,CAAC;YACH,mBAAmB;YACnB,MAAM,gBAAgB,GAAG,8CAA4B,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACzE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;gBAC9B,MAAM,QAAQ,GAAG,yCAAyC,gBAAgB,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC3F,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,yCAAyC,EAAE;oBAC5D,SAAS;oBACT,KAAK,EAAE,QAAQ;iBAChB,CAAC,CAAC;gBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YACxD,CAAC;YAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,uCAAqB,CAAC,QAAQ,EAAE,CAAC;YAEtE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,2CAA2C,EAAE;gBAC9D,SAAS;gBACT,QAAQ,EAAE,OAAO,CAAC,SAAS;gBAC3B,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,IAAI;gBACpC,GAAG;aACJ,CAAC,CAAC;YAEH,gCAAgC;YAChC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,UAAU,CAC1B,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EACxB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE;oBAC1D,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;qBAC9C;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;oBAC7B,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC1B,CAAC,CAAC;gBAEH,YAAY,CAAC,SAAS,CAAC,CAAC;gBAExB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,sDAAsD;oBACtD,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,CAAC;oBACrE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,2CAA2C,EAAE;wBAC9D,SAAS;wBACT,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,KAAK,EAAE,SAAS;qBACjB,CAAC,CAAC;oBACH,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,SAAS;wBACT,KAAK,EAAE,QAAQ,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE;qBAC/C,CAAC;gBACJ,CAAC;gBAED,iBAAiB;gBACjB,MAAM,YAAY,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAEhB,CAAC;gBAC5B,MAAM,WAAW,GAAG,+CAA6B,CAAC,SAAS,CACzD,YAAY,CAAC,IAAI,IAAI,YAAY,CAClC,CAAC;gBAEF,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBACzB,IAAI,CAAC,MAAM,CAAC,MAAM,CAChB,+CAA+C,EAC/C;wBACE,SAAS;wBACT,QAAQ,EAAE,YAAY;qBACvB,CACF,CAAC;oBACF,iDAAiD;oBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;gBACtC,CAAC;gBAED,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,0CAA0C,EAAE;oBAC7D,SAAS;oBACT,UAAU,EAAE,WAAW,CAAC,IAAI,CAAC,UAAU;iBACxC,CAAC,CAAC;gBAEH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YACtC,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,SAAS,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,uBAAuB;YACvB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,yCAAyC,EAAE;oBAC5D,SAAS;oBACT,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;iBACjC,CAAC,CAAC;gBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC;YACnE,CAAC;YAED,sBAAsB;YACtB,MAAM,QAAQ,GACZ,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YAC3D,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,wCAAwC,EAAE;gBAC3D,SAAS;gBACT,KAAK,EAAE,QAAQ;aAChB,CAAC,CAAC;YACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;QACxD,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACH,oBAAoB,CAAC,OAA+B;QAClD,iDAAiD;QACjD,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC5C,qEAAqE;YACrE,mBAAmB;YACnB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,sDAAsD,EAAE;gBACzE,SAAS,EAAE,OAAO,CAAC,UAAU;gBAC7B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA7JD,gEA6JC;AAED;;;;GAIG;AACH,SAAgB,gCAAgC,CAAC,OAKhD;IACC,2BAA2B;IAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACvC,OAAO,CAAC,MAAM,EAAE,CACd,+EAA+E,EAC/E;YACE,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM;YAC3B,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM;SAC5B,CACF,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,IAAI,0BAA0B,CAAC;QACpC,OAAO,EAAE,OAAO,CAAC,MAAM;QACvB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tool Context Builder
|
|
3
|
+
*
|
|
4
|
+
* Builds ToolExecutionContext for tool handlers by resolving IDP tokens
|
|
5
|
+
* based on tool protection configuration and user identity.
|
|
6
|
+
*
|
|
7
|
+
* @package @kya-os/mcp-i-core
|
|
8
|
+
*/
|
|
9
|
+
import type { ToolExecutionContext } from "@kya-os/contracts/config";
|
|
10
|
+
import type { IdpTokenResolver } from "../identity/idp-token-resolver.js";
|
|
11
|
+
import type { ToolProtection } from "../types/tool-protection.js";
|
|
12
|
+
import type { OAuthConfigService } from "./oauth-config.service.js";
|
|
13
|
+
import type { ProviderResolver } from "./provider-resolver.js";
|
|
14
|
+
export interface ToolContextBuilderConfig {
|
|
15
|
+
/** IDP token resolver for resolving tokens from User DID */
|
|
16
|
+
tokenResolver: IdpTokenResolver;
|
|
17
|
+
/** OAuth config service for fetching provider configurations */
|
|
18
|
+
configService: OAuthConfigService;
|
|
19
|
+
/** Provider resolver for resolving OAuth providers for tools */
|
|
20
|
+
providerResolver: ProviderResolver;
|
|
21
|
+
/** Project ID for fetching OAuth config */
|
|
22
|
+
projectId: string;
|
|
23
|
+
/** Optional logger callback for diagnostics */
|
|
24
|
+
logger?: (message: string, data?: unknown) => void;
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Builder for tool execution context
|
|
28
|
+
*
|
|
29
|
+
* Resolves IDP tokens and builds context for tool handlers.
|
|
30
|
+
* Phase 1: Uses configured provider as temporary fallback.
|
|
31
|
+
* Phase 2+: Requires explicit oauthProvider on tool protection.
|
|
32
|
+
*/
|
|
33
|
+
export declare class ToolContextBuilder {
|
|
34
|
+
private config;
|
|
35
|
+
constructor(config: ToolContextBuilderConfig);
|
|
36
|
+
/**
|
|
37
|
+
* Build tool execution context
|
|
38
|
+
*
|
|
39
|
+
* @param toolName - Name of the tool being executed
|
|
40
|
+
* @param userDid - User DID (optional, required for OAuth)
|
|
41
|
+
* @param sessionId - Session ID (optional)
|
|
42
|
+
* @param delegationToken - Delegation token (optional)
|
|
43
|
+
* @param toolProtection - Tool protection configuration (optional)
|
|
44
|
+
* @returns Tool execution context or undefined if not needed
|
|
45
|
+
*/
|
|
46
|
+
buildContext(toolName: string, userDid: string | undefined, sessionId: string | undefined, delegationToken: string | undefined, toolProtection: ToolProtection | null): Promise<ToolExecutionContext | undefined>;
|
|
47
|
+
/**
|
|
48
|
+
* Resolve OAuth provider for a tool
|
|
49
|
+
*
|
|
50
|
+
* Phase 2: Uses ProviderResolver with priority-based resolution
|
|
51
|
+
*
|
|
52
|
+
* @param toolProtection - Tool protection configuration
|
|
53
|
+
* @returns Provider name or throws error if not found
|
|
54
|
+
*/
|
|
55
|
+
private resolveProvider;
|
|
56
|
+
}
|
|
57
|
+
//# sourceMappingURL=tool-context-builder.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-context-builder.d.ts","sourceRoot":"","sources":["../../src/services/tool-context-builder.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAC1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAG/D,MAAM,WAAW,wBAAwB;IACvC,4DAA4D;IAC5D,aAAa,EAAE,gBAAgB,CAAC;IAEhC,gEAAgE;IAChE,aAAa,EAAE,kBAAkB,CAAC;IAElC,gEAAgE;IAChE,gBAAgB,EAAE,gBAAgB,CAAC;IAEnC,2CAA2C;IAC3C,SAAS,EAAE,MAAM,CAAC;IAElB,+CAA+C;IAC/C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD;AAED;;;;;;GAMG;AACH,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,MAAM,CAEZ;gBAEU,MAAM,EAAE,wBAAwB;IAU5C;;;;;;;;;OASG;IACG,YAAY,CAChB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,eAAe,EAAE,MAAM,GAAG,SAAS,EACnC,cAAc,EAAE,cAAc,GAAG,IAAI,GACpC,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC;IAqE5C;;;;;;;OAOG;YACW,eAAe;CAsB9B"}
|