Added new graph methods for vertex

Macrometacorp · Mar 23, 2023 · bae6a6a · bae6a6a
1 parent d812022
commit bae6a6a
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 11 deletions.
diff --git a/src/client.ts b/src/client.ts
@@ -512,8 +512,7 @@ export class C8Client extends Fabric {
     return streamApp.activateStreamApplication(active);
   }
 
-  //************
-  // Graphs Methods
+  //------------ Graph ----------
   getGraphs(): Promise<Graph[]> {
     return this.graphs();
   }
@@ -549,6 +548,22 @@ export class C8Client extends Fabric {
     return graph.addEdgeDefinition(definition);
   }
 
+  removeEdgeDefinition(graphName: string, edgeCollectionName: string) {
+    const graph = this.graph(graphName);
+    return graph.removeEdgeDefinition(edgeCollectionName);
+  }
+
+  getEdge(
+    graphName: string,
+    collectionName: string,
+    documentHandle: DocumentHandle,
+    opts: any = {}
+  ) {
+    const graph = this.graph(graphName);
+    const graphEdgeCollection = graph.edgeCollection(collectionName);
+    return graphEdgeCollection.document(documentHandle, opts);
+  }
+
   updateEdge(
     graphName: string,
     collectionName: string,
@@ -617,18 +632,83 @@ export class C8Client extends Fabric {
     return graph.listVertexCollections();
   }
 
-  addVertexToCollection(
+  async addVertexCollection(graphName: string, collectionName: string) {
+    const graph = this.graph(graphName);
+    return graph.addVertexCollection(collectionName);
+  }
+
+  async removeVertexCollection(
+    graphName: string,
+    collectionName: string,
+    dropCollection: boolean = false
+  ) {
+    const graph = this.graph(graphName);
+    return graph.removeVertexCollection(collectionName, dropCollection);
+  }
+
+  addVertexToVertexCollection(
     graphName: string,
     collectionName: string,
     properties: any = {},
     returnNew: boolean = false
   ) {
     const graph = this.graph(graphName);
-    return graph.addVertexToCollection(collectionName, properties, returnNew);
+    return graph.addVertexToVertexCollection(
+      collectionName,
+      properties,
+      returnNew
+    );
+  }
+
+  removeVertexFromVertexCollection(
+    graphName: string,
+    collectionName: string,
+    documentHandle: DocumentHandle,
+    opts: any = {}
+  ) {
+    const graph = this.graph(graphName);
+    return graph.vertexCollection(collectionName).remove(documentHandle, opts);
+  }
+
+  getVertexFromVertexCollection(
+    graphName: string,
+    collectionName: string,
+    documentHandle: DocumentHandle,
+    opts: any = {}
+  ) {
+    const graph = this.graph(graphName);
+    return graph
+      .vertexCollection(collectionName)
+      .document(documentHandle, opts);
+  }
+
+  updateVertexFromVertexCollection(
+    graphName: string,
+    collectionName: string,
+    documentHandle: DocumentHandle,
+    newValue: any,
+    opts: any = {}
+  ) {
+    const graph = this.graph(graphName);
+    return graph
+      .vertexCollection(collectionName)
+      .update(documentHandle, newValue, opts);
+  }
+
+  replaceVertexFromVertexCollection(
+    graphName: string,
+    collectionName: string,
+    documentHandle: DocumentHandle,
+    newValue: any,
+    opts: any = {}
+  ) {
+    const graph = this.graph(graphName);
+    return graph
+      .vertexCollection(collectionName)
+      .replace(documentHandle, newValue, opts);
   }
 
-  //************
-  // Users Methods
+  //------------ User ----------
   hasUser(userName: string) {
     const user = this.user(userName);
     return user.hasUser();

diff --git a/src/graph.ts b/src/graph.ts
@@ -456,10 +456,7 @@ export class Graph {
     );
   }
 
-  //todo New Edge Endpoints
-
-  // todo Add Vertex to given collection
-  addVertexToCollection(
+  addVertexToVertexCollection(
     collectionName: string,
     properties: any = {},
     returnNew: boolean = false
@@ -479,7 +476,6 @@ export class Graph {
     );
   }
 
-  // todo Add new Edge to given collection
   addEdgeToEdgeCollection(
     edgeCollectionName: string,
     properties: any = {},