Rye can update self to a special git branch

CHANGELOG.md

@@ -16,6 +16,8 @@ _Unreleased_
 - Updating `rye` will now also ensure that the self-venv is updated.  Previously
   this was deferred until the next `sync`.  #863
 
+- The `self update` command now accepts `--branch`.  #864
+
 ## 0.28.0
 
 Released on 2024-03-07

docs/guide/commands/self/update.md

@@ -25,6 +25,12 @@ Compile a specific revision:
 $ rye self update --rev 08910bc9b3b7c72a3d3ac694c4f3412259161477
 ```
 
+Compile latest development version:
+
+```
+$ rye self update --branch main
+```
+
 ## Arguments
 
 _no arguments_
@@ -37,6 +43,8 @@ _no arguments_
 
 * `--rev <REV>`: Update to a specific git rev
 
+* `--branch <BRANCH>`: Update to a specific git branch
+
 * `--force`: Force reinstallation
 
 * `-h, --help`: Print help (see a summary with '-h')

rye/src/cli/rye.rs

@@ -75,6 +75,9 @@ pub struct UpdateCommand {
     /// Update to a specific git rev.
     #[arg(long, conflicts_with = "tag")]
     rev: Option<String>,
+    /// Update to a specific git branch.
+    #[arg(long, conflicts_with = "tag", conflicts_with = "rev")]
+    branch: Option<String>,
     /// Force reinstallation
     #[arg(long)]
     force: bool,
@@ -192,7 +195,7 @@ fn update(args: UpdateCommand) -> Result<(), Error> {
     let current_exe = env::current_exe()?;
 
     // git based installation with cargo
-    if args.rev.is_some() || args.tag.is_some() {
+    if args.rev.is_some() || args.tag.is_some() || args.branch.is_some() {
         let mut cmd = Command::new("cargo");
         let tmp = tempdir()?;
         cmd.arg("install")
@@ -214,6 +217,9 @@ fn update(args: UpdateCommand) -> Result<(), Error> {
         } else if let Some(ref tag) = args.tag {
             cmd.arg("--tag");
             cmd.arg(tag);
+        } else if let Some(ref branch) = args.branch {
+            cmd.arg("--branch");
+            cmd.arg(branch);
         }
         if args.force {
             cmd.arg("--force");