More hashbang support (#97)

Author: kurtmckee

changelog.d/20260219_071227_kurtmckee_more_hashbang_support.rst

@@ -0,0 +1,7 @@
+Default styles
+--------------
+
+*   Support ``.fish`` (Fish shell).
+*   Support ``.zsh`` (zsh shell).
+*   Recognize fish, Powershell, R, and zsh file types
+    by their hashbang executable name.

changelog.d/20260219_071459_kurtmckee_more_hashbang_support.rst

@@ -0,0 +1,8 @@
+Fixed
+-----
+
+*   Strip the executable extension (like ``.exe``) before removing version numbers
+    when identifying a file type based on its hashbang.
+
+    This resolves a bug that prevented identification of version-specific
+    executable names, like ``python3.14.exe``, on Windows platforms.

src/chipshot/reader/identity.py

@@ -38,16 +38,16 @@ def handle(info: FileInfo, config: dict[str, typing.Any]) -> None:
             info.identity = config["interpreters"][name]
             return
 
-        # Look for a version-less name.
-        name_without_version = name.rstrip("0123456789.")
-        if name_without_version in config["interpreters"]:
-            log.debug(f"{info.path}: Found versioned interpreter '{name}'")
-            info.identity = config["interpreters"][name_without_version]
-            return
-
-        # Try removing the extension (such as "python.exe" on Windows).
+        # Try removing the extension (such as ".exe" on Windows).
         stem = path.stem
         if stem in config["interpreters"]:
             log.debug(f"{info.path}: Found base interpreter '{name}'")
             info.identity = config["interpreters"][stem]
             return
+
+        # Look for a version-less name.
+        name_without_version = stem.rstrip("0123456789.")
+        if name_without_version in config["interpreters"]:
+            log.debug(f"{info.path}: Found versioned interpreter '{name}'")
+            info.identity = config["interpreters"][name_without_version]
+            return

src/chipshot/resources/default-config.toml

@@ -117,6 +117,10 @@ style = "slash-slash"
 prologue = "css-charset"
 style = "slash-star"
 
+[extensions.fish]
+prologue = "hashbang"
+style = "hash"
+
 [extensions.gd]
 style = "hash"
 
@@ -230,20 +234,28 @@ prologue = "xhtml_declaration_and_doctype"
 style = "xml"
 prologue = "xml_declaration"
 
+[extensions.zsh]
+style = "hash"
+prologue = "hashbang"
+
 
 # Interpreters
 # ------------
 
 [interpreters]
 bash = "shell"
+fish = "shell"
 gorun = "go"
 node = "javascript"
 nodejs = "javascript"
 perl = "perl"
+powershell = "powershell"
 pwsh = "powershell"
 python = "python"
+rscript = "r"
 ruby = "ruby"
 sh = "shell"
+zsh = "shell"
 
 
 # Types
@@ -269,6 +281,10 @@ style = "hash"
 prologue = "hashbang-with-encoding"
 style = "hash"
 
+[types.r]
+prologue = "hashbang"
+style = "hash"
+
 [types.ruby]
 prologue = "hashbang-with-encoding"
 style = "hash"

tests/test_reader_identity.py

@@ -19,7 +19,14 @@
         pytest.param("#!/usr/bin/python", "python", id="base executable name"),
         pytest.param("#!/usr/bin/python3.11", "python", id="trailing version"),
         pytest.param("#!/usr/bin/env python", "python", id="env"),
-        pytest.param(r"#!C:\Program Files\Python39\python.exe", "python", id="windows"),
+        pytest.param(
+            r'#!"C:\Program Files\Python39\python3.9.exe"', "python", id="windows"
+        ),
+        pytest.param(
+            "#!/c/Windows/System32/WindowsPowerShell/v1.0/powershell.exe -File",
+            "powershell",
+            id="language-powershell",
+        ),
         # Loop exhaustion
         pytest.param("#!env unknown -c", "", id="loop exhaustion"),
     ),