add arguments to godbolt

Author: tinaun

src/godbolt.rs

@@ -1,3 +1,4 @@
+use crate::{api, commands::Args};
 pub enum Compilation {
     Success { asm: String },
     Error { stderr: String },
@@ -32,17 +33,39 @@ struct GodboltResponse {
     asm: GodboltOutput,
 }
 
+pub fn help(args: Args) -> Result<(), crate::Error> {
+    let message = "Compile Rust code using https://rust.godbolt.org/. Full optimizations are applied unless overriden.
+```?godbolt flags={} rustc={} ``\u{200B}`code``\u{200B}` ```
+Optional arguments:
+    \tflags: flags to pass to rustc invocation. Defaults to \"-Copt-level=3 --edition=2018\".
+    \trustc: compiler version to invoke. Defaults to `nightly`. Possible values: `nightly`, `beta` or full version like `1.45.2`.
+    ";
+
+    api::send_reply(&args, &message)?;
+    Ok(())
+}
+
 /// Compile a given Rust source code file on Godbolt using the latest nightly compiler with
-/// full optimizations (-O3)
+/// full optimizations (-O3) by default
 /// Returns a multiline string with the pretty printed assembly
 pub fn compile_rust_source(
     http: &reqwest::blocking::Client,
     source_code: &str,
+    flags: &str,
+    rustc: &str,
 ) -> Result<Compilation, crate::Error> {
+    let cv = rustc_to_godbolt(rustc);
+    let cv = match cv {
+        Ok(c) => c,
+        Err(e) => {
+            return Ok(Compilation::Error { stderr: e });
+        }
+    };
+
     let response: GodboltResponse = http
         .execute(
-            http.post("https://godbolt.org/api/compiler/nightly/compile")
-                .query(&[("options", "-Copt-level=3")])
+            http.post(&format!("https://godbolt.org/api/compiler/{}/compile", cv))
+                .query(&[("options", &flags)])
                 .header(reqwest::header::ACCEPT, "application/json")
                 .body(source_code.to_owned())
                 .build()?,
@@ -61,3 +84,21 @@ pub fn compile_rust_source(
         }
     })
 }
+
+// converts a rustc version number to a godbolt compiler id
+fn rustc_to_godbolt(rustc_version: &str) -> Result<String, String> {
+    match rustc_version {
+        "beta" => Ok("beta".to_string()),
+        "nightly" => Ok("nightly".to_string()),
+        // this heuristic is barebones but catches most obviously wrong things
+        // it doesn't know anything about valid rustc versions
+        ver if ver.contains('.') && !ver.contains(|c: char| c.is_alphabetic()) => {
+            let mut godbolt_version = "r".to_string();
+            for segment in ver.split('.') {
+                godbolt_version.push_str(segment);
+            }
+            Ok(godbolt_version)
+        }
+        other => Err(format!("invalid rustc version: `{}`", other)),
+    }
+}

src/main.rs

@@ -149,12 +149,15 @@ fn app() -> Result<(), Error> {
         });
     }
 
-    cmds.add("?godbolt ```\ncode```", |args| {
+    cmds.add("?godbolt flags={} version={} ```\ncode```", |args| {
+        let flags = args.params.get("flags").unwrap_or(&"false");
+        let rustc = args.params.get("rustc").unwrap_or(&"nightly");
+
         let code = args
             .params
             .get("code")
             .ok_or("Unable to retrieve param: code")?;
-        let (lang, text) = match godbolt::compile_rust_source(args.http, code)? {
+        let (lang, text) = match godbolt::compile_rust_source(args.http, code, flags, rustc)? {
             godbolt::Compilation::Success { asm } => ("x86asm", asm),
             godbolt::Compilation::Error { stderr } => ("rust", stderr),
         };
@@ -169,12 +172,7 @@ fn app() -> Result<(), Error> {
         Ok(())
     });
     cmds.help("?godbolt", "View assembly using Godbolt", |args| {
-        api::send_reply(
-            &args,
-            "Compile Rust code using https://rust.godbolt.org. Full optimizations are applied. \
-            ```?godbolt ``\u{200B}`code``\u{200B}` ```",
-        )?;
-        Ok(())
+        godbolt::help(args)
     });
 
     // Slow mode.