toolhome 0.5.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 +21 -0
- package/README.md +295 -0
- package/README.zh.md +281 -0
- package/dist/server/app.d.ts +16 -0
- package/dist/server/app.d.ts.map +1 -0
- package/dist/server/app.js +166 -0
- package/dist/server/app.js.map +1 -0
- package/dist/server/cli/main.d.ts +3 -0
- package/dist/server/cli/main.d.ts.map +1 -0
- package/dist/server/cli/main.js +587 -0
- package/dist/server/cli/main.js.map +1 -0
- package/dist/server/cli-plane/allow-list.d.ts +18 -0
- package/dist/server/cli-plane/allow-list.d.ts.map +1 -0
- package/dist/server/cli-plane/allow-list.js +28 -0
- package/dist/server/cli-plane/allow-list.js.map +1 -0
- package/dist/server/cli-plane/cli-service.d.ts +41 -0
- package/dist/server/cli-plane/cli-service.d.ts.map +1 -0
- package/dist/server/cli-plane/cli-service.js +282 -0
- package/dist/server/cli-plane/cli-service.js.map +1 -0
- package/dist/server/cli-plane/frames.d.ts +26 -0
- package/dist/server/cli-plane/frames.d.ts.map +1 -0
- package/dist/server/cli-plane/frames.js +12 -0
- package/dist/server/cli-plane/frames.js.map +1 -0
- package/dist/server/cli-plane/models.d.ts +174 -0
- package/dist/server/cli-plane/models.d.ts.map +1 -0
- package/dist/server/cli-plane/models.js +169 -0
- package/dist/server/cli-plane/models.js.map +1 -0
- package/dist/server/cli-plane/routes.d.ts +32 -0
- package/dist/server/cli-plane/routes.d.ts.map +1 -0
- package/dist/server/cli-plane/routes.js +94 -0
- package/dist/server/cli-plane/routes.js.map +1 -0
- package/dist/server/cli-plane/runner.d.ts +41 -0
- package/dist/server/cli-plane/runner.d.ts.map +1 -0
- package/dist/server/cli-plane/runner.js +163 -0
- package/dist/server/cli-plane/runner.js.map +1 -0
- package/dist/server/cli-plane/status.d.ts +12 -0
- package/dist/server/cli-plane/status.d.ts.map +1 -0
- package/dist/server/cli-plane/status.js +24 -0
- package/dist/server/cli-plane/status.js.map +1 -0
- package/dist/server/config/harness-import.d.ts +44 -0
- package/dist/server/config/harness-import.d.ts.map +1 -0
- package/dist/server/config/harness-import.js +166 -0
- package/dist/server/config/harness-import.js.map +1 -0
- package/dist/server/config.d.ts +20 -0
- package/dist/server/config.d.ts.map +1 -0
- package/dist/server/config.js +80 -0
- package/dist/server/config.js.map +1 -0
- package/dist/server/control/client.d.ts +24 -0
- package/dist/server/control/client.d.ts.map +1 -0
- package/dist/server/control/client.js +115 -0
- package/dist/server/control/client.js.map +1 -0
- package/dist/server/control/control-api.d.ts +39 -0
- package/dist/server/control/control-api.d.ts.map +1 -0
- package/dist/server/control/control-api.js +291 -0
- package/dist/server/control/control-api.js.map +1 -0
- package/dist/server/control/control-service.d.ts +713 -0
- package/dist/server/control/control-service.d.ts.map +1 -0
- package/dist/server/control/control-service.js +693 -0
- package/dist/server/control/control-service.js.map +1 -0
- package/dist/server/control/openapi.d.ts +2 -0
- package/dist/server/control/openapi.d.ts.map +1 -0
- package/dist/server/control/openapi.js +370 -0
- package/dist/server/control/openapi.js.map +1 -0
- package/dist/server/data-plane/data-plane.d.ts +20 -0
- package/dist/server/data-plane/data-plane.d.ts.map +1 -0
- package/dist/server/data-plane/data-plane.js +335 -0
- package/dist/server/data-plane/data-plane.js.map +1 -0
- package/dist/server/data-plane/gateway-server.d.ts +13 -0
- package/dist/server/data-plane/gateway-server.d.ts.map +1 -0
- package/dist/server/data-plane/gateway-server.js +796 -0
- package/dist/server/data-plane/gateway-server.js.map +1 -0
- package/dist/server/data-plane/legacy-session.d.ts +18 -0
- package/dist/server/data-plane/legacy-session.d.ts.map +1 -0
- package/dist/server/data-plane/legacy-session.js +137 -0
- package/dist/server/data-plane/legacy-session.js.map +1 -0
- package/dist/server/data-plane/projection.d.ts +18 -0
- package/dist/server/data-plane/projection.d.ts.map +1 -0
- package/dist/server/data-plane/projection.js +44 -0
- package/dist/server/data-plane/projection.js.map +1 -0
- package/dist/server/data-plane/registry.d.ts +31 -0
- package/dist/server/data-plane/registry.d.ts.map +1 -0
- package/dist/server/data-plane/registry.js +129 -0
- package/dist/server/data-plane/registry.js.map +1 -0
- package/dist/server/data-plane/task-extension.d.ts +7 -0
- package/dist/server/data-plane/task-extension.d.ts.map +1 -0
- package/dist/server/data-plane/task-extension.js +33 -0
- package/dist/server/data-plane/task-extension.js.map +1 -0
- package/dist/server/data-plane/virtualization.d.ts +35 -0
- package/dist/server/data-plane/virtualization.d.ts.map +1 -0
- package/dist/server/data-plane/virtualization.js +228 -0
- package/dist/server/data-plane/virtualization.js.map +1 -0
- package/dist/server/domain/errors.d.ts +9 -0
- package/dist/server/domain/errors.d.ts.map +1 -0
- package/dist/server/domain/errors.js +19 -0
- package/dist/server/domain/errors.js.map +1 -0
- package/dist/server/domain/models.d.ts +720 -0
- package/dist/server/domain/models.d.ts.map +1 -0
- package/dist/server/domain/models.js +327 -0
- package/dist/server/domain/models.js.map +1 -0
- package/dist/server/main.d.ts +2 -0
- package/dist/server/main.d.ts.map +1 -0
- package/dist/server/main.js +37 -0
- package/dist/server/main.js.map +1 -0
- package/dist/server/manage/management-mcp.d.ts +27 -0
- package/dist/server/manage/management-mcp.d.ts.map +1 -0
- package/dist/server/manage/management-mcp.js +431 -0
- package/dist/server/manage/management-mcp.js.map +1 -0
- package/dist/server/market/catalog.d.ts +111 -0
- package/dist/server/market/catalog.d.ts.map +1 -0
- package/dist/server/market/catalog.js +784 -0
- package/dist/server/market/catalog.js.map +1 -0
- package/dist/server/market/installers.d.ts +64 -0
- package/dist/server/market/installers.d.ts.map +1 -0
- package/dist/server/market/installers.js +181 -0
- package/dist/server/market/installers.js.map +1 -0
- package/dist/server/market/market-service.d.ts +146 -0
- package/dist/server/market/market-service.d.ts.map +1 -0
- package/dist/server/market/market-service.js +930 -0
- package/dist/server/market/market-service.js.map +1 -0
- package/dist/server/observability/call-recorder.d.ts +22 -0
- package/dist/server/observability/call-recorder.d.ts.map +1 -0
- package/dist/server/observability/call-recorder.js +99 -0
- package/dist/server/observability/call-recorder.js.map +1 -0
- package/dist/server/observability/logger.d.ts +16 -0
- package/dist/server/observability/logger.d.ts.map +1 -0
- package/dist/server/observability/logger.js +25 -0
- package/dist/server/observability/logger.js.map +1 -0
- package/dist/server/security/api-keys.d.ts +15 -0
- package/dist/server/security/api-keys.d.ts.map +1 -0
- package/dist/server/security/api-keys.js +35 -0
- package/dist/server/security/api-keys.js.map +1 -0
- package/dist/server/security/auth-service.d.ts +26 -0
- package/dist/server/security/auth-service.d.ts.map +1 -0
- package/dist/server/security/auth-service.js +76 -0
- package/dist/server/security/auth-service.js.map +1 -0
- package/dist/server/security/control-session.d.ts +9 -0
- package/dist/server/security/control-session.d.ts.map +1 -0
- package/dist/server/security/control-session.js +51 -0
- package/dist/server/security/control-session.js.map +1 -0
- package/dist/server/security/credential-materializer.d.ts +20 -0
- package/dist/server/security/credential-materializer.d.ts.map +1 -0
- package/dist/server/security/credential-materializer.js +82 -0
- package/dist/server/security/credential-materializer.js.map +1 -0
- package/dist/server/security/cursor-codec.d.ts +13 -0
- package/dist/server/security/cursor-codec.d.ts.map +1 -0
- package/dist/server/security/cursor-codec.js +43 -0
- package/dist/server/security/cursor-codec.js.map +1 -0
- package/dist/server/security/oauth-routes.d.ts +21 -0
- package/dist/server/security/oauth-routes.d.ts.map +1 -0
- package/dist/server/security/oauth-routes.js +100 -0
- package/dist/server/security/oauth-routes.js.map +1 -0
- package/dist/server/security/oauth-server.d.ts +16 -0
- package/dist/server/security/oauth-server.d.ts.map +1 -0
- package/dist/server/security/oauth-server.js +544 -0
- package/dist/server/security/oauth-server.js.map +1 -0
- package/dist/server/security/secret-box.d.ts +7 -0
- package/dist/server/security/secret-box.d.ts.map +1 -0
- package/dist/server/security/secret-box.js +27 -0
- package/dist/server/security/secret-box.js.map +1 -0
- package/dist/server/security/secure-action.d.ts +21 -0
- package/dist/server/security/secure-action.d.ts.map +1 -0
- package/dist/server/security/secure-action.js +69 -0
- package/dist/server/security/secure-action.js.map +1 -0
- package/dist/server/storage/sqlite-store.d.ts +70 -0
- package/dist/server/storage/sqlite-store.d.ts.map +1 -0
- package/dist/server/storage/sqlite-store.js +1522 -0
- package/dist/server/storage/sqlite-store.js.map +1 -0
- package/dist/server/storage/store.d.ts +82 -0
- package/dist/server/storage/store.d.ts.map +1 -0
- package/dist/server/storage/store.js +2 -0
- package/dist/server/storage/store.js.map +1 -0
- package/dist/server/upstream/adapter.d.ts +52 -0
- package/dist/server/upstream/adapter.d.ts.map +1 -0
- package/dist/server/upstream/adapter.js +807 -0
- package/dist/server/upstream/adapter.js.map +1 -0
- package/dist/server/upstream/credential-resolver.d.ts +14 -0
- package/dist/server/upstream/credential-resolver.d.ts.map +1 -0
- package/dist/server/upstream/credential-resolver.js +56 -0
- package/dist/server/upstream/credential-resolver.js.map +1 -0
- package/dist/server/upstream/extension-transport.d.ts +17 -0
- package/dist/server/upstream/extension-transport.d.ts.map +1 -0
- package/dist/server/upstream/extension-transport.js +207 -0
- package/dist/server/upstream/extension-transport.js.map +1 -0
- package/dist/server/upstream/manager.d.ts +23 -0
- package/dist/server/upstream/manager.d.ts.map +1 -0
- package/dist/server/upstream/manager.js +324 -0
- package/dist/server/upstream/manager.js.map +1 -0
- package/dist/server/upstream/oauth-provider.d.ts +38 -0
- package/dist/server/upstream/oauth-provider.d.ts.map +1 -0
- package/dist/server/upstream/oauth-provider.js +225 -0
- package/dist/server/upstream/oauth-provider.js.map +1 -0
- package/dist/server/upstream/oauth-refresh.d.ts +65 -0
- package/dist/server/upstream/oauth-refresh.d.ts.map +1 -0
- package/dist/server/upstream/oauth-refresh.js +240 -0
- package/dist/server/upstream/oauth-refresh.js.map +1 -0
- package/dist/server/upstream/oauth-routes.d.ts +17 -0
- package/dist/server/upstream/oauth-routes.d.ts.map +1 -0
- package/dist/server/upstream/oauth-routes.js +87 -0
- package/dist/server/upstream/oauth-routes.js.map +1 -0
- package/dist/server/upstream/oauth-service.d.ts +143 -0
- package/dist/server/upstream/oauth-service.d.ts.map +1 -0
- package/dist/server/upstream/oauth-service.js +217 -0
- package/dist/server/upstream/oauth-service.js.map +1 -0
- package/dist/server/upstream/stable-json.d.ts +3 -0
- package/dist/server/upstream/stable-json.d.ts.map +1 -0
- package/dist/server/upstream/stable-json.js +17 -0
- package/dist/server/upstream/stable-json.js.map +1 -0
- package/package.json +69 -0
- package/skills/toolhome/SKILL.md +238 -0
- package/skills/toolhome/references/cli-reference.md +242 -0
- package/skills/toolhome/references/deployment.md +113 -0
- package/skills/toolhome/references/market-guide.md +128 -0
- package/skills/toolhome/references/oauth-guide.md +96 -0
- package/skills/toolhome/references/troubleshooting.md +133 -0
|
@@ -0,0 +1,324 @@
|
|
|
1
|
+
import { ProtocolError, } from '@modelcontextprotocol/client';
|
|
2
|
+
import { AppError, errorMessage } from '../domain/errors.js';
|
|
3
|
+
import { UpstreamAdapter, } from './adapter.js';
|
|
4
|
+
export class UpstreamManager {
|
|
5
|
+
#adapters = new Map();
|
|
6
|
+
#refreshing = new Map();
|
|
7
|
+
#listeners = new Set();
|
|
8
|
+
#recovering = new Map();
|
|
9
|
+
#store;
|
|
10
|
+
#credentials;
|
|
11
|
+
#logger;
|
|
12
|
+
#closed = false;
|
|
13
|
+
constructor(store, credentials, logger) {
|
|
14
|
+
this.#store = store;
|
|
15
|
+
this.#credentials = credentials;
|
|
16
|
+
this.#logger = logger;
|
|
17
|
+
}
|
|
18
|
+
subscribe(listener) {
|
|
19
|
+
this.#listeners.add(listener);
|
|
20
|
+
return () => this.#listeners.delete(listener);
|
|
21
|
+
}
|
|
22
|
+
async execute(serverId, request, context, clientCapabilities, transforms = {}) {
|
|
23
|
+
const server = this.#requireServer(serverId, true);
|
|
24
|
+
const adapter = await this.#adapterFor(server);
|
|
25
|
+
try {
|
|
26
|
+
const result = await adapter.execute(request, {
|
|
27
|
+
context,
|
|
28
|
+
clientCapabilities,
|
|
29
|
+
...transforms,
|
|
30
|
+
});
|
|
31
|
+
this.#markReady(server, adapter);
|
|
32
|
+
return result;
|
|
33
|
+
}
|
|
34
|
+
catch (error) {
|
|
35
|
+
if (this.#isConnectionFailure(error))
|
|
36
|
+
this.#markFailure(server, error);
|
|
37
|
+
throw error;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
async notify(serverId, notification, context, clientCapabilities) {
|
|
41
|
+
const server = this.#requireServer(serverId, true);
|
|
42
|
+
const adapter = await this.#adapterFor(server);
|
|
43
|
+
await adapter.notify(notification, clientCapabilities, { context, clientCapabilities });
|
|
44
|
+
}
|
|
45
|
+
async notifyDetached(serverId, notification, clientCapabilities) {
|
|
46
|
+
const server = this.#requireServer(serverId, true);
|
|
47
|
+
const adapter = await this.#adapterFor(server);
|
|
48
|
+
await adapter.notify(notification, clientCapabilities);
|
|
49
|
+
}
|
|
50
|
+
async subscribeResources(serverId, uris, clientCapabilities) {
|
|
51
|
+
const server = this.#requireServer(serverId, true);
|
|
52
|
+
const adapter = await this.#adapterFor(server);
|
|
53
|
+
return adapter.subscribeResources(uris, clientCapabilities);
|
|
54
|
+
}
|
|
55
|
+
refresh(serverId) {
|
|
56
|
+
const existing = this.#refreshing.get(serverId);
|
|
57
|
+
if (existing)
|
|
58
|
+
return existing;
|
|
59
|
+
const operation = this.#refreshNow(serverId).finally(() => this.#refreshing.delete(serverId));
|
|
60
|
+
this.#refreshing.set(serverId, operation);
|
|
61
|
+
return operation;
|
|
62
|
+
}
|
|
63
|
+
async restart(serverId) {
|
|
64
|
+
const server = this.#requireServer(serverId, false);
|
|
65
|
+
const refreshing = this.#refreshing.get(serverId);
|
|
66
|
+
const managed = this.#adapters.get(serverId);
|
|
67
|
+
if (managed)
|
|
68
|
+
await managed.adapter.restart();
|
|
69
|
+
await refreshing?.catch(() => undefined);
|
|
70
|
+
const state = this.#store.getRuntimeState(serverId);
|
|
71
|
+
this.#saveRuntime(server, {
|
|
72
|
+
status: 'connecting',
|
|
73
|
+
processId: null,
|
|
74
|
+
restartCount: (state?.restartCount ?? 0) + 1,
|
|
75
|
+
lastError: null,
|
|
76
|
+
});
|
|
77
|
+
return this.refresh(serverId);
|
|
78
|
+
}
|
|
79
|
+
async remove(serverId) {
|
|
80
|
+
const managed = this.#adapters.get(serverId);
|
|
81
|
+
this.#adapters.delete(serverId);
|
|
82
|
+
if (managed)
|
|
83
|
+
await managed.adapter.close();
|
|
84
|
+
}
|
|
85
|
+
async invalidate(serverId) {
|
|
86
|
+
const refreshing = this.#refreshing.get(serverId);
|
|
87
|
+
await this.remove(serverId);
|
|
88
|
+
await refreshing?.catch(() => undefined);
|
|
89
|
+
this.#store.deleteSnapshot(serverId);
|
|
90
|
+
const server = this.#store.getServer(serverId);
|
|
91
|
+
if (server) {
|
|
92
|
+
this.#saveRuntime(server, {
|
|
93
|
+
status: server.enabled ? 'unknown' : 'disabled',
|
|
94
|
+
protocolVersion: null,
|
|
95
|
+
protocolEra: null,
|
|
96
|
+
processId: null,
|
|
97
|
+
lastError: null,
|
|
98
|
+
});
|
|
99
|
+
}
|
|
100
|
+
this.#emitCapabilityChanges(serverId);
|
|
101
|
+
}
|
|
102
|
+
async close() {
|
|
103
|
+
this.#closed = true;
|
|
104
|
+
const adapters = [...this.#adapters.values()];
|
|
105
|
+
this.#adapters.clear();
|
|
106
|
+
await Promise.allSettled(adapters.map((item) => item.adapter.close()));
|
|
107
|
+
await Promise.allSettled([...this.#refreshing.values(), ...this.#recovering.values()]);
|
|
108
|
+
}
|
|
109
|
+
async #refreshNow(serverId) {
|
|
110
|
+
const server = this.#requireServer(serverId, false);
|
|
111
|
+
const adapter = await this.#adapterFor(server);
|
|
112
|
+
this.#saveRuntime(server, { status: 'connecting', lastError: null });
|
|
113
|
+
try {
|
|
114
|
+
const previous = this.#store.getSnapshot(serverId);
|
|
115
|
+
const snapshot = await adapter.discoverSnapshot(previous?.version ?? 0);
|
|
116
|
+
this.#store.saveSnapshot(snapshot);
|
|
117
|
+
if (previous?.fingerprint !== snapshot.fingerprint)
|
|
118
|
+
this.#emitCapabilityChanges(serverId);
|
|
119
|
+
this.#saveRuntime(server, {
|
|
120
|
+
status: server.enabled ? 'ready' : 'disabled',
|
|
121
|
+
protocolVersion: snapshot.protocolVersion,
|
|
122
|
+
protocolEra: snapshot.protocolEra,
|
|
123
|
+
processId: adapter.processId(),
|
|
124
|
+
lastSuccessAt: new Date().toISOString(),
|
|
125
|
+
lastError: null,
|
|
126
|
+
});
|
|
127
|
+
this.#store.appendEvent({
|
|
128
|
+
level: 'info',
|
|
129
|
+
type: 'server.refreshed',
|
|
130
|
+
serverId,
|
|
131
|
+
message: `Refreshed ${server.slug}`,
|
|
132
|
+
detail: {
|
|
133
|
+
protocolVersion: snapshot.protocolVersion,
|
|
134
|
+
protocolEra: snapshot.protocolEra,
|
|
135
|
+
tools: snapshot.tools.length,
|
|
136
|
+
resources: snapshot.resources.length,
|
|
137
|
+
prompts: snapshot.prompts.length,
|
|
138
|
+
},
|
|
139
|
+
});
|
|
140
|
+
return snapshot;
|
|
141
|
+
}
|
|
142
|
+
catch (error) {
|
|
143
|
+
this.#markFailure(server, error);
|
|
144
|
+
throw error;
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
async #adapterFor(server) {
|
|
148
|
+
if (this.#closed)
|
|
149
|
+
throw new AppError('upstream_closed', 'Upstream manager is closed', 503);
|
|
150
|
+
const existing = this.#adapters.get(server.id);
|
|
151
|
+
if (existing?.updatedAt === server.updatedAt)
|
|
152
|
+
return existing.adapter;
|
|
153
|
+
if (existing)
|
|
154
|
+
await existing.adapter.close();
|
|
155
|
+
const adapter = new UpstreamAdapter(server, this.#credentials, this.#logger, (event) => this.#onEvent(event));
|
|
156
|
+
this.#adapters.set(server.id, { updatedAt: server.updatedAt, adapter });
|
|
157
|
+
return adapter;
|
|
158
|
+
}
|
|
159
|
+
#onEvent(event) {
|
|
160
|
+
if (this.#closed)
|
|
161
|
+
return;
|
|
162
|
+
if (event.type === 'stderr') {
|
|
163
|
+
this.#store.appendEvent({
|
|
164
|
+
level: 'debug',
|
|
165
|
+
type: 'server.stderr',
|
|
166
|
+
serverId: event.serverId,
|
|
167
|
+
message: event.message,
|
|
168
|
+
detail: {},
|
|
169
|
+
});
|
|
170
|
+
return;
|
|
171
|
+
}
|
|
172
|
+
if (event.type === 'connection_closed') {
|
|
173
|
+
if (this.#recovering.has(event.serverId))
|
|
174
|
+
return;
|
|
175
|
+
const operation = this.#recover(event.serverId).finally(() => this.#recovering.delete(event.serverId));
|
|
176
|
+
this.#recovering.set(event.serverId, operation);
|
|
177
|
+
return;
|
|
178
|
+
}
|
|
179
|
+
if (event.type === 'resource_updated') {
|
|
180
|
+
for (const listener of this.#listeners)
|
|
181
|
+
listener(event);
|
|
182
|
+
return;
|
|
183
|
+
}
|
|
184
|
+
void this.refresh(event.serverId)
|
|
185
|
+
.then(() => undefined)
|
|
186
|
+
.catch((error) => {
|
|
187
|
+
this.#logger.warn('Capability refresh after list change failed', {
|
|
188
|
+
serverId: event.serverId,
|
|
189
|
+
error: errorMessage(error),
|
|
190
|
+
});
|
|
191
|
+
});
|
|
192
|
+
}
|
|
193
|
+
async #recover(serverId) {
|
|
194
|
+
const server = this.#store.getServer(serverId);
|
|
195
|
+
if (!server)
|
|
196
|
+
return;
|
|
197
|
+
const refreshing = this.#refreshing.get(serverId);
|
|
198
|
+
await this.remove(serverId);
|
|
199
|
+
await refreshing?.catch(() => undefined);
|
|
200
|
+
if (this.#closed)
|
|
201
|
+
return;
|
|
202
|
+
const state = this.#store.getRuntimeState(serverId);
|
|
203
|
+
const restart = server.kind === 'home' && server.enabled && server.settings.restart !== 'never';
|
|
204
|
+
this.#store.appendEvent({
|
|
205
|
+
level: 'warn',
|
|
206
|
+
type: 'server.connection_closed',
|
|
207
|
+
serverId,
|
|
208
|
+
message: `Connection to ${server.slug} closed`,
|
|
209
|
+
detail: { restart },
|
|
210
|
+
});
|
|
211
|
+
if (!restart) {
|
|
212
|
+
this.#saveRuntime(server, {
|
|
213
|
+
status: server.enabled ? 'unreachable' : 'disabled',
|
|
214
|
+
processId: null,
|
|
215
|
+
lastError: 'Upstream connection closed',
|
|
216
|
+
});
|
|
217
|
+
return;
|
|
218
|
+
}
|
|
219
|
+
this.#saveRuntime(server, {
|
|
220
|
+
status: 'connecting',
|
|
221
|
+
processId: null,
|
|
222
|
+
restartCount: (state?.restartCount ?? 0) + 1,
|
|
223
|
+
lastError: null,
|
|
224
|
+
});
|
|
225
|
+
try {
|
|
226
|
+
await this.refresh(serverId);
|
|
227
|
+
}
|
|
228
|
+
catch (error) {
|
|
229
|
+
this.#saveRuntime(server, {
|
|
230
|
+
status: 'start-failed',
|
|
231
|
+
processId: null,
|
|
232
|
+
lastError: errorMessage(error),
|
|
233
|
+
});
|
|
234
|
+
}
|
|
235
|
+
}
|
|
236
|
+
#emitCapabilityChanges(serverId) {
|
|
237
|
+
for (const listener of this.#listeners) {
|
|
238
|
+
listener({ type: 'tools_changed', serverId });
|
|
239
|
+
listener({ type: 'prompts_changed', serverId });
|
|
240
|
+
listener({ type: 'resources_changed', serverId });
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
#requireServer(serverId, requireEnabled) {
|
|
244
|
+
const server = this.#store.getServer(serverId);
|
|
245
|
+
if (!server)
|
|
246
|
+
throw new AppError('server_not_found', 'Server not found', 404);
|
|
247
|
+
if (requireEnabled && !server.enabled) {
|
|
248
|
+
throw new AppError('server_disabled', `Server ${server.slug} is disabled`, 503);
|
|
249
|
+
}
|
|
250
|
+
return server;
|
|
251
|
+
}
|
|
252
|
+
#markReady(server, adapter) {
|
|
253
|
+
const state = this.#store.getRuntimeState(server.id);
|
|
254
|
+
if (state?.status === 'ready')
|
|
255
|
+
return;
|
|
256
|
+
const snapshot = this.#store.getSnapshot(server.id);
|
|
257
|
+
this.#saveRuntime(server, {
|
|
258
|
+
status: server.enabled ? 'ready' : 'disabled',
|
|
259
|
+
protocolVersion: snapshot?.protocolVersion ?? null,
|
|
260
|
+
protocolEra: snapshot?.protocolEra ?? null,
|
|
261
|
+
processId: adapter.processId(),
|
|
262
|
+
lastSuccessAt: new Date().toISOString(),
|
|
263
|
+
lastError: null,
|
|
264
|
+
});
|
|
265
|
+
}
|
|
266
|
+
#markFailure(server, error) {
|
|
267
|
+
const message = errorMessage(error);
|
|
268
|
+
const lower = message.toLowerCase();
|
|
269
|
+
const status = lower.includes('401') || lower.includes('unauthorized') ? 'auth-required' : 'unreachable';
|
|
270
|
+
this.#saveRuntime(server, { status, lastError: message });
|
|
271
|
+
this.#store.appendEvent({
|
|
272
|
+
level: 'error',
|
|
273
|
+
type: 'server.error',
|
|
274
|
+
serverId: server.id,
|
|
275
|
+
message: `Server ${server.slug} failed`,
|
|
276
|
+
detail: { error: message },
|
|
277
|
+
});
|
|
278
|
+
}
|
|
279
|
+
#saveRuntime(server, patch) {
|
|
280
|
+
const current = this.#store.getRuntimeState(server.id);
|
|
281
|
+
return this.#store.saveRuntimeState({
|
|
282
|
+
serverId: server.id,
|
|
283
|
+
status: patch.status ?? current?.status ?? 'unknown',
|
|
284
|
+
protocolVersion: patch.protocolVersion === undefined
|
|
285
|
+
? (current?.protocolVersion ?? null)
|
|
286
|
+
: patch.protocolVersion,
|
|
287
|
+
protocolEra: patch.protocolEra === undefined ? (current?.protocolEra ?? null) : patch.protocolEra,
|
|
288
|
+
processId: patch.processId === undefined ? (current?.processId ?? null) : patch.processId,
|
|
289
|
+
restartCount: patch.restartCount ?? current?.restartCount ?? 0,
|
|
290
|
+
lastSuccessAt: patch.lastSuccessAt ?? current?.lastSuccessAt ?? null,
|
|
291
|
+
lastError: patch.lastError === undefined ? (current?.lastError ?? null) : patch.lastError,
|
|
292
|
+
updatedAt: new Date().toISOString(),
|
|
293
|
+
});
|
|
294
|
+
}
|
|
295
|
+
#isConnectionFailure(error) {
|
|
296
|
+
if (ProtocolError.isInstance(error))
|
|
297
|
+
return false;
|
|
298
|
+
if (error instanceof AppError) {
|
|
299
|
+
if (error.code === 'upstream_jsonrpc_error')
|
|
300
|
+
return false;
|
|
301
|
+
if ([
|
|
302
|
+
'upstream_closed',
|
|
303
|
+
'upstream_restarted',
|
|
304
|
+
'upstream_stream_ended',
|
|
305
|
+
'upstream_timeout',
|
|
306
|
+
].includes(error.code)) {
|
|
307
|
+
return true;
|
|
308
|
+
}
|
|
309
|
+
}
|
|
310
|
+
const message = errorMessage(error).toLowerCase();
|
|
311
|
+
return [
|
|
312
|
+
'econnrefused',
|
|
313
|
+
'econnreset',
|
|
314
|
+
'enotfound',
|
|
315
|
+
'fetch failed',
|
|
316
|
+
'socket',
|
|
317
|
+
'connection closed',
|
|
318
|
+
'unauthorized',
|
|
319
|
+
'status 401',
|
|
320
|
+
'status 403',
|
|
321
|
+
].some((fragment) => message.includes(fragment));
|
|
322
|
+
}
|
|
323
|
+
}
|
|
324
|
+
//# sourceMappingURL=manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../src/upstream/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,aAAa,GAId,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAI7D,OAAO,EACL,eAAe,GAIhB,MAAM,cAAc,CAAC;AAUtB,MAAM,OAAO,eAAe;IACjB,SAAS,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC9C,WAAW,GAAG,IAAI,GAAG,EAAuC,CAAC;IAC7D,UAAU,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC9C,WAAW,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC/C,MAAM,CAAQ;IACd,YAAY,CAAqB;IACjC,OAAO,CAAS;IACzB,OAAO,GAAG,KAAK,CAAC;IAEhB,YAAY,KAAY,EAAE,WAA+B,EAAE,MAAc;QACvE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,SAAS,CAAC,QAA+B;QACvC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,OAAwB,EACxB,OAAsB,EACtB,kBAAsC,EACtC,UAAU,GAAqB,EAAE;QAEjC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;gBAC5C,OAAO;gBACP,kBAAkB;gBAClB,GAAG,UAAU;aACd,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACjC,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC;gBAAE,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACvE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAAgB,EAChB,YAA0B,EAC1B,OAAsB,EACtB,kBAAsC;QAEtC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,kBAAkB,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAC1F,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,QAAgB,EAChB,YAA0B,EAC1B,kBAAsC;QAEtC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,QAAgB,EAChB,IAAc,EACd,kBAAsC;QAEtC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,kBAAkB,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;IAC9D,CAAC;IAED,OAAO,CAAC,QAAgB;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC9F,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC1C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB;QAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,OAAO;YAAE,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC7C,MAAM,UAAU,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,MAAM,EAAE,YAAY;YACpB,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC;YAC5C,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,OAAO;YAAE,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAgB;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5B,MAAM,UAAU,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU;gBAC/C,eAAe,EAAE,IAAI;gBACrB,WAAW,EAAE,IAAI;gBACjB,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,MAAM,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACvE,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/C,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,IAAI,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,QAAQ,EAAE,WAAW,KAAK,QAAQ,CAAC,WAAW;gBAAE,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;YAC1F,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU;gBAC7C,eAAe,EAAE,QAAQ,CAAC,eAAe;gBACzC,WAAW,EAAE,QAAQ,CAAC,WAAW;gBACjC,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;gBAC9B,aAAa,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACvC,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;gBACtB,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,kBAAkB;gBACxB,QAAQ;gBACR,OAAO,EAAE,aAAa,MAAM,CAAC,IAAI,EAAE;gBACnC,MAAM,EAAE;oBACN,eAAe,EAAE,QAAQ,CAAC,eAAe;oBACzC,WAAW,EAAE,QAAQ,CAAC,WAAW;oBACjC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM;oBAC5B,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,MAAM;oBACpC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;iBACjC;aACF,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACjC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAoB;QACpC,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,QAAQ,CAAC,iBAAiB,EAAE,4BAA4B,EAAE,GAAG,CAAC,CAAC;QAC3F,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC/C,IAAI,QAAQ,EAAE,SAAS,KAAK,MAAM,CAAC,SAAS;YAAE,OAAO,QAAQ,CAAC,OAAO,CAAC;QACtE,IAAI,QAAQ;YAAE,MAAM,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC7C,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CACrF,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CACrB,CAAC;QACF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QACxE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,QAAQ,CAAC,KAAoB;QAC3B,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;gBACtB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,eAAe;gBACrB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,MAAM,EAAE,EAAE;aACX,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACvC,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAAE,OAAO;YACjD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAC3D,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CACxC,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YACtC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,UAAU;gBAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QACD,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;aAC9B,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACrB,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE;gBAC/D,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC;aAC3B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5B,MAAM,UAAU,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,KAAK,MAAM,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,KAAK,OAAO,CAAC;QAChG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YACtB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,0BAA0B;YAChC,QAAQ;YACR,OAAO,EAAE,iBAAiB,MAAM,CAAC,IAAI,SAAS;YAC9C,MAAM,EAAE,EAAE,OAAO,EAAE;SACpB,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,UAAU;gBACnD,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,4BAA4B;aACxC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,MAAM,EAAE,YAAY;YACpB,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC;YAC5C,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QACH,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACxB,MAAM,EAAE,cAAc;gBACtB,SAAS,EAAE,IAAI;gBACf,SAAS,EAAE,YAAY,CAAC,KAAK,CAAC;aAC/B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,sBAAsB,CAAC,QAAgB;QACrC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACvC,QAAQ,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC9C,QAAQ,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChD,QAAQ,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,cAAuB;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,GAAG,CAAC,CAAC;QAC7E,IAAI,cAAc,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,IAAI,QAAQ,CAAC,iBAAiB,EAAE,UAAU,MAAM,CAAC,IAAI,cAAc,EAAE,GAAG,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU,CAAC,MAAoB,EAAE,OAAwB;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrD,IAAI,KAAK,EAAE,MAAM,KAAK,OAAO;YAAE,OAAO;QACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;YACxB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU;YAC7C,eAAe,EAAE,QAAQ,EAAE,eAAe,IAAI,IAAI;YAClD,WAAW,EAAE,QAAQ,EAAE,WAAW,IAAI,IAAI;YAC1C,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;YAC9B,aAAa,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACvC,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC;IAED,YAAY,CAAC,MAAoB,EAAE,KAAc;QAC/C,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,MAAM,GACV,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,CAAC;QAC5F,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YACtB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,cAAc;YACpB,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,OAAO,EAAE,UAAU,MAAM,CAAC,IAAI,SAAS;YACvC,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,YAAY,CAAC,MAAoB,EAAE,KAA4B;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAClC,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,OAAO,EAAE,MAAM,IAAI,SAAS;YACpD,eAAe,EACb,KAAK,CAAC,eAAe,KAAK,SAAS;gBACjC,CAAC,CAAC,CAAC,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC;gBACpC,CAAC,CAAC,KAAK,CAAC,eAAe;YAC3B,WAAW,EACT,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW;YACtF,SAAS,EAAE,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS;YACzF,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,OAAO,EAAE,YAAY,IAAI,CAAC;YAC9D,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,OAAO,EAAE,aAAa,IAAI,IAAI;YACpE,SAAS,EAAE,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS;YACzF,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAAC,KAAc;QACjC,IAAI,aAAa,CAAC,UAAU,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAClD,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,IAAI,KAAK,CAAC,IAAI,KAAK,wBAAwB;gBAAE,OAAO,KAAK,CAAC;YAC1D,IACE;gBACE,iBAAiB;gBACjB,oBAAoB;gBACpB,uBAAuB;gBACvB,kBAAkB;aACnB,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EACtB,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAClD,OAAO;YACL,cAAc;YACd,YAAY;YACZ,WAAW;YACX,cAAc;YACd,QAAQ;YACR,mBAAmB;YACnB,cAAc;YACd,YAAY;YACZ,YAAY;SACb,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnD,CAAC;CACF"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { type OAuthClientInformationContext, type OAuthClientMetadata, type OAuthClientProvider, type OAuthDiscoveryState, type StoredOAuthClientInformation, type StoredOAuthTokens } from '@modelcontextprotocol/client';
|
|
2
|
+
import type { Store } from '../storage/store.js';
|
|
3
|
+
export interface StoredOAuthProviderOptions {
|
|
4
|
+
/**
|
|
5
|
+
* Whether to advertise URL-based client metadata (RFC 9728
|
|
6
|
+
* `client_id_metadata_document_supported`) to upstream authorization
|
|
7
|
+
* servers. When disabled, dynamic client registration is used instead.
|
|
8
|
+
* Some providers (e.g. Cloudflare's hosted MCP) fail to fetch metadata
|
|
9
|
+
* documents from proxied origins, so DCR is the reliable fallback.
|
|
10
|
+
*/
|
|
11
|
+
urlClientId?: boolean;
|
|
12
|
+
}
|
|
13
|
+
export declare class StoredOAuthProvider implements OAuthClientProvider {
|
|
14
|
+
#private;
|
|
15
|
+
readonly clientMetadataUrl?: string;
|
|
16
|
+
constructor(store: Store, credentialId: string, publicUrl: URL, options?: StoredOAuthProviderOptions);
|
|
17
|
+
get redirectUrl(): URL;
|
|
18
|
+
get clientMetadata(): OAuthClientMetadata;
|
|
19
|
+
validateResourceURL(serverUrl: string | URL, resource?: string): Promise<URL | undefined>;
|
|
20
|
+
state(): Promise<string>;
|
|
21
|
+
clientInformation(context?: OAuthClientInformationContext): StoredOAuthClientInformation | undefined;
|
|
22
|
+
saveClientInformation(clientInformation: StoredOAuthClientInformation): void;
|
|
23
|
+
tokens(context?: OAuthClientInformationContext): StoredOAuthTokens | undefined;
|
|
24
|
+
saveTokens(tokens: StoredOAuthTokens, context?: OAuthClientInformationContext): void;
|
|
25
|
+
redirectToAuthorization(authorizationUrl: URL): void;
|
|
26
|
+
saveCodeVerifier(codeVerifier: string): void;
|
|
27
|
+
codeVerifier(): string;
|
|
28
|
+
saveAuthorizationServerUrl(authorizationServerUrl: string): void;
|
|
29
|
+
authorizationServerUrl(): string | undefined;
|
|
30
|
+
saveResourceUrl(resourceUrl: string): void;
|
|
31
|
+
resourceUrl(): string | undefined;
|
|
32
|
+
saveDiscoveryState(discoveryState: OAuthDiscoveryState): void;
|
|
33
|
+
discoveryState(): OAuthDiscoveryState | undefined;
|
|
34
|
+
invalidateCredentials(scope: 'all' | 'client' | 'tokens' | 'verifier' | 'discovery'): void;
|
|
35
|
+
authorizationUrl(): string | undefined;
|
|
36
|
+
validateCallbackState(state: string): void;
|
|
37
|
+
}
|
|
38
|
+
//# sourceMappingURL=oauth-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"oauth-provider.d.ts","sourceRoot":"","sources":["../../../src/upstream/oauth-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,6BAA6B,EAClC,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,EACxB,KAAK,4BAA4B,EACjC,KAAK,iBAAiB,EACvB,MAAM,8BAA8B,CAAC;AAItC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAIjD,MAAM,WAAW,0BAA0B;IACzC;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED,qBAAa,mBAAoB,YAAW,mBAAmB;;IAC7D,QAAQ,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAKpC,YACE,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,GAAG,EACd,OAAO,GAAE,0BAA+B,EAYzC;IAED,IAAI,WAAW,IAAI,GAAG,CAErB;IAED,IAAI,cAAc,IAAI,mBAAmB,CAaxC;IAEK,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,GAAG,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,GAAG,SAAS,CAAC,CAuB9F;IAEK,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,CAI7B;IAED,iBAAiB,CACf,OAAO,CAAC,EAAE,6BAA6B,GACtC,4BAA4B,GAAG,SAAS,CAc1C;IAED,qBAAqB,CAAC,iBAAiB,EAAE,4BAA4B,GAAG,IAAI,CAM3E;IAED,MAAM,CAAC,OAAO,CAAC,EAAE,6BAA6B,GAAG,iBAAiB,GAAG,SAAS,CAgB7E;IAED,UAAU,CAAC,MAAM,EAAE,iBAAiB,EAAE,OAAO,CAAC,EAAE,6BAA6B,GAAG,IAAI,CAoBnF;IAED,uBAAuB,CAAC,gBAAgB,EAAE,GAAG,GAAG,IAAI,CAKnD;IAED,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAE3C;IAED,YAAY,IAAI,MAAM,CAMrB;IAED,0BAA0B,CAAC,sBAAsB,EAAE,MAAM,GAAG,IAAI,CAE/D;IAED,sBAAsB,IAAI,MAAM,GAAG,SAAS,CAE3C;IAED,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAEzC;IAED,WAAW,IAAI,MAAM,GAAG,SAAS,CAEhC;IAED,kBAAkB,CAAC,cAAc,EAAE,mBAAmB,GAAG,IAAI,CAE5D;IAED,cAAc,IAAI,mBAAmB,GAAG,SAAS,CAEhD;IAED,qBAAqB,CAAC,KAAK,EAAE,KAAK,GAAG,QAAQ,GAAG,QAAQ,GAAG,UAAU,GAAG,WAAW,GAAG,IAAI,CAiCzF;IAED,gBAAgB,IAAI,MAAM,GAAG,SAAS,CAErC;IAED,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAQzC;CAeF"}
|
|
@@ -0,0 +1,225 @@
|
|
|
1
|
+
import {} from '@modelcontextprotocol/client';
|
|
2
|
+
import { randomBytes } from 'node:crypto';
|
|
3
|
+
import { AppError } from '../domain/errors.js';
|
|
4
|
+
import { oauthCredentialPayloadSchema } from '../domain/models.js';
|
|
5
|
+
const FLOW_TTL_MS = 10 * 60 * 1_000;
|
|
6
|
+
export class StoredOAuthProvider {
|
|
7
|
+
clientMetadataUrl;
|
|
8
|
+
#store;
|
|
9
|
+
#credentialId;
|
|
10
|
+
#publicUrl;
|
|
11
|
+
constructor(store, credentialId, publicUrl, options = {}) {
|
|
12
|
+
this.#store = store;
|
|
13
|
+
this.#credentialId = credentialId;
|
|
14
|
+
this.#publicUrl = publicUrl;
|
|
15
|
+
if (publicUrl.protocol === 'https:' && (options.urlClientId ?? true)) {
|
|
16
|
+
this.clientMetadataUrl = new URL(`/oauth/upstream/client/${credentialId}`, publicUrl).toString();
|
|
17
|
+
}
|
|
18
|
+
this.#payload();
|
|
19
|
+
}
|
|
20
|
+
get redirectUrl() {
|
|
21
|
+
return new URL(`/oauth/upstream/callback/${this.#credentialId}`, this.#publicUrl);
|
|
22
|
+
}
|
|
23
|
+
get clientMetadata() {
|
|
24
|
+
const payload = this.#payload();
|
|
25
|
+
return {
|
|
26
|
+
redirect_uris: [this.redirectUrl.toString()],
|
|
27
|
+
token_endpoint_auth_method: payload.clientSecret ? 'client_secret_basic' : 'none',
|
|
28
|
+
grant_types: ['authorization_code', 'refresh_token'],
|
|
29
|
+
response_types: ['code'],
|
|
30
|
+
client_name: 'ToolHome',
|
|
31
|
+
client_uri: this.#publicUrl.toString(),
|
|
32
|
+
software_id: 'toolhome',
|
|
33
|
+
software_version: '0.1.0',
|
|
34
|
+
...(payload.scope === undefined ? {} : { scope: payload.scope }),
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
async validateResourceURL(serverUrl, resource) {
|
|
38
|
+
// The SDK discovers protected-resource metadata from the path-scoped
|
|
39
|
+
// well-known document, which some providers (e.g. Notion) publish with a
|
|
40
|
+
// different resource identifier than their authorization server accepts.
|
|
41
|
+
// Prefer the origin's well-known declaration when it is available, and
|
|
42
|
+
// fall back to the SDK-discovered value otherwise.
|
|
43
|
+
try {
|
|
44
|
+
const origin = new URL('/', serverUrl);
|
|
45
|
+
const response = await fetch(new URL('/.well-known/oauth-protected-resource', origin), {
|
|
46
|
+
headers: { Accept: 'application/json' },
|
|
47
|
+
redirect: 'follow',
|
|
48
|
+
});
|
|
49
|
+
if (response.ok) {
|
|
50
|
+
const body = (await response.json());
|
|
51
|
+
if (typeof body.resource === 'string' && body.resource.length > 0) {
|
|
52
|
+
return new URL(body.resource);
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
catch {
|
|
57
|
+
// Fall through to the SDK-discovered resource.
|
|
58
|
+
}
|
|
59
|
+
if (resource !== undefined)
|
|
60
|
+
return new URL(resource);
|
|
61
|
+
return new URL(serverUrl);
|
|
62
|
+
}
|
|
63
|
+
async state() {
|
|
64
|
+
const state = randomBytes(32).toString('base64url');
|
|
65
|
+
this.#update({ state });
|
|
66
|
+
return state;
|
|
67
|
+
}
|
|
68
|
+
clientInformation(context) {
|
|
69
|
+
const payload = this.#payload();
|
|
70
|
+
const saved = payload.clientInformation;
|
|
71
|
+
if (saved) {
|
|
72
|
+
if (context && saved.issuer && saved.issuer !== context.issuer)
|
|
73
|
+
return undefined;
|
|
74
|
+
return saved;
|
|
75
|
+
}
|
|
76
|
+
if (!payload.clientId)
|
|
77
|
+
return undefined;
|
|
78
|
+
if (context && payload.issuer && payload.issuer !== context.issuer)
|
|
79
|
+
return undefined;
|
|
80
|
+
return {
|
|
81
|
+
client_id: payload.clientId,
|
|
82
|
+
...(payload.clientSecret === undefined ? {} : { client_secret: payload.clientSecret }),
|
|
83
|
+
...(payload.issuer === undefined ? {} : { issuer: payload.issuer }),
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
saveClientInformation(clientInformation) {
|
|
87
|
+
this.#update({
|
|
88
|
+
clientInformation,
|
|
89
|
+
clientId: clientInformation.client_id,
|
|
90
|
+
clientSecret: clientInformation.client_secret,
|
|
91
|
+
});
|
|
92
|
+
}
|
|
93
|
+
tokens(context) {
|
|
94
|
+
const payload = this.#payload();
|
|
95
|
+
if (!payload.accessToken)
|
|
96
|
+
return undefined;
|
|
97
|
+
if (context && payload.issuer && payload.issuer !== context.issuer)
|
|
98
|
+
return undefined;
|
|
99
|
+
const expiresIn = payload.expiresAt
|
|
100
|
+
? Math.max(0, Math.floor((Date.parse(payload.expiresAt) - Date.now()) / 1_000))
|
|
101
|
+
: payload.expiresIn;
|
|
102
|
+
return {
|
|
103
|
+
access_token: payload.accessToken,
|
|
104
|
+
token_type: payload.tokenType,
|
|
105
|
+
...(payload.refreshToken === undefined ? {} : { refresh_token: payload.refreshToken }),
|
|
106
|
+
...(payload.idToken === undefined ? {} : { id_token: payload.idToken }),
|
|
107
|
+
...(payload.scope === undefined ? {} : { scope: payload.scope }),
|
|
108
|
+
...(expiresIn === undefined ? {} : { expires_in: expiresIn }),
|
|
109
|
+
...(payload.issuer === undefined ? {} : { issuer: payload.issuer }),
|
|
110
|
+
};
|
|
111
|
+
}
|
|
112
|
+
saveTokens(tokens, context) {
|
|
113
|
+
const current = this.#payload();
|
|
114
|
+
const expiresAt = tokens.expires_in === undefined
|
|
115
|
+
? undefined
|
|
116
|
+
: new Date(Date.now() + tokens.expires_in * 1_000).toISOString();
|
|
117
|
+
this.#update({
|
|
118
|
+
accessToken: tokens.access_token,
|
|
119
|
+
refreshToken: tokens.refresh_token ?? current.refreshToken,
|
|
120
|
+
idToken: tokens.id_token,
|
|
121
|
+
tokenType: tokens.token_type,
|
|
122
|
+
expiresIn: tokens.expires_in,
|
|
123
|
+
expiresAt,
|
|
124
|
+
scope: tokens.scope ?? current.scope,
|
|
125
|
+
issuer: tokens.issuer ?? context?.issuer ?? current.issuer,
|
|
126
|
+
authorizationUrl: undefined,
|
|
127
|
+
pendingExpiresAt: undefined,
|
|
128
|
+
state: undefined,
|
|
129
|
+
codeVerifier: undefined,
|
|
130
|
+
});
|
|
131
|
+
}
|
|
132
|
+
redirectToAuthorization(authorizationUrl) {
|
|
133
|
+
this.#update({
|
|
134
|
+
authorizationUrl: authorizationUrl.toString(),
|
|
135
|
+
pendingExpiresAt: new Date(Date.now() + FLOW_TTL_MS).toISOString(),
|
|
136
|
+
});
|
|
137
|
+
}
|
|
138
|
+
saveCodeVerifier(codeVerifier) {
|
|
139
|
+
this.#update({ codeVerifier });
|
|
140
|
+
}
|
|
141
|
+
codeVerifier() {
|
|
142
|
+
const verifier = this.#payload().codeVerifier;
|
|
143
|
+
if (!verifier) {
|
|
144
|
+
throw new AppError('oauth_flow_expired', 'OAuth authorization flow has expired', 400);
|
|
145
|
+
}
|
|
146
|
+
return verifier;
|
|
147
|
+
}
|
|
148
|
+
saveAuthorizationServerUrl(authorizationServerUrl) {
|
|
149
|
+
this.#update({ authorizationServerUrl });
|
|
150
|
+
}
|
|
151
|
+
authorizationServerUrl() {
|
|
152
|
+
return this.#payload().authorizationServerUrl;
|
|
153
|
+
}
|
|
154
|
+
saveResourceUrl(resourceUrl) {
|
|
155
|
+
this.#update({ resourceUrl });
|
|
156
|
+
}
|
|
157
|
+
resourceUrl() {
|
|
158
|
+
return this.#payload().resourceUrl;
|
|
159
|
+
}
|
|
160
|
+
saveDiscoveryState(discoveryState) {
|
|
161
|
+
this.#update({ discoveryState });
|
|
162
|
+
}
|
|
163
|
+
discoveryState() {
|
|
164
|
+
return this.#payload().discoveryState;
|
|
165
|
+
}
|
|
166
|
+
invalidateCredentials(scope) {
|
|
167
|
+
if (scope === 'all' || scope === 'client') {
|
|
168
|
+
this.#update({
|
|
169
|
+
clientInformation: undefined,
|
|
170
|
+
clientId: undefined,
|
|
171
|
+
clientSecret: undefined,
|
|
172
|
+
});
|
|
173
|
+
}
|
|
174
|
+
if (scope === 'all' || scope === 'tokens') {
|
|
175
|
+
this.#update({
|
|
176
|
+
accessToken: undefined,
|
|
177
|
+
refreshToken: undefined,
|
|
178
|
+
idToken: undefined,
|
|
179
|
+
expiresAt: undefined,
|
|
180
|
+
expiresIn: undefined,
|
|
181
|
+
issuer: undefined,
|
|
182
|
+
});
|
|
183
|
+
}
|
|
184
|
+
if (scope === 'all' || scope === 'verifier') {
|
|
185
|
+
this.#update({
|
|
186
|
+
authorizationUrl: undefined,
|
|
187
|
+
pendingExpiresAt: undefined,
|
|
188
|
+
state: undefined,
|
|
189
|
+
codeVerifier: undefined,
|
|
190
|
+
});
|
|
191
|
+
}
|
|
192
|
+
if (scope === 'all' || scope === 'discovery') {
|
|
193
|
+
this.#update({
|
|
194
|
+
discoveryState: undefined,
|
|
195
|
+
authorizationServerUrl: undefined,
|
|
196
|
+
resourceUrl: undefined,
|
|
197
|
+
});
|
|
198
|
+
}
|
|
199
|
+
}
|
|
200
|
+
authorizationUrl() {
|
|
201
|
+
return this.#payload().authorizationUrl;
|
|
202
|
+
}
|
|
203
|
+
validateCallbackState(state) {
|
|
204
|
+
const payload = this.#payload();
|
|
205
|
+
if (!payload.state || payload.state !== state) {
|
|
206
|
+
throw new AppError('oauth_state_mismatch', 'OAuth callback state is invalid', 400);
|
|
207
|
+
}
|
|
208
|
+
if (!payload.pendingExpiresAt || Date.parse(payload.pendingExpiresAt) <= Date.now()) {
|
|
209
|
+
throw new AppError('oauth_flow_expired', 'OAuth authorization flow has expired', 400);
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
#payload() {
|
|
213
|
+
const payload = this.#store.getCredentialPayload(this.#credentialId);
|
|
214
|
+
if (!payload || payload.type !== 'oauth') {
|
|
215
|
+
throw new AppError('oauth_credential_invalid', 'OAuth credential is unavailable', 404);
|
|
216
|
+
}
|
|
217
|
+
return payload;
|
|
218
|
+
}
|
|
219
|
+
#update(changes) {
|
|
220
|
+
const next = oauthCredentialPayloadSchema.parse({ ...this.#payload(), ...changes });
|
|
221
|
+
this.#store.updateCredential(this.#credentialId, { payload: next });
|
|
222
|
+
return next;
|
|
223
|
+
}
|
|
224
|
+
}
|
|
225
|
+
//# sourceMappingURL=oauth-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"oauth-provider.js","sourceRoot":"","sources":["../../../src/upstream/oauth-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAON,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,EAAE,4BAA4B,EAA+B,MAAM,qBAAqB,CAAC;AAGhG,MAAM,WAAW,GAAG,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;AAapC,MAAM,OAAO,mBAAmB;IACrB,iBAAiB,CAAU;IAC3B,MAAM,CAAQ;IACd,aAAa,CAAS;IACtB,UAAU,CAAM;IAEzB,YACE,KAAY,EACZ,YAAoB,EACpB,SAAc,EACd,OAAO,GAA+B,EAAE;QAExC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,SAAS,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,EAAE,CAAC;YACrE,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,CAC9B,0BAA0B,YAAY,EAAE,EACxC,SAAS,CACV,CAAC,QAAQ,EAAE,CAAC;QACf,CAAC;QACD,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,GAAG,CAAC,4BAA4B,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IACpF,CAAC;IAED,IAAI,cAAc;QAChB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,OAAO;YACL,aAAa,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;YAC5C,0BAA0B,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,MAAM;YACjF,WAAW,EAAE,CAAC,oBAAoB,EAAE,eAAe,CAAC;YACpD,cAAc,EAAE,CAAC,MAAM,CAAC;YACxB,WAAW,EAAE,UAAU;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE;YACtC,WAAW,EAAE,UAAU;YACvB,gBAAgB,EAAE,OAAO;YACzB,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;SACjE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,SAAuB,EAAE,QAAiB;QAClE,qEAAqE;QACrE,yEAAyE;QACzE,yEAAyE;QACzE,uEAAuE;QACvE,mDAAmD;QACnD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACvC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,uCAAuC,EAAE,MAAM,CAAC,EAAE;gBACrF,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;gBACvC,QAAQ,EAAE,QAAQ;aACnB,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAChB,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA2B,CAAC;gBAC/D,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClE,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,+CAA+C;QACjD,CAAC;QACD,IAAI,QAAQ,KAAK,SAAS;YAAE,OAAO,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrD,OAAO,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACxB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iBAAiB,CACf,OAAuC;QAEvC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,KAAK,GAAG,OAAO,CAAC,iBAAiB,CAAC;QACxC,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,OAAO,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM;gBAAE,OAAO,SAAS,CAAC;YACjF,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,QAAQ;YAAE,OAAO,SAAS,CAAC;QACxC,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QACrF,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,QAAQ;YAC3B,GAAG,CAAC,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC;YACtF,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;SACpE,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,iBAA+C;QACnE,IAAI,CAAC,OAAO,CAAC;YACX,iBAAiB;YACjB,QAAQ,EAAE,iBAAiB,CAAC,SAAS;YACrC,YAAY,EAAE,iBAAiB,CAAC,aAAa;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,OAAuC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,WAAW;YAAE,OAAO,SAAS,CAAC;QAC3C,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QACrF,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS;YACjC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC;YAC/E,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;QACtB,OAAO;YACL,YAAY,EAAE,OAAO,CAAC,WAAW;YACjC,UAAU,EAAE,OAAO,CAAC,SAAS;YAC7B,GAAG,CAAC,OAAO,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC;YACtF,GAAG,CAAC,OAAO,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;YACvE,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;YAChE,GAAG,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;YAC7D,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;SACpE,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,MAAyB,EAAE,OAAuC;QAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,SAAS,GACb,MAAM,CAAC,UAAU,KAAK,SAAS;YAC7B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,UAAU,GAAG,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QACrE,IAAI,CAAC,OAAO,CAAC;YACX,WAAW,EAAE,MAAM,CAAC,YAAY;YAChC,YAAY,EAAE,MAAM,CAAC,aAAa,IAAI,OAAO,CAAC,YAAY;YAC1D,OAAO,EAAE,MAAM,CAAC,QAAQ;YACxB,SAAS,EAAE,MAAM,CAAC,UAAU;YAC5B,SAAS,EAAE,MAAM,CAAC,UAAU;YAC5B,SAAS;YACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK;YACpC,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,MAAM;YAC1D,gBAAgB,EAAE,SAAS;YAC3B,gBAAgB,EAAE,SAAS;YAC3B,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,SAAS;SACxB,CAAC,CAAC;IACL,CAAC;IAED,uBAAuB,CAAC,gBAAqB;QAC3C,IAAI,CAAC,OAAO,CAAC;YACX,gBAAgB,EAAE,gBAAgB,CAAC,QAAQ,EAAE;YAC7C,gBAAgB,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC,WAAW,EAAE;SACnE,CAAC,CAAC;IACL,CAAC;IAED,gBAAgB,CAAC,YAAoB;QACnC,IAAI,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,YAAY;QACV,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC;QAC9C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,QAAQ,CAAC,oBAAoB,EAAE,sCAAsC,EAAE,GAAG,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,0BAA0B,CAAC,sBAA8B;QACvD,IAAI,CAAC,OAAO,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,sBAAsB;QACpB,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,sBAAsB,CAAC;IAChD,CAAC;IAED,eAAe,CAAC,WAAmB;QACjC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;IAChC,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,WAAW,CAAC;IACrC,CAAC;IAED,kBAAkB,CAAC,cAAmC;QACpD,IAAI,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,cAAc,CAAC;IACxC,CAAC;IAED,qBAAqB,CAAC,KAA6D;QACjF,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC;gBACX,iBAAiB,EAAE,SAAS;gBAC5B,QAAQ,EAAE,SAAS;gBACnB,YAAY,EAAE,SAAS;aACxB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC;gBACX,WAAW,EAAE,SAAS;gBACtB,YAAY,EAAE,SAAS;gBACvB,OAAO,EAAE,SAAS;gBAClB,SAAS,EAAE,SAAS;gBACpB,SAAS,EAAE,SAAS;gBACpB,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,UAAU,EAAE,CAAC;YAC5C,IAAI,CAAC,OAAO,CAAC;gBACX,gBAAgB,EAAE,SAAS;gBAC3B,gBAAgB,EAAE,SAAS;gBAC3B,KAAK,EAAE,SAAS;gBAChB,YAAY,EAAE,SAAS;aACxB,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;YAC7C,IAAI,CAAC,OAAO,CAAC;gBACX,cAAc,EAAE,SAAS;gBACzB,sBAAsB,EAAE,SAAS;gBACjC,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,gBAAgB,CAAC;IAC1C,CAAC;IAED,qBAAqB,CAAC,KAAa;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;YAC9C,MAAM,IAAI,QAAQ,CAAC,sBAAsB,EAAE,iCAAiC,EAAE,GAAG,CAAC,CAAC;QACrF,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,gBAAgB,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YACpF,MAAM,IAAI,QAAQ,CAAC,oBAAoB,EAAE,sCAAsC,EAAE,GAAG,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IAED,QAAQ;QACN,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACrE,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzC,MAAM,IAAI,QAAQ,CAAC,0BAA0B,EAAE,iCAAiC,EAAE,GAAG,CAAC,CAAC;QACzF,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO,CAAC,OAAgC;QACtC,MAAM,IAAI,GAAG,4BAA4B,CAAC,KAAK,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|