queenshow/src/modules/editor/components/Viewport/Slider/SliderRight/CompTree.tsx

import { TreeToolbars } from "@/modules/editor/objects/Toolbars";
import { css } from "@linaria/core";
import { Image } from "@queenjs/ui";
import { useReactive } from "@queenjs/use";
import { Tree, TreeProps } from "ant-design-vue";
import { defineComponent, nextTick, watch } from "vue";
import { string, bool } from "vue-types";
import { useEditor } from "../../../..";
import { DesignComp } from "../../../../objects/DesignTemp/DesignComp";
import { TreeDataItem } from "ant-design-vue/lib/tree";
import { CompObject } from "@/modules/editor/controllers/SelectCtrl/compObj";

type TreeItem = {
  key: string;
  title: string;
  value: string;
  children: TreeItem[];
};

export const CompTree = defineComponent({
  setup() {
    const { store, actions, helper, controls } = useEditor();
    const { compUICtrl } = controls;

    const state = useReactive(() => ({
      expandedKeys: [] as string[],
      treeData() {
        const rootComp = helper.findRootComp();

        let deep = 0;
        function getCompChildren(ids: string[]): TreeItem[] {
          deep += 1;
          if (deep >2 ) ids = ids.reverse();
         
          return ids.map((id) => {
            const comp = helper.findComp(id) as DesignComp;
            return {
              key: comp.id,
              title:
                comp.title ||
                compUICtrl.state.components.get(comp.compKey)?.name ||
                "未命名",
              value: comp.id,
              children: getCompChildren(
                comp.getChildIds(),
              ),
            };
          });
        }
        return getCompChildren(rootComp?.id ? [rootComp.id] : []);
      },
    }));

    watch(
      () => store.currCompId,
      () => {
        // expandedKeys需要等Tree的内部状态更新后再赋值
        nextTick(() => {
          state.expandedKeys = store.currCompId
            ? helper.getCompTrees(store.currCompId).map((comp) => comp.id)
            : [];
        });
      }
    );
    const onDrop = (info: any) => {
      const dragNode = info.dragNode; //被拖拽
      const dropNode = info.node; //拖拽落下

      const dragKey = dragNode.key;
      const dropKey = dropNode.key;

      const dragPos = dragNode.pos.split("-");
      const dropPos = dropNode.pos.split("-");

      const dragLevel = dragPos.length;
      const dropLevel = dropPos.length;
      if (
        dragLevel >= 5 ||
        dragKey == "root" ||
        dragLevel < dropLevel ||
        dragLevel - dropLevel > 1
      ) {
        return;
      }
      let parentComp: any = null;
      if (dragLevel > dropLevel) {
        parentComp = helper.findComp(dropKey);
      } else {
        parentComp = helper.findParentComp(dropKey);
      }
      const currComp = helper.findComp(dragKey);
      const currParent = helper.findParentComp(dragKey);

      if (!currComp) return;
      const currParentComp = helper.findParentComp(dragKey);
      const index = currParentComp?.children.default?.indexOf(currComp.id);
      if (index != -1) {
        currParentComp?.children.default?.splice(index as number, 1);
        helper.extendStreamCard(currParentComp?.id || "");
      }
      actions.pickComp("");
      if (!info.dropToGap) {
        parentComp?.children.default == parentComp?.children.default || [];
        if (currComp.compKey == "Container") {
          parentComp?.children.default?.unshift(currComp?.id);
        } else {
          parentComp?.children.default?.push(currComp?.id);
        }
        if (currParent?.id != parentComp.id) {
          const len = parentComp?.children.default?.length;
          const index = currComp.compKey == "Container" ? 0 : len - 1;
          const revert = currComp.compKey == "Container" ? false : true;
          sortCompInCard(parentComp, index, revert);
        }
      } else {
        if (currComp.compKey == "Container") {
          parentComp?.children.default?.splice(
            info.dropPosition,
            0,
            currComp.id
          );
        } else {
          const len = parentComp?.children.default.length;
          parentComp?.children.default?.splice(
            len - info.dropPosition,
            0,
            currComp.id
          );
        }

        if (currParent?.id != parentComp.id) {
          const revert = currComp.compKey == "Container" ? false : true;
          sortCompInCard(parentComp, info.dropPosition, revert);
        }
      }
      nextTick(() => {
        actions.pickComp(currComp.id);
        helper.extendStreamCard(parentComp.id);
      });
    };
    const sortCompInCard = (
      parentComp: DesignComp,
      index: number,
      revert = false
    ) => {
      if (parentComp?.compKey != "Container") {
        return;
      }
      const children = parentComp.children.default || [];
      if (children?.length > 0) {
        const prveCompId = revert ? children[index + 1] : children[index - 1];
        const nowCompId = children[index];
        const nowComp = helper.findComp(nowCompId);
        if (!prveCompId) {
          const nowCompBound = helper.getCardCompBound(nowCompId);
          if (!nowComp) {
            return;
          }
          const nowObj = new CompObject(nowComp);
          nowObj.worldTransform.translate(-nowCompBound.x, -nowCompBound.y);
          nowComp.layout.transformMatrix = nowObj.worldTransform.getMatrixStr();
          return;
        }
        const prevCompBound = helper.getCardCompBound(prveCompId);
        const prevComp = helper.findComp(prveCompId);
        if (!prevComp || !nowComp) {
          return;
        }
        const obj = new CompObject(prevComp);
        const yOffset = prevCompBound.h;
        obj.worldTransform.translate(0, yOffset);
        nowComp.layout.transformMatrix = obj.worldTransform.getMatrixStr();
      }
    };

    return () => (
      <div class="!overflow-x-auto">
        <Tree
          class={treeStyle}
          treeData={state.treeData}
          v-model={[state.expandedKeys, "expandedKeys"]}
          selectedKeys={[store.currCompId]}
          blockNode={true}
          draggable={true}
          onDrop={onDrop}
          onSelect={(ids) => {
            const id =
              (ids[0] as string) || state.expandedKeys.at(-2) || "root";
            actions.pickComp(id);
          }}
        >
          {{
            title: (data: any) => {
              return <CompNode title={data.title} id={data.key} />;
            },
          }}
        </Tree>
      </div>
    );
  },
});

const CompNode = defineComponent({
  props: {
    id: string().isRequired,
    title: string().isRequired,
  },
  setup(props) {
    const editor = useEditor();
    const comp = editor.helper.findComp(props.id);
    return () => {
      if (!comp) return;
      const compOpts = editor.controls.compUICtrl.state.components.get(
        comp.compKey
      );
      const actions = TreeToolbars[comp.compKey] || TreeToolbars.default;
      const thumbnail =
        comp.compKey === "Image"
          ? comp.value.url
          : comp.thumbnail || compOpts?.thumbnail;
      return (
        <div class={nodeStyle}>
          <Image src={thumbnail} size={240} />
          <span class="w-1/1 flex-1 whitespace-nowrap mr-30px">
            {props.title}
          </span>
          <span class="space-x-4px whitespace-nowrap">
            {actions.map((action, i) =>
              action.getVisible.call(editor, comp) ? (
                <action.component
                  key={i}
                  class="p-4px"
                  value={action.getValue?.(comp)}
                  onClick={(e: MouseEvent) => {
                    e.stopPropagation();
                    action.onClick.call(editor, comp);
                  }}
                />
              ) : null
            )}
          </span>
        </div>
      );
    };
  },
});

const treeStyle = css`
  .ant-tree-switcher {
    display: flex;
    justify-content: center;
    align-items: center;
  }
`;

const nodeStyle = css`
  display: flex;
  align-items: center;
  padding: 6px;
  @apply rounded;

  > img {
    width: 24px;
    height: 24px;
    margin-right: 6px;
    object-fit: contain;
  }
`;