meson_install.py: Start by checking if the shebang is directly runnable

If it's just runnable as-is, then we don't need to do fancy basename detection
and such to find the interpretor to use for running the script.

1 files changed, 10 insertions, 6 deletions

diff --git a/mesonbuild/scripts/meson_install.py b/mesonbuild/scripts/meson_install.py
index 792af6c..8e3d0ca 100644
--- a/mesonbuild/scripts/meson_install.py
+++ b/mesonbuild/scripts/meson_install.py
@@ -118,10 +118,14 @@ def run_install_script(d):
         if platform.system().lower() == 'windows' and suffix != '.bat':
             first_line = open(script).readline().strip()
             if first_line.startswith('#!'):
-                commands = first_line[2:].split('#')[0].strip().split()
-                commands[0] = shutil.which(commands[0].split('/')[-1])
-                if commands[0] is None:
-                    raise RuntimeError("Don't know how to run script %s." % script)
+                if shutil.which(first_line[2:]):
+                    commands = [first_line[2:]]
+                else:
+                    commands = first_line[2:].split('#')[0].strip().split()
+                    commands[0] = shutil.which(commands[0].split('/')[-1])
+                    if commands[0] is None:
+                        commands
+                        raise RuntimeError("Don't know how to run script %s." % script)
                 final_command = commands + [script] + i.cmd_arr[1:]
         else:
             final_command = i.cmd_arr
@@ -129,8 +133,8 @@ def run_install_script(d):
             rc = subprocess.call(final_command, env=child_env)
             if rc != 0:
                 sys.exit(rc)
-        except Exception:
-            print('Failed to run install script:', i.cmd_arr[0])
+        except:
+            print('Failed to run install script:', *i.cmd_arr)
             sys.exit(1)
 
 def is_elf_platform():