Browse Source

add types

AykutSarac 3 years ago
parent
commit
f54337b04d
2 changed files with 10 additions and 4 deletions
  1. 3 3
      src/components/Graph/index.tsx
  2. 7 1
      src/containers/LiveEditor/helpers.ts

+ 3 - 3
src/components/Graph/index.tsx

@@ -1,12 +1,12 @@
 import React from "react";
 import React from "react";
-import { Canvas } from "reaflow";
+import { Canvas, EdgeData, NodeData } from "reaflow";
 import { CustomNode } from "src/components/CustomNode";
 import { CustomNode } from "src/components/CustomNode";
 import { useConfig } from "src/hocs/config";
 import { useConfig } from "src/hocs/config";
 import { getEdgeNodes } from "src/containers/LiveEditor/helpers";
 import { getEdgeNodes } from "src/containers/LiveEditor/helpers";
 
 
 export const Graph = () => {
 export const Graph = () => {
-  const [nodes, setNodes] = React.useState([]);
-  const [edges, setEdges] = React.useState([]);
+  const [nodes, setNodes] = React.useState<NodeData[]>([]);
+  const [edges, setEdges] = React.useState<EdgeData[]>([]);
   const [size, setSize] = React.useState({
   const [size, setSize] = React.useState({
     width: 2000,
     width: 2000,
     height: 2000,
     height: 2000,

+ 7 - 1
src/containers/LiveEditor/helpers.ts

@@ -14,7 +14,13 @@ const toString = (value: string | object) => {
   return String(value);
   return String(value);
 };
 };
 
 
-export function getEdgeNodes(graph: any, isExpanded: boolean = true): any {
+export function getEdgeNodes(
+  graph: string,
+  isExpanded: boolean = true
+): {
+  nodes: NodeData[];
+  edges: EdgeData[];
+} {
   graph = JSON.parse(graph);
   graph = JSON.parse(graph);
   const elements = parser(graph);
   const elements = parser(graph);