Switch rustup_wrapper to Rust implementation

For better Windows portability

build.py

@@ -638,10 +638,16 @@ def setup_rustup(args):
     for src in cargo_bin.iterdir():
         tgt = wrapper_dir / src.name
         tgt.symlink_to(src)
-    # Replace rustup with python script
-    wrapper = wrapper_dir / "rustup"
-    wrapper.unlink()
-    cp(Path("scripts", "rustup_wrapper.py"), wrapper)
+
+    # Build rustup_wrapper
+    wrapper_src = Path("tools", "rustup_wrapper")
+    cargo_toml = wrapper_src / "Cargo.toml"
+    execv([cargo, "build", "--release", f"--manifest-path={cargo_toml}"])
+
+    # Replace rustup with wrapper
+    wrapper = wrapper_dir / (f"rustup{EXE_EXT}")
+    wrapper.unlink(missing_ok=True)
+    cp(wrapper_src / "target" / "release" / (f"rustup_wrapper{EXE_EXT}"), wrapper)
     wrapper.chmod(0o755)
 
 

tools/rustup_wrapper/.gitignore

@@ -0,0 +1 @@
+target/

tools/rustup_wrapper/Cargo.toml

@@ -0,0 +1,9 @@
+[package]
+name = "rustup_wrapper"
+version = "0.0.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+home = "0.5"

tools/rustup_wrapper/src/main.rs

@@ -0,0 +1,43 @@
+use std::env;
+use std::process::{Command, Stdio};
+
+use home::cargo_home;
+
+/********************************
+ * Why do we need this wrapper?
+ ********************************
+ *
+ * The command `rustup component list` does not work with custom toolchains:
+ * > error: toolchain 'magisk' does not support components
+ *
+ * However, this command is used by several IDEs to determine available
+ * components that can be used, such as clippy, rustfmt etc.
+ * In this script, we simply redirect the output when using the nightly
+ * channel if any `component` command failed.
+*/
+
+fn main() -> std::io::Result<()> {
+    let rustup = cargo_home()?.join("bin").join("rustup");
+    let argv: Vec<String> = env::args().skip(1).collect();
+
+    if argv.iter().any(|s| s == "component") {
+        let status = Command::new(&rustup)
+            .args(&argv)
+            .stdout(Stdio::null())
+            .stderr(Stdio::null())
+            .status()?;
+        if !status.success() {
+            let mut cmd = Command::new(&rustup);
+            cmd.arg("+nightly");
+            if argv[0].starts_with('+') {
+                cmd.args(argv.iter().skip(1));
+            } else {
+                cmd.args(&argv);
+            }
+            return cmd.status().map(|_| ());
+        }
+    }
+
+    // Simply pass through
+    Command::new(&rustup).args(argv.iter()).status().map(|_| ())
+}