lowcode-engine/packages/shell/src/drag-object.ts

import { dragObjectSymbol } from './symbols';
import { IPublicModelDragObject, DragObject as InnerDragObject, DragNodeDataObject } from '@alilc/lowcode-types';
import Node from './node';

export default class DragObject implements IPublicModelDragObject {
  private readonly [dragObjectSymbol]: InnerDragObject;

  constructor(dragObject: InnerDragObject) {
    this[dragObjectSymbol] = dragObject;
  }

  static create(dragObject: InnerDragObject): IPublicModelDragObject | null {
    if (!dragObject) {
      return null;
    }
    return new DragObject(dragObject);
  }

  get type(): any {
    return this[dragObjectSymbol].type;
  }

  get nodes(): any {
    const { nodes } = this[dragObjectSymbol];
    if (!nodes) {
      return null;
    }
    return nodes.map(Node.create);
  }

  get data(): any {
    return (this[dragObjectSymbol] as DragNodeDataObject).data;
  }
}