From cbff3f8366857d5c87cef3b69765f4de3f1fbe27 Mon Sep 17 00:00:00 2001
From: Rodrigo <rodrigosilvanader@gmail.com>
Date: Sun, 16 Jun 2024 20:57:09 -0300
Subject: [PATCH] feat: Add SelectivePassThroughComponent

---
 .../experimental/SelectivePassThrough.py      | 75 +++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 src/backend/base/langflow/components/experimental/SelectivePassThrough.py

diff --git a/src/backend/base/langflow/components/experimental/SelectivePassThrough.py b/src/backend/base/langflow/components/experimental/SelectivePassThrough.py
new file mode 100644
index 000000000..9e623b3e6
--- /dev/null
+++ b/src/backend/base/langflow/components/experimental/SelectivePassThrough.py
@@ -0,0 +1,75 @@
+from langflow.custom import Component
+from langflow.field_typing import Text
+from langflow.inputs import BoolInput, DropdownInput, StrInput
+from langflow.template import Output
+
+class SelectivePassThroughComponent(Component):
+    display_name = "Selective Pass Through"
+    description = "Passes the specified value if a specified condition is met."
+    icon = "filter"
+
+    inputs = [
+        StrInput(
+            name="input_value",
+            display_name="Input Value",
+            info="The primary input value to evaluate.",
+        ),
+        StrInput(
+            name="comparison_value",
+            display_name="Comparison Value",
+            info="The value to compare against the input value.",
+        ),
+        DropdownInput(
+            name="operator",
+            display_name="Operator",
+            options=["equals", "not equals", "contains", "starts with", "ends with"],
+            info="Condition to evaluate the input value.",
+        ),
+        StrInput(
+            name="value_to_pass",
+            display_name="Value to Pass",
+            info="The value to pass if the condition is met.",
+        ),
+        BoolInput(
+            name="case_sensitive",
+            display_name="Case Sensitive",
+            info="If true, the comparison will be case sensitive.",
+            value=False,
+            advanced=True,
+        ),
+    ]
+
+    outputs = [
+        Output(display_name="Passed Output", name="passed_output", method="pass_through"),
+    ]
+
+    def evaluate_condition(self, input_value: str, comparison_value: str, operator: str, case_sensitive: bool) -> bool:
+        if not case_sensitive:
+            input_value = input_value.lower()
+            comparison_value = comparison_value.lower()
+
+        if operator == "equals":
+            return input_value == comparison_value
+        elif operator == "not equals":
+            return input_value != comparison_value
+        elif operator == "contains":
+            return comparison_value in input_value
+        elif operator == "starts with":
+            return input_value.startswith(comparison_value)
+        elif operator == "ends with":
+            return input_value.endswith(comparison_value)
+        return False
+
+    def pass_through(self) -> Text:
+        input_value = self.input_value
+        comparison_value = self.comparison_value
+        operator = self.operator
+        value_to_pass = self.value_to_pass
+        case_sensitive = self.case_sensitive
+
+        if self.evaluate_condition(input_value, comparison_value, operator, case_sensitive):
+            self.status = value_to_pass
+            return value_to_pass
+        else:
+            self.status = ""
+            return ""