index.d.ts make

Author: thebestnom

index.d.ts

@@ -1,3 +1,7 @@
+import { DataSet } from "vis-data";
+import { Node as VisNode, Edge as VisEdge } from "vis-network";
+import { Node as Neo4jNode, Relationship as Neo4jRelationship } from "neo4j-driver";
+
 export const NEOVIS_DEFAULT_CONFIG: unique symbol;
 
 export interface ILabelConfig {
@@ -35,10 +39,18 @@ export interface INeovisConfig {
     trust?: "TRUST_ALL_CERTIFICATES" | "TRUST_SYSTEM_CA_SIGNED_CERTIFICATES";
 }
 
+export interface INode extends VisNode {
+    raw: Neo4jNode
+}
+
+export interface IEdge extends VisEdge {
+    raw: Neo4jRelationship
+}
+
 declare class Neovis {
-    nodes: any;
-    edges: any;
     constructor(config: INeovisConfig);
+    get nodes(): DataSet<INode>;
+    get edges(): DataSet<IEdge>;
     render(): void;
     clearNetwork(): void;
     registerOnEvent(eventType: string, handler: (event: any) => void): void;

package.json

@@ -60,6 +60,7 @@
   "dependencies": {
     "@babel/runtime-corejs3": "^7.8.4",
     "neo4j-driver": "^4.1.0",
+    "vis-data": "^7.0.0",
     "vis-network": "^7.3.5"
   },
   "jest": {

yarn.lock

@@ -6475,6 +6475,11 @@ [email protected]:
     core-util-is "1.0.2"
     extsprintf "^1.2.0"
 
+vis-data@^7.0.0:
+  version "7.0.0"
+  resolved "https://registry.yarnpkg.com/vis-data/-/vis-data-7.0.0.tgz#71ff8ec06e4d0b99f1c8cdecfdf97309eb4562cf"
+  integrity sha512-qKpyAQ9UMT0QygLbCulgabKkgfo8aVkuWSqhvEiaah/iy/Dvj17iMFChUU+UIioorWlweXcp5ziXoMLIl7hTAg==
+
 vis-network@^7.3.5:
   version "7.10.2"
   resolved "https://registry.yarnpkg.com/vis-network/-/vis-network-7.10.2.tgz#b318f1907cf006d9640c4c31a262e0782405a3cf"