| ... | ... | @@ -1,8 +1,10 @@ |
| 1 | import * as DateTime from "effect/DateTime"; |
| 1 | 2 | import * as Effect from "effect/Effect"; |
| 2 | 3 | import * as FileSystem from "effect/FileSystem"; |
| 3 | 4 | import * as Path from "effect/Path"; |
| 4 | 5 | import { |
| 5 | 6 | type ClientOrchestrationCommand, |
| 7 | type IsoDateTime, |
| 6 | 8 | type OrchestrationCommand, |
| 7 | 9 | OrchestrationDispatchCommandError, |
| 8 | 10 | PROVIDER_SEND_TURN_MAX_IMAGE_BYTES, |
| ... | ... | @@ -13,8 +15,38 @@ import { ServerConfig } from "../config.ts"; |
| 13 | 15 | import { parseBase64DataUrl } from "../imageMime.ts"; |
| 14 | 16 | import * as WorkspacePaths from "../workspace/WorkspacePaths.ts"; |
| 15 | 17 | |
| 18 | export const canonicalizeClientCommandTimestamps = ( |
| 19 | command: ClientOrchestrationCommand, |
| 20 | receivedAt: IsoDateTime, |
| 21 | ): ClientOrchestrationCommand => { |
| 22 | const canonicalCommand = |
| 23 | "createdAt" in command |
| 24 | ? { |
| 25 | ...command, |
| 26 | createdAt: receivedAt, |
| 27 | } |
| 28 | : command; |
| 29 | |
| 30 | if (canonicalCommand.type !== "thread.turn.start" || !canonicalCommand.bootstrap?.createThread) { |
| 31 | return canonicalCommand; |
| 32 | } |
| 33 | |
| 34 | return { |
| 35 | ...canonicalCommand, |
| 36 | bootstrap: { |
| 37 | ...canonicalCommand.bootstrap, |
| 38 | createThread: { |
| 39 | ...canonicalCommand.bootstrap.createThread, |
| 40 | createdAt: receivedAt, |
| 41 | }, |
| 42 | }, |
| 43 | }; |
| 44 | }; |
| 45 | |
| 16 | 46 | export const normalizeDispatchCommand = (command: ClientOrchestrationCommand) => |
| 17 | 47 | Effect.gen(function* () { |
| 48 | const receivedAt = DateTime.formatIso(yield* DateTime.now); |
| 49 | const canonicalCommand = canonicalizeClientCommandTimestamps(command, receivedAt); |
| 18 | 50 | const fileSystem = yield* FileSystem.FileSystem; |
| 19 | 51 | const path = yield* Path.Path; |
| 20 | 52 | const serverConfig = yield* ServerConfig; |
| ... | ... | @@ -47,30 +79,33 @@ export const normalizeDispatchCommand = (command: ClientOrchestrationCommand) => |
| 47 | 79 | ), |
| 48 | 80 | ); |
| 49 | 81 | |
| 50 | | if (command.type === "project.create") { |
| 82 | if (canonicalCommand.type === "project.create") { |
| 51 | 83 | return { |
| 52 | | ...command, |
| 84 | ...canonicalCommand, |
| 53 | 85 | workspaceRoot: yield* normalizeProjectWorkspaceRootForCreate( |
| 54 | | command.workspaceRoot, |
| 55 | | command.createWorkspaceRootIfMissing, |
| 86 | canonicalCommand.workspaceRoot, |
| 87 | canonicalCommand.createWorkspaceRootIfMissing, |
| 56 | 88 | ), |
| 57 | | createWorkspaceRootIfMissing: command.createWorkspaceRootIfMissing === true, |
| 89 | createWorkspaceRootIfMissing: canonicalCommand.createWorkspaceRootIfMissing === true, |
| 58 | 90 | } satisfies OrchestrationCommand; |
| 59 | 91 | } |
| 60 | 92 | |
| 61 | | if (command.type === "project.meta.update" && command.workspaceRoot !== undefined) { |
| 93 | if ( |
| 94 | canonicalCommand.type === "project.meta.update" && |
| 95 | canonicalCommand.workspaceRoot !== undefined |
| 96 | ) { |
| 62 | 97 | return { |
| 63 | | ...command, |
| 64 | | workspaceRoot: yield* normalizeProjectWorkspaceRoot(command.workspaceRoot), |
| 98 | ...canonicalCommand, |
| 99 | workspaceRoot: yield* normalizeProjectWorkspaceRoot(canonicalCommand.workspaceRoot), |
| 65 | 100 | } satisfies OrchestrationCommand; |
| 66 | 101 | } |
| 67 | 102 | |
| 68 | | if (command.type !== "thread.turn.start") { |
| 69 | | return command as OrchestrationCommand; |
| 103 | if (canonicalCommand.type !== "thread.turn.start") { |
| 104 | return canonicalCommand as OrchestrationCommand; |
| 70 | 105 | } |
| 71 | 106 | |
| 72 | 107 | const normalizedAttachments = yield* Effect.forEach( |
| 73 | | command.message.attachments, |
| 108 | canonicalCommand.message.attachments, |
| 74 | 109 | (attachment) => |
| 75 | 110 | Effect.gen(function* () { |
| 76 | 111 | const parsed = parseBase64DataUrl(attachment.dataUrl); |
| ... | ... | @@ -87,7 +122,7 @@ export const normalizeDispatchCommand = (command: ClientOrchestrationCommand) => |
| 87 | 122 | }); |
| 88 | 123 | } |
| 89 | 124 | |
| 90 | | const attachmentId = createAttachmentId(command.threadId); |
| 125 | const attachmentId = createAttachmentId(canonicalCommand.threadId); |
| 91 | 126 | if (!attachmentId) { |
| 92 | 127 | return yield* new OrchestrationDispatchCommandError({ |
| 93 | 128 | message: "Failed to create a safe attachment id.", |
| ... | ... | @@ -135,9 +170,9 @@ export const normalizeDispatchCommand = (command: ClientOrchestrationCommand) => |
| 135 | 170 | ); |
| 136 | 171 | |
| 137 | 172 | return { |
| 138 | | ...command, |
| 173 | ...canonicalCommand, |
| 139 | 174 | message: { |
| 140 | | ...command.message, |
| 175 | ...canonicalCommand.message, |
| 141 | 176 | attachments: normalizedAttachments, |
| 142 | 177 | }, |
| 143 | 178 | } satisfies OrchestrationCommand; |