fix: 🐛 add updateCellId api (antvis#1739)

TomScavo · Jan 8, 2022 · 78cdb3b · 78cdb3b
1 parent 0dd0c13
commit 78cdb3b
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 0 deletions.
diff --git a/packages/x6/src/graph/graph.ts b/packages/x6/src/graph/graph.ts
@@ -496,6 +496,10 @@ export class Graph extends Basecoat<EventArgs> {
     return result
   }
 
+  updateCellId(cell: Cell, newId: string) {
+    return this.model.updateCellId(cell, newId)
+  }
+
   // #endregion
 
   // #region view

diff --git a/packages/x6/src/model/model.ts b/packages/x6/src/model/model.ts
@@ -306,6 +306,36 @@ export class Model extends Basecoat<Model.EventArgs> {
     return null
   }
 
+  updateCellId(cell: Cell, newId: string) {
+    this.startBatch('update', { id: newId })
+    cell.prop('id', newId)
+    const newCell = cell.clone({ keepId: true })
+    this.addCell(newCell)
+
+    // update connected edge terminal
+    const edges = this.getConnectedEdges(cell)
+    edges.forEach((edge) => {
+      const sourceCell = edge.getSourceCell()
+      const targetCell = edge.getTargetCell()
+      if (sourceCell === cell) {
+        edge.setSource({
+          ...edge.getSource(),
+          cell: newId,
+        })
+      }
+      if (targetCell === cell) {
+        edge.setTarget({
+          ...edge.getTarget(),
+          cell: newId,
+        })
+      }
+    })
+
+    this.removeCell(cell)
+    this.startBatch('update', { id: newId })
+    return newCell
+  }
+
   removeCells(cells: (Cell | string)[], options: Cell.RemoveOptions = {}) {
     if (cells.length) {
       return this.batchUpdate('remove', () => {

diff --git a/sites/x6-sites/docs/api/graph/model.en.md b/sites/x6-sites/docs/api/graph/model.en.md
@@ -282,6 +282,21 @@ getCellById(id: string)
 |------|--------|:----:|--------|-------------|
 | cell | string |  ✓   |        | 节点/边的 ID。 |
 
+### updateCellId(...)
+
+```sign
+updateCellId(cell: Cell, newId: string)
+```
+
+更新节点或者边的 ID，会返回新创建的节点/边。
+
+<span class="tag-param">参数<span>
+
+| 名称 | 类型   | 必选 | 默认值 | 描述          |
+|------|--------|:----:|--------|-------------|
+| cell | Cell   |  ✓   |        | 节点/边。 |
+| newId |string |  ✓   |        | 新的 ID。 |
+
 ### getCells()
 
 ```sign

diff --git a/sites/x6-sites/docs/api/graph/model.zh.md b/sites/x6-sites/docs/api/graph/model.zh.md
@@ -282,6 +282,21 @@ getCellById(id: string)
 |------|--------|:----:|--------|-------------|
 | cell | string |  ✓   |        | 节点/边的 ID。 |
 
+### updateCellId(...)
+
+```sign
+updateCellId(cell: Cell, newId: string)
+```
+
+更新节点或者边的 ID，会返回新创建的节点/边。
+
+<span class="tag-param">参数<span>
+
+| 名称 | 类型   | 必选 | 默认值 | 描述          |
+|------|--------|:----:|--------|-------------|
+| cell | Cell   |  ✓   |        | 节点/边。 |
+| newId |string |  ✓   |        | 新的 ID。 |
+
 ### getCells()
 
 ```sign