@xingular/harness 0.0.0-stage → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +269 -2
- package/dist/cli.js +11046 -0
- package/dist/terminal-pi.js +326 -0
- package/package.json +41 -4
- package/src/accounts/accounts.test.ts +223 -0
- package/src/accounts/accounts.ts +522 -0
- package/src/accounts/env.ts +204 -0
- package/src/accounts/secrets.ts +50 -0
- package/src/accounts/store.ts +338 -0
- package/src/capabilities/api.ts +75 -0
- package/src/capabilities/runSnapshot.ts +59 -0
- package/src/capabilities/types.ts +34 -0
- package/src/cli.ts +110 -0
- package/src/core/bus.ts +34 -0
- package/src/core/changes.ts +33 -0
- package/src/core/env.ts +27 -0
- package/src/core/errors.ts +32 -0
- package/src/core/frontmatter.ts +47 -0
- package/src/core/fs.test.ts +38 -0
- package/src/core/fs.ts +92 -0
- package/src/core/index.ts +193 -0
- package/src/core/path.ts +45 -0
- package/src/core/projects.ts +90 -0
- package/src/core/text.ts +39 -0
- package/src/harness.ts +289 -0
- package/src/index.ts +69 -0
- package/src/mcp/connections.ts +441 -0
- package/src/mcp/manager.ts +130 -0
- package/src/mcp/oauth.ts +195 -0
- package/src/mcp/service.test.ts +376 -0
- package/src/mcp/service.ts +567 -0
- package/src/mcp/types.ts +105 -0
- package/src/memory/api.test.ts +82 -0
- package/src/memory/api.ts +287 -0
- package/src/memory/blocks.test.ts +78 -0
- package/src/memory/blocks.ts +406 -0
- package/src/memory/caller.ts +57 -0
- package/src/memory/dispatch.ts +59 -0
- package/src/memory/facts.test.ts +41 -0
- package/src/memory/facts.ts +248 -0
- package/src/memory/graph.ts +103 -0
- package/src/memory/health.test.ts +130 -0
- package/src/memory/health.ts +74 -0
- package/src/memory/maintainer.ts +272 -0
- package/src/memory/maintenance.test.ts +176 -0
- package/src/memory/maintenance.ts +330 -0
- package/src/memory/migrate.ts +79 -0
- package/src/memory/operations.ts +138 -0
- package/src/memory/proposals.test.ts +168 -0
- package/src/memory/proposals.ts +576 -0
- package/src/memory/recall.test.ts +104 -0
- package/src/memory/recall.ts +271 -0
- package/src/memory/references.ts +77 -0
- package/src/memory/runContext.ts +93 -0
- package/src/memory/secrets.ts +6 -0
- package/src/memory/store.test.ts +66 -0
- package/src/memory/store.ts +385 -0
- package/src/memory/wiki.test.ts +47 -0
- package/src/memory/wiki.ts +422 -0
- package/src/memory/wikiIndex.test.ts +88 -0
- package/src/memory/wikiIndex.ts +116 -0
- package/src/plugins/api.ts +181 -0
- package/src/plugins/attachments.test.ts +84 -0
- package/src/plugins/attachments.ts +128 -0
- package/src/plugins/builtin.ts +82 -0
- package/src/plugins/catalogue.test.ts +254 -0
- package/src/plugins/catalogue.ts +250 -0
- package/src/plugins/cli.ts +184 -0
- package/src/plugins/discover.ts +151 -0
- package/src/plugins/firstParty.test.ts +143 -0
- package/src/plugins/firstParty.ts +129 -0
- package/src/plugins/hooks.test.ts +250 -0
- package/src/plugins/hooks.ts +239 -0
- package/src/plugins/identity.ts +37 -0
- package/src/plugins/install.test.ts +107 -0
- package/src/plugins/install.ts +106 -0
- package/src/plugins/inventory.ts +123 -0
- package/src/plugins/manifest.ts +44 -0
- package/src/plugins/plugins.test.ts +205 -0
- package/src/plugins/plugins.ts +184 -0
- package/src/plugins/proposal.ts +121 -0
- package/src/plugins/runtime.test.ts +40 -0
- package/src/plugins/runtime.ts +82 -0
- package/src/plugins/trust.ts +142 -0
- package/src/plugins/types.ts +130 -0
- package/src/prompt/compile.test.ts +261 -0
- package/src/prompt/compile.ts +732 -0
- package/src/server/methods.ts +308 -0
- package/src/server/remote.ts +77 -0
- package/src/server/rpc.ts +116 -0
- package/src/server/serve.ts +28 -0
- package/src/server/server.test.ts +174 -0
- package/src/sessions/approvals.ts +118 -0
- package/src/sessions/contracts.ts +299 -0
- package/src/sessions/conversation.test.ts +150 -0
- package/src/sessions/extensionSession.test.ts +74 -0
- package/src/sessions/extensionUi.test.ts +65 -0
- package/src/sessions/extensionUi.ts +102 -0
- package/src/sessions/history.ts +51 -0
- package/src/sessions/lineageJoin.ts +75 -0
- package/src/sessions/policy.ts +32 -0
- package/src/sessions/pricing.ts +60 -0
- package/src/sessions/reasoningSummaries.ts +59 -0
- package/src/sessions/registry.ts +294 -0
- package/src/sessions/runIndex.ts +99 -0
- package/src/sessions/runStore.test.ts +152 -0
- package/src/sessions/runStore.ts +332 -0
- package/src/sessions/runner.ts +556 -0
- package/src/sessions/runtimeResources.test.ts +61 -0
- package/src/sessions/sessions.test.ts +434 -0
- package/src/sessions/sessions.ts +295 -0
- package/src/skills/registry.test.ts +104 -0
- package/src/skills/registry.ts +394 -0
- package/src/terminal/cli.ts +149 -0
- package/src/terminal/client.ts +74 -0
- package/src/terminal/controller.ts +227 -0
- package/src/terminal/piExtension.ts +208 -0
- package/src/terminal/pluginControl.test.ts +146 -0
- package/src/terminal/pluginMenu.ts +112 -0
- package/src/terminal/socket.ts +182 -0
- package/src/terminal/standalone.test.ts +115 -0
- package/src/terminal/standalone.ts +126 -0
- package/src/terminal/terminal.test.ts +302 -0
- package/src/terminal/terminals.ts +429 -0
- package/src/tools/askUser.ts +45 -0
- package/src/tools/definitions.ts +187 -0
- package/src/tools/memoryTools.ts +22 -0
- package/src/tools/pluginPropose.ts +41 -0
- package/src/tools/types.ts +36 -0
- package/src/web/safeRequest.ts +171 -0
- package/src/web/text.ts +53 -0
- package/src/web/tools.ts +134 -0
- package/src/web/web.test.ts +141 -0
package/src/mcp/oauth.ts
ADDED
|
@@ -0,0 +1,195 @@
|
|
|
1
|
+
// OAuth sign-in for a remote connection. State is kept with the connection's other secrets.
|
|
2
|
+
// Only an interactive provider (a sign-in a person started) may open a browser; a session that
|
|
3
|
+
// needs a new sign-in fails instead.
|
|
4
|
+
import { auth, UnauthorizedError } from '@modelcontextprotocol/client'
|
|
5
|
+
import type { OAuthClientInformationContext, OAuthClientMetadata, OAuthClientProvider, OAuthDiscoveryState, StoredOAuthClientInformation, StoredOAuthTokens } from '@modelcontextprotocol/client'
|
|
6
|
+
import { randomUUID } from 'node:crypto'
|
|
7
|
+
import { createServer } from 'node:http'
|
|
8
|
+
|
|
9
|
+
import { HarnessError } from '../core/errors'
|
|
10
|
+
import { MCP_OAUTH_PORT, MCP_OAUTH_REDIRECT_URL, type HttpConfig } from './types'
|
|
11
|
+
|
|
12
|
+
interface Saved {
|
|
13
|
+
client?: StoredOAuthClientInformation
|
|
14
|
+
tokens?: StoredOAuthTokens
|
|
15
|
+
discovery?: OAuthDiscoveryState
|
|
16
|
+
staticClient?: StoredOAuthClientInformation
|
|
17
|
+
staticIssuer?: string
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
/** Where a connection's OAuth state is read and written, and how a sign-in page is shown. */
|
|
21
|
+
export interface OAuthHost {
|
|
22
|
+
load(connectionId: string): Promise<Record<string, unknown>>
|
|
23
|
+
save(connectionId: string, state: Record<string, unknown>): Promise<void>
|
|
24
|
+
openUrl(url: string): Promise<void> | void
|
|
25
|
+
/** The name shown on the provider's consent page. */
|
|
26
|
+
clientName: string
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
export class StoredOAuthProvider implements OAuthClientProvider {
|
|
30
|
+
readonly expectedState = randomUUID()
|
|
31
|
+
private verifier: string | null = null
|
|
32
|
+
|
|
33
|
+
constructor(
|
|
34
|
+
private readonly host: OAuthHost,
|
|
35
|
+
private readonly config: Pick<HttpConfig, 'id' | 'scopes'>,
|
|
36
|
+
private readonly interactive: boolean
|
|
37
|
+
) {}
|
|
38
|
+
|
|
39
|
+
get redirectUrl(): string {
|
|
40
|
+
return MCP_OAUTH_REDIRECT_URL
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
get clientMetadata(): OAuthClientMetadata {
|
|
44
|
+
return {
|
|
45
|
+
client_name: this.host.clientName,
|
|
46
|
+
redirect_uris: [MCP_OAUTH_REDIRECT_URL],
|
|
47
|
+
grant_types: ['authorization_code', 'refresh_token'],
|
|
48
|
+
response_types: ['code'],
|
|
49
|
+
token_endpoint_auth_method: 'none',
|
|
50
|
+
...(this.config.scopes.length ? { scope: this.config.scopes.join(' ') } : {})
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
state(): string {
|
|
55
|
+
return this.expectedState
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
private load(): Promise<Saved> {
|
|
59
|
+
return this.host.load(this.config.id) as Promise<Saved>
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
private async change(edit: (saved: Saved) => Saved): Promise<void> {
|
|
63
|
+
await this.host.save(this.config.id, edit(await this.load()) as Record<string, unknown>)
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
async clientInformation(context?: OAuthClientInformationContext): Promise<StoredOAuthClientInformation | undefined> {
|
|
67
|
+
const saved = await this.load()
|
|
68
|
+
if (!saved.staticClient) return saved.client
|
|
69
|
+
// A client the user registered themselves (providers without dynamic registration) wins. It is
|
|
70
|
+
// tied to the authorization server seen at sign-in, so a session never offers it to another one.
|
|
71
|
+
if (this.interactive && context && saved.staticIssuer !== context.issuer) await this.change((current) => ({ ...current, staticIssuer: context.issuer }))
|
|
72
|
+
const issuer = this.interactive ? context?.issuer : saved.staticIssuer
|
|
73
|
+
return issuer ? { ...saved.staticClient, issuer } : saved.staticClient
|
|
74
|
+
}
|
|
75
|
+
|
|
76
|
+
async saveClientInformation(client: StoredOAuthClientInformation): Promise<void> {
|
|
77
|
+
await this.change((saved) => ({ ...saved, client }))
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
async tokens(): Promise<StoredOAuthTokens | undefined> {
|
|
81
|
+
return (await this.load()).tokens
|
|
82
|
+
}
|
|
83
|
+
|
|
84
|
+
async saveTokens(tokens: StoredOAuthTokens): Promise<void> {
|
|
85
|
+
await this.change((saved) => ({ ...saved, tokens }))
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
async discoveryState(): Promise<OAuthDiscoveryState | undefined> {
|
|
89
|
+
return (await this.load()).discovery
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
async saveDiscoveryState(discovery: OAuthDiscoveryState): Promise<void> {
|
|
93
|
+
await this.change((saved) => ({ ...saved, discovery }))
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
saveCodeVerifier(verifier: string): void {
|
|
97
|
+
this.verifier = verifier
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
codeVerifier(): string {
|
|
101
|
+
if (!this.verifier) throw new UnauthorizedError('Sign-in was not started from this app.')
|
|
102
|
+
return this.verifier
|
|
103
|
+
}
|
|
104
|
+
|
|
105
|
+
async invalidateCredentials(scope: 'all' | 'client' | 'tokens' | 'verifier' | 'discovery'): Promise<void> {
|
|
106
|
+
if (scope === 'verifier' || scope === 'all') this.verifier = null
|
|
107
|
+
if (scope === 'verifier') return
|
|
108
|
+
await this.change((saved) => (scope === 'all' ? {} : { ...saved, [scope]: undefined }))
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
async redirectToAuthorization(url: URL): Promise<void> {
|
|
112
|
+
if (!this.interactive) throw new UnauthorizedError('Sign in again in Connections.')
|
|
113
|
+
// Google only returns a refresh token when offline access is requested with a consent prompt.
|
|
114
|
+
if (url.hostname === 'accounts.google.com') {
|
|
115
|
+
url.searchParams.set('access_type', 'offline')
|
|
116
|
+
url.searchParams.set('prompt', 'consent')
|
|
117
|
+
}
|
|
118
|
+
await this.host.openUrl(url.toString())
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
const page = (title: string, message: string): string => `<!doctype html><meta charset="utf-8"><title>${title}</title><body style="font:16px system-ui;margin:20vh auto;max-width:28rem;text-align:center"><p>${message}</p></body>`
|
|
123
|
+
|
|
124
|
+
// One sign-in owns the loopback port. Starting another ends the one still waiting, so a closed
|
|
125
|
+
// browser tab never blocks a retry.
|
|
126
|
+
let cancelActive: (() => Promise<void>) | null = null
|
|
127
|
+
|
|
128
|
+
/** Waits on the fixed loopback address for the provider to send the browser back. */
|
|
129
|
+
function listenForCallback(expectedState: string, clientName: string): Promise<{ result: Promise<{ code: string; iss?: string }>; close: () => Promise<void> }> {
|
|
130
|
+
return new Promise((ready, failed) => {
|
|
131
|
+
let settle: { resolve: (value: { code: string; iss?: string }) => void; reject: (error: Error) => void }
|
|
132
|
+
const result = new Promise<{ code: string; iss?: string }>((resolve, reject) => {
|
|
133
|
+
settle = { resolve, reject }
|
|
134
|
+
})
|
|
135
|
+
result.catch(() => undefined)
|
|
136
|
+
const server = createServer((request, response) => {
|
|
137
|
+
const url = new URL(request.url ?? '/', MCP_OAUTH_REDIRECT_URL)
|
|
138
|
+
const reply = (status: number, message: string): void => {
|
|
139
|
+
response.writeHead(status, { 'content-type': 'text/html; charset=utf-8' })
|
|
140
|
+
response.end(page(clientName, message))
|
|
141
|
+
}
|
|
142
|
+
if (url.pathname !== new URL(MCP_OAUTH_REDIRECT_URL).pathname) return reply(404, 'Not found.')
|
|
143
|
+
// Anything on this machine can reach the port; only the request carrying this sign-in's state counts.
|
|
144
|
+
if (url.searchParams.get('state') !== expectedState) return reply(400, `This sign-in link is not the one ${clientName} is waiting for.`)
|
|
145
|
+
const code = url.searchParams.get('code')
|
|
146
|
+
if (!code) {
|
|
147
|
+
reply(400, 'Sign-in was not completed. You can close this window.')
|
|
148
|
+
return settle.reject(new HarnessError('mcp_unauthorized', url.searchParams.get('error') === 'access_denied' ? 'Sign-in was cancelled.' : 'The provider did not complete the sign-in.'))
|
|
149
|
+
}
|
|
150
|
+
reply(200, `Signed in. You can close this window and return to ${clientName}.`)
|
|
151
|
+
settle.resolve({ code, iss: url.searchParams.get('iss') ?? undefined })
|
|
152
|
+
})
|
|
153
|
+
const timer = setTimeout(() => settle.reject(new HarnessError('mcp_unauthorized', 'Sign-in timed out. Try again.')), 300000)
|
|
154
|
+
server.once('error', (error: NodeJS.ErrnoException) => {
|
|
155
|
+
clearTimeout(timer)
|
|
156
|
+
failed(new HarnessError('invalid_request', error.code === 'EADDRINUSE' ? `Port ${MCP_OAUTH_PORT} is in use. Finish or cancel the other sign-in, or quit the program using that port, then try again.` : 'The sign-in listener could not start.'))
|
|
157
|
+
})
|
|
158
|
+
const closed = new Promise<void>((resolve) => server.once('close', resolve))
|
|
159
|
+
const close = (): Promise<void> => {
|
|
160
|
+
clearTimeout(timer)
|
|
161
|
+
if (cancelActive === cancel) cancelActive = null
|
|
162
|
+
server.close()
|
|
163
|
+
server.closeAllConnections()
|
|
164
|
+
return closed
|
|
165
|
+
}
|
|
166
|
+
const cancel = (): Promise<void> => {
|
|
167
|
+
settle.reject(new HarnessError('mcp_unauthorized', 'Sign-in was cancelled.'))
|
|
168
|
+
return close()
|
|
169
|
+
}
|
|
170
|
+
server.listen(MCP_OAUTH_PORT, '127.0.0.1', () => {
|
|
171
|
+
cancelActive = cancel
|
|
172
|
+
ready({ result, close })
|
|
173
|
+
})
|
|
174
|
+
})
|
|
175
|
+
}
|
|
176
|
+
|
|
177
|
+
/** Opens the provider's sign-in page and waits for the browser to return to the loopback address. */
|
|
178
|
+
export async function signIn(host: OAuthHost, config: HttpConfig): Promise<void> {
|
|
179
|
+
const provider = new StoredOAuthProvider(host, config, true)
|
|
180
|
+
const scope = config.scopes.join(' ') || undefined
|
|
181
|
+
await cancelActive?.()
|
|
182
|
+
const callback = await listenForCallback(provider.expectedState, host.clientName)
|
|
183
|
+
try {
|
|
184
|
+
if ((await auth(provider, { serverUrl: config.url, scope, forceReauthorization: true })) === 'REDIRECT') {
|
|
185
|
+
const { code, iss } = await callback.result
|
|
186
|
+
if ((await auth(provider, { serverUrl: config.url, scope, authorizationCode: code, iss })) !== 'AUTHORIZED') throw new HarnessError('mcp_unauthorized', 'The provider did not complete the sign-in.')
|
|
187
|
+
}
|
|
188
|
+
} catch (error) {
|
|
189
|
+
if (error instanceof HarnessError) throw error
|
|
190
|
+
// Provider responses can quote credentials; report only that it failed.
|
|
191
|
+
throw new HarnessError('mcp_unauthorized', `${config.name} sign-in failed. Check the connection's address and OAuth client, then try again.`)
|
|
192
|
+
} finally {
|
|
193
|
+
await callback.close()
|
|
194
|
+
}
|
|
195
|
+
}
|
|
@@ -0,0 +1,376 @@
|
|
|
1
|
+
import { createServer } from 'node:http'
|
|
2
|
+
import type { AddressInfo } from 'node:net'
|
|
3
|
+
import { mkdtempSync } from 'node:fs'
|
|
4
|
+
import { tmpdir } from 'node:os'
|
|
5
|
+
import { dirname, join } from 'node:path'
|
|
6
|
+
import { fileURLToPath } from 'node:url'
|
|
7
|
+
import { McpServer, WebStandardStreamableHTTPServerTransport } from '@modelcontextprotocol/server'
|
|
8
|
+
import { afterEach, describe, expect, it } from 'vitest'
|
|
9
|
+
|
|
10
|
+
import { memorySecretStore } from '../accounts/secrets'
|
|
11
|
+
import { createHarness, type Harness } from '../harness'
|
|
12
|
+
import type { RunEvent } from '../sessions/contracts'
|
|
13
|
+
import type { ToolPolicy } from '../sessions/policy'
|
|
14
|
+
import type { PiSessionLike } from '../sessions/runner'
|
|
15
|
+
import { allowed, redactJsonField, scrubSensitive, scrubWith, setAgentScope, validateScopeIds, webUrl } from './connections'
|
|
16
|
+
import type { BrokerSession } from './service'
|
|
17
|
+
import { compactTools, pageTools } from './service'
|
|
18
|
+
import { connectionSchema, type Connection } from './types'
|
|
19
|
+
|
|
20
|
+
const fixture = join(dirname(fileURLToPath(import.meta.url)), '../../tests/mcp-fixture.mjs')
|
|
21
|
+
const connection = (fields: Partial<Connection>): Connection => connectionSchema.parse(fields)
|
|
22
|
+
const scope = { agentId: 'agent-test', projectId: 'project-test' }
|
|
23
|
+
|
|
24
|
+
describe('who may reach a connection', () => {
|
|
25
|
+
it('global access applies to future agents and honors overrides', () => {
|
|
26
|
+
const item = connection({ enabled: true, global: true })
|
|
27
|
+
expect(allowed(item, scope)).toBe(true)
|
|
28
|
+
expect(allowed({ ...item, excludedAgentIds: ['agent-test'] }, scope)).toBe(false)
|
|
29
|
+
expect(allowed({ ...item, excludedProjectIds: ['project-test'] }, scope)).toBe(false)
|
|
30
|
+
expect(allowed({ ...item, enabled: false }, scope)).toBe(false)
|
|
31
|
+
})
|
|
32
|
+
|
|
33
|
+
it('local access requires an explicit agent or project', () => {
|
|
34
|
+
const item = connection({ enabled: true })
|
|
35
|
+
expect(allowed(item, scope)).toBe(false)
|
|
36
|
+
expect(allowed({ ...item, agentIds: ['agent-test'] }, scope)).toBe(true)
|
|
37
|
+
expect(allowed({ ...item, projectIds: ['project-test'] }, scope)).toBe(true)
|
|
38
|
+
})
|
|
39
|
+
|
|
40
|
+
it('giving an agent its connections is exact and preserves other scopes', () => {
|
|
41
|
+
const creator = { agentId: 'creator' }
|
|
42
|
+
const all = [
|
|
43
|
+
connection({ id: 'global', name: 'Global', enabled: true, global: true, agentIds: ['existing'] }),
|
|
44
|
+
connection({ id: 'scoped', name: 'Scoped', enabled: true, agentIds: ['creator', 'existing'] }),
|
|
45
|
+
connection({ id: 'private', name: 'Private', enabled: true, agentIds: ['existing'] })
|
|
46
|
+
]
|
|
47
|
+
expect(() => validateScopeIds(all, creator, ['scoped'])).not.toThrow()
|
|
48
|
+
expect(() => validateScopeIds(all, creator, ['private'])).toThrow('outside your connection scope')
|
|
49
|
+
expect(() => validateScopeIds(all, creator, ['missing'])).toThrow()
|
|
50
|
+
setAgentScope(all, 'helper', ['scoped'])
|
|
51
|
+
const helper = { agentId: 'helper' }
|
|
52
|
+
expect(all.map((item) => allowed(item, helper))).toEqual([false, true, false])
|
|
53
|
+
expect(all.every((item) => item.agentIds.includes('existing'))).toBe(true)
|
|
54
|
+
expect([allowed(all[0], creator), allowed(all[1], creator)]).toEqual([true, true])
|
|
55
|
+
all[1].enabled = false
|
|
56
|
+
expect(() => validateScopeIds(all, creator, ['scoped'])).toThrow()
|
|
57
|
+
setAgentScope(all, 'helper', [])
|
|
58
|
+
expect(all.some((item) => allowed(item, helper))).toBe(false)
|
|
59
|
+
})
|
|
60
|
+
|
|
61
|
+
it('remote servers and sign-in pages must use TLS unless local', () => {
|
|
62
|
+
for (const accepted of ['https://mcp.linear.app/mcp', 'http://127.0.0.1:8080/mcp', 'http://localhost:8080/mcp']) expect(webUrl(accepted)).not.toBeNull()
|
|
63
|
+
for (const rejected of ['http://mcp.example.com/mcp', 'file:///etc/passwd', 'javascript:alert(1)', 'mcp.linear.app']) expect(webUrl(rejected)).toBeNull()
|
|
64
|
+
})
|
|
65
|
+
|
|
66
|
+
it('connections saved before remote servers still load', () => {
|
|
67
|
+
const saved = connection(JSON.parse('{"id":"old","name":"Old","kind":"local","command":"server","args":[],"envKeys":["KEY"],"enabled":true}'))
|
|
68
|
+
expect([saved.url, saved.authState, saved.catalogId, saved.headerKeys.length, saved.auth, saved.envKeys]).toEqual([null, null, null, 0, '', ['KEY']])
|
|
69
|
+
})
|
|
70
|
+
})
|
|
71
|
+
|
|
72
|
+
describe('what is never persisted', () => {
|
|
73
|
+
it('drops tokens and stored secrets, including from JSON carried as text', () => {
|
|
74
|
+
const result = JSON.stringify(scrubWith({ structuredContent: { approvalToken: 'preview-secret', data: 'fixture-secret' }, content: [{ text: '{"approvalToken":"preview-secret","data":"fixture-secret"}' }] }, ['fixture-secret']))
|
|
75
|
+
expect(result).not.toContain('preview-secret')
|
|
76
|
+
expect(result).not.toContain('fixture-secret')
|
|
77
|
+
expect(result).toContain('[redacted]')
|
|
78
|
+
expect(JSON.stringify(scrubWith('{"approvalToken":"truncated-preview', []))).not.toContain('truncated-preview')
|
|
79
|
+
const evidence = JSON.stringify(scrubSensitive({ apiKey: 'top-secret', Authorization: 'Bearer private', nested: { password: 'not saved', query: 'keep this' } }))
|
|
80
|
+
for (const leaked of ['top-secret', 'private', 'not saved']) expect(evidence).not.toContain(leaked)
|
|
81
|
+
expect(evidence).toContain('keep this')
|
|
82
|
+
expect(JSON.stringify(scrubSensitive('{"password":"cut-off-secret'))).not.toContain('cut-off-secret')
|
|
83
|
+
expect(redactJsonField('{"token" : "abc", "other": "token"}', 'token')).toBe('{"token" : "[redacted]", "other": "token"}')
|
|
84
|
+
})
|
|
85
|
+
|
|
86
|
+
it('lists tools as names and summaries, a page at a time', () => {
|
|
87
|
+
const tools = Array.from({ length: 45 }, (_, index) => ({ name: `tool_${index}`, description: index === 3 ? 'x'.repeat(300) : `Tool ${index}`, inputSchema: { type: 'object' } }))
|
|
88
|
+
expect(compactTools(tools, 'tool_3')).toEqual([tools[3]])
|
|
89
|
+
const page = pageTools(tools)
|
|
90
|
+
expect([page.tools.length, page.total, page.more, page.nextOffset]).toEqual([40, 45, 5, 40])
|
|
91
|
+
expect(page.tools[0]).not.toHaveProperty('inputSchema')
|
|
92
|
+
expect((page.tools[3].description ?? '').length).toBeLessThanOrEqual(161)
|
|
93
|
+
expect(pageTools(tools, { query: 'TOOL_44' }).tools.map((tool) => tool.name)).toEqual(['tool_44'])
|
|
94
|
+
})
|
|
95
|
+
})
|
|
96
|
+
|
|
97
|
+
// ---- A real server, a real run -------------------------------------------------------------
|
|
98
|
+
|
|
99
|
+
const open: Harness[] = []
|
|
100
|
+
const servers: Array<() => void> = []
|
|
101
|
+
afterEach(async () => {
|
|
102
|
+
for (const harness of open.splice(0)) await harness.close()
|
|
103
|
+
for (const close of servers.splice(0)) close()
|
|
104
|
+
})
|
|
105
|
+
|
|
106
|
+
const tick = (ms = 30) => new Promise((resolve) => setTimeout(resolve, ms))
|
|
107
|
+
|
|
108
|
+
/** A harness with one live run whose session keeps going until the test releases it. */
|
|
109
|
+
async function live(policy: ToolPolicy = {}) {
|
|
110
|
+
let release: () => void = () => undefined
|
|
111
|
+
const stateDir = mkdtempSync(join(tmpdir(), 'harness-mcp-state-'))
|
|
112
|
+
const secrets = memorySecretStore()
|
|
113
|
+
const session: PiSessionLike = { sessionId: 'native', subscribe: () => () => undefined, prompt: () => new Promise<void>((resolve) => (release = resolve)), sendCustomMessage: async () => undefined, abort: async () => release(), dispose: () => undefined }
|
|
114
|
+
const harness = createHarness({ dataRoot: mkdtempSync(join(tmpdir(), 'harness-mcp-data-')), stateDir, secrets, policy, sessionFactory: async () => session })
|
|
115
|
+
open.push(harness)
|
|
116
|
+
harness.accounts.create('Main', 'apiKey')
|
|
117
|
+
const events: RunEvent[] = []
|
|
118
|
+
harness.runs.listen((event) => void events.push(event))
|
|
119
|
+
await harness.sessions.run({ run: { id: 'run-1', sessionId: 'session-1' }, cwd: harness.env.root, sessionDir: join(harness.env.root, 'history'), text: 'go', caller: { agentId: 'agent-1', actor: 'Chat' }, tools: { allow: [] } })
|
|
120
|
+
const granted: BrokerSession = harness.apps.grant({ sessionId: 'session-1', runId: 'run-1', caller: { agentId: 'agent-1', actor: 'Chat' }, readOnly: false, depth: 0 })
|
|
121
|
+
return { harness, events, granted, secrets, release: () => release() }
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
const local = { name: 'Fixture', kind: 'local', command: process.execPath, args: [fixture], enabled: true, global: true, env: { MCP_FIXTURE_VALUE: 'personal task outside repo' } }
|
|
125
|
+
|
|
126
|
+
describe('connected apps for a session', () => {
|
|
127
|
+
it('discovers and calls a real stdio server, asking before a change and recording a redacted trace', async () => {
|
|
128
|
+
const { harness, events, granted, secrets } = await live()
|
|
129
|
+
const saved = await harness.apps.save({ ...local, readTools: ['read_record'] })
|
|
130
|
+
expect(saved.envKeys).toEqual(['MCP_FIXTURE_VALUE'])
|
|
131
|
+
expect(JSON.stringify(harness.apps.connections.list())).not.toContain('personal task')
|
|
132
|
+
expect([...secrets.values.keys()]).toEqual([`mcp--${saved.id}/MCP_FIXTURE_VALUE`])
|
|
133
|
+
|
|
134
|
+
const tested = await harness.apps.test(saved.id)
|
|
135
|
+
expect([tested.serverName, tested.error, tested.tools.length]).toEqual(['fixture', null, 2])
|
|
136
|
+
|
|
137
|
+
const listed = await harness.apps.list(granted)
|
|
138
|
+
expect(listed.errors).toEqual([])
|
|
139
|
+
const read = listed.tools.find((tool) => tool.tool === 'read_record')!
|
|
140
|
+
expect(read.name).toMatch(/^mcp_[0-9a-f]{12}_read_record$/)
|
|
141
|
+
// A read runs without asking; its result is stored with the connection's secret value removed.
|
|
142
|
+
expect(await harness.apps.invoke(granted, read.name, {})).toMatchObject({ content: [{ type: 'text', text: 'personal task outside repo' }] })
|
|
143
|
+
const trace = harness.apps.runContext('run-1')
|
|
144
|
+
expect(trace).toHaveLength(1)
|
|
145
|
+
expect(trace[0]).toMatchObject({ connectionName: 'Fixture', tool: 'read_record', ok: true })
|
|
146
|
+
expect(trace[0].result).toContain('[redacted]')
|
|
147
|
+
expect(trace[0].result).not.toContain('personal task')
|
|
148
|
+
expect(events.filter((event) => event.type === 'approval.requested')).toHaveLength(0)
|
|
149
|
+
|
|
150
|
+
// A change waits for a person, and does not run when they say no.
|
|
151
|
+
const change = listed.tools.find((tool) => tool.tool === 'change_record')!
|
|
152
|
+
const denied = harness.apps.invoke(granted, change.name, {})
|
|
153
|
+
await tick(200)
|
|
154
|
+
const request = events.find((event) => event.type === 'approval.requested')!.payload as { approvalId: string; target: string; action: string }
|
|
155
|
+
expect(request.approvalId.startsWith('mcp-')).toBe(true)
|
|
156
|
+
expect([request.target, request.action]).toEqual(['Fixture', 'change_record\n{}'])
|
|
157
|
+
harness.runs.resolveApproval('run-1', request.approvalId, 'denied')
|
|
158
|
+
await expect(denied).rejects.toMatchObject({ code: 'mcp_denied' })
|
|
159
|
+
expect(harness.apps.runContext('run-1')).toHaveLength(1)
|
|
160
|
+
|
|
161
|
+
// "Always allow" for one tool keeps the connection as it was tested.
|
|
162
|
+
expect(harness.apps.connections.allowTool(saved.id, 'change_record').readTools).toEqual(['read_record', 'change_record'])
|
|
163
|
+
expect(await harness.apps.invoke(granted, change.name, {})).toMatchObject({ content: [{ type: 'text', text: 'changed' }] })
|
|
164
|
+
expect(harness.apps.connections.find(saved.id).checkedAt).toBe(tested.checkedAt)
|
|
165
|
+
})
|
|
166
|
+
|
|
167
|
+
it('offers the two broker tools and the application\'s own tools through them', async () => {
|
|
168
|
+
const { harness, granted } = await live()
|
|
169
|
+
harness.apps.expose({ list: async () => [{ name: 'get_capabilities', description: 'What you can do', inputSchema: { type: 'object' } }], has: (name) => name === 'get_capabilities', call: async () => ({ tools: ['a'] }) })
|
|
170
|
+
const [list, call] = harness.apps.brokerTools(granted)
|
|
171
|
+
expect([list.name, call.name]).toEqual(['mcp_list_tools', 'mcp_call_tool'])
|
|
172
|
+
const listed = JSON.parse((await list.execute({})) as string)
|
|
173
|
+
expect(listed.tools).toEqual([{ name: 'get_capabilities', description: 'What you can do', origin: 'app', connectionId: '__app__', tool: 'get_capabilities' }])
|
|
174
|
+
expect(listed.note).toContain('Names and summaries only')
|
|
175
|
+
expect(JSON.parse((await list.execute({ tool: 'get_capabilities' })) as string).tools[0].inputSchema).toEqual({ type: 'object' })
|
|
176
|
+
expect(JSON.parse((await call.execute({ name: 'get_capabilities', arguments: {} })) as string)).toEqual({ content: [{ type: 'text', text: '{"tools":["a"]}' }], structuredContent: { tools: ['a'] } })
|
|
177
|
+
})
|
|
178
|
+
|
|
179
|
+
it('reports an unavailable executable without exposing diagnostics, and marks the connection', async () => {
|
|
180
|
+
const { harness, granted } = await live()
|
|
181
|
+
const saved = await harness.apps.save({ ...local, command: '/missing/mcp-fixture' })
|
|
182
|
+
const result = await harness.apps.list(granted)
|
|
183
|
+
expect(result.tools).toEqual([])
|
|
184
|
+
expect(result.errors[0].message).toContain('Settings')
|
|
185
|
+
expect(harness.apps.connections.find(saved.id).error).toContain('executable and arguments')
|
|
186
|
+
})
|
|
187
|
+
|
|
188
|
+
it('lets the application narrow what a session reaches, and nothing is reachable once the run ends', async () => {
|
|
189
|
+
let allow = true
|
|
190
|
+
const { harness, granted, release } = await live({ connections: async ({ connectionIds }) => (allow ? connectionIds : []) })
|
|
191
|
+
await harness.apps.save({ ...local, readTools: ['read_record'] })
|
|
192
|
+
expect((await harness.apps.list(granted)).tools).toHaveLength(2)
|
|
193
|
+
allow = false
|
|
194
|
+
expect((await harness.apps.list(granted)).tools).toEqual([])
|
|
195
|
+
allow = true
|
|
196
|
+
const read = (await harness.apps.list(granted)).tools.find((tool) => tool.tool === 'read_record')!
|
|
197
|
+
release()
|
|
198
|
+
await tick()
|
|
199
|
+
expect(() => harness.apps.session(granted.token)).toThrow('no longer authorized')
|
|
200
|
+
await expect(harness.apps.invoke(granted, read.name, {})).rejects.toMatchObject({ code: 'run_not_active' })
|
|
201
|
+
})
|
|
202
|
+
|
|
203
|
+
it('a read-only session may only read', async () => {
|
|
204
|
+
const { harness } = await live()
|
|
205
|
+
await harness.apps.save({ ...local, readTools: ['read_record'] })
|
|
206
|
+
const readOnly = harness.apps.grant({ sessionId: 'session-1', runId: 'run-1', caller: { agentId: 'agent-1', actor: 'Chat' }, readOnly: true, depth: 0 })
|
|
207
|
+
const tools = (await harness.apps.list(readOnly)).tools
|
|
208
|
+
await expect(harness.apps.invoke(readOnly, tools.find((tool) => tool.tool === 'change_record')!.name, {})).rejects.toMatchObject({ code: 'mcp_read_only' })
|
|
209
|
+
expect(await harness.apps.invoke(readOnly, tools.find((tool) => tool.tool === 'read_record')!.name, {})).toMatchObject({ content: [{ type: 'text' }] })
|
|
210
|
+
})
|
|
211
|
+
})
|
|
212
|
+
|
|
213
|
+
/** A stateless Streamable HTTP server; `status` makes it refuse every request instead. */
|
|
214
|
+
async function remoteFixture(status?: number) {
|
|
215
|
+
const requests: Array<Record<string, unknown>> = []
|
|
216
|
+
const server = createServer(async (request, response) => {
|
|
217
|
+
requests.push(request.headers)
|
|
218
|
+
if (status || request.method !== 'POST') return void response.writeHead(status ?? 405).end()
|
|
219
|
+
const chunks: Buffer[] = []
|
|
220
|
+
for await (const chunk of request) chunks.push(chunk as Buffer)
|
|
221
|
+
const mcp = new McpServer({ name: 'remote-fixture', version: '1' })
|
|
222
|
+
mcp.registerTool('read_record', { description: 'Read remote record' }, async () => ({ content: [{ type: 'text', text: 'remote record' }] }))
|
|
223
|
+
const transport = new WebStandardStreamableHTTPServerTransport({ sessionIdGenerator: undefined, enableJsonResponse: true })
|
|
224
|
+
await mcp.connect(transport)
|
|
225
|
+
const reply = await transport.handleRequest(new Request(`http://127.0.0.1${request.url}`, { method: 'POST', headers: request.headers as Record<string, string>, body: Buffer.concat(chunks) }))
|
|
226
|
+
response.writeHead(reply.status, Object.fromEntries(reply.headers)).end(Buffer.from(await reply.arrayBuffer()))
|
|
227
|
+
})
|
|
228
|
+
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve))
|
|
229
|
+
servers.push(() => {
|
|
230
|
+
server.close()
|
|
231
|
+
server.closeAllConnections()
|
|
232
|
+
})
|
|
233
|
+
return { url: `http://127.0.0.1:${(server.address() as AddressInfo).port}/mcp`, requests }
|
|
234
|
+
}
|
|
235
|
+
|
|
236
|
+
describe('remote connected apps', () => {
|
|
237
|
+
const remote = { name: 'Remote', kind: 'remote', enabled: true, global: true, readTools: ['read_record'] }
|
|
238
|
+
|
|
239
|
+
it('reaches a Streamable HTTP server with its saved headers, which stay out of the file', async () => {
|
|
240
|
+
const server = await remoteFixture()
|
|
241
|
+
const { harness, granted } = await live()
|
|
242
|
+
const saved = await harness.apps.save({ ...remote, url: server.url, auth: 'headers', headers: { 'X-Api-Key': 'fixture-key' } })
|
|
243
|
+
expect([saved.headerKeys, saved.command, saved.authState]).toEqual([['X-Api-Key'], '', null])
|
|
244
|
+
const read = (await harness.apps.list(granted)).tools.find((tool) => tool.tool === 'read_record')!
|
|
245
|
+
expect(await harness.apps.invoke(granted, read.name, {})).toMatchObject({ content: [{ type: 'text', text: 'remote record' }] })
|
|
246
|
+
expect(server.requests.every((headers) => headers['x-api-key'] === 'fixture-key')).toBe(true)
|
|
247
|
+
expect(JSON.stringify(harness.apps.connections.list())).not.toContain('fixture-key')
|
|
248
|
+
})
|
|
249
|
+
|
|
250
|
+
it('reports an unreachable server without naming its credentials', async () => {
|
|
251
|
+
const server = await remoteFixture(500)
|
|
252
|
+
const { harness, granted } = await live()
|
|
253
|
+
const saved = await harness.apps.save({ ...remote, url: server.url, auth: 'headers', headers: { Authorization: 'Bearer fixture-key' } })
|
|
254
|
+
const result = await harness.apps.list(granted)
|
|
255
|
+
expect(result.tools).toEqual([])
|
|
256
|
+
expect(JSON.stringify(result)).not.toContain('fixture-key')
|
|
257
|
+
expect(harness.apps.connections.find(saved.id).error).toContain('address and sign-in')
|
|
258
|
+
})
|
|
259
|
+
|
|
260
|
+
it('asks for sign-in instead of contacting an OAuth server without saved tokens, and never opens a browser from a run', async () => {
|
|
261
|
+
const server = await remoteFixture()
|
|
262
|
+
const { harness, granted } = await live()
|
|
263
|
+
const opened: unknown[] = []
|
|
264
|
+
harness.bus.on((notice) => void (notice.type === 'connections.openUrl' && opened.push(notice)))
|
|
265
|
+
const saved = await harness.apps.save({ ...remote, url: server.url, auth: 'oauth' })
|
|
266
|
+
expect(saved.authState).toBe('signed_out')
|
|
267
|
+
const result = await harness.apps.list(granted)
|
|
268
|
+
expect(result.errors[0].message).toContain('sign in again')
|
|
269
|
+
expect(server.requests).toEqual([])
|
|
270
|
+
expect(opened).toEqual([])
|
|
271
|
+
expect(harness.apps.connections.find(saved.id).authState).toBe('expired')
|
|
272
|
+
})
|
|
273
|
+
|
|
274
|
+
it('rejects what a connection must not be', async () => {
|
|
275
|
+
const { harness } = await live()
|
|
276
|
+
await expect(harness.apps.save({ ...remote, url: 'http://mcp.example.com/mcp', auth: 'none' })).rejects.toThrow('https://')
|
|
277
|
+
await expect(harness.apps.save({ name: '', kind: 'local', command: 'x' })).rejects.toThrow('A name, executable')
|
|
278
|
+
await expect(harness.apps.save({ ...local, env: { 'BAD NAME': 'x' } })).rejects.toThrow('environment variable')
|
|
279
|
+
await expect(harness.apps.save({ ...remote, url: 'https://x.example/mcp', auth: 'headers', headers: { 'X-Key': 'a\nb' } })).rejects.toThrow('header')
|
|
280
|
+
const workspace = await harness.apps.save({ name: 'Xingular Workspace', kind: 'workspace', command: process.execPath, enabled: true, global: true })
|
|
281
|
+
expect(() => harness.apps.connections.allowTool(workspace.id, 'x_workspace_mutation_apply')).toThrow('one at a time')
|
|
282
|
+
await harness.apps.remove(workspace.id)
|
|
283
|
+
expect(harness.apps.connections.list()).toEqual([])
|
|
284
|
+
})
|
|
285
|
+
})
|
|
286
|
+
|
|
287
|
+
/** A protected MCP server together with its authorization server, as hosted providers run them. */
|
|
288
|
+
async function oauthFixture() {
|
|
289
|
+
const seen = { registered: 0, tokenRequests: [] as string[], authorizeUrls: [] as URL[] }
|
|
290
|
+
const server = createServer(async (request, response) => {
|
|
291
|
+
const base = `http://127.0.0.1:${(server.address() as AddressInfo).port}`
|
|
292
|
+
const url = new URL(request.url ?? '/', base)
|
|
293
|
+
const chunks: Buffer[] = []
|
|
294
|
+
for await (const chunk of request) chunks.push(chunk as Buffer)
|
|
295
|
+
const body = Buffer.concat(chunks)
|
|
296
|
+
const json = (status: number, value: unknown) => response.writeHead(status, { 'content-type': 'application/json' }).end(JSON.stringify(value))
|
|
297
|
+
if (url.pathname.startsWith('/.well-known/oauth-protected-resource')) return json(200, { resource: `${base}/mcp`, authorization_servers: [base] })
|
|
298
|
+
if (url.pathname.startsWith('/.well-known/oauth-authorization-server')) return json(200, { issuer: base, authorization_endpoint: `${base}/authorize`, token_endpoint: `${base}/token`, registration_endpoint: `${base}/register`, response_types_supported: ['code'], grant_types_supported: ['authorization_code', 'refresh_token'], code_challenge_methods_supported: ['S256'], token_endpoint_auth_methods_supported: ['none', 'client_secret_post', 'client_secret_basic'] })
|
|
299
|
+
if (url.pathname === '/register') {
|
|
300
|
+
seen.registered += 1
|
|
301
|
+
return json(201, { ...JSON.parse(body.toString()), client_id: 'registered-client' })
|
|
302
|
+
}
|
|
303
|
+
if (url.pathname === '/authorize') {
|
|
304
|
+
seen.authorizeUrls.push(url)
|
|
305
|
+
return response.writeHead(302, { location: `${url.searchParams.get('redirect_uri')}?code=fixture-code&state=${url.searchParams.get('state')}` }).end()
|
|
306
|
+
}
|
|
307
|
+
if (url.pathname === '/token') {
|
|
308
|
+
seen.tokenRequests.push(`${request.headers.authorization ?? ''} ${body.toString()}`)
|
|
309
|
+
return json(200, { access_token: 'fixture-access', token_type: 'Bearer', expires_in: 3600, refresh_token: 'fixture-refresh' })
|
|
310
|
+
}
|
|
311
|
+
if (url.pathname !== '/mcp' || request.method !== 'POST') return response.writeHead(404).end()
|
|
312
|
+
if (request.headers.authorization !== 'Bearer fixture-access') return response.writeHead(401, { 'www-authenticate': `Bearer resource_metadata="${base}/.well-known/oauth-protected-resource"` }).end()
|
|
313
|
+
const mcp = new McpServer({ name: 'oauth-fixture', version: '1' })
|
|
314
|
+
mcp.registerTool('read_record', { description: 'Read protected record' }, async () => ({ content: [{ type: 'text', text: 'protected record' }] }))
|
|
315
|
+
const transport = new WebStandardStreamableHTTPServerTransport({ sessionIdGenerator: undefined, enableJsonResponse: true })
|
|
316
|
+
await mcp.connect(transport)
|
|
317
|
+
const reply = await transport.handleRequest(new Request(`${base}/mcp`, { method: 'POST', headers: request.headers as Record<string, string>, body }))
|
|
318
|
+
response.writeHead(reply.status, Object.fromEntries(reply.headers)).end(Buffer.from(await reply.arrayBuffer()))
|
|
319
|
+
})
|
|
320
|
+
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve))
|
|
321
|
+
servers.push(() => {
|
|
322
|
+
server.close()
|
|
323
|
+
server.closeAllConnections()
|
|
324
|
+
})
|
|
325
|
+
return { url: `http://127.0.0.1:${(server.address() as AddressInfo).port}/mcp`, seen }
|
|
326
|
+
}
|
|
327
|
+
|
|
328
|
+
describe('signing in to a remote connected app', () => {
|
|
329
|
+
it('signs in through the browser with dynamic registration, then a session uses the saved tokens', async () => {
|
|
330
|
+
const server = await oauthFixture()
|
|
331
|
+
const { harness, granted, secrets } = await live()
|
|
332
|
+
let opened = 0
|
|
333
|
+
// Stands in for the application: a browser that approves at once.
|
|
334
|
+
harness.bus.on((notice) => {
|
|
335
|
+
if (notice.type !== 'connections.openUrl') return
|
|
336
|
+
opened += 1
|
|
337
|
+
void fetch(String(notice.url))
|
|
338
|
+
})
|
|
339
|
+
const saved = await harness.apps.save({ name: 'OAuth', kind: 'remote', enabled: true, global: true, url: server.url, auth: 'oauth', oauthScopes: ['read'], readTools: ['read_record'] })
|
|
340
|
+
expect((await harness.apps.signIn(saved.id)).authState).toBe('signed_in')
|
|
341
|
+
expect(JSON.parse(secrets.values.get(`mcp--${saved.id}/oauth`)!).tokens).toMatchObject({ access_token: 'fixture-access', refresh_token: 'fixture-refresh' })
|
|
342
|
+
expect(server.seen.registered).toBe(1)
|
|
343
|
+
expect(server.seen.authorizeUrls[0].searchParams.get('scope')).toBe('read')
|
|
344
|
+
expect(server.seen.authorizeUrls[0].searchParams.get('code_challenge_method')).toBe('S256')
|
|
345
|
+
const listed = await harness.apps.list(granted)
|
|
346
|
+
expect(listed.errors).toEqual([])
|
|
347
|
+
const read = listed.tools.find((tool) => tool.tool === 'read_record')!
|
|
348
|
+
await harness.apps.invoke(granted, read.name, {})
|
|
349
|
+
expect(opened).toBe(1)
|
|
350
|
+
// The tokens a result might quote are removed from what is stored.
|
|
351
|
+
expect(harness.apps.connections.scrub({ content: [{ text: 'fixture-access fixture-refresh kept' }] }, saved.id)).toEqual({ content: [{ text: '[redacted] [redacted] kept' }] })
|
|
352
|
+
expect((await harness.apps.signOut(saved.id)).authState).toBe('signed_out')
|
|
353
|
+
expect(secrets.values.has(`mcp--${saved.id}/oauth`)).toBe(false)
|
|
354
|
+
})
|
|
355
|
+
|
|
356
|
+
it('uses a client the user registered, and ignores a callback that does not carry this sign-in\'s state', async () => {
|
|
357
|
+
const server = await oauthFixture()
|
|
358
|
+
const { harness } = await live()
|
|
359
|
+
let forged = 0
|
|
360
|
+
harness.bus.on((notice) => {
|
|
361
|
+
if (notice.type !== 'connections.openUrl') return
|
|
362
|
+
void (async () => {
|
|
363
|
+
forged = (await fetch('http://127.0.0.1:47823/oauth/callback?code=stolen&state=wrong')).status
|
|
364
|
+
await fetch(String(notice.url))
|
|
365
|
+
})()
|
|
366
|
+
})
|
|
367
|
+
const saved = await harness.apps.save({ name: 'Own client', kind: 'remote', enabled: true, global: true, url: server.url, auth: 'oauth', oauthClient: 'user', oauthCredentials: { clientId: 'own-client', clientSecret: 'own-secret' } })
|
|
368
|
+
await harness.apps.signIn(saved.id)
|
|
369
|
+
expect(forged).toBe(400)
|
|
370
|
+
expect(server.seen.registered).toBe(0)
|
|
371
|
+
expect(server.seen.authorizeUrls[0].searchParams.get('client_id')).toBe('own-client')
|
|
372
|
+
expect(server.seen.tokenRequests).toHaveLength(1)
|
|
373
|
+
expect(server.seen.tokenRequests[0]).toContain('fixture-code')
|
|
374
|
+
expect(server.seen.tokenRequests[0]).toMatch(/own-client|b3duLWNsaWVudDpvd24tc2VjcmV0/)
|
|
375
|
+
})
|
|
376
|
+
})
|