1 files changed, 19 insertions, 0 deletions
diff --git a/utils/run_with_display.sh b/utils/run_with_display.sh
new file mode 100755
index 0000000..5f1f3a0
--- /dev/null
+++ b/utils/run_with_display.sh
@@ -0,0 +1,19 @@
+#!/bin/bash
+
+run_with_display() {
+    output=$("$@" 2>&1)
+    exit_status=$?
+
+    if [[ $exit_status -ne 0 && ("$output" =~ "cannot open display" || "$output" =~ "DISPLAY environment variable is missing") ]]; then
+        DISPLAY=:0 "$@"
+    else
+        echo "$output"
+        return $exit_status
+    fi
+}
+
+# Call this script with any command you want to run
+command=$1
+shift
+run_with_display "$command" "$@"
+