feat(BUG): Azure CosmosDB Integration

server/node/db.cjs

@@ -0,0 +1,107 @@
+const { Container, ContainerResponse, CosmosClient } = require("@azure/cosmos");
+const { array, mixed, object, string } = require("yup");
+
+class CosmosStore {
+  /**
+   * @constructor
+   * @param {string} endpoint - Azure Cosmos Endpoint
+   * @param {string} key - Azure Key
+   * @param {string} database - Cosmos Database ID
+   * @param {string} container - Cosmos Container ID
+   * @throws {Error}
+   */
+  constructor(endpoint, key, database, container) {
+    if (typeof endpoint != "string") throw new Error("Invalid endpoint type");
+    if (typeof key != "string") throw new Error("Invalid key type");
+    if (typeof database != "string") throw new Error("Invalid database type");
+    if (typeof container != "string") throw new Error("Invalid container type");
+
+    this.$endpoint = endpoint;
+    this.$key = key;
+    this.$databaseName = database;
+    this.$containerName = container;
+
+    this.$containerRun = null;
+    this.$client = new CosmosClient({ endpoint, key });
+  }
+
+  /**
+   * Gets a container, create if not exists.
+   * @returns {Promise<Container>}
+   * @throws {Error}
+   */
+  async resolveContainer() {
+    if (this.$containerRun) {
+      return this.$containerRun;
+    } else {
+      const { database } = await this.$client.databases.createIfNotExists({
+        id: this.$databaseName,
+      });
+
+      const { container } = await database.containers.createIfNotExists({
+        id: this.$containerName,
+        partitionKey: {
+          paths: ["/id"],
+          kind: "Hash",
+          version: 2,
+        },
+      });
+
+      this.$containerRun = container;
+      return container;
+    }
+  }
+
+  /**
+   * List all data in container.
+   * @returns {Promise<Array<{ name: string, content: Buffer }>>}
+   * @throws {Error}
+   */
+  async listData() {
+    const c = await this.resolveContainer();
+    const list = await c.items.readAll().fetchAll();
+    return list.resources;
+  }
+
+  /**
+   * Creates a data in container.
+   * @param {string} name - Name of the data
+   * @param {Buffer} content - Content
+   * @throws {Error}
+   */
+  async createData(name, content) {
+    const c = await this.resolveContainer();
+
+    const item = {
+      id: name,
+      content: content,
+    };
+
+    await c.items.create(item);
+  }
+
+  /**
+   * Removes a data in container by name.
+   * @param {string} name - Name of the data to remove
+   * @throws {Error}
+   */
+  async removeData(name) {
+    const c = await this.resolveContainer();
+    await c.item(name).delete();
+  }
+
+  /**
+   * Gets a data in container by name.
+   * @param {string} name - Name of the data to get
+   * @throws {Error}
+   */
+  async getData(name) {
+    const c = await this.resolveContainer();
+    const data = await c.item(name).read();
+
+    console.log(data);
+    return data.content;
+  }
+}
+
+exports.CosmosStore = CosmosStore;