@@ -6,37 +6,50 @@ export function useGetters(state: State): ComputedGetters {
6
6
const edgeIds = computed ( ( ) => state . edges . map ( ( e ) => e . id ) )
7
7
8
8
const getNode : ComputedGetters [ 'getNode' ] = computed ( ( ) => ( id : string ) => {
9
- if ( state . nodes && ! nodeIds . value . length ) return state . nodes . find ( ( node ) => node . id === id )
9
+ if ( state . nodes && ! nodeIds . value . length ) {
10
+ return state . nodes . find ( ( node ) => node . id === id )
11
+ }
12
+
10
13
return state . nodes [ nodeIds . value . indexOf ( id ) ]
11
14
} )
12
15
13
16
const getEdge : ComputedGetters [ 'getEdge' ] = computed ( ( ) => ( id : string ) => {
14
- if ( state . edges && ! edgeIds . value . length ) return state . edges . find ( ( edge ) => edge . id === id )
17
+ if ( state . edges && ! edgeIds . value . length ) {
18
+ return state . edges . find ( ( edge ) => edge . id === id )
19
+ }
20
+
15
21
return state . edges [ edgeIds . value . indexOf ( id ) ]
16
22
} )
17
23
18
- const getEdgeTypes = computed ( ( ) => {
24
+ const getEdgeTypes : ComputedGetters [ 'getEdgeTypes' ] = computed ( ( ) => {
19
25
const edgeTypes : Record < string , any > = {
20
26
...defaultEdgeTypes ,
21
27
...state . edgeTypes ,
22
28
}
29
+
23
30
const keys = Object . keys ( edgeTypes )
31
+
24
32
state . edges ?. forEach ( ( e ) => e . type && ! keys . includes ( e . type ) && ( edgeTypes [ e . type ] = e . type ) )
33
+
25
34
return edgeTypes
26
35
} )
27
36
28
- const getNodeTypes = computed ( ( ) => {
37
+ const getNodeTypes : ComputedGetters [ 'getNodeTypes' ] = computed ( ( ) => {
29
38
const nodeTypes : Record < string , any > = {
30
39
...defaultNodeTypes ,
31
40
...state . nodeTypes ,
32
41
}
42
+
33
43
const keys = Object . keys ( nodeTypes )
44
+
34
45
state . nodes ?. forEach ( ( n ) => n . type && ! keys . includes ( n . type ) && ( nodeTypes [ n . type ] = n . type ) )
46
+
35
47
return nodeTypes
36
48
} )
37
49
38
- const getNodes = computed < GraphNode [ ] > ( ( ) => {
50
+ const getNodes : ComputedGetters [ 'getNodes' ] = computed ( ( ) => {
39
51
const nodes = state . nodes . filter ( ( n ) => ! n . hidden )
52
+
40
53
return state . onlyRenderVisibleElements
41
54
? nodes &&
42
55
getNodesInside (
@@ -67,7 +80,7 @@ export function useGetters(state: State): ComputedGetters {
67
80
return ! e . hidden && ! target . hidden && ! source . hidden
68
81
}
69
82
70
- const getEdges = computed < GraphEdge [ ] > ( ( ) => {
83
+ const getEdges : ComputedGetters [ 'getEdges' ] = computed ( ( ) => {
71
84
if ( ! state . onlyRenderVisibleElements ) return state . edges . filter ( ( edge ) => edgeHidden ( edge ) )
72
85
73
86
return state . edges . filter ( ( e ) => {
0 commit comments