From 475a1ed6d6951e689e1f71a6f98d6739dbd2c0d6 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 27 Aug 2024 05:36:13 -0700
Subject: [PATCH] feat: Add --install-main-branch switch

---
 aider/args.py         | 6 ++++++
 aider/main.py         | 5 +++++
 aider/versioncheck.py | 4 ++--
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/aider/args.py b/aider/args.py
index b2d3aeadc..f228bbfd2 100644
--- a/aider/args.py
+++ b/aider/args.py
@@ -503,6 +503,12 @@ def get_parser(default_config_files, git_root):
         help="Check for new aider versions on launch",
         default=True,
     )
+    group.add_argument(
+        "--install-main-branch",
+        action="store_true",
+        help="Install the latest version from the main branch",
+        default=False,
+    )
     group.add_argument(
         "--apply",
         metavar="FILE",
diff --git a/aider/main.py b/aider/main.py
index 2310de715..8fe68ac70 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -430,6 +430,11 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
         update_available = check_version(io, just_check=True, verbose=args.verbose)
         return 0 if not update_available else 1
 
+    if args.install_main_branch:
+        from aider.versioncheck import install_from_main_branch
+        success = install_from_main_branch(io)
+        return 0 if success else 1
+
     if args.check_update:
         check_version(io, verbose=args.verbose)
 
diff --git a/aider/versioncheck.py b/aider/versioncheck.py
index a71a9471a..38c91c20c 100644
--- a/aider/versioncheck.py
+++ b/aider/versioncheck.py
@@ -16,8 +16,8 @@ def install_from_main_branch(io):
     """
     return utils.check_pip_install_extra(
         io,
-        None
-        "Install the latest version from the main branch?",
+        "aider",
+        "Installing the latest version from the main branch...",
         ["--upgrade", "git+https://github.com/paul-gauthier/aider.git"],
     )