dmi: Add QR code generation (#226)

Author: itsmeow

Cargo.lock

@@ -78,6 +78,7 @@ indexmap = { version = "2.10.0", optional = true, features = [
     "rayon",
 ] }
 ordered-float = { version = "5.0.0", optional = true, features = ["serde"] }
+qrcode = { version = "0.14.1", optional = true, features = ["image", "svg"]}
 
 [features]
 default = [
@@ -137,7 +138,7 @@ all = [
 acreplace = ["aho-corasick"]
 batchnoise = ["dbpnoise"]
 cellularnoise = ["rand", "rayon"]
-dmi = ["png", "image", "serde_repr"]
+dmi = ["png", "image", "qrcode", "serde_repr"]
 file = []
 git = ["gix", "chrono"]
 hash = [

Cargo.toml

@@ -37,3 +37,6 @@
  * `metadata` must be a json_encode'd list in the metadata format listed above.
  */
 #define rustg_dmi_inject_metadata(path, metadata) RUSTG_CALL(RUST_G, "dmi_inject_metadata")(path, metadata)
+
+#define rustg_create_qr_code_png(path, data) RUSTG_CALL(RUST_G, "create_qr_code_png")(path, data)
+#define rustg_create_qr_code_svg(data) RUSTG_CALL(RUST_G, "create_qr_code_svg")(data)

dmsrc/dmi.dm

@@ -3,7 +3,9 @@ use dmi::{
     error::DmiError,
     icon::{Icon, Looping},
 };
+use image::Rgba;
 use png::{text_metadata::ZTXtChunk, Decoder, Encoder, OutputInfo, Reader};
+use qrcode::{render::svg, QrCode};
 use serde::{Deserialize, Serialize};
 use serde_repr::{Deserialize_repr, Serialize_repr};
 use std::{
@@ -302,3 +304,24 @@ fn inject_metadata(path: &str, metadata: &str) -> Result<()> {
     encoder.write_header()?.write_image_data(&raw_image_data)?;
     Ok(())
 }
+
+byond_fn!(fn create_qr_code_png(path, data) {
+    let code = match QrCode::new(data.as_bytes()) {
+        Ok(code) => code,
+        Err(err) => return Some(format!("Error: Could not read data into QR code: {err}"))
+    };
+    let image = code.render::<Rgba<u8>>().build();
+    match image.save(path) {
+        Ok(_) => Some(String::from(path)),
+        Err(err) => Some(format!("Error: Could not write QR code image to path: {err}"))
+    }
+});
+
+byond_fn!(fn create_qr_code_svg(data) {
+    let code = match QrCode::new(data.as_bytes()) {
+        Ok(code) => code,
+        Err(err) => return Some(format!("Error: Could not read data into QR code: {err}"))
+    };
+    let svg_xml = code.render::<svg::Color>().build();
+    Some(svg_xml)
+});