drawing-board/src/board/core/base/entity-factory.ts

import { getChangePart } from "../../shared";
import {
  Entity,
  EntityKey,
  EntityShape,
  EntityTree,
  EntityType,
  EntityTypeInstance,
} from "./entity";

export const entityFactory = <
  P extends Entity,
  T,
  S extends EntityShape,
  C extends EntityType<T, S, EntityTree<P>>
>(
  attrib: T,
  key: EntityKey,
  Type: C,
  parent?: P,
  extra?: (self: EntityTypeInstance<C>) => void,
  created?: (self: EntityTypeInstance<C>) => void
): EntityTypeInstance<C> => {
  const entity = new Type({
    attrib,
    key,
  }) as EntityTypeInstance<C>;

  extra && extra(entity);
  parent.addChild(entity);
  created && created(entity);
  return entity;
};

export type IncEntitys<
  P extends Entity,
  T,
  C extends EntityType<T, S, EntityTree<P>>,
  S extends EntityShape
> = {
  adds: EntityTypeInstance<C>[];
  dels: EntityTypeInstance<C>[];
  upds: EntityTypeInstance<C>[];
};

export type IncEntitysFactory<
  P extends Entity,
  T,
  C extends EntityType<T, S, EntityTree<P>>,
  S extends EntityShape = EntityShape
> = (attribs: T[]) => IncEntitys<P, T, C, S>;

// 增量工厂
export const incEntitysFactoryGenerate = <
  P extends Entity,
  T,
  S extends EntityShape,
  C extends EntityType<T, S, EntityTree<P>>
>(
  Type: C,
  parent?: P,
  extra?: (self: EntityTypeInstance<C>) => void,
  created?: (self: EntityTypeInstance<C>) => void
): IncEntitysFactory<P, T, C, S> => {
  let oldKeys: string[] = [];
  let useIndex = false;
  let inited = false;

  const cache: { [key in string]: EntityTypeInstance<C> } = {};

  const findAttrib = (attribs: any[], key: string) => {
    if (useIndex) {
      return attribs[key];
    } else {
      return attribs.find((attrib) => attrib.id === key);
    }
  };

  const destory = (key: string) => {
    const delEntity = cache[key];
    delEntity.destory();
    delete cache[key];
    return delEntity;
  };

  const add = (attrib: T, key: string) => {
    const addEntity = entityFactory(attrib, key, Type, parent, extra, created);
    return (cache[key] = addEntity);
  };

  return (attribsRaw: T[]) => {
    const attribs = attribsRaw as any[];
    if (!inited && attribs.length && typeof attribs[0] === "object") {
      useIndex = "id" in attribs[0];
      inited = true;
    }

    if (
      !useIndex &&
      attribs.length &&
      attribs.some(
        (item) =>
          typeof item !== "object" ||
          (typeof item.id !== "string" && typeof item.id !== "number")
      )
    ) {
      throw "attribs 不合法，缺少id";
    }
    const newKeys = attribs.map((attrib, ndx) => (useIndex ? ndx : attrib.id));

    if (new Set(newKeys).size !== newKeys.length) {
      throw "attribs 的id不合法 不可重复";
    }

    const { addPort, delPort, holdPort } = getChangePart(newKeys, oldKeys);

    const dels = delPort.map(destory);
    const adds = addPort.map((key) => add(findAttrib(attribs, key), key));
    const upds = holdPort.map((key) => {
      const newAttrib = findAttrib(attribs, key);
      cache[key].setAttrib(newAttrib);
      return cache[key];
    });
    oldKeys = newKeys;

    return {
      adds,
      dels,
      upds,
    };
  };
};

export type SingleEntity<
  P extends Entity,
  T,
  C extends EntityType<T, S, EntityTree<P>>,
  S extends EntityShape
> = {
  add?: EntityTypeInstance<C>;
  del?: EntityTypeInstance<C>;
  upd?: EntityTypeInstance<C>;
};

export type SingleEntityFactory<
  P extends Entity,
  T,
  C extends EntityType<T, S, EntityTree<P>>,
  S extends EntityShape = EntityShape
> = (data: { attrib: T; key?: string }) => SingleEntity<P, T, C, S>;

export const singleEntityFactory = <
  P extends Entity,
  T,
  S extends EntityShape,
  C extends EntityType<T, S, EntityTree<P>>
>(
  Type: C,
  parent?: P,
  extra?: (self: EntityTypeInstance<C>) => void,
  created?: (self: EntityTypeInstance<C>) => void
): SingleEntityFactory<P, T, C, S> => {
  let entity: EntityTypeInstance<C> | null = null;

  return ({ attrib, key }) => {
    const result: SingleEntity<P, T, C, S> = {};

    if (entity) {
      if (!key) {
        entity.destory();
        result.del = entity;
        entity = null;
      } else if (key !== entity.getKey()) {
        console.error("entity 主键更改 销毁重建");
        entity.destory();
        result.del = entity;
        result.add = entityFactory(attrib, key, Type, parent, extra, created);
        entity = result.add;
      } else {
        entity.setAttrib(attrib);
        result.upd = entity;
      }
    } else if (key && attrib) {
      result.add = entityFactory(attrib, key, Type, parent, extra, created);
      entity = result.add;
    }

    return result;
  };
};