From 2a1999fe2048cf7220390f706f6b642e3e662fc7 Mon Sep 17 00:00:00 2001
From: Alok Singh <8325708+alok@users.noreply.github.com>
Date: Mon, 13 May 2024 05:52:13 -0700
Subject: [PATCH] Add rye support (#799)

Rye is a new cargo-like package manager for python by @mitsuhiko.
---
 src/config.rs        | 1 +
 src/main.rs          | 1 +
 src/steps/generic.rs | 7 +++++++
 3 files changed, 9 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index 447c6d49..a2a61cf7 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -133,6 +133,7 @@ pub enum Step {
     Rtcl,
     RubyGems,
     Rustup,
+    Rye,
     Scoop,
     Sdkman,
     SelfUpdate,
diff --git a/src/main.rs b/src/main.rs
index 4c12d5ac..75ca73c1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -344,6 +344,7 @@ fn run() -> Result<()> {
     // The following update function should be executed on all OSes.
     runner.execute(Step::Fossil, "fossil", || generic::run_fossil(&ctx))?;
     runner.execute(Step::Elan, "elan", || generic::run_elan(&ctx))?;
+    runner.execute(Step::Rye, "rye", || generic::run_rye(&ctx))?;
     runner.execute(Step::Rustup, "rustup", || generic::run_rustup(&ctx))?;
     runner.execute(Step::Juliaup, "juliaup", || generic::run_juliaup(&ctx))?;
     runner.execute(Step::Dotnet, ".NET", || generic::run_dotnet_upgrade(&ctx))?;
diff --git a/src/steps/generic.rs b/src/steps/generic.rs
index 2db01bae..731db822 100644
--- a/src/steps/generic.rs
+++ b/src/steps/generic.rs
@@ -227,6 +227,13 @@ pub fn run_rustup(ctx: &ExecutionContext) -> Result<()> {
     ctx.run_type().execute(rustup).arg("update").status_checked()
 }
 
+pub fn run_rye(ctx: &ExecutionContext) -> Result<()> {
+    let rye = require("rye")?;
+
+    print_separator("Rye");
+    ctx.run_type().execute(rye).args(["self", "update"]).status_checked()
+}
+
 pub fn run_elan(ctx: &ExecutionContext) -> Result<()> {
     let elan = require("elan")?;