Added kill process script

script/kill-process/customize.py

@@ -0,0 +1,52 @@
+from mlc import utils
+import os
+import subprocess
+
+
+def generate_kill_command(env):
+    kill_cmd = ""
+
+    if env.get("MLC_KILL_PROCESS_GROUP"):
+        if env.get("MLC_KILL_PROCESS_ID"):
+            process_id = env["MLC_KILL_PROCESS_ID"]
+            # Find process group and kill all processes in it
+            kill_cmd = f"pkill -g $(ps -o pgid= -p {process_id} | tr -d ' ')"
+
+        elif env.get("MLC_KILL_PROCESS_NAME"):
+            process_name = env["MLC_KILL_PROCESS_NAME"]
+            # Find process group of the first matching process name and kill
+            # all in that group
+            kill_cmd = f"pkill -g $(pgrep -o {process_name} | xargs ps -o pgid= -p | tr -d ' ')"
+
+    else:
+        if env.get("MLC_KILL_PROCESS_ID"):
+            process_id = env["MLC_KILL_PROCESS_ID"]
+            kill_cmd = f"kill {process_id}"  # Kill a single process by ID
+
+        elif env.get("MLC_KILL_PROCESS_NAME"):
+            process_name = env["MLC_KILL_PROCESS_NAME"]
+            # Kill all processes matching the name
+            kill_cmd = f"pkill {process_name}"
+
+    env["MLC_RUN_CMD"] = kill_cmd if kill_cmd else "echo 'No valid input provided'"
+
+
+def preprocess(i):
+
+    env = i['env']
+    state = i['state']
+    os_info = i['os_info']
+
+    generate_kill_command(env)
+
+    return {'return': 0}
+
+
+def postprocess(i):
+
+    env = i['env']
+    state = i['state']
+
+    os_info = i['os_info']
+
+    return {'return': 0}

script/kill-process/meta.yaml

@@ -0,0 +1,22 @@
+alias: kill-process
+automation_alias: script
+automation_uid: 5b4e0237da074764
+category: MLC Script Template
+deps: []
+new_env_keys: []
+new_state_keys: []
+post_deps: []
+posthook_deps: []
+prehook_deps: []
+tags:
+- kill
+- process
+tests:
+  run_inputs: []
+uid: 27b5651692454cb6
+input_mapping:
+  group: MLC_KILL_PROCESS_GROUP
+  pid: MLC_KILL_PROCESS_ID
+  pname: MLC_KILL_PROCESS_NAME
+
+

script/kill-process/run.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+function exit_if_error() {
+  test $? -eq 0 || exit $?
+}
+
+function run() {
+  echo "Running: "
+  echo "$1"
+  echo ""
+  if [[ ${MLC_FAKE_RUN} != 'yes' ]]; then
+    eval "$1"
+    exit_if_error
+  fi
+}
+
+#Add your run commands here...
+run "$MLC_RUN_CMD"