Fix spaces in scripts shebang

src/poetry/masonry/builders/editable.py

@@ -9,6 +9,8 @@
 from pathlib import Path
 from typing import TYPE_CHECKING
 
+from installer.scripts import Script
+from installer.utils import get_launcher_kind
 from poetry.core.masonry.builders.builder import Builder
 from poetry.core.masonry.builders.sdist import SdistBuilder
 from poetry.core.masonry.utils.package_include import PackageInclude
@@ -27,15 +29,6 @@
     from poetry.poetry import Poetry
     from poetry.utils.env import Env
 
-SCRIPT_TEMPLATE = """\
-#!{python}
-import sys
-from {module} import {callable_holder}
-
-if __name__ == '__main__':
-    sys.exit({callable_}())
-"""
-
 WINDOWS_CMD_TEMPLATE = """\
 @echo off\r\n"{python}" "%~dp0\\{script}" %*\r\n
 """
@@ -176,21 +169,16 @@ def _add_scripts(self) -> list[Path]:
 
                 raise ValueError(msg)
 
-            callable_holder = callable_.split(".", 1)[0]
-
             script_file = scripts_path.joinpath(name)
             self._debug(
                 f"  - Adding the <c2>{name}</c2> script to <b>{scripts_path}</b>"
             )
             with script_file.open("w", encoding="utf-8") as f:
                 f.write(
-                    decode(
-                        SCRIPT_TEMPLATE.format(
-                            python=self._env.python,
-                            module=module,
-                            callable_holder=callable_holder,
-                            callable_=callable_,
-                        )
+                    str(
+                        Script(name, module, callable_, "console")
+                        .generate(str(self._env.python), get_launcher_kind())[1]
+                        .decode()
                     )
                 )
 

tests/masonry/builders/test_editable_builder.py

@@ -213,38 +213,79 @@
 
     baz_script = f"""\
 #!{tmp_venv.python}
+# -*- coding: utf-8 -*-
+import re
 import sys
 from bar import baz
-
-if __name__ == '__main__':
+if __name__ == "__main__":
+    sys.argv[0] = re.sub(r"(-script\\.pyw|\\.exe)?$", "", sys.argv[0])
     sys.exit(baz.boom.bim())
 """
 
     assert baz_script == tmp_venv._bin_dir.joinpath("baz").read_text(encoding="utf-8")
 
     foo_script = f"""\
 #!{tmp_venv.python}
+# -*- coding: utf-8 -*-
+import re
 import sys
 from foo import bar
-
-if __name__ == '__main__':
+if __name__ == "__main__":
+    sys.argv[0] = re.sub(r"(-script\\.pyw|\\.exe)?$", "", sys.argv[0])
     sys.exit(bar())
 """
 
     assert foo_script == tmp_venv._bin_dir.joinpath("foo").read_text(encoding="utf-8")
 
     fox_script = f"""\
 #!{tmp_venv.python}
+# -*- coding: utf-8 -*-
+import re
 import sys
 from fuz.foo import bar
-
-if __name__ == '__main__':
+if __name__ == "__main__":
+    sys.argv[0] = re.sub(r"(-script\\.pyw|\\.exe)?$", "", sys.argv[0])
     sys.exit(bar.baz())
 """
 
     assert fox_script == tmp_venv._bin_dir.joinpath("fox").read_text(encoding="utf-8")
 
 
+def test_builder_generates_scripts_with_escaped_shebang_when_spaces_in_path(
+    simple_poetry: Poetry,
+    tmp_path: Path,
+    fixture_dir: FixtureDirGetter,
+) -> None:
+    simple_poetry = Factory().create_poetry(fixture_dir("simple_project"))
+    env_manager = EnvManager(simple_poetry)
+    venv_path = tmp_path / "path with space" / "venv"
+    env_manager.build_venv(venv_path)
+    tmp_venv = VirtualEnv(venv_path)
+
+    builder = EditableBuilder(simple_poetry, tmp_venv, NullIO())
+
+    builder.build()
+
+    assert tmp_venv._bin_dir.joinpath("baz").exists()
+
+    expected_shebang = f"""/bin/sh
+'''exec' '{tmp_path}/path with space/venv/bin/python' "$0" "$@"
+' '''"""
+
+    baz_script = f"""\
+#!{expected_shebang}
+# -*- coding: utf-8 -*-
+import re
+import sys
+from bar import baz
+if __name__ == "__main__":
+    sys.argv[0] = re.sub(r"(-script\\.pyw|\\.exe)?$", "", sys.argv[0])
+    sys.exit(baz.boom.bim())
+"""
+
+    assert baz_script == tmp_venv._bin_dir.joinpath("baz").read_text(encoding="utf-8")
+
+
 def test_builder_falls_back_on_setup_and_pip_for_packages_with_build_scripts(
     mocker: MockerFixture, extended_poetry: Poetry, tmp_path: Path
 ) -> None: