-
Notifications
You must be signed in to change notification settings - Fork 10
/
compile-graph.js
43 lines (31 loc) · 1.23 KB
/
compile-graph.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import uniq from 'lodash.uniq';
import identity from 'lodash.identity';
import nodeSpecs from './nodes';
export default ({ nodes, edges }) => {
let inject = [];
const createNode = (nodeId) => {
const graphNode = nodes.get(nodeId);
const nodeMetadata = graphNode.get('metadata');
const nodeInstance = new nodeSpecs[graphNode.get('type')](nodeMetadata && nodeMetadata.toJS());
inject.push(graphNode.get('type'));
nodeInstance.getSpec().inlets.forEach(({ id }) => {
const matchingInlet = edges.find(edge => {
return edge.getIn([ 'to', 'id' ]) === nodeId && edge.getIn([ 'to', 'inlet' ]) === id;
});
if (matchingInlet) {
nodeInstance.connect({
[id]: createNode(matchingInlet.getIn([ 'from', 'id' ]))
});
}
});
return nodeInstance;
};
const outputNodeType = 'sysOutput';
const outputNode = nodes.find(node => node.get('type') === outputNodeType);
const outputEdge = outputNode && edges.find(edge => edge.getIn([ 'to', 'id' ]) === outputNode.get('id'));
if (!outputEdge) { return; }
return {
model: createNode(outputEdge.getIn([ 'from', 'id' ])).generate(),
inject: uniq(inject).map(type => nodeSpecs[type].frag).filter(identity)
};
};