AbstractActionNodeFactory.ts

import * as React from 'react';
import { DiagramEngine, AbstractNodeFactory } from 'storm-react-diagrams';

import { BaseNodeModel } from '../models/BaseNodeModel';
import { BaseNodeWidget } from '../widgets/BaseNodeWidget';

export class AbstractActionNodeFactory<T extends BaseNodeModel, U extends BaseNodeWidget> extends AbstractNodeFactory<T> {

	constructor(nodeType: string, public nodeModel: (new (...args: any[]) => T), public nodeWidget: (new (...args: any[]) => U)) {
		super(nodeType);
	}

	generateReactWidget(diagramEngine: DiagramEngine, node: T): JSX.Element {
		return React.createElement(this.nodeWidget, {
			node: node,
			diagramEngine: diagramEngine
		});
	}

	getNewInstance(initialConfig?: any): T {
		return new this.nodeModel();
	}
}