Sync engine (#298)

Fixes #261
2023-01-13 15:41:29 +01:00
parent de6f531e91
commit a56e14bff1
51 changed files with 1033 additions and 1418 deletions
@@ -0,0 +1,131 @@
+import { Plug } from "../../plugos/plug.ts";
+import {
+  FileData,
+  FileEncoding,
+  SpacePrimitives,
+} from "../../common/spaces/space_primitives.ts";
+import { FileMeta } from "../../common/types.ts";
+import {
+  NamespaceOperation,
+  PageNamespaceHook,
+} from "../hooks/page_namespace.ts";
+import { base64DecodeDataUrl } from "../../plugos/asset_bundle/base64.ts";
+
+export class PlugSpacePrimitives implements SpacePrimitives {
+  constructor(
+    private wrapped: SpacePrimitives,
+    private hook: PageNamespaceHook,
+    private env?: string,
+  ) {}
+
+  performOperation(
+    type: NamespaceOperation,
+    pageName: string,
+    ...args: any[]
+  ): Promise<any> | false {
+    for (
+      const { operation, pattern, plug, name, env } of this.hook.spaceFunctions
+    ) {
+      if (
+        operation === type && pageName.match(pattern) &&
+        (!this.env || (env && env === this.env))
+      ) {
+        return plug.invoke(name, [pageName, ...args]);
+      }
+    }
+    return false;
+  }
+
+  async fetchFileList(): Promise<FileMeta[]> {
+    const allFiles: FileMeta[] = [];
+    for (const { plug, name, operation } of this.hook.spaceFunctions) {
+      if (operation === "listFiles") {
+        try {
+          for (const pm of await plug.invoke(name, [])) {
+            allFiles.push(pm);
+          }
+        } catch (e) {
+          console.error("Error listing files", e);
+        }
+      }
+    }
+    const files = await this.wrapped.fetchFileList();
+    for (const pm of files) {
+      allFiles.push(pm);
+    }
+    return allFiles;
+  }
+
+  async readFile(
+    name: string,
+    encoding: FileEncoding,
+  ): Promise<{ data: FileData; meta: FileMeta }> {
+    const wantArrayBuffer = encoding === "arraybuffer";
+    const result: { data: FileData; meta: FileMeta } | false = await this
+      .performOperation(
+        "readFile",
+        name,
+        wantArrayBuffer ? "dataurl" : encoding,
+      );
+    if (result) {
+      if (wantArrayBuffer) {
+        return {
+          data: base64DecodeDataUrl(result.data as string),
+          meta: result.meta,
+        };
+      } else {
+        return result;
+      }
+    }
+    return this.wrapped.readFile(name, encoding);
+  }
+
+  getFileMeta(name: string): Promise<FileMeta> {
+    const result = this.performOperation("getFileMeta", name);
+    if (result) {
+      return result;
+    }
+    return this.wrapped.getFileMeta(name);
+  }
+
+  writeFile(
+    name: string,
+    encoding: FileEncoding,
+    data: FileData,
+    selfUpdate?: boolean,
+  ): Promise<FileMeta> {
+    const result = this.performOperation(
+      "writeFile",
+      name,
+      encoding,
+      data,
+      selfUpdate,
+    );
+    if (result) {
+      return result;
+    }
+
+    return this.wrapped.writeFile(name, encoding, data, selfUpdate);
+  }
+
+  deleteFile(name: string): Promise<void> {
+    const result = this.performOperation("deleteFile", name);
+    if (result) {
+      return result;
+    }
+    return this.wrapped.deleteFile(name);
+  }
+
+  proxySyscall(plug: Plug<any>, name: string, args: any[]): Promise<any> {
+    return this.wrapped.proxySyscall(plug, name, args);
+  }
+
+  invokeFunction(
+    plug: Plug<any>,
+    env: string,
+    name: string,
+    args: any[],
+  ): Promise<any> {
+    return this.wrapped.invokeFunction(plug, env, name, args);
+  }
+}