feat(waf): 规则编辑器一键格式化节点布局

按从开始节点出发的层次从左到右整理坐标，并 fitView 到画布。
2026-10-11 01:36:37 +08:00 · 2026-07-19 12:49:42 +08:00
parent 32113eb790
commit fd9348b7bd
4 changed files with 208 additions and 2 deletions
@@ -0,0 +1,76 @@
+import { describe, expect, it } from 'vitest';
+
+import type { WAFRuleGraph } from '@/lib/services/openflare';
+
+import { layoutRuleGraph } from './graph-layout';
+
+const sampleGraph: WAFRuleGraph = {
+  schema_version: 1,
+  nodes: [
+    {
+      id: 'start',
+      type: 'start',
+      position: { x: 500, y: 400 },
+      config: {},
+    },
+    {
+      id: 'match',
+      type: 'ip_match',
+      position: { x: 10, y: 10 },
+      config: { ips: [], cidrs: [], ip_group_ids: [] },
+    },
+    {
+      id: 'allow',
+      type: 'allow',
+      position: { x: 0, y: 0 },
+      config: {},
+    },
+    {
+      id: 'block',
+      type: 'block',
+      position: { x: 99, y: 99 },
+      config: { status_code: 403, response_body: '' },
+    },
+  ],
+  edges: [
+    {
+      id: 'e1',
+      source: 'start',
+      source_handle: 'next',
+      target: 'match',
+    },
+    {
+      id: 'e2',
+      source: 'match',
+      source_handle: 'true',
+      target: 'allow',
+    },
+    {
+      id: 'e3',
+      source: 'match',
+      source_handle: 'false',
+      target: 'block',
+    },
+  ],
+};
+
+describe('layoutRuleGraph', () => {
+  it('places start left of match and match left of terminals', () => {
+    const laid = layoutRuleGraph(sampleGraph);
+    const byId = Object.fromEntries(laid.nodes.map((n) => [n.id, n]));
+    expect(byId.start.position.x).toBeLessThan(byId.match.position.x);
+    expect(byId.match.position.x).toBeLessThan(byId.allow.position.x);
+    expect(byId.match.position.x).toBeLessThan(byId.block.position.x);
+  });
+
+  it('keeps edges unchanged', () => {
+    const laid = layoutRuleGraph(sampleGraph);
+    expect(laid.edges).toEqual(sampleGraph.edges);
+  });
+
+  it('separates sibling terminals on y axis', () => {
+    const laid = layoutRuleGraph(sampleGraph);
+    const byId = Object.fromEntries(laid.nodes.map((n) => [n.id, n]));
+    expect(byId.allow.position.y).not.toBe(byId.block.position.y);
+  });
+});