@progmruansilva/omp-web 1.202608.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 +181 -0
- package/dist/cli.js +1015 -0
- package/dist/cli.js.map +1 -0
- package/dist/client/assets/CodeViewer-Da66baHj.js +4 -0
- package/dist/client/assets/TerminalPanel-Bj9an79F.js +126 -0
- package/dist/client/assets/UnifiedDiffViewer-eK_kynTO.js +34 -0
- package/dist/client/assets/index-DytaeWNd.js +3070 -0
- package/dist/client/assets/vendor-editor-core-CL2gOCIg.js +12 -0
- package/dist/client/assets/vendor-editor-languages-zOM9I4iZ.js +26 -0
- package/dist/client/assets/vendor-editor-legacy-B4QLsWF8.js +1 -0
- package/dist/client/assets/vendor-terminal-DDGTF8rc.css +1 -0
- package/dist/client/assets/vendor-terminal-DjQ08hXu.js +16 -0
- package/dist/client/favicon-16x16.png +0 -0
- package/dist/client/favicon-180x180.png +0 -0
- package/dist/client/favicon-192x192.png +0 -0
- package/dist/client/favicon-32x32.png +0 -0
- package/dist/client/favicon-512x512.png +0 -0
- package/dist/client/favicon.ico +0 -0
- package/dist/client/favicon.png +0 -0
- package/dist/client/favicon.svg +11 -0
- package/dist/client/index.html +67 -0
- package/dist/client/manifest.webmanifest +36 -0
- package/dist/config.js +272 -0
- package/dist/config.js.map +1 -0
- package/dist/docker/ompWebDockerCommandPlan.js +211 -0
- package/dist/docker/ompWebDockerCommandPlan.js.map +1 -0
- package/dist/omp-web-plugins/info/omp-web-plugin.js +51 -0
- package/dist/omp-web-plugins/info/package.json +9 -0
- package/dist/omp-web-plugins/schedule-prompts/omp-web-plugin.js +43 -0
- package/dist/omp-web-plugins/schedule-prompts/package.json +9 -0
- package/dist/omp-web-plugins/schedule-prompts/schedulePromptsPanelElement.js +242 -0
- package/dist/omp-web-plugins/schedule-prompts/types.js +2 -0
- package/dist/omp-web-plugins/updates/omp-web-plugin.js +173 -0
- package/dist/omp-web-plugins/updates/package.json +9 -0
- package/dist/omp-web-plugins/updates/updatesLogic.js +97 -0
- package/dist/ompWebVersionReport.js +217 -0
- package/dist/ompWebVersionReport.js.map +1 -0
- package/dist/plugin-api/unstable.d.ts +22 -0
- package/dist/plugin-api.d.ts +197 -0
- package/dist/server/activity/workspaceActivityRoutes.js +4 -0
- package/dist/server/activity/workspaceActivityRoutes.js.map +1 -0
- package/dist/server/activity/workspaceActivityService.js +98 -0
- package/dist/server/activity/workspaceActivityService.js.map +1 -0
- package/dist/server/app.js +162 -0
- package/dist/server/app.js.map +1 -0
- package/dist/server/configRoutes.js +270 -0
- package/dist/server/configRoutes.js.map +1 -0
- package/dist/server/diagnostics/nodePtySpawnHelper.js +130 -0
- package/dist/server/diagnostics/nodePtySpawnHelper.js.map +1 -0
- package/dist/server/git/gitEnv.js +15 -0
- package/dist/server/git/gitEnv.js.map +1 -0
- package/dist/server/git/gitService.js +119 -0
- package/dist/server/git/gitService.js.map +1 -0
- package/dist/server/gitRoutes.js +23 -0
- package/dist/server/gitRoutes.js.map +1 -0
- package/dist/server/index.js +7 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/machines/machineClient.js +162 -0
- package/dist/server/machines/machineClient.js.map +1 -0
- package/dist/server/machines/machinePluginProxyRoutes.js +157 -0
- package/dist/server/machines/machinePluginProxyRoutes.js.map +1 -0
- package/dist/server/machines/machineProxyRoutes.js +127 -0
- package/dist/server/machines/machineProxyRoutes.js.map +1 -0
- package/dist/server/machines/machineRoutes.js +54 -0
- package/dist/server/machines/machineRoutes.js.map +1 -0
- package/dist/server/machines/machineService.js +230 -0
- package/dist/server/machines/machineService.js.map +1 -0
- package/dist/server/machines/machineStore.js +123 -0
- package/dist/server/machines/machineStore.js.map +1 -0
- package/dist/server/machines/proxyUtils.js +30 -0
- package/dist/server/machines/proxyUtils.js.map +1 -0
- package/dist/server/ompWebPluginService.js +279 -0
- package/dist/server/ompWebPluginService.js.map +1 -0
- package/dist/server/ompWebStatus.js +601 -0
- package/dist/server/ompWebStatus.js.map +1 -0
- package/dist/server/ompWebStatusCache.js +31 -0
- package/dist/server/ompWebStatusCache.js.map +1 -0
- package/dist/server/piPackageRoutes.js +82 -0
- package/dist/server/piPackageRoutes.js.map +1 -0
- package/dist/server/piPackageService.js +65 -0
- package/dist/server/piPackageService.js.map +1 -0
- package/dist/server/projects/directorySuggestions.js +37 -0
- package/dist/server/projects/directorySuggestions.js.map +1 -0
- package/dist/server/projects/projectService.js +31 -0
- package/dist/server/projects/projectService.js.map +1 -0
- package/dist/server/push/PushNotificationService.js +137 -0
- package/dist/server/push/PushNotificationService.js.map +1 -0
- package/dist/server/push/pushRoutes.js +55 -0
- package/dist/server/push/pushRoutes.js.map +1 -0
- package/dist/server/realtime/sessionEventHub.js +39 -0
- package/dist/server/realtime/sessionEventHub.js.map +1 -0
- package/dist/server/sessiond/sessionProxyRoutes.js +61 -0
- package/dist/server/sessiond/sessionProxyRoutes.js.map +1 -0
- package/dist/server/sessiond.js +92 -0
- package/dist/server/sessiond.js.map +1 -0
- package/dist/server/sessions/attachmentService.js +112 -0
- package/dist/server/sessions/attachmentService.js.map +1 -0
- package/dist/server/sessions/authProviderOptions.js +52 -0
- package/dist/server/sessions/authProviderOptions.js.map +1 -0
- package/dist/server/sessions/authRoutes.js +59 -0
- package/dist/server/sessions/authRoutes.js.map +1 -0
- package/dist/server/sessions/authService.js +105 -0
- package/dist/server/sessions/authService.js.map +1 -0
- package/dist/server/sessions/builtinCommands.js +29 -0
- package/dist/server/sessions/builtinCommands.js.map +1 -0
- package/dist/server/sessions/editPreview.js +196 -0
- package/dist/server/sessions/editPreview.js.map +1 -0
- package/dist/server/sessions/messagePaging.js +41 -0
- package/dist/server/sessions/messagePaging.js.map +1 -0
- package/dist/server/sessions/oauthLoginFlowService.js +136 -0
- package/dist/server/sessions/oauthLoginFlowService.js.map +1 -0
- package/dist/server/sessions/piSessionManagerGateway.js +97 -0
- package/dist/server/sessions/piSessionManagerGateway.js.map +1 -0
- package/dist/server/sessions/piSessionService.js +2401 -0
- package/dist/server/sessions/piSessionService.js.map +1 -0
- package/dist/server/sessions/schedulePrompt/schedulePromptService.js +46 -0
- package/dist/server/sessions/schedulePrompt/schedulePromptService.js.map +1 -0
- package/dist/server/sessions/schedulePrompt/scheduler.js +315 -0
- package/dist/server/sessions/schedulePrompt/scheduler.js.map +1 -0
- package/dist/server/sessions/schedulePrompt/storage.js +118 -0
- package/dist/server/sessions/schedulePrompt/storage.js.map +1 -0
- package/dist/server/sessions/schedulePrompt/tool.js +252 -0
- package/dist/server/sessions/schedulePrompt/tool.js.map +1 -0
- package/dist/server/sessions/schedulePrompt/types.js +2 -0
- package/dist/server/sessions/schedulePrompt/types.js.map +1 -0
- package/dist/server/sessions/sessionArchiveStore.js +254 -0
- package/dist/server/sessions/sessionArchiveStore.js.map +1 -0
- package/dist/server/sessions/sessionArchiveTree.js +35 -0
- package/dist/server/sessions/sessionArchiveTree.js.map +1 -0
- package/dist/server/sessions/sessionCleanup.js +157 -0
- package/dist/server/sessions/sessionCleanup.js.map +1 -0
- package/dist/server/sessions/sessionCommandService.js +260 -0
- package/dist/server/sessions/sessionCommandService.js.map +1 -0
- package/dist/server/sessions/sessionNameGenerator.js +61 -0
- package/dist/server/sessions/sessionNameGenerator.js.map +1 -0
- package/dist/server/sessions/sessionRoutes.js +344 -0
- package/dist/server/sessions/sessionRoutes.js.map +1 -0
- package/dist/server/sessions/sessionRuntimeStore.js +2 -0
- package/dist/server/sessions/sessionRuntimeStore.js.map +1 -0
- package/dist/server/sessions/spawnSessionTool.js +41 -0
- package/dist/server/sessions/spawnSessionTool.js.map +1 -0
- package/dist/server/sessions/spawnSubsessionTool.js +203 -0
- package/dist/server/sessions/spawnSubsessionTool.js.map +1 -0
- package/dist/server/sessions/spawnTargetResolver.js +36 -0
- package/dist/server/sessions/spawnTargetResolver.js.map +1 -0
- package/dist/server/sessions/subsessionTranscript.js +233 -0
- package/dist/server/sessions/subsessionTranscript.js.map +1 -0
- package/dist/server/storage/projectStore.js +83 -0
- package/dist/server/storage/projectStore.js.map +1 -0
- package/dist/server/terminalProxyRoutes.js +140 -0
- package/dist/server/terminalProxyRoutes.js.map +1 -0
- package/dist/server/terminals/terminalRoutes.js +153 -0
- package/dist/server/terminals/terminalRoutes.js.map +1 -0
- package/dist/server/terminals/terminalService.js +320 -0
- package/dist/server/terminals/terminalService.js.map +1 -0
- package/dist/server/terminals/terminalSize.js +17 -0
- package/dist/server/terminals/terminalSize.js.map +1 -0
- package/dist/server/types.js +2 -0
- package/dist/server/types.js.map +1 -0
- package/dist/server/utils.js +8 -0
- package/dist/server/utils.js.map +1 -0
- package/dist/server/webSocketBridge.js +32 -0
- package/dist/server/webSocketBridge.js.map +1 -0
- package/dist/server/workingDirectory.js +44 -0
- package/dist/server/workingDirectory.js.map +1 -0
- package/dist/server/workspaceExplorerRoutes.js +109 -0
- package/dist/server/workspaceExplorerRoutes.js.map +1 -0
- package/dist/server/workspaces/effectivePathAccess.js +27 -0
- package/dist/server/workspaces/effectivePathAccess.js.map +1 -0
- package/dist/server/workspaces/fileContentService.js +181 -0
- package/dist/server/workspaces/fileContentService.js.map +1 -0
- package/dist/server/workspaces/fileSuggestions.js +512 -0
- package/dist/server/workspaces/fileSuggestions.js.map +1 -0
- package/dist/server/workspaces/fileTreeService.js +27 -0
- package/dist/server/workspaces/fileTreeService.js.map +1 -0
- package/dist/server/workspaces/gitWorktreeDiscovery.js +34 -0
- package/dist/server/workspaces/gitWorktreeDiscovery.js.map +1 -0
- package/dist/server/workspaces/imagePreviewService.js +40 -0
- package/dist/server/workspaces/imagePreviewService.js.map +1 -0
- package/dist/server/workspaces/pathAccessPolicy.js +87 -0
- package/dist/server/workspaces/pathAccessPolicy.js.map +1 -0
- package/dist/server/workspaces/pathSafety.js +54 -0
- package/dist/server/workspaces/pathSafety.js.map +1 -0
- package/dist/server/workspaces/projectOmpWebConfig.js +56 -0
- package/dist/server/workspaces/projectOmpWebConfig.js.map +1 -0
- package/dist/server/workspaces/workspaceContext.js +8 -0
- package/dist/server/workspaces/workspaceContext.js.map +1 -0
- package/dist/server/workspaces/workspaceDeletionRoutes.js +111 -0
- package/dist/server/workspaces/workspaceDeletionRoutes.js.map +1 -0
- package/dist/server/workspaces/workspaceService.js +39 -0
- package/dist/server/workspaces/workspaceService.js.map +1 -0
- package/dist/sessiond/config.js +9 -0
- package/dist/sessiond/config.js.map +1 -0
- package/dist/sessiond/sessionDaemonClient.js +65 -0
- package/dist/sessiond/sessionDaemonClient.js.map +1 -0
- package/dist/shared/activity.js +26 -0
- package/dist/shared/activity.js.map +1 -0
- package/dist/shared/apiTypes.d.ts +688 -0
- package/dist/shared/apiTypes.js +11 -0
- package/dist/shared/apiTypes.js.map +1 -0
- package/dist/shared/capabilities.js +52 -0
- package/dist/shared/capabilities.js.map +1 -0
- package/dist/shared/federatedRoutes.js +79 -0
- package/dist/shared/federatedRoutes.js.map +1 -0
- package/dist/shared/machinePluginIds.js +41 -0
- package/dist/shared/machinePluginIds.js.map +1 -0
- package/dist/shared/ompWebStatusParsing.js +100 -0
- package/dist/shared/ompWebStatusParsing.js.map +1 -0
- package/dist/shared/pluginIds.js +5 -0
- package/dist/shared/pluginIds.js.map +1 -0
- package/dist/shared/promptAttachments.js +90 -0
- package/dist/shared/promptAttachments.js.map +1 -0
- package/dist/shared/thinkingLevels.d.ts +19 -0
- package/dist/shared/thinkingLevels.js +24 -0
- package/dist/shared/thinkingLevels.js.map +1 -0
- package/dist/shared/utils.js +4 -0
- package/dist/shared/utils.js.map +1 -0
- package/dist/shared/workspaceDeletion.js +12 -0
- package/dist/shared/workspaceDeletion.js.map +1 -0
- package/dist/shared/workspaceFiles.js +3 -0
- package/dist/shared/workspaceFiles.js.map +1 -0
- package/docs/assets/favicon.svg +11 -0
- package/docs/assets/omp-web-banner.png +0 -0
- package/docs/assets/omp-web-desktop.png +0 -0
- package/docs/assets/omp-web-mobile.png +0 -0
- package/docs/assets/omp-web-tablet.png +0 -0
- package/docs/config.md +208 -0
- package/docs/plugins.md +916 -0
- package/extensions/omp-web.ts +133 -0
- package/install.sh +5 -0
- package/package.json +141 -0
- package/plugin-api/unstable.d.ts +1 -0
- package/plugin-api.d.ts +1 -0
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { mkdir, rm } from "node:fs/promises";
|
|
3
|
+
import { dirname } from "node:path";
|
|
4
|
+
import Fastify from "fastify";
|
|
5
|
+
import fastifyWebsocket from "@fastify/websocket";
|
|
6
|
+
import { WorkspaceActivityService } from "./activity/workspaceActivityService.js";
|
|
7
|
+
import { registerWorkspaceActivityRoutes } from "./activity/workspaceActivityRoutes.js";
|
|
8
|
+
import { SessionEventHub } from "./realtime/sessionEventHub.js";
|
|
9
|
+
import { AuthService } from "./sessions/authService.js";
|
|
10
|
+
import { registerAuthRoutes } from "./sessions/authRoutes.js";
|
|
11
|
+
import { PiSessionService } from "./sessions/piSessionService.js";
|
|
12
|
+
import { registerSessionRoutes } from "./sessions/sessionRoutes.js";
|
|
13
|
+
import { ProjectScopedSpawnTargetResolver } from "./sessions/spawnTargetResolver.js";
|
|
14
|
+
import { ProjectService } from "./projects/projectService.js";
|
|
15
|
+
import { ProjectStore } from "./storage/projectStore.js";
|
|
16
|
+
import { WorkspaceService } from "./workspaces/workspaceService.js";
|
|
17
|
+
import { sessiondSocketPath } from "../sessiond/config.js";
|
|
18
|
+
import { TerminalService } from "./terminals/terminalService.js";
|
|
19
|
+
import { registerTerminalRoutes } from "./terminals/terminalRoutes.js";
|
|
20
|
+
import { getOmpWebRuntimeComponent } from "./ompWebStatus.js";
|
|
21
|
+
import { SESSIOND_RUNTIME_CAPABILITIES } from "../shared/capabilities.js";
|
|
22
|
+
import { effectiveOmpWebConfig, maxUploadBytes, spawnSessionsEnabled, subsessionsEnabled } from "../config.js";
|
|
23
|
+
import { PushNotificationService } from "./push/PushNotificationService.js";
|
|
24
|
+
import { registerPushRoutes } from "./push/pushRoutes.js";
|
|
25
|
+
const { config } = effectiveOmpWebConfig();
|
|
26
|
+
const app = Fastify({ logger: true, bodyLimit: maxUploadBytes(process.env, config) });
|
|
27
|
+
await app.register(fastifyWebsocket);
|
|
28
|
+
const eventHub = new SessionEventHub();
|
|
29
|
+
const workspaceActivity = new WorkspaceActivityService(eventHub);
|
|
30
|
+
const auth = await AuthService.create();
|
|
31
|
+
const spawnTargets = spawnSessionsEnabled(process.env, config)
|
|
32
|
+
? new ProjectScopedSpawnTargetResolver({ projects: new ProjectService(new ProjectStore()), workspaces: new WorkspaceService() })
|
|
33
|
+
: undefined;
|
|
34
|
+
const pushService = new PushNotificationService((msg) => { app.log.warn({ service: "push" }, msg); });
|
|
35
|
+
const sessions = new PiSessionService(eventHub, {
|
|
36
|
+
modelRegistry: auth.modelRegistry,
|
|
37
|
+
workspaceActivity,
|
|
38
|
+
logger: app.log,
|
|
39
|
+
pushService,
|
|
40
|
+
...(spawnTargets === undefined ? {} : { spawnTargets }),
|
|
41
|
+
subsessionsEnabled: spawnTargets !== undefined && subsessionsEnabled(process.env, config),
|
|
42
|
+
});
|
|
43
|
+
auth.subscribe((change) => { sessions.applyAuthChange(change); });
|
|
44
|
+
const terminals = new TerminalService(eventHub, workspaceActivity);
|
|
45
|
+
registerWorkspaceActivityRoutes(app, workspaceActivity);
|
|
46
|
+
registerAuthRoutes(app, auth);
|
|
47
|
+
registerSessionRoutes(app, sessions, eventHub);
|
|
48
|
+
registerTerminalRoutes(app, terminals);
|
|
49
|
+
registerPushRoutes(app, pushService);
|
|
50
|
+
app.get("/health", () => {
|
|
51
|
+
const runtime = getOmpWebRuntimeComponent("sessiond", SESSIOND_RUNTIME_CAPABILITIES);
|
|
52
|
+
return {
|
|
53
|
+
ok: true,
|
|
54
|
+
activeSessions: sessions.activeCount(),
|
|
55
|
+
checkedAt: new Date().toISOString(),
|
|
56
|
+
version: {
|
|
57
|
+
component: runtime.component,
|
|
58
|
+
label: runtime.label,
|
|
59
|
+
...(runtime.runtimeVersion === undefined ? {} : { runtimeVersion: runtime.runtimeVersion }),
|
|
60
|
+
stale: false,
|
|
61
|
+
available: runtime.available,
|
|
62
|
+
},
|
|
63
|
+
};
|
|
64
|
+
});
|
|
65
|
+
app.get("/runtime", () => getOmpWebRuntimeComponent("sessiond", SESSIOND_RUNTIME_CAPABILITIES));
|
|
66
|
+
let shuttingDown = false;
|
|
67
|
+
async function shutdown(signal) {
|
|
68
|
+
if (shuttingDown)
|
|
69
|
+
return;
|
|
70
|
+
shuttingDown = true;
|
|
71
|
+
app.log.info({ signal }, "shutting down session daemon");
|
|
72
|
+
terminals.dispose();
|
|
73
|
+
auth.dispose();
|
|
74
|
+
await sessions.dispose();
|
|
75
|
+
await app.close();
|
|
76
|
+
}
|
|
77
|
+
process.once("SIGINT", (signal) => { void shutdown(signal); });
|
|
78
|
+
process.once("SIGTERM", (signal) => { void shutdown(signal); });
|
|
79
|
+
const portValue = process.env["OMP_WEB_SESSIOND_PORT"];
|
|
80
|
+
const port = portValue !== undefined && portValue !== "" ? Number(portValue) : undefined;
|
|
81
|
+
const host = process.env["OMP_WEB_SESSIOND_HOST"] ?? "127.0.0.1";
|
|
82
|
+
if (port !== undefined) {
|
|
83
|
+
await app.listen({ port, host });
|
|
84
|
+
}
|
|
85
|
+
else {
|
|
86
|
+
const path = sessiondSocketPath();
|
|
87
|
+
await mkdir(dirname(path), { recursive: true });
|
|
88
|
+
await rm(path, { force: true });
|
|
89
|
+
await app.listen({ path });
|
|
90
|
+
process.on("exit", () => void rm(path, { force: true }));
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=sessiond.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessiond.js","sourceRoot":"","sources":["../../src/server/sessiond.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,gBAAgB,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,+BAA+B,EAAE,MAAM,uCAAuC,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAClE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,gCAAgC,EAAE,MAAM,mCAAmC,CAAC;AACrF,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAC/G,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,qBAAqB,EAAE,CAAC;AAC3C,MAAM,GAAG,GAAG,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;AACtF,MAAM,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;AAErC,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;AACvC,MAAM,iBAAiB,GAAG,IAAI,wBAAwB,CAAC,QAAQ,CAAC,CAAC;AACjE,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;AACxC,MAAM,YAAY,GAAG,oBAAoB,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC;IAC5D,CAAC,CAAC,IAAI,gCAAgC,CAAC,EAAE,QAAQ,EAAE,IAAI,cAAc,CAAC,IAAI,YAAY,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,gBAAgB,EAAE,EAAE,CAAC;IAChI,CAAC,CAAC,SAAS,CAAC;AACd,MAAM,WAAW,GAAG,IAAI,uBAAuB,CAAC,CAAC,GAAG,EAAE,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACtG,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,QAAQ,EAAE;IAC9C,aAAa,EAAE,IAAI,CAAC,aAAa;IACjC,iBAAiB;IACjB,MAAM,EAAE,GAAG,CAAC,GAAG;IACf,WAAW;IACX,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC;IACvD,kBAAkB,EAAE,YAAY,KAAK,SAAS,IAAI,kBAAkB,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC;CAC1F,CAAC,CAAC;AACH,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAClE,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;AACnE,+BAA+B,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;AACxD,kBAAkB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;AAC9B,qBAAqB,CAAC,GAAG,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;AAC/C,sBAAsB,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;AACvC,kBAAkB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;AAErC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE;IACtB,MAAM,OAAO,GAAG,yBAAyB,CAAC,UAAU,EAAE,6BAA6B,CAAC,CAAC;IACrF,OAAO;QACL,EAAE,EAAE,IAAI;QACR,cAAc,EAAE,QAAQ,CAAC,WAAW,EAAE;QACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,OAAO,EAAE;YACP,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,GAAG,CAAC,OAAO,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3F,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,yBAAyB,CAAC,UAAU,EAAE,6BAA6B,CAAC,CAAC,CAAC;AAEhG,IAAI,YAAY,GAAG,KAAK,CAAC;AACzB,KAAK,UAAU,QAAQ,CAAC,MAAsB;IAC5C,IAAI,YAAY;QAAE,OAAO;IACzB,YAAY,GAAG,IAAI,CAAC;IACpB,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,8BAA8B,CAAC,CAAC;IACzD,SAAS,CAAC,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,OAAO,EAAE,CAAC;IACf,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;IACzB,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;AACpB,CAAC;AAED,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/D,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE,GAAG,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEhE,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;AACvD,MAAM,IAAI,GAAG,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACzF,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,WAAW,CAAC;AAEjE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;IACvB,MAAM,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;AACnC,CAAC;KAAM,CAAC;IACN,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;IAClC,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAChD,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAChC,MAAM,GAAG,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3B,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}
|
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import { mkdir, realpath, writeFile } from "node:fs/promises";
|
|
2
|
+
import { basename, extname, join } from "node:path";
|
|
3
|
+
import { extensionForImageMimeType } from "../../shared/promptAttachments.js";
|
|
4
|
+
import { ensureInside, isNodeErrorWithCode, resolveParentInsideWorkspace } from "../workspaces/pathSafety.js";
|
|
5
|
+
/**
|
|
6
|
+
* Default workspace-relative folder used when saving pasted/dropped
|
|
7
|
+
* attachments for the agent to read with its own tools.
|
|
8
|
+
*/
|
|
9
|
+
export const DEFAULT_ATTACHMENT_FOLDER = ".omp-web/attachments";
|
|
10
|
+
/**
|
|
11
|
+
* Convert validated attachments into pi-compatible inline image content.
|
|
12
|
+
*
|
|
13
|
+
* Mirrors pi's own CLI/TUI behaviour: each image is run through pi's
|
|
14
|
+
* `resizeImage` so it fits within pi's max dimensions and inline byte budget
|
|
15
|
+
* (2000x2000, ~4.5MB base64). Images that cannot be resized below the limit
|
|
16
|
+
* are dropped, matching pi's `[Image omitted]` behaviour.
|
|
17
|
+
*/
|
|
18
|
+
export function attachmentsToInlineImages(attachments) {
|
|
19
|
+
const results = [];
|
|
20
|
+
for (const attachment of attachments) {
|
|
21
|
+
results.push({
|
|
22
|
+
image: { type: "image", data: attachment.data, mimeType: attachment.mimeType },
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
return results;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Write attachments into a workspace folder and return their relative paths.
|
|
29
|
+
* Filenames are collision-safe and stay inside the workspace root.
|
|
30
|
+
*/
|
|
31
|
+
export async function saveAttachmentsToWorkspace(cwd, attachments, options = {}) {
|
|
32
|
+
const folder = options.folder ?? DEFAULT_ATTACHMENT_FOLDER;
|
|
33
|
+
const now = options.now ?? (() => new Date());
|
|
34
|
+
const { root, target: requestedFolderTarget, relativePath: normalizedFolder } = await resolveParentInsideWorkspace(cwd, folder);
|
|
35
|
+
await mkdir(requestedFolderTarget, { recursive: true });
|
|
36
|
+
const folderTarget = await realpath(requestedFolderTarget);
|
|
37
|
+
ensureInside(root, folderTarget);
|
|
38
|
+
const stamp = timestamp(now());
|
|
39
|
+
const saved = [];
|
|
40
|
+
for (const [index, attachment] of attachments.entries()) {
|
|
41
|
+
const bytes = Buffer.from(attachment.data, "base64");
|
|
42
|
+
const filename = await writeUniqueAttachmentFile(folderTarget, attachmentFilename(attachment, stamp, index), bytes);
|
|
43
|
+
const relativePath = normalizedFolder === "" ? filename : `${normalizedFolder}/${filename}`;
|
|
44
|
+
saved.push({ path: relativePath, mimeType: attachment.mimeType, size: bytes.byteLength });
|
|
45
|
+
}
|
|
46
|
+
return saved;
|
|
47
|
+
}
|
|
48
|
+
async function writeUniqueAttachmentFile(folderTarget, filename, bytes) {
|
|
49
|
+
for (let attempt = 0; attempt < 100; attempt += 1) {
|
|
50
|
+
const candidate = attempt === 0 ? filename : addCollisionSuffix(filename, attempt + 1);
|
|
51
|
+
try {
|
|
52
|
+
await writeFile(join(folderTarget, candidate), bytes, { flag: "wx" });
|
|
53
|
+
return candidate;
|
|
54
|
+
}
|
|
55
|
+
catch (error) {
|
|
56
|
+
if (!isNodeErrorWithCode(error, "EEXIST"))
|
|
57
|
+
throw error;
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
throw new Error("Unable to choose a unique attachment filename");
|
|
61
|
+
}
|
|
62
|
+
function addCollisionSuffix(filename, suffix) {
|
|
63
|
+
const extension = extname(filename);
|
|
64
|
+
const stem = filename.slice(0, filename.length - extension.length);
|
|
65
|
+
return `${stem}-${String(suffix)}${extension}`;
|
|
66
|
+
}
|
|
67
|
+
function attachmentFilename(attachment, stamp, index) {
|
|
68
|
+
const originalName = sanitizeOriginalFilename(attachment.name) ?? fallbackAttachmentFilename(attachment);
|
|
69
|
+
return `attachment-${stamp}-${String(index + 1)}-${originalName}`;
|
|
70
|
+
}
|
|
71
|
+
function fallbackAttachmentFilename(attachment) {
|
|
72
|
+
if (attachment.kind === "image")
|
|
73
|
+
return `image.${extensionForImageMimeType(attachment.mimeType)}`;
|
|
74
|
+
return "file.bin";
|
|
75
|
+
}
|
|
76
|
+
const MAX_ORIGINAL_FILENAME_LENGTH = 96;
|
|
77
|
+
function sanitizeOriginalFilename(name) {
|
|
78
|
+
const trimmed = name?.trim();
|
|
79
|
+
if (trimmed === undefined || trimmed === "")
|
|
80
|
+
return undefined;
|
|
81
|
+
const leaf = basename(trimmed.replace(/\\/g, "/"));
|
|
82
|
+
const sanitized = stripControlCharacters(leaf)
|
|
83
|
+
.normalize("NFKC")
|
|
84
|
+
.replace(/[^A-Za-z0-9._-]+/g, "-")
|
|
85
|
+
.replace(/-+/g, "-")
|
|
86
|
+
.replace(/-+\./g, ".")
|
|
87
|
+
.replace(/^\.+/, "")
|
|
88
|
+
.replace(/[.-]+$/, "");
|
|
89
|
+
if (sanitized === "")
|
|
90
|
+
return undefined;
|
|
91
|
+
return truncateFilename(sanitized, MAX_ORIGINAL_FILENAME_LENGTH);
|
|
92
|
+
}
|
|
93
|
+
function stripControlCharacters(value) {
|
|
94
|
+
return Array.from(value).filter((character) => {
|
|
95
|
+
const codePoint = character.codePointAt(0);
|
|
96
|
+
return codePoint !== undefined && codePoint > 0x1f && codePoint !== 0x7f;
|
|
97
|
+
}).join("");
|
|
98
|
+
}
|
|
99
|
+
function truncateFilename(filename, maxLength) {
|
|
100
|
+
if (filename.length <= maxLength)
|
|
101
|
+
return filename;
|
|
102
|
+
const extension = extname(filename);
|
|
103
|
+
if (extension.length >= maxLength)
|
|
104
|
+
return filename.slice(0, maxLength);
|
|
105
|
+
const stem = filename.slice(0, filename.length - extension.length);
|
|
106
|
+
return `${stem.slice(0, maxLength - extension.length)}${extension}`;
|
|
107
|
+
}
|
|
108
|
+
function timestamp(date) {
|
|
109
|
+
const pad = (value, length = 2) => String(value).padStart(length, "0");
|
|
110
|
+
return `${String(date.getFullYear())}${pad(date.getMonth() + 1)}${pad(date.getDate())}-${pad(date.getHours())}${pad(date.getMinutes())}${pad(date.getSeconds())}-${pad(date.getMilliseconds(), 3)}`;
|
|
111
|
+
}
|
|
112
|
+
//# sourceMappingURL=attachmentService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"attachmentService.js","sourceRoot":"","sources":["../../../src/server/sessions/attachmentService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAIpD,OAAO,EAAE,yBAAyB,EAAE,MAAM,mCAAmC,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAE9G;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;AAQhE;;;;;;;GAOG;AACH,MAAM,UAAU,yBAAyB,CAAC,WAAoC;IAC7E,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QACtC,OAAO,CAAC,IAAI,CAAC;YACZ,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE;SAC9E,CAAC,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AASD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC/C,GAAW,EACX,WAA+B,EAC/B,UAAkC,EAAE;IAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,yBAAyB,CAAC;IAC3D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAC9C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,qBAAqB,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,4BAA4B,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAChI,MAAM,KAAK,CAAC,qBAAqB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,qBAAqB,CAAC,CAAC;IAC3D,YAAY,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC;IAC/B,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,MAAM,yBAAyB,CAAC,YAAY,EAAE,kBAAkB,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,KAAK,CAAC,CAAC;QACpH,MAAM,YAAY,GAAG,gBAAgB,KAAK,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,gBAAgB,IAAI,QAAQ,EAAE,CAAC;QAC5F,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;IAC3F,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,KAAK,UAAU,yBAAyB,CAAC,YAAoB,EAAE,QAAgB,EAAE,KAAa;IAC7F,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,GAAG,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QACnD,MAAM,SAAS,GAAG,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC;YACJ,MAAM,SAAS,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC;QAClB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,CAAC;gBAAE,MAAM,KAAK,CAAC;QACxD,CAAC;IACF,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAgB,EAAE,MAAc;IAC3D,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;IACnE,OAAO,GAAG,IAAI,IAAI,MAAM,CAAC,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,kBAAkB,CAAC,UAA4B,EAAE,KAAa,EAAE,KAAa;IACrF,MAAM,YAAY,GAAG,wBAAwB,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,0BAA0B,CAAC,UAAU,CAAC,CAAC;IACzG,OAAO,cAAc,KAAK,IAAI,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,YAAY,EAAE,CAAC;AACnE,CAAC;AAED,SAAS,0BAA0B,CAAC,UAA4B;IAC/D,IAAI,UAAU,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,SAAS,yBAAyB,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;IAClG,OAAO,UAAU,CAAC;AACnB,CAAC;AAED,MAAM,4BAA4B,GAAG,EAAE,CAAC;AAExC,SAAS,wBAAwB,CAAC,IAAwB;IACzD,MAAM,OAAO,GAAG,IAAI,EAAE,IAAI,EAAE,CAAC;IAC7B,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,KAAK,EAAE;QAAE,OAAO,SAAS,CAAC;IAC9D,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;IACnD,MAAM,SAAS,GAAG,sBAAsB,CAAC,IAAI,CAAC;SAC5C,SAAS,CAAC,MAAM,CAAC;SACjB,OAAO,CAAC,mBAAmB,EAAE,GAAG,CAAC;SACjC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;SACnB,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACxB,IAAI,SAAS,KAAK,EAAE;QAAE,OAAO,SAAS,CAAC;IACvC,OAAO,gBAAgB,CAAC,SAAS,EAAE,4BAA4B,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAa;IAC5C,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE;QAC7C,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAC3C,OAAO,SAAS,KAAK,SAAS,IAAI,SAAS,GAAG,IAAI,IAAI,SAAS,KAAK,IAAI,CAAC;IAC1E,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,SAAiB;IAC5D,IAAI,QAAQ,CAAC,MAAM,IAAI,SAAS;QAAE,OAAO,QAAQ,CAAC;IAClD,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpC,IAAI,SAAS,CAAC,MAAM,IAAI,SAAS;QAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACvE,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;IACnE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,SAAS,EAAE,CAAC;AACrE,CAAC;AAED,SAAS,SAAS,CAAC,IAAU;IAC5B,MAAM,GAAG,GAAG,CAAC,KAAa,EAAE,MAAM,GAAG,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/E,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;AACrM,CAAC"}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
const OAUTH_ONLY_PROVIDERS = new Set(["github-copilot", "openai-codex"]);
|
|
2
|
+
export function getLoginProviderOptions(modelRegistry, authType) {
|
|
3
|
+
const oauthProviders = modelRegistry.authStorage.getOAuthProviders();
|
|
4
|
+
const oauthProviderIds = new Set(oauthProviders.map((provider) => provider.id));
|
|
5
|
+
const options = oauthProviders.map((provider) => ({
|
|
6
|
+
id: provider.id,
|
|
7
|
+
name: provider.name,
|
|
8
|
+
authType: "oauth",
|
|
9
|
+
status: modelRegistry.getProviderAuthStatus(provider.id),
|
|
10
|
+
}));
|
|
11
|
+
const modelProviders = new Set(modelRegistry.getAll().map((model) => model.provider));
|
|
12
|
+
for (const providerId of modelProviders) {
|
|
13
|
+
if (!isApiKeyLoginProvider(providerId, oauthProviderIds))
|
|
14
|
+
continue;
|
|
15
|
+
options.push({
|
|
16
|
+
id: providerId,
|
|
17
|
+
name: modelRegistry.getProviderDisplayName(providerId),
|
|
18
|
+
authType: "api_key",
|
|
19
|
+
status: modelRegistry.getProviderAuthStatus(providerId),
|
|
20
|
+
});
|
|
21
|
+
}
|
|
22
|
+
return filterAndSort(options, authType);
|
|
23
|
+
}
|
|
24
|
+
export function getLogoutProviderOptions(modelRegistry) {
|
|
25
|
+
const options = [];
|
|
26
|
+
for (const providerId of modelRegistry.authStorage.list()) {
|
|
27
|
+
const credential = modelRegistry.authStorage.get(providerId);
|
|
28
|
+
if (credential === undefined)
|
|
29
|
+
continue;
|
|
30
|
+
options.push({
|
|
31
|
+
id: providerId,
|
|
32
|
+
name: modelRegistry.getProviderDisplayName(providerId),
|
|
33
|
+
authType: credential.type,
|
|
34
|
+
status: modelRegistry.getProviderAuthStatus(providerId),
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
return filterAndSort(options);
|
|
38
|
+
}
|
|
39
|
+
export function isApiKeyLoginProvider(providerId, oauthProviderIds) {
|
|
40
|
+
if (OAUTH_ONLY_PROVIDERS.has(providerId))
|
|
41
|
+
return false;
|
|
42
|
+
if (providerId === "anthropic")
|
|
43
|
+
return true;
|
|
44
|
+
if (oauthProviderIds.has(providerId))
|
|
45
|
+
return false;
|
|
46
|
+
return true;
|
|
47
|
+
}
|
|
48
|
+
function filterAndSort(options, authType) {
|
|
49
|
+
const filtered = authType === undefined ? options : options.filter((option) => option.authType === authType);
|
|
50
|
+
return filtered.sort((a, b) => a.name.localeCompare(b.name) || a.authType.localeCompare(b.authType) || a.id.localeCompare(b.id));
|
|
51
|
+
}
|
|
52
|
+
//# sourceMappingURL=authProviderOptions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"authProviderOptions.js","sourceRoot":"","sources":["../../../src/server/sessions/authProviderOptions.ts"],"names":[],"mappings":"AAEA,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC,CAAC,gBAAgB,EAAE,cAAc,CAAC,CAAC,CAAC;AAazE,MAAM,UAAU,uBAAuB,CAAC,aAAwC,EAAE,QAAmB;IACnG,MAAM,cAAc,GAAG,aAAa,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAC;IACrE,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;IAChF,MAAM,OAAO,GAAyB,cAAc,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACtE,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,QAAQ,EAAE,OAAO;QACjB,MAAM,EAAE,aAAa,CAAC,qBAAqB,CAAC,QAAQ,CAAC,EAAE,CAAC;KACzD,CAAC,CAAC,CAAC;IAEJ,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IACtF,KAAK,MAAM,UAAU,IAAI,cAAc,EAAE,CAAC;QACxC,IAAI,CAAC,qBAAqB,CAAC,UAAU,EAAE,gBAAgB,CAAC;YAAE,SAAS;QACnE,OAAO,CAAC,IAAI,CAAC;YACX,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,aAAa,CAAC,sBAAsB,CAAC,UAAU,CAAC;YACtD,QAAQ,EAAE,SAAS;YACnB,MAAM,EAAE,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,aAAwC;IAC/E,MAAM,OAAO,GAAyB,EAAE,CAAC;IACzC,KAAK,MAAM,UAAU,IAAI,aAAa,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;QAC1D,MAAM,UAAU,GAAG,aAAa,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC7D,IAAI,UAAU,KAAK,SAAS;YAAE,SAAS;QACvC,OAAO,CAAC,IAAI,CAAC;YACX,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,aAAa,CAAC,sBAAsB,CAAC,UAAU,CAAC;YACtD,QAAQ,EAAE,UAAU,CAAC,IAAI;YACzB,MAAM,EAAE,aAAa,CAAC,qBAAqB,CAAC,UAAU,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IACD,OAAO,aAAa,CAAC,OAAO,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,UAAkB,EAAE,gBAAqC;IAC7F,IAAI,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC;QAAE,OAAO,KAAK,CAAC;IACvD,IAAI,UAAU,KAAK,WAAW;QAAE,OAAO,IAAI,CAAC;IAC5C,IAAI,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC;QAAE,OAAO,KAAK,CAAC;IACnD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,aAAa,CAAC,OAA6B,EAAE,QAAmB;IACvE,MAAM,QAAQ,GAAG,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;IAC7G,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACnI,CAAC"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
export function registerAuthRoutes(app, auth, prefix = "") {
|
|
2
|
+
app.get(`${prefix}/auth/providers`, async (request, reply) => {
|
|
3
|
+
try {
|
|
4
|
+
return await auth.authProviders(request.query.mode ?? "login", request.query.authType);
|
|
5
|
+
}
|
|
6
|
+
catch (error) {
|
|
7
|
+
return reply.code(404).send({ error: error instanceof Error ? error.message : String(error) });
|
|
8
|
+
}
|
|
9
|
+
});
|
|
10
|
+
app.post(`${prefix}/auth/api-key`, async (request, reply) => {
|
|
11
|
+
try {
|
|
12
|
+
return await auth.saveApiKey(request.body.providerId, request.body.key);
|
|
13
|
+
}
|
|
14
|
+
catch (error) {
|
|
15
|
+
return reply.code(400).send({ error: error instanceof Error ? error.message : String(error) });
|
|
16
|
+
}
|
|
17
|
+
});
|
|
18
|
+
app.post(`${prefix}/auth/logout`, async (request, reply) => {
|
|
19
|
+
try {
|
|
20
|
+
return await auth.logoutProvider(request.body.providerId);
|
|
21
|
+
}
|
|
22
|
+
catch (error) {
|
|
23
|
+
return reply.code(400).send({ error: error instanceof Error ? error.message : String(error) });
|
|
24
|
+
}
|
|
25
|
+
});
|
|
26
|
+
app.post(`${prefix}/auth/oauth`, async (request, reply) => {
|
|
27
|
+
try {
|
|
28
|
+
return auth.startOAuthLogin(request.body.providerId);
|
|
29
|
+
}
|
|
30
|
+
catch (error) {
|
|
31
|
+
return reply.code(400).send({ error: error instanceof Error ? error.message : String(error) });
|
|
32
|
+
}
|
|
33
|
+
});
|
|
34
|
+
app.get(`${prefix}/auth/oauth/:flowId`, async (request, reply) => {
|
|
35
|
+
try {
|
|
36
|
+
return auth.oauthFlow(request.params.flowId);
|
|
37
|
+
}
|
|
38
|
+
catch (error) {
|
|
39
|
+
return reply.code(404).send({ error: error instanceof Error ? error.message : String(error) });
|
|
40
|
+
}
|
|
41
|
+
});
|
|
42
|
+
app.post(`${prefix}/auth/oauth/:flowId/respond`, async (request, reply) => {
|
|
43
|
+
try {
|
|
44
|
+
return auth.respondToOAuthFlow(request.params.flowId, request.body.requestId, request.body.value);
|
|
45
|
+
}
|
|
46
|
+
catch (error) {
|
|
47
|
+
return reply.code(400).send({ error: error instanceof Error ? error.message : String(error) });
|
|
48
|
+
}
|
|
49
|
+
});
|
|
50
|
+
app.post(`${prefix}/auth/oauth/:flowId/cancel`, async (request, reply) => {
|
|
51
|
+
try {
|
|
52
|
+
return auth.cancelOAuthFlow(request.params.flowId);
|
|
53
|
+
}
|
|
54
|
+
catch (error) {
|
|
55
|
+
return reply.code(400).send({ error: error instanceof Error ? error.message : String(error) });
|
|
56
|
+
}
|
|
57
|
+
});
|
|
58
|
+
}
|
|
59
|
+
//# sourceMappingURL=authRoutes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"authRoutes.js","sourceRoot":"","sources":["../../../src/server/sessions/authRoutes.ts"],"names":[],"mappings":"AAGA,MAAM,UAAU,kBAAkB,CAAC,GAAoB,EAAE,IAAiB,EAAE,MAAM,GAAG,EAAE;IACrF,GAAG,CAAC,GAAG,CAAiF,GAAG,MAAM,iBAAiB,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QAC3I,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,IAAI,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACzF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAgD,GAAG,MAAM,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QACzG,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAmC,GAAG,MAAM,cAAc,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QAC3F,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAmC,GAAG,MAAM,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QAC1F,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAiC,GAAG,MAAM,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QAC/F,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAA6E,GAAG,MAAM,6BAA6B,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QACpJ,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAiC,GAAG,MAAM,4BAA4B,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QACvG,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjG,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
import { AuthStorage, ModelRegistry, SqliteAuthCredentialStore } from "@oh-my-pi/pi-coding-agent";
|
|
2
|
+
import { getLoginProviderOptions, getLogoutProviderOptions } from "./authProviderOptions.js";
|
|
3
|
+
import { OAuthLoginFlowService } from "./oauthLoginFlowService.js";
|
|
4
|
+
/** Adapt omp ModelRegistry to omp-web's AuthProviderModelRegistry interface. */
|
|
5
|
+
function toAuthProviderModelRegistry(mr) {
|
|
6
|
+
const authStorage = mr.authStorage;
|
|
7
|
+
return {
|
|
8
|
+
authStorage: {
|
|
9
|
+
getOAuthProviders: () => [],
|
|
10
|
+
list: () => authStorage.list(),
|
|
11
|
+
get: (provider) => authStorage.get(provider),
|
|
12
|
+
},
|
|
13
|
+
getAll: () => mr.getAll().map((m) => ({ provider: m.provider })),
|
|
14
|
+
getProviderDisplayName: (provider) => mr.getProviderBaseUrl(provider) ?? provider,
|
|
15
|
+
getProviderAuthStatus: () => ({ configured: true }),
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export class AuthService {
|
|
19
|
+
constructor(deps) {
|
|
20
|
+
this.listeners = new Set();
|
|
21
|
+
this.modelRegistry = deps.modelRegistry;
|
|
22
|
+
this.authFlows = deps.authFlows ?? new OAuthLoginFlowService();
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Create an AuthService with the given or default model registry.
|
|
26
|
+
* The default path uses an in-memory auth storage.
|
|
27
|
+
*/
|
|
28
|
+
static async create(deps = {}) {
|
|
29
|
+
const modelRegistry = deps.modelRegistry ?? await createDefaultModelRegistry();
|
|
30
|
+
return new AuthService({ modelRegistry, ...(deps.authFlows === undefined ? {} : { authFlows: deps.authFlows }) });
|
|
31
|
+
}
|
|
32
|
+
subscribe(listener) {
|
|
33
|
+
this.listeners.add(listener);
|
|
34
|
+
return () => {
|
|
35
|
+
this.listeners.delete(listener);
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
dispose() {
|
|
39
|
+
this.authFlows.dispose();
|
|
40
|
+
this.listeners.clear();
|
|
41
|
+
}
|
|
42
|
+
async authProviders(mode, authType) {
|
|
43
|
+
await this.modelRegistry.refresh();
|
|
44
|
+
const adapted = toAuthProviderModelRegistry(this.modelRegistry);
|
|
45
|
+
const providers = mode === "logout" ? getLogoutProviderOptions(adapted) : getLoginProviderOptions(adapted, authType);
|
|
46
|
+
return { providers };
|
|
47
|
+
}
|
|
48
|
+
async saveApiKey(providerId, key) {
|
|
49
|
+
if (key.trim() === "")
|
|
50
|
+
throw new Error("API key is required");
|
|
51
|
+
await this.modelRegistry.authStorage.set(providerId, { type: "api_key", key });
|
|
52
|
+
await this.refreshAuthState();
|
|
53
|
+
return { accepted: true };
|
|
54
|
+
}
|
|
55
|
+
async logoutProvider(providerId) {
|
|
56
|
+
await this.modelRegistry.authStorage.logout(providerId);
|
|
57
|
+
await this.refreshAuthState({ removedProviderId: providerId });
|
|
58
|
+
return { accepted: true };
|
|
59
|
+
}
|
|
60
|
+
startOAuthLogin(providerId) {
|
|
61
|
+
const provider = this.requireOAuthLoginProvider(providerId);
|
|
62
|
+
const view = this.authFlows.start({
|
|
63
|
+
providerId,
|
|
64
|
+
providerName: provider.name,
|
|
65
|
+
authStorage: this.modelRegistry.authStorage,
|
|
66
|
+
});
|
|
67
|
+
return { ...view, providerId, providerName: provider.name };
|
|
68
|
+
}
|
|
69
|
+
oauthFlow(flowId) {
|
|
70
|
+
const view = this.authFlows.get(flowId);
|
|
71
|
+
return { ...view, providerId: "", providerName: "" };
|
|
72
|
+
}
|
|
73
|
+
respondToOAuthFlow(flowId, requestId, value) {
|
|
74
|
+
const view = this.authFlows.respond(flowId, requestId, value);
|
|
75
|
+
return { ...view, providerId: "", providerName: "" };
|
|
76
|
+
}
|
|
77
|
+
cancelOAuthFlow(flowId) {
|
|
78
|
+
this.authFlows.cancel(flowId);
|
|
79
|
+
const view = this.authFlows.get(flowId);
|
|
80
|
+
return { ...view, providerId: "", providerName: "" };
|
|
81
|
+
}
|
|
82
|
+
async refreshAuthState(change = {}) {
|
|
83
|
+
await this.modelRegistry.authStorage.reload();
|
|
84
|
+
await this.modelRegistry.refresh();
|
|
85
|
+
this.emit(change);
|
|
86
|
+
}
|
|
87
|
+
emit(change) {
|
|
88
|
+
for (const listener of this.listeners)
|
|
89
|
+
listener(change);
|
|
90
|
+
}
|
|
91
|
+
requireOAuthLoginProvider(providerId) {
|
|
92
|
+
const adapted = toAuthProviderModelRegistry(this.modelRegistry);
|
|
93
|
+
const provider = getLoginProviderOptions(adapted, "oauth").find((option) => option.id === providerId);
|
|
94
|
+
if (provider === undefined)
|
|
95
|
+
throw new Error(`OAuth provider not found: ${providerId}`);
|
|
96
|
+
return provider;
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
async function createDefaultModelRegistry() {
|
|
100
|
+
const store = await SqliteAuthCredentialStore.open();
|
|
101
|
+
const authStorage = new AuthStorage(store);
|
|
102
|
+
await authStorage.reload();
|
|
103
|
+
return new ModelRegistry(authStorage);
|
|
104
|
+
}
|
|
105
|
+
//# sourceMappingURL=authService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"authService.js","sourceRoot":"","sources":["../../../src/server/sessions/authService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAC;AAElG,OAAO,EAAE,uBAAuB,EAAE,wBAAwB,EAAkC,MAAM,0BAA0B,CAAC;AAC7H,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,gFAAgF;AAChF,SAAS,2BAA2B,CAAC,EAAiB;IACrD,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC;IACnC,OAAO;QACN,WAAW,EAAE;YACZ,iBAAiB,EAAE,GAAmC,EAAE,CAAC,EAAE;YAC3D,IAAI,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE;YAC9B,GAAG,EAAE,CAAC,QAAgB,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC;SACpD;QACD,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;QAChE,sBAAsB,EAAE,CAAC,QAAgB,EAAE,EAAE,CAAC,EAAE,CAAC,kBAAkB,CAAC,QAAQ,CAAC,IAAI,QAAQ;QACzF,qBAAqB,EAAE,GAAuB,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;KACvE,CAAC;AACH,CAAC;AASD,MAAM,OAAO,WAAW;IAKvB,YAAY,IAAyE;QAFpE,cAAS,GAAG,IAAI,GAAG,EAAsB,CAAC;QAG1D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,IAAI,qBAAqB,EAAE,CAAC;IAChE,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAA6E,EAAE;QAClG,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,MAAM,0BAA0B,EAAE,CAAC;QAC/E,OAAO,IAAI,WAAW,CAAC,EAAE,aAAa,EAAE,GAAG,CAAC,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACnH,CAAC;IAED,SAAS,CAAC,QAA4B;QACrC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE;YACX,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC;IACH,CAAC;IAED,OAAO;QACN,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACzB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,IAAwB,EAAE,QAAmB;QAChE,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,2BAA2B,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAChE,MAAM,SAAS,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,uBAAuB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACrH,OAAO,EAAE,SAAS,EAAE,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,UAAkB,EAAE,GAAW;QAC/C,IAAI,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC9D,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;QACxF,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAkB;QACtC,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACxD,MAAM,IAAI,CAAC,gBAAgB,CAAC,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,CAAC;QAC/D,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,eAAe,CAAC,UAAkB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YACjC,UAAU;YACV,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW;SAC3C,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC7D,CAAC;IAED,SAAS,CAAC,MAAc;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,OAAO,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;IACtD,CAAC;IAED,kBAAkB,CAAC,MAAc,EAAE,SAAiB,EAAE,KAAa;QAClE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;IACtD,CAAC;IACD,eAAe,CAAC,MAAc;QAC7B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACxC,OAAO,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,SAAqB,EAAE;QACrD,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;QAC9C,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnB,CAAC;IAEO,IAAI,CAAC,MAAkB;QAC9B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS;YAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IAEO,yBAAyB,CAAC,UAAkB;QACnD,MAAM,OAAO,GAAG,2BAA2B,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAChE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;QACtG,IAAI,QAAQ,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;QACvF,OAAO,QAAQ,CAAC;IACjB,CAAC;CACD;AAED,KAAK,UAAU,0BAA0B;IACxC,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAAC,IAAI,EAAE,CAAC;IACrD,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAC3C,MAAM,WAAW,CAAC,MAAM,EAAE,CAAC;IAC3B,OAAO,IAAI,aAAa,CAAC,WAAW,CAAC,CAAC;AACvC,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
export const BUILTIN_COMMANDS = [
|
|
2
|
+
{ name: "settings", description: "Open settings menu", source: "builtin" },
|
|
3
|
+
{ name: "model", description: "Select model", source: "builtin" },
|
|
4
|
+
{ name: "scoped-models", description: "Enable/disable models for cycling", source: "builtin" },
|
|
5
|
+
{ name: "export", description: "Export session", source: "builtin" },
|
|
6
|
+
{ name: "import", description: "Import and resume a session from JSONL", source: "builtin" },
|
|
7
|
+
{ name: "share", description: "Share session as a secret GitHub gist", source: "builtin" },
|
|
8
|
+
{ name: "copy", description: "Copy last agent message", source: "builtin" },
|
|
9
|
+
{ name: "name", description: "Set session display name", source: "builtin" },
|
|
10
|
+
{ name: "session", description: "Show session info and stats", source: "builtin" },
|
|
11
|
+
{ name: "changelog", description: "Show changelog entries", source: "builtin" },
|
|
12
|
+
{ name: "hotkeys", description: "Show keyboard shortcuts", source: "builtin" },
|
|
13
|
+
{ name: "fork", description: "Create a new fork from a previous user message", source: "builtin" },
|
|
14
|
+
{ name: "clone", description: "Duplicate current session at current position", source: "builtin" },
|
|
15
|
+
{ name: "tree", description: "Navigate session tree", source: "builtin" },
|
|
16
|
+
{ name: "login", description: "Configure provider authentication", source: "builtin" },
|
|
17
|
+
{ name: "logout", description: "Remove provider authentication", source: "builtin" },
|
|
18
|
+
{ name: "new", description: "Start a new session", source: "builtin" },
|
|
19
|
+
{ name: "compact", description: "Manually compact session context", source: "builtin" },
|
|
20
|
+
{ name: "resume", description: "Resume a different session", source: "builtin" },
|
|
21
|
+
{ name: "reload", description: "Reload Pi runtime resources for this session", source: "builtin" },
|
|
22
|
+
{ name: "quit", description: "Quit pi", source: "builtin" },
|
|
23
|
+
{ name: "plan", description: "Toggle plan mode", source: "builtin" },
|
|
24
|
+
{ name: "advisor", description: "Toggle advisor", source: "builtin" },
|
|
25
|
+
];
|
|
26
|
+
export function isBuiltinCommand(name) {
|
|
27
|
+
return BUILTIN_COMMANDS.some((command) => command.name === name);
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=builtinCommands.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"builtinCommands.js","sourceRoot":"","sources":["../../../src/server/sessions/builtinCommands.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,gBAAgB,GAAoB;IAC/C,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,EAAE,SAAS,EAAE;IAC1E,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE;IACjE,EAAE,IAAI,EAAE,eAAe,EAAE,WAAW,EAAE,mCAAmC,EAAE,MAAM,EAAE,SAAS,EAAE;IAC9F,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,EAAE;IACpE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wCAAwC,EAAE,MAAM,EAAE,SAAS,EAAE;IAC5F,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,uCAAuC,EAAE,MAAM,EAAE,SAAS,EAAE;IAC1F,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,EAAE,SAAS,EAAE;IAC3E,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,0BAA0B,EAAE,MAAM,EAAE,SAAS,EAAE;IAC5E,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,6BAA6B,EAAE,MAAM,EAAE,SAAS,EAAE;IAClF,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,wBAAwB,EAAE,MAAM,EAAE,SAAS,EAAE;IAC/E,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,EAAE,SAAS,EAAE;IAC9E,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,gDAAgD,EAAE,MAAM,EAAE,SAAS,EAAE;IAClG,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,+CAA+C,EAAE,MAAM,EAAE,SAAS,EAAE;IAClG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,uBAAuB,EAAE,MAAM,EAAE,SAAS,EAAE;IACzE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,mCAAmC,EAAE,MAAM,EAAE,SAAS,EAAE;IACtF,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gCAAgC,EAAE,MAAM,EAAE,SAAS,EAAE;IACpF,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,qBAAqB,EAAE,MAAM,EAAE,SAAS,EAAE;IACtE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,kCAAkC,EAAE,MAAM,EAAE,SAAS,EAAE;IACvF,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4BAA4B,EAAE,MAAM,EAAE,SAAS,EAAE;IAChF,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8CAA8C,EAAE,MAAM,EAAE,SAAS,EAAE;IAClG,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE;IAC3D,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,MAAM,EAAE,SAAS,EAAE;IACpE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,EAAE;CACtE,CAAC;AAEF,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;AACnE,CAAC"}
|