From d555284a01d0d3eba8921cdd192cff9a1e0da6b7 Mon Sep 17 00:00:00 2001
From: Ian Fijolek <ian@iamthefij.com>
Date: Mon, 22 May 2023 17:09:55 -0700
Subject: [PATCH] Avoid installing pre-release versions unless explicitly asked

---
 release_gitter.py | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/release_gitter.py b/release_gitter.py
index cc5e81f..4cf378a 100755
--- a/release_gitter.py
+++ b/release_gitter.py
@@ -157,8 +157,8 @@ def read_version(from_tags: bool = False, fetch: bool = False) -> str | None:
 
 def fetch_release(
     remote: GitRemoteInfo,
-    version: str | None = None
-    # TODO: Accept an argument for pre-release
+    version: str | None = None,
+    pre_release=False,
 ) -> dict[Any, Any]:
     """Fetches a release object from a Github repo
 
@@ -174,14 +174,21 @@ def fetch_release(
 
     # Return the latest if requested
     if version is None or version == "latest":
-        return result.json()[0]
+        for release in result.json():
+            if release["prerelease"] and not pre_release:
+                continue
+
+            return release
 
     # Return matching version
     for release in result.json():
         if release["tag_name"].endswith(version):
             return release
 
-    raise ValueError(f"Could not find release version ending in {version}")
+    raise ValueError(
+        f"Could not find release version ending in {version}."
+        f"{ ' Is it a pre-release?' if not pre_release else ''}"
+    )
 
 
 def match_asset(
@@ -447,6 +454,11 @@ def _parse_args(args: list[str] | None = None) -> argparse.Namespace:
         "--version",
         help="Release version to download. If not provied, it will look for project metadata",
     )
+    parser.add_argument(
+        "--prerelease",
+        action="store_true",
+        help="Include pre-release versions in search",
+    )
     parser.add_argument(
         "--version-git-tag",
         "-t",
@@ -527,9 +539,14 @@ def download_release(
     system_mapping: dict[str, str] | None = None,
     arch_mapping: dict[str, str] | None = None,
     extract_files: list[str] | None = None,
+    pre_release=False,
 ) -> list[Path]:
     """Convenience method for fetching, downloading and extracting a release"""
-    release = fetch_release(remote_info, version=version)
+    release = fetch_release(
+        remote_info,
+        version=version,
+        pre_release=pre_release,
+    )
     asset = match_asset(
         release,
         format,
@@ -550,7 +567,9 @@ def main():
     args = _parse_args()
 
     release = fetch_release(
-        GitRemoteInfo(args.hostname, args.owner, args.repo), args.version
+        GitRemoteInfo(args.hostname, args.owner, args.repo),
+        version=args.version,
+        pre_release=args.prerelease,
     )
     asset = match_asset(
         release,