<template>
  <div v-if="visible">
    <svg id="mySvg" v-if="visible"></svg>
  </div>
</template>

<script>
import dagreD3 from "dagre-d3";
import * as d3 from "d3";

export default {
  name: "degraD3",
  props: {
    nodes: {
      type:Array,
      default:()=>[]
    },
    edges:{
      type:Array,
      default:()=>[]
    },
    rankdir:{
      type: String,
      default: "DL",
    }
  },
  data() { 
    return {
      g:null,
      visible:false,
      isSaving: false,
      type:""
    };
  },
  watch: {
    nodes: {
      handler(val) {
        console.log(val);
        this.updateGraph();
      },
      deep: true,
    },
    edges: {
      handler(val) {
        console.log(val);
        this.updateGraph();
      },
      deep: true,
    },
  },
  mounted() {},
  methods: {
    open(){
      this.visible = true;
      this.handleWatch();
    },
    handleWatch() {
      let that = this;
      that.$nextTick(() => {
        if(that.g!==null){
          that.updateGraph();     
        }else{
          that.g = new dagreD3.graphlib.Graph().setGraph({
            rankdir: that.rankdir,
            nodesep: 40,
            edgesep: 25, //两条线之间的距离
            ranksep: 20, //节点之间的距离
            marginx: 80,
            marginy: 10,
          });
          // 添加节点
          that.nodes.forEach((item) => {
            if(item.id&& item.label){
              that.g.setNode(item.id, {
                // 节点标签
                label: item.label,
                // 节点形状
                shape: item.shape,
                toolText:  item.label,
                //节点样式
                style: "fill:#fff;stroke:#000",
                labelStyle: "fill:#000;",
                rx: 5, //矩形节点圆角度
                ry: 5,
              });
            }
          });
          //流转得到线  链接关系
          that.edges.forEach((transition0) => {
            that.g.setEdge(transition0.source,transition0.target,{
                label: transition0.label,// 边标签
                style: "fill:#ffffff;stroke:#c0c1c3;stroke-width:1.5px", // 根据后台数据来改变连线的颜色
              });
          });
          // 创建渲染器
          let render = new dagreD3.render();
          // 选择 svg 并添加一个g元素作为绘图容器.
          let svg = d3.select("#mySvg");
          let inner = svg.append("g");
          that.inner = inner;
          // 在绘图容器上运行渲染器生成流程图.
          render(inner, that.g);
          let mySvgHeight =document.getElementsByClassName("nodes")[0].getBoundingClientRect().height + 50;
          let mySvgWdith =document.getElementsByClassName("output")[0].getBoundingClientRect().width+150 ;
          document.getElementById('mySvg').setAttribute("height", mySvgHeight);
          document.getElementById('mySvg').setAttribute("width", mySvgWdith);
          svg.selectAll("g.node").on("click", function(event, d) {
            svg.selectAll("g.node").style("fill", "white");
            svg.selectAll("g.node")._groups[0].forEach(item=>{
              console.log('item',item);
              d3.select(item).select("rect").style("fill", "#fff");  // 修改为白色
            })
            // 改变节点的颜色
            const node = d3.select(this);
            const currentColor = node.select("rect").style("fill")
            // 判断当前颜色，如果是默认颜色，就修改为新的颜色
            if (currentColor === "rgb(255, 255, 255)") { // 如果当前是白色
              node.select("rect").style("fill", "#f00");  // 修改为红色
            } else {
              node.select("rect").style("fill", "#fff");  // 否则恢复白色
            }
            // 调用自定义方法
            onNodeClick(d);
          });
          // 自定义方法：在节点点击时触发
          function onNodeClick(nodeData) {
            let batch = "";
            that.nodes.forEach((item) => {
              if (item.id === nodeData){
                batch = item.label;
              }
            })
            that.$emit("nodeClick", batch);
            // 你可以在这里处理更多的逻辑，例如显示一个提示框，更新图形状态等
          }
        }
      })
    },
    // 假设你有一个方法来更新节点和边的数据
    updateGraph() {
      let that = this;
      // 清空现有图形
      that.g.nodes().forEach(function (v) {
        that.g.removeNode(v);
      });
      that.g.edges().forEach(function (e) {
        that.g.removeEdge(e);
      });

      // 重新添加节点
      that.nodes.forEach((item) => {
        if (item.id && item.label) {
          that.g.setNode(item.id, {
            label: item.label,
            shape: item.shape,
            toolText: item.label,
            style: "fill:#fff;stroke:#000",
            labelStyle: "fill:#000;",
            rx: 5,
            ry: 5,
          });
        }
      });

      // 重新添加边
      that.edges.forEach((transition0) => {
        that.g.setEdge(transition0.source, transition0.target, {
          label: transition0.label,
          style: "fill:#ffffff;stroke:#c0c1c3;stroke-width:1.5px",
        });
      });
      let render = new dagreD3.render();
      // 渲染更新后的图形
      render(that.inner, that.g);
      // 获取新的 SVG 尺寸
      let mySvgHeight = document.getElementsByClassName("nodes")[0]?.getBoundingClientRect().height + 50;
      let mySvgWdith = document.getElementsByClassName("output")[0]?.getBoundingClientRect().width + 150;

      // 检查节点和边的容器是否存在
      if (mySvgHeight && mySvgWdith) {
        // 设置 SVG 尺寸
        document.getElementById('mySvg').setAttribute("height", mySvgHeight);
        document.getElementById('mySvg').setAttribute("width", mySvgWdith);
      } else {
        console.log("元素尚未加载，无法更新 SVG 尺寸");
      }
    },
    scaleUp() {
      var svg = document.getElementById("mySvg");
      svg.style.transform = "scale(0.5)";
    }
  },
};
</script>
<style scoped>
.treeMain {
  height: 280px;
  overflow: auto;
  border: 1px solid #dcdfe6;
  margin-bottom: 10px;
}

::-webkit-scrollbar {
  width: 10px;
}

::-webkit-scrollbar-track {
  -webkit-box-shadow: inset 0 0 6px rgba(0, 0, 0, 0.2);
  background-color: #fefefe;
  border-radius: 5px;
}

::-webkit-scrollbar-thumb {
  border-radius: 5px;
  -webkit-box-shadow: inset 0 0 6px rgba(0, 0, 0, 0.5);
  background-color: #f5f5f5;
}

.svgMark {
  width: 100%;
  height: 100%;
  position: fixed;
  top: 0;
  right: 0;
  bottom: 0;
  left: 0;
  overflow: auto;
  margin: 0;
  z-index: 2000;
  background: rgba(0, 0, 0, 0.3);
}

.svgWrapper {
  background: #fff;
  width: 800px;
  margin: 5vh auto;
  height: 90vh;
  text-align: center;
  border-radius: 2px;
  overflow-y: scroll;
}

.svgItem {
  padding: 20px 40px 0;
  font-family: Helvetica Neue, Helvetica, PingFang SC, Hiragino Sans GB,
    Microsoft YaHei, Arial, sans-serif;
  font-size: 18px;
  display: flex;
  justify-content: space-between;
}

svg {
  font-size: 14px;
}

.node rect {
  stroke: #606266;
  fill: #fff;
}

.edgePath path {
  stroke: #606266;
  fill: #333;
  stroke-width: 1.5px;
}

g.conditions > rect {
  fill: #00ffd0;
  stroke: #000;
}

.el-icon-close {
  cursor: pointer;
}
</style>

