From 90306b97b5189589714226dc1b855143ce64b61c Mon Sep 17 00:00:00 2001
From: Felix Stupp <felix.stupp@outlook.com>
Date: Sun, 31 Jul 2022 09:50:47 +0000
Subject: [PATCH] Add exec_shell with shell searching

---
 podman-compose-backup.py | 42 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/podman-compose-backup.py b/podman-compose-backup.py
index cfb9340..0e1dc00 100644
--- a/podman-compose-backup.py
+++ b/podman-compose-backup.py
@@ -32,6 +32,7 @@ import shlex
 from subprocess import CompletedProcess, PIPE, run
 import sys
 from typing import (
+    Callable,
     Dict,
     Iterable,
     List,
@@ -144,6 +145,47 @@ def exec_cmd(
     return res
 
 
+def process_tester(
+    exec: Callable[[CommandArgs], CompletedProcess]
+) -> Callable[[CommandArgs], bool]:
+    return lambda command: exec(command).returncode == 0
+
+
+def search_shell_with(tester: Callable[[CommandArgs], bool]) -> str:
+    for shell in DETECTED_SHELLS:
+        command = CommandArgs([shell, "-c", "true"])
+        res = exec_cmd(command, check=False)
+        if tester(command):
+            return command[0]
+    # TODO specialize
+    raise Exception(
+        f"Could not find an acceptable shell on this host, searched for {DETECTED_SHELLS}"
+    )
+
+
+@cache
+def search_shell() -> str:
+    return search_shell_with(
+        process_tester(
+            lambda command: exec_cmd(
+                command=command,
+                check=False,
+                capture_stdout=False,
+            )
+        )
+    )
+
+
+def exec_shell(
+    shell_cmd: ShellCommand, check: bool = True, capture_stdout: bool = True
+) -> CompletedProcess:
+    return exec_cmd(
+        CommandArgs([search_shell(), "-c", shell_cmd]),
+        check=check,
+        capture_stdout=capture_stdout,
+    )
+
+
 @wraps(print)
 def error(*args, **kwargs):
     ret = print(*args, file=sys.stderr, **kwargs)