112 lines
2.7 KiB
TypeScript
112 lines
2.7 KiB
TypeScript
import { createHooks } from './hooks'
|
|
import {
|
|
ConnectionMode,
|
|
State,
|
|
PanOnScrollMode,
|
|
DefaultNodeTypes,
|
|
DefaultEdgeTypes,
|
|
ConnectionLineType,
|
|
FlowOptions,
|
|
} from '~/types'
|
|
import { DefaultNode, InputNode, OutputNode, BezierEdge, SmoothStepEdge, StepEdge, StraightEdge } from '~/components'
|
|
|
|
export const defaultNodeTypes: DefaultNodeTypes = {
|
|
input: InputNode,
|
|
default: DefaultNode,
|
|
output: OutputNode,
|
|
}
|
|
|
|
export const defaultEdgeTypes: DefaultEdgeTypes = {
|
|
default: BezierEdge,
|
|
straight: StraightEdge,
|
|
step: StepEdge,
|
|
smoothstep: SmoothStepEdge,
|
|
}
|
|
|
|
const isDef = <T>(val: T): val is NonNullable<T> => typeof val !== 'undefined'
|
|
export default (opts?: FlowOptions): State => {
|
|
const state: State = {
|
|
nodes: [],
|
|
edges: [],
|
|
|
|
paneReady: false,
|
|
initialized: false,
|
|
instance: undefined,
|
|
|
|
dimensions: {
|
|
width: 0,
|
|
height: 0,
|
|
},
|
|
transform: [0, 0, 1],
|
|
|
|
d3Zoom: undefined,
|
|
d3Selection: undefined,
|
|
d3ZoomHandler: undefined,
|
|
minZoom: 0.5,
|
|
maxZoom: 2,
|
|
translateExtent: [
|
|
[Number.NEGATIVE_INFINITY, Number.NEGATIVE_INFINITY],
|
|
[Number.POSITIVE_INFINITY, Number.POSITIVE_INFINITY],
|
|
],
|
|
|
|
nodeExtent: [
|
|
[Number.NEGATIVE_INFINITY, Number.NEGATIVE_INFINITY],
|
|
[Number.POSITIVE_INFINITY, Number.POSITIVE_INFINITY],
|
|
],
|
|
preventScrolling: true,
|
|
zoomOnScroll: true,
|
|
zoomOnPinch: true,
|
|
zoomOnDoubleClick: true,
|
|
panOnScroll: false,
|
|
panOnScrollSpeed: 0.5,
|
|
panOnScrollMode: PanOnScrollMode.Free,
|
|
paneMovable: true,
|
|
edgeUpdaterRadius: 10,
|
|
onlyRenderVisibleElements: false,
|
|
defaultZoom: 1,
|
|
defaultPosition: [0, 0],
|
|
|
|
nodesSelectionActive: false,
|
|
selectionActive: false,
|
|
selectedNodesBbox: { x: 0, y: 0, width: 0, height: 0 },
|
|
|
|
defaultMarkerColor: '#b1b1b7',
|
|
connectionLineStyle: {},
|
|
connectionLineType: ConnectionLineType.Bezier,
|
|
connectionNodeId: undefined,
|
|
connectionHandleId: undefined,
|
|
connectionHandleType: 'source',
|
|
connectionPosition: { x: NaN, y: NaN },
|
|
connectionMode: ConnectionMode.Loose,
|
|
|
|
snapGrid: [15, 15],
|
|
snapToGrid: false,
|
|
|
|
edgesUpdatable: false,
|
|
nodesConnectable: true,
|
|
nodesDraggable: true,
|
|
elementsSelectable: true,
|
|
selectNodesOnDrag: true,
|
|
multiSelectionActive: false,
|
|
selectionKeyCode: 'Shift',
|
|
multiSelectionKeyCode: 'Meta',
|
|
zoomActivationKeyCode: undefined,
|
|
deleteKeyCode: 'Backspace',
|
|
|
|
hooks: createHooks(),
|
|
|
|
applyDefault: true,
|
|
|
|
vueFlowVersion: typeof __VUE_FLOW_VERSION__ !== 'undefined' ? __VUE_FLOW_VERSION__ : '-',
|
|
}
|
|
|
|
if (opts) {
|
|
Object.keys(opts).forEach((o) => {
|
|
const option = opts[o as keyof typeof opts]
|
|
if (isDef(option)) (state as any)[o] = option
|
|
})
|
|
}
|
|
|
|
return state
|
|
}
|