src/logic/model/NodeFactory.ts - master - gamma/Gamma-Editor

src/logic/model/NodeFactory.ts
48 строк2 KB
Vlavimir Solodovnikov
# Save changes
09 июн 2022, 02:46
09 июн 2022, 02:46777b915
import NodeTypeEnum from 'logic/model/NodeTypeEnum';
import SimpleContainerNode from 'logic/model/page/Containers/SimpleContainerNode';
import NodeBase from 'logic/model/page/NodeBase';
import ButtonNode from 'logic/model/page/Nodes/ButtonNode';
import TextNode from 'logic/model/page/Nodes/TextNode';
import PageDocument from 'logic/model/page/PageDocument';
import INodeFactory from 'logic/model/INodeFactory';
import IContainerNode from 'logic/model/page/IContainerNode';
import ColumnsContainerNode from 'logic/model/page/Containers/ColumnsContainerNode';
import ImageNode from 'logic/model/page/Nodes/ImageNode';

interface NodeConstructor {
  new (): NodeBase;
}

export default class NodeFactory implements INodeFactory {
  private static _instance: NodeFactory | null = null;

  private _nodeMapTable: { [nodeType: number]: NodeConstructor } = {};

  private constructor() {
    this._nodeMapTable[NodeTypeEnum.PageDocumentNodeType] = PageDocument;
    this._nodeMapTable[NodeTypeEnum.TextNodeType] = TextNode;
    this._nodeMapTable[NodeTypeEnum.ButtonNodeType] = ButtonNode;
    this._nodeMapTable[NodeTypeEnum.SimpleContainerNodeType] = SimpleContainerNode;
    this._nodeMapTable[NodeTypeEnum.ColumnsContainerNodeType] = ColumnsContainerNode;
    this._nodeMapTable[NodeTypeEnum.ImageNodeType] = ImageNode;
  }

  public static getInstance(): NodeFactory {
    if (!NodeFactory._instance) NodeFactory._instance = new NodeFactory();
    return NodeFactory._instance;
  }

  public createNodeByType<T extends NodeBase>(type: NodeTypeEnum, parent: IContainerNode = null): T {
    const constructor = this._nodeMapTable[type];

    if (constructor) {
      const node = new constructor() as T;
      node.parent = parent;
      const parentId = parent ? `${parent.document.id}-` : 'id';
      node.id = `${parentId}${Math.random().toString(27).slice(2)}`;
      return node;
    }

    return null;
  }
}