From 989201e8f63c5e78e7337e2e2510c3e9019a8525 Mon Sep 17 00:00:00 2001
From: _Kerman <KermanX@qq.com>
Date: Fri, 10 Nov 2023 17:52:33 +0800
Subject: [PATCH] feat: allow to display and hide then socket

---
 packages/northstar/src/blocks/special/imp.ts | 40 ++++++++++++++++----
 1 file changed, 32 insertions(+), 8 deletions(-)

diff --git a/packages/northstar/src/blocks/special/imp.ts b/packages/northstar/src/blocks/special/imp.ts
index 07ed4aa..632baf7 100644
--- a/packages/northstar/src/blocks/special/imp.ts
+++ b/packages/northstar/src/blocks/special/imp.ts
@@ -9,11 +9,11 @@ import {
   PATH_IN_TRIANGLE,
   PATH_OUT_TRIANGLE,
   SingleOutSocket,
-  Socket,
   UseSocket,
   UsedSockets,
-  blockCtors,
+  blockCtors
 } from "@quasi-dev/visual-flow";
+import { PropsData } from "../../utils/props";
 import {
   multiInSocketToOutput,
   singleOutSocketToOutput,
@@ -21,6 +21,11 @@ import {
 import { FuncBlockBase } from "./FuncBlockBase.r";
 
 export class ImpBlock extends FuncBlockBase {
+  cloneTo(target: this): this {
+    super.cloneTo(target);
+    target.hasThen = this.hasThen;
+    return target;
+  }
   type: FuncBlockTypes = "imp";
 
   boardHeight = 80;
@@ -29,11 +34,13 @@ export class ImpBlock extends FuncBlockBase {
   outputLabel = "retVal";
   useTextarea = true;
 
+  hasThen = false;
+
   get whenSocket() {
     return this.getSocketByName("when") as MultiInSocket;
   }
   get thenSocket() {
-    return this.getSocketByName("then") as SingleOutSocket;
+    return this.getSocketByName("then") as SingleOutSocket | undefined;
   }
 
   socketUpdater(useSocket: UseSocket, usedSockets: UsedSockets): void {
@@ -43,11 +50,13 @@ export class ImpBlock extends FuncBlockBase {
       path: PATH_IN_TRIANGLE,
       direction: Direction.LEFT,
     });
-    useSocket("then", SingleOutSocket, {
-      type: "E",
-      path: PATH_OUT_TRIANGLE,
-      direction: Direction.RIGHT,
-    });
+    if (this.hasThen) {
+      useSocket("then", SingleOutSocket, {
+        type: "E",
+        path: PATH_OUT_TRIANGLE,
+        direction: Direction.RIGHT,
+      });
+    }
     usedSockets.find(([n]) => n === "output")![1].disabled =
       this.inputValue.value.match(/\breturn\b/g) === null;
   }
@@ -58,6 +67,21 @@ export class ImpBlock extends FuncBlockBase {
     return [...matches].map((match) => match[0].slice(1));
   }
 
+  getProps(): PropsData {
+    return [
+      {
+        name: "has then",
+        type: "switch",
+        getVal: () => {
+          return this.hasThen;
+        },
+        setVal: (val) => {
+          this.hasThen = val;
+        },
+      },
+    ];
+  }
+
   toOutput(): ImpBlockOutput {
     return {
       ...(super.toOutput() as FuncBlockOutput),