fix(site): add utils directory

Author: yhlchao

packages/site/src/app.tsx

@@ -5,8 +5,8 @@ import Share from './component/Share';
 import Setting from './component/Setting';
 import Links from './component/Links';
 import './index.less';
-import CustomNode from './lib/utils/registerNode';
-import CustomListener from './lib/utils/addListener';
+import CustomNode from './utils/registerNode';
+import CustomListener from './utils/addListener';
 
 LogicFlow.use(Snapshot);
 LogicFlow.use(Dnd);

packages/site/src/utils/addListener.ts

@@ -0,0 +1,26 @@
+import LogicFlow from '@logicflow/core';
+
+class CustomListener {
+  lf: LogicFlow;
+
+  constructor(lf: LogicFlow) {
+    this.lf = lf;
+  }
+
+  addAll() {
+    this.addNodeListener();
+    // ...
+  }
+
+  addNodeListener() {
+    const { lf } = this;
+    lf.on('node:click', ({ data }) => {
+      console.log('单击', data);
+    });
+    lf.on('node:dbclick', ({ data }) => {
+      console.log('双击', data);
+    });
+  }
+}
+
+export default CustomListener;

packages/site/src/utils/registerNode.ts

@@ -0,0 +1,41 @@
+import LogicFlow from '@logicflow/core';
+
+class CustomNode {
+  lf: LogicFlow;
+
+  constructor(lf: LogicFlow) {
+    this.lf = lf;
+  }
+
+  // 注册所有的自定义节点
+  registerAll() {
+    this.registerRect();
+    // ...
+  }
+
+  registerRect() {
+    this.lf.register('rect', ({ RectNode, RectNodeModel }) => {
+      class CustomRectModel extends RectNodeModel {
+        getConnectedSourceRules() {
+          const rules = super.getConnectedSourceRules();
+          const rule = {
+            message: '不满足连线的校验规则',
+            validate: (source, target) => {
+              console.log(target);
+              // 校验规则
+              return true;
+            },
+          };
+          rules.push(rule);
+          return rules;
+        }
+      }
+      return {
+        view: RectNode,
+        model: CustomRectModel,
+      };
+    });
+  }
+}
+
+export default CustomNode;