process: enable stream_command to filter by pattern

Author: ismaelsadeeq

src/warnet/process.py

@@ -1,14 +1,15 @@
+import re
 import subprocess
 
-
 def run_command(command: str) -> str:
     result = subprocess.run(command, shell=True, capture_output=True, text=True, executable="bash")
     if result.returncode != 0:
         raise Exception(result.stderr)
     return result.stdout
 
 
-def stream_command(command: str) -> bool:
+def stream_command(command: str, grep_pattern: str = "") -> bool:
+    """Stream output and apply an optional pattern filter."""
     process = subprocess.Popen(
         ["bash", "-c", command],
         stdout=subprocess.PIPE,
@@ -18,10 +19,15 @@ def stream_command(command: str) -> bool:
         universal_newlines=True,
     )
 
-    message = ""
+    pattern = re.compile(grep_pattern) if grep_pattern else None
+
+    # Only display lines matching the pattern if grep is specified
     for line in iter(process.stdout.readline, ""):
-        message += line
-        print(line, end="")
+        if pattern:
+            if pattern.search(line):
+                print(line, end="")
+        else:
+            print(line, end="")
 
     process.stdout.close()
     return_code = process.wait()