-
Notifications
You must be signed in to change notification settings - Fork 89
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
30 changed files
with
837 additions
and
445 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,162 @@ | ||
| import { Context } from "deco/deco.ts"; | ||
| import { Resolvable } from "deco/engine/core/resolver.ts"; | ||
| import { lazySchemaFor } from "deco/engine/schema/lazy.ts"; | ||
| import { MetaInfo, toManifestBlocks } from "deco/runtime/fresh/routes/_meta.ts"; | ||
| import { fjp } from "../../deps.ts"; | ||
| import { storage } from "../../fsStorage.ts"; | ||
|
|
||
| export interface Props { | ||
| /** Environment name to connect to */ | ||
| name: string; | ||
| } | ||
|
|
||
| interface PatchState { | ||
| type: "patch-state"; | ||
| payload: fjp.Operation[]; | ||
| } | ||
|
|
||
| interface FetchState { | ||
| type: "fetch-state"; | ||
| } | ||
|
|
||
| interface StatePatched { | ||
| type: "state-patched"; | ||
| payload: fjp.Operation[]; | ||
| // Maybe add data and user info in here | ||
| metadata?: unknown; | ||
| } | ||
|
|
||
| interface StateFetched { | ||
| type: "state-fetched"; | ||
| payload: State; | ||
| } | ||
|
|
||
| type Acked<T> = T & { ack: string }; | ||
|
|
||
| interface State { | ||
| blocks: Record<string, Resolvable>; | ||
| manifest: MetaInfo["manifest"]; | ||
| schema: MetaInfo["schema"]; | ||
| etag: string; | ||
| } | ||
|
|
||
| type Commands = PatchState | FetchState; | ||
| type Events = StatePatched | StateFetched; | ||
|
|
||
| const subscribers: WebSocket[] = []; | ||
|
|
||
| const fetchState = async (): Promise<State> => { | ||
| const blocks = await storage.state({ forceFresh: true }); | ||
|
|
||
| const active = Context.active(); | ||
| const lazySchema = lazySchemaFor(active); | ||
|
|
||
| const [schema, runtime] = await Promise.all([ | ||
| lazySchema.value, | ||
| active.runtime, | ||
| ]); | ||
|
|
||
| return { | ||
| blocks, | ||
| manifest: toManifestBlocks(runtime!.manifest), | ||
| schema, | ||
| etag: await storage.revision(), | ||
| }; | ||
| }; | ||
|
|
||
| const saveState = ({ blocks }: State): Promise<void> => storage.update(blocks); | ||
|
|
||
| // Apply patch and save state ATOMICALLY! | ||
| // This is easily done on play. On production, however, we probably | ||
| // need a distributed queue | ||
| let queue = Promise.resolve<fjp.Operation[]>([]); | ||
| const patchState = (decofileOps: fjp.Operation[]) => { | ||
| queue = queue.catch(() => null).then(async () => { | ||
| const state = await fetchState(); | ||
| const observer = fjp.observe<State>(state); | ||
|
|
||
| try { | ||
| await saveState(decofileOps.reduce(fjp.applyReducer, state)); | ||
|
|
||
| // Wait for a while before fetching the state | ||
| const newState = await new Promise<State>((resolve) => | ||
| setTimeout(() => resolve(fetchState()), 1e3) | ||
| ); | ||
|
|
||
| state.manifest = newState.manifest; | ||
| state.schema = newState.schema; | ||
| state.etag = newState.etag; | ||
|
|
||
| return fjp.generate(observer, true); | ||
| } finally { | ||
| fjp.unobserve(state, observer); | ||
| } | ||
| }); | ||
|
|
||
| return queue; | ||
| }; | ||
|
|
||
| const action = (_props: Props, req: Request) => { | ||
| const { socket, response } = Deno.upgradeWebSocket(req); | ||
|
|
||
| const broadcast = (event: Acked<Events>) => { | ||
| const message = JSON.stringify(event); | ||
| subscribers.forEach((s) => s.send(message)); | ||
| }; | ||
| const send = (event: Acked<Events>) => socket.send(JSON.stringify(event)); | ||
| const parse = (event: MessageEvent<string>): Acked<Commands> => | ||
| JSON.parse(event.data); | ||
|
|
||
| const open = () => subscribers.push(socket); | ||
| const close = () => subscribers.splice(subscribers.indexOf(socket), 1); | ||
| const message = async (event: MessageEvent<string>) => { | ||
| const data = parse(event); | ||
|
|
||
| const { ack } = data; | ||
|
|
||
| if (data.type === "patch-state") { | ||
| try { | ||
| const { payload: decofileOps } = data; | ||
|
|
||
| const allOps = await patchState(decofileOps); | ||
|
|
||
| // Broadcast changes | ||
| broadcast({ | ||
| type: "state-patched", | ||
| payload: allOps, | ||
| metadata: {}, // TODO: add metadata | ||
| ack, | ||
| }); | ||
| } catch ({ name, operation }) { | ||
| console.error({ name, operation }); | ||
| } | ||
| } else if (data.type === "fetch-state") { | ||
| send({ | ||
| type: "state-fetched", | ||
| payload: await fetchState(), | ||
| ack, | ||
| }); | ||
| } else { | ||
| console.error("UNKNOWN EVENT", event); | ||
| } | ||
| }; | ||
|
|
||
| /** | ||
| * Handles the WebSocket connection on open event. | ||
| */ | ||
| socket.onopen = open; | ||
| /** | ||
| * Handles the WebSocket connection on close event. | ||
| */ | ||
| socket.onclose = close; | ||
|
|
||
| /** | ||
| * Handles the WebSocket connection on message event. | ||
| * @param {MessageEvent} event - The WebSocket message event. | ||
| */ | ||
| socket.onmessage = (e) => message(e).catch(() => {}); | ||
|
|
||
| return response; | ||
| }; | ||
|
|
||
| export default action; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.