Merge pull request #215 from wbkd/develop

fix(edges): add step edge to standard types
This commit is contained in:
Moritz
2020-05-11 22:51:16 +02:00
committed by GitHub
+6 -10
View File
@@ -2,27 +2,23 @@ import { ComponentType } from 'react';
import StraightEdge from '../../components/Edges/StraightEdge'; import StraightEdge from '../../components/Edges/StraightEdge';
import BezierEdge from '../../components/Edges/BezierEdge'; import BezierEdge from '../../components/Edges/BezierEdge';
import StepEdge from '../../components/Edges/StepEdge';
import wrapEdge from '../../components/Edges/wrapEdge'; import wrapEdge from '../../components/Edges/wrapEdge';
import { EdgeTypesType, EdgeCompProps } from '../../types'; import { EdgeTypesType, EdgeCompProps } from '../../types';
export function createEdgeTypes(edgeTypes: EdgeTypesType): EdgeTypesType { export function createEdgeTypes(edgeTypes: EdgeTypesType): EdgeTypesType {
const standardTypes: EdgeTypesType = { const standardTypes: EdgeTypesType = {
default: wrapEdge((edgeTypes.default || BezierEdge) as ComponentType< default: wrapEdge((edgeTypes.default || BezierEdge) as ComponentType<EdgeCompProps>),
EdgeCompProps straight: wrapEdge((edgeTypes.bezier || StraightEdge) as ComponentType<EdgeCompProps>),
>), step: wrapEdge((edgeTypes.step || StepEdge) as ComponentType<EdgeCompProps>),
straight: wrapEdge((edgeTypes.bezier || StraightEdge) as ComponentType<
EdgeCompProps
>),
}; };
const wrappedTypes = {} as EdgeTypesType; const wrappedTypes = {} as EdgeTypesType;
const specialTypes: EdgeTypesType = Object.keys(edgeTypes) const specialTypes: EdgeTypesType = Object.keys(edgeTypes)
.filter(k => !['default', 'bezier'].includes(k)) .filter((k) => !['default', 'bezier'].includes(k))
.reduce((res, key) => { .reduce((res, key) => {
res[key] = wrapEdge((edgeTypes[key] || BezierEdge) as ComponentType< res[key] = wrapEdge((edgeTypes[key] || BezierEdge) as ComponentType<EdgeCompProps>);
EdgeCompProps
>);
return res; return res;
}, wrappedTypes); }, wrappedTypes);