shotgun_ruby 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 843c3ccc8ae2090623976c3d8561be3842d2ce26af5e105c6667de0e85d73992
+  data.tar.gz: 70079ea8b4c5d108722ccde7387b4221614973612150c7ce4580390bad663eb5
+SHA512:
+  metadata.gz: 4c0d5a3a4ff75bdec530dbe994faf12180f85894f84797e04fa93546fa464afd193da555f77ac1cfa61cd862ba6128dd915ec7bb0c06f695bd249d85113d419a
+  data.tar.gz: c12e2df1282b5b19c8f07cc9f0b1f947e7cf802b150a83f22ed34db78b6f48a97b01419c1942c159044a7cde6e3c20800638f1448f162d92b484ed7851b549d4

data/ext/shotgun_ruby/Cargo.lock

@@ -0,0 +1,410 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "adler32"
+version = "1.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "aae1277d39aeec15cb388266ecc24b11c80469deae6067e17a1a7aa9e5c1f234"
+
+[[package]]
+name = "aho-corasick"
+version = "0.7.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc936419f96fa211c1b9166887b38e5e40b19958e5b895be7c1f93adec7071ac"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "autocfg"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d468802bab17cbc0cc575e9b053f41e72aa36bfa6b7f55e3529ffa43161b97fa"
+
+[[package]]
+name = "bindgen"
+version = "0.60.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "062dddbc1ba4aca46de6338e2bf87771414c335f7b2f2036e8f3e9befebf88e6"
+dependencies = [
+ "bitflags",
+ "cexpr",
+ "clang-sys",
+ "lazy_static",
+ "lazycell",
+ "peeking_take_while",
+ "proc-macro2",
+ "quote",
+ "regex",
+ "rustc-hash",
+ "shlex",
+]
+
+[[package]]
+name = "bitflags"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
+
+[[package]]
+name = "bytemuck"
+version = "1.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "17febce684fd15d89027105661fec94afb475cb995fbc59d2865198446ba2eea"
+
+[[package]]
+name = "byteorder"
+version = "1.4.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "14c189c53d098945499cdfa7ecc63567cf3886b3332b312a5b4585d8d3a6a610"
+
+[[package]]
+name = "cexpr"
+version = "0.6.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6fac387a98bb7c37292057cffc56d62ecb629900026402633ae9160df93a8766"
+dependencies = [
+ "nom",
+]
+
+[[package]]
+name = "cfg-if"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+
+[[package]]
+name = "clang-sys"
+version = "1.6.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c688fc74432808e3eb684cae8830a86be1d66a2bd58e1f248ed0960a590baf6f"
+dependencies = [
+ "glob",
+ "libc",
+ "libloading",
+]
+
+[[package]]
+name = "color_quant"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3d7b894f5411737b7867f4827955924d7c254fc9f4d91a6aad6b097804b1018b"
+
+[[package]]
+name = "crc32fast"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b540bd8bc810d3885c6ea91e2018302f68baba2129ab3e88f32389ee9370880d"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "deflate"
+version = "0.8.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "73770f8e1fe7d64df17ca66ad28994a0a623ea497fa69486e14984e715c5d174"
+dependencies = [
+ "adler32",
+ "byteorder",
+]
+
+[[package]]
+name = "getopts"
+version = "0.2.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "14dbbfd5c71d70241ecf9e6f13737f7b5ce823821063188d7e46c41d371eebd5"
+dependencies = [
+ "unicode-width",
+]
+
+[[package]]
+name = "glob"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d2fabcfbdc87f4758337ca535fb41a6d701b65693ce38287d856d1674551ec9b"
+
+[[package]]
+name = "image"
+version = "0.23.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "24ffcb7e7244a9bf19d35bf2883b9c080c4ced3c07a9895572178cdb8f13f6a1"
+dependencies = [
+ "bytemuck",
+ "byteorder",
+ "color_quant",
+ "num-iter",
+ "num-rational",
+ "num-traits",
+ "png",
+]
+
+[[package]]
+name = "lazy_static"
+version = "1.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
+
+[[package]]
+name = "lazycell"
+version = "1.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "830d08ce1d1d941e6b30645f1a0eb5643013d835ce3779a5fc208261dbe10f55"
+
+[[package]]
+name = "libc"
+version = "0.2.140"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "99227334921fae1a979cf0bfdfcc6b3e5ce376ef57e16fb6fb3ea2ed6095f80c"
+
+[[package]]
+name = "libloading"
+version = "0.7.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b67380fd3b2fbe7527a606e18729d21c6f3951633d0500574c4dc22d2d638b9f"
+dependencies = [
+ "cfg-if",
+ "winapi",
+]
+
+[[package]]
+name = "memchr"
+version = "2.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2dffe52ecf27772e601905b7522cb4ef790d2cc203488bbd0e2fe85fcb74566d"
+
+[[package]]
+name = "minimal-lexical"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68354c5c6bd36d73ff3feceb05efa59b6acb7626617f4962be322a825e61f79a"
+
+[[package]]
+name = "miniz_oxide"
+version = "0.3.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "791daaae1ed6889560f8c4359194f56648355540573244a5448a83ba1ecc7435"
+dependencies = [
+ "adler32",
+]
+
+[[package]]
+name = "nom"
+version = "7.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d273983c5a657a70a3e8f2a01329822f3b8c8172b73826411a55751e404a0a4a"
+dependencies = [
+ "memchr",
+ "minimal-lexical",
+]
+
+[[package]]
+name = "num-integer"
+version = "0.1.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "225d3389fb3509a24c93f5c29eb6bde2586b98d9f016636dff58d7c6f7569cd9"
+dependencies = [
+ "autocfg",
+ "num-traits",
+]
+
+[[package]]
+name = "num-iter"
+version = "0.1.43"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7d03e6c028c5dc5cac6e2dec0efda81fc887605bb3d884578bb6d6bf7514e252"
+dependencies = [
+ "autocfg",
+ "num-integer",
+ "num-traits",
+]
+
+[[package]]
+name = "num-rational"
+version = "0.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "12ac428b1cb17fce6f731001d307d351ec70a6d202fc2e60f7d4c5e42d8f4f07"
+dependencies = [
+ "autocfg",
+ "num-integer",
+ "num-traits",
+]
+
+[[package]]
+name = "num-traits"
+version = "0.2.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "578ede34cf02f8924ab9447f50c28075b4d3e5b269972345e7e0372b38c6cdcd"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "peeking_take_while"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "19b17cddbe7ec3f8bc800887bab5e717348c95ea2ca0b1bf0837fb964dc67099"
+
+[[package]]
+name = "pkg-config"
+version = "0.3.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6ac9a59f73473f1b8d852421e59e64809f025994837ef743615c6d0c5b305160"
+
+[[package]]
+name = "png"
+version = "0.16.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3c3287920cb847dee3de33d301c463fba14dda99db24214ddf93f83d3021f4c6"
+dependencies = [
+ "bitflags",
+ "crc32fast",
+ "deflate",
+ "miniz_oxide",
+]
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.54"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e472a104799c74b514a57226160104aa483546de37e839ec50e3c2e41dd87534"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.26"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4424af4bf778aae2051a77b60283332f386554255d722233d09fbfc7e30da2fc"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rb-sys"
+version = "0.9.70"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dd532ce4ab93abcabd4b92d0eb1df5c02f64d3624b713e2bcf6a4186847f2c65"
+dependencies = [
+ "rb-sys-build",
+]
+
+[[package]]
+name = "rb-sys-build"
+version = "0.9.70"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6959d6f87715ddb2854d833039aa22b06bbab3c3d63bb1df22e6da198e8390c5"
+dependencies = [
+ "bindgen",
+ "lazy_static",
+ "proc-macro2",
+ "quote",
+ "regex",
+ "shell-words",
+ "syn",
+]
+
+[[package]]
+name = "regex"
+version = "1.7.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8b1f693b24f6ac912f4893ef08244d70b6067480d2f1a46e950c9691e6749d1d"
+dependencies = [
+ "aho-corasick",
+ "memchr",
+ "regex-syntax",
+]
+
+[[package]]
+name = "regex-syntax"
+version = "0.6.29"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f162c6dd7b008981e4d40210aca20b4bd0f9b60ca9271061b07f78537722f2e1"
+
+[[package]]
+name = "rustc-hash"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "08d43f7aa6b08d49f382cde6a7982047c3426db949b1424bc4b7ec9ae12c6ce2"
+
+[[package]]
+name = "shell-words"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "24188a676b6ae68c3b2cb3a01be17fbf7240ce009799bb56d5b1409051e78fde"
+
+[[package]]
+name = "shlex"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "43b2853a4d09f215c24cc5489c992ce46052d359b5109343cbafbf26bc62f8a3"
+
+[[package]]
+name = "shotgun_ruby"
+version = "0.1.0"
+dependencies = [
+ "getopts",
+ "image",
+ "libc",
+ "num-traits",
+ "rb-sys",
+ "x11",
+]
+
+[[package]]
+name = "syn"
+version = "1.0.109"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e5464a87b239f13a63a501f2701565754bae92d243d4bb7eb12f6d57d2269bf4"
+
+[[package]]
+name = "unicode-width"
+version = "0.1.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c0edd1e5b14653f783770bce4a4dabb4a5108a5370a5f5d8cfe8710c361f6c8b"
+
+[[package]]
+name = "winapi"
+version = "0.3.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
+dependencies = [
+ "winapi-i686-pc-windows-gnu",
+ "winapi-x86_64-pc-windows-gnu",
+]
+
+[[package]]
+name = "winapi-i686-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
+
+[[package]]
+name = "winapi-x86_64-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
+
+[[package]]
+name = "x11"
+version = "2.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "502da5464ccd04011667b11c435cb992822c2c0dbde1770c988480d312a0db2e"
+dependencies = [
+ "libc",
+ "pkg-config",
+]

data/ext/shotgun_ruby/Cargo.toml

@@ -0,0 +1,27 @@
+[package]
+name = "shotgun_ruby"
+version = "0.1.0"
+edition = "2018"
+
+[dependencies]
+rb-sys = { version = "0.9.68", features = ["global-allocator"] }
+
+getopts = "0.2"
+libc = "0.2"
+num-traits = "0.2"
+
+[dependencies.image]
+default-features = false
+version = "0.23"
+features = ["png", "pnm"]
+
+[dependencies.x11]
+version = "2.18"
+features = ["xlib", "xrandr"]
+
+
+[lib]
+crate-type = ["cdylib"]
+
+[features]
+test-feature = []

data/ext/shotgun_ruby/extconf.rb

@@ -0,0 +1,34 @@
+# Use local rb_sys gem (only needed for developing in this repo)
+$LOAD_PATH.unshift(File.expand_path("../../../../gem/lib", __dir__))
+
+# We need to require mkmf *first* this since `rake-compiler` injects code here for cross compilation
+require "mkmf"
+require "rb_sys/mkmf"
+
+create_rust_makefile("shotgun_ruby/shotgun_ruby") do |r|
+  # Create debug builds in dev. Make sure that release gems are compiled with
+  # `RB_SYS_CARGO_PROFILE=release` (optional)
+  r.profile = ENV.fetch("RB_SYS_CARGO_PROFILE", :dev).to_sym
+
+  # Can be overridden with `RB_SYS_CARGO_FEATURES` env var (optional)
+  r.features = ["test-feature"]
+
+  # You can add whatever env vars you want to the env hash (optional)
+  r.env = {"FOO" => "BAR"}
+
+  # If your Cargo.toml is in a different directory, you can specify it here (optional)
+  r.ext_dir = "."
+
+  # Extra flags to pass to the $RUSTFLAGS environment variable (optional)
+  r.extra_rustflags << "--cfg=some_nested_config_var_for_crate"
+
+  # Force a rust toolchain to be installed via rustup (optional)
+  # You can also set the env var `RB_SYS_FORCE_INSTALL_RUST_TOOLCHAIN=true`
+  r.force_install_rust_toolchain = false
+
+  # Extra args to pass to the `cargo rustc` command (optional)
+  r.extra_cargo_args << "--quiet"
+
+  # Extra args to pass to the `rustc` command (optional)
+  r.extra_rustc_args = []
+end

data/ext/shotgun_ruby/src/lib.rs

@@ -0,0 +1,101 @@
+use rb_sys::*;
+use std::ffi::{CStr, CString};
+use std::os::raw::c_long;
+
+use std::fs::File;
+use std::path::Path;
+use x11::xlib;
+
+
+mod util;
+mod xwrap;
+use crate::xwrap::Display;
+
+fn print_type_of<T>(_: &T) {
+    println!("{}", std::any::type_name::<T>())
+}
+
+unsafe extern "C" fn take_screenshot(_klass: RubyValue, mut winid: RubyValue, mut filepath: RubyValue) -> RubyValue {
+
+    let window_id = CStr::from_ptr(rb_string_value_cstr(&mut winid)).to_str().unwrap();
+    let path = CStr::from_ptr(rb_string_value_cstr(&mut filepath)).to_str().unwrap();
+
+    let display = match Display::open(None) {
+        Some(d) => d,
+        None => {
+            eprintln!("Failed to open display");
+            return 1;
+        }
+    };
+
+    let window = match util::parse_int::<xlib::Window>(&window_id){
+        Ok(r) => r,
+        Err(_) => {
+            eprintln!("Window ID is not a valid integer");
+            eprintln!("Accepted values are decimal, hex (0x*), octal (0o*) and binary (0b*)");
+            return 1;
+        }
+    };
+
+    let output_ext = "pam";
+    let output_format = image::ImageOutputFormat::Pnm(image::pnm::PNMSubtype::ArbitraryMap);
+    
+    let window_rect = display.get_window_rect(window);
+    let sel = util::Rect {
+        x: 0,
+        y: 0,
+        w: window_rect.w,
+        h: window_rect.h,
+    };
+
+    let image = match display.get_image(window, sel, xwrap::ALL_PLANES, xlib::ZPixmap) {
+        Some(i) => i,
+        None => {
+            eprintln!("Failed to get image from X");
+            return 1;
+        }
+    };
+
+    let image = match image.into_image_buffer() {
+        Some(i) => image::DynamicImage::ImageRgba8(i),
+        None => {
+            eprintln!(
+                "Failed to convert captured framebuffer, only 24/32 \
+                      bit (A)RGB8 is supported"
+            );
+            return 1;
+        }
+    };
+
+    match File::create(&Path::new(&path)) {
+        Ok(mut f) => image
+            .write_to(&mut f, output_format)
+            .expect("Writing to file failed"),
+        Err(e) => {
+            eprintln!("Failed to create {}: {}", path, e);
+            return 1;
+        }
+    }
+
+    let output = CString::new(path).unwrap();
+    let size = path.len() as c_long;
+
+    rb_utf8_str_new(output.as_ptr(), size)
+}
+
+
+#[allow(non_snake_case)]
+#[no_mangle]
+extern "C" fn Init_shotgun_ruby() {
+    let name = CString::new("ShotgunRuby").unwrap();
+    let function_name = CString::new("screenshot").unwrap();
+
+    unsafe {
+        let klass = rb_define_module(name.as_ptr());
+        let callback = std::mem::transmute::<
+            unsafe extern "C" fn(RubyValue, RubyValue, RubyValue) -> RubyValue,
+            unsafe extern "C" fn() -> RubyValue,
+        >(take_screenshot);
+        rb_define_module_function(klass, function_name.as_ptr(), Some(callback), 2);
+    }
+}

data/ext/shotgun_ruby/src/util.rs

@@ -0,0 +1,51 @@
+use std::cmp;
+
+#[derive(Copy, Clone, Debug)]
+pub struct Rect {
+    pub x: i32,
+    pub y: i32,
+    pub w: i32,
+    pub h: i32,
+}
+
+#[derive(Copy, Clone, Debug)]
+pub struct Point {
+    pub x: i32,
+    pub y: i32,
+}
+
+impl Rect {
+    pub fn intersection(&self, other: Rect) -> Option<Rect> {
+        let ix = cmp::max(self.x, other.x);
+        let iy = cmp::max(self.y, other.y);
+        let iw = cmp::min(self.x + self.w, other.x + other.w) - ix;
+        let ih = cmp::min(self.y + self.h, other.y + other.h) - iy;
+
+        if iw > 0 && ih > 0 {
+            Some(Rect {
+                x: ix,
+                y: iy,
+                w: iw,
+                h: ih,
+            })
+        } else {
+            None
+        }
+    }
+
+    pub fn contains(&self, pos: Point) -> bool {
+        pos.x >= self.x && pos.x < self.x + self.w && pos.y >= self.y && pos.y < self.y + self.h
+    }
+}
+
+pub fn parse_int<T: num_traits::Num>(string: &str) -> Result<T, T::FromStrRadixErr> {
+    if string.len() < 2 {
+        return T::from_str_radix(string, 10);
+    }
+    match &string[..2] {
+        "0x" | "0X" => T::from_str_radix(&string[2..], 16),
+        "0o" | "0O" => T::from_str_radix(&string[2..], 8),
+        "0b" | "0B" => T::from_str_radix(&string[2..], 2),
+        _ => T::from_str_radix(string, 10),
+    }
+}

data/ext/shotgun_ruby/src/xwrap.rs

@@ -0,0 +1,327 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+use std::ffi;
+use std::mem;
+use std::os::raw;
+use std::ptr;
+use std::slice;
+
+use image::Pixel;
+use image::Rgba;
+use image::RgbaImage;
+use libc;
+use x11::xlib;
+use x11::xrandr;
+
+use crate::util;
+
+pub const ALL_PLANES: libc::c_ulong = !0;
+
+pub struct Display {
+    handle: *mut xlib::Display,
+}
+
+pub struct Image {
+    handle: *mut xlib::XImage,
+}
+
+pub struct ScreenRectIter<'a> {
+    dpy: &'a Display,
+    res: *mut xrandr::XRRScreenResources,
+    crtcs: &'a [xrandr::RRCrtc],
+    i: usize,
+}
+
+impl Display {
+    pub fn open(name: Option<ffi::CString>) -> Option<Display> {
+        unsafe {
+            let name = match name {
+                None => ptr::null(),
+                Some(cstr) => cstr.as_ptr(),
+            };
+            let d = xlib::XOpenDisplay(name);
+
+            if d.is_null() {
+                return None;
+            }
+
+            Some(Display { handle: d })
+        }
+    }
+
+    pub fn get_default_root(&self) -> xlib::Window {
+        unsafe { xlib::XDefaultRootWindow(self.handle) }
+    }
+
+    pub fn get_window_rect(&self, window: xlib::Window) -> util::Rect {
+        unsafe {
+            let mut attrs = mem::MaybeUninit::uninit();
+            xlib::XGetWindowAttributes(self.handle, window, attrs.as_mut_ptr());
+            let attrs = attrs.assume_init();
+
+            let mut root = 0;
+            let mut parent = 0;
+            let mut children: *mut xlib::Window = ptr::null_mut();
+            let mut nchildren = 0;
+            xlib::XQueryTree(
+                self.handle,
+                window,
+                &mut root,
+                &mut parent,
+                &mut children,
+                &mut nchildren,
+            );
+            if !children.is_null() {
+                xlib::XFree(children as *mut raw::c_void);
+            }
+
+            let mut x = attrs.x;
+            let mut y = attrs.y;
+
+            if parent != 0 {
+                let mut child = 0;
+                xlib::XTranslateCoordinates(
+                    self.handle,
+                    parent,
+                    root,
+                    attrs.x,
+                    attrs.y,
+                    &mut x,
+                    &mut y,
+                    &mut child,
+                );
+            }
+
+            util::Rect {
+                x: x,
+                y: y,
+                w: attrs.width,
+                h: attrs.height,
+            }
+        }
+    }
+
+    pub fn get_image(
+        &self,
+        window: xlib::Window,
+        rect: util::Rect,
+        plane_mask: libc::c_ulong,
+        format: libc::c_int,
+    ) -> Option<Image> {
+        unsafe {
+            let image = xlib::XGetImage(
+                self.handle,
+                window,
+                rect.x,
+                rect.y,
+                rect.w as libc::c_uint,
+                rect.h as libc::c_uint,
+                plane_mask,
+                format,
+            );
+
+            if image.is_null() {
+                return None;
+            }
+
+            Some(Image::from_raw_ximage(image))
+        }
+    }
+
+    pub fn get_screen_rects(&self, root: xlib::Window) -> Option<ScreenRectIter<'_>> {
+        unsafe {
+            let xrr_res = xrandr::XRRGetScreenResourcesCurrent(self.handle, root);
+
+            if xrr_res.is_null() {
+                return None;
+            }
+
+            Some(ScreenRectIter {
+                dpy: &self,
+                res: xrr_res,
+                crtcs: slice::from_raw_parts((*xrr_res).crtcs, (*xrr_res).ncrtc as usize),
+                i: 0,
+            })
+        }
+    }
+
+    pub fn get_cursor_position(&self, window: xlib::Window) -> Option<util::Point> {
+        let mut x = 0;
+        let mut y = 0;
+
+        unsafe {
+            if xlib::XQueryPointer(
+                self.handle,
+                window,
+                &mut 0,
+                &mut 0,
+                &mut x,
+                &mut y,
+                &mut 0,
+                &mut 0,
+                &mut 0,
+            ) == 0
+            {
+                return None;
+            }
+        }
+
+        Some(util::Point { x, y })
+    }
+}
+
+impl Drop for Display {
+    fn drop(&mut self) {
+        unsafe {
+            xlib::XCloseDisplay(self.handle);
+        }
+    }
+}
+
+impl Image {
+    pub fn from_raw_ximage(ximage: *mut xlib::XImage) -> Image {
+        Image { handle: ximage }
+    }
+
+    pub fn into_image_buffer(&self) -> Option<RgbaImage> {
+        unsafe {
+            // Extract values from the XImage into our own scope
+            macro_rules! get {
+                ($($a:ident),+) => ($(let $a = (*self.handle).$a;)+);
+            }
+            get!(
+                width,
+                height,
+                byte_order,
+                depth,
+                bytes_per_line,
+                bits_per_pixel,
+                red_mask,
+                green_mask,
+                blue_mask
+            );
+
+            // Pixel size
+            let stride = match (depth, bits_per_pixel) {
+                (24, 24) => 3,
+                (24, 32) | (32, 32) => 4,
+                _ => return None,
+            };
+
+            // Compute subpixel offsets into each pixel according the the bitmasks X gives us
+            // Only 8 bit, byte-aligned values are supported
+            // Truncate masks to the lower 32 bits as that is the maximum pixel size
+            macro_rules! channel_offset {
+                ($mask:expr) => {
+                    match (byte_order, $mask & 0xFFFFFFFF) {
+                        (0, 0xFF) | (1, 0xFF000000) => 0,
+                        (0, 0xFF00) | (1, 0xFF0000) => 1,
+                        (0, 0xFF0000) | (1, 0xFF00) => 2,
+                        (0, 0xFF000000) | (1, 0xFF) => 3,
+                        _ => return None,
+                    }
+                };
+            }
+            let red_offset = channel_offset!(red_mask);
+            let green_offset = channel_offset!(green_mask);
+            let blue_offset = channel_offset!(blue_mask);
+            let alpha_offset = channel_offset!(!(red_mask | green_mask | blue_mask));
+
+            // Wrap the pixel buffer into a slice
+            let size = (bytes_per_line * height) as usize;
+            let data = slice::from_raw_parts((*self.handle).data as *const u8, size);
+
+            // Finally, generate the image object
+            Some(RgbaImage::from_fn(width as u32, height as u32, |x, y| {
+                macro_rules! subpixel {
+                    ($channel_offset:ident) => {
+                        data[(y * bytes_per_line as u32 + x * stride as u32 + $channel_offset)
+                            as usize]
+                    };
+                }
+                Rgba::from_channels(
+                    subpixel!(red_offset),
+                    subpixel!(green_offset),
+                    subpixel!(blue_offset),
+                    // Make the alpha channel fully opaque if none is provided
+                    if depth == 24 {
+                        0xFF
+                    } else {
+                        subpixel!(alpha_offset)
+                    },
+                )
+            }))
+        }
+    }
+}
+
+impl Drop for Image {
+    fn drop(&mut self) {
+        unsafe {
+            xlib::XDestroyImage(self.handle);
+        }
+    }
+}
+
+impl<'a> Iterator for ScreenRectIter<'a> {
+    type Item = util::Rect;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        if self.i >= self.crtcs.len() {
+            return None;
+        }
+
+        unsafe {
+            // TODO Handle failure here?
+            let crtc = xrandr::XRRGetCrtcInfo((*self.dpy).handle, self.res, self.crtcs[self.i]);
+            let x = (*crtc).x;
+            let y = (*crtc).y;
+            let w = (*crtc).width;
+            let h = (*crtc).height;
+            xrandr::XRRFreeCrtcInfo(crtc);
+
+            self.i += 1;
+
+            //Some((w as i32, h as i32, x as i32, y as i32))
+            Some(util::Rect {
+                x: x,
+                y: y,
+                w: w as i32,
+                h: h as i32,
+            })
+        }
+    }
+}
+
+impl<'a> Drop for ScreenRectIter<'a> {
+    fn drop(&mut self) {
+        unsafe {
+            xrandr::XRRFreeScreenResources(self.res);
+        }
+    }
+}
+
+pub fn parse_geometry(g: ffi::CString) -> util::Rect {
+    unsafe {
+        let mut x = 0;
+        let mut y = 0;
+        let mut w = 0;
+        let mut h = 0;
+        xlib::XParseGeometry(
+            g.as_ptr() as *const raw::c_char,
+            &mut x,
+            &mut y,
+            &mut w,
+            &mut h,
+        );
+
+        util::Rect {
+            x: x,
+            y: y,
+            w: w as i32,
+            h: h as i32,
+        }
+    }
+}

data/lib/shotgun_ruby.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+begin
+  RUBY_VERSION =~ /(\d+\.\d+)/
+  require "shotgun_ruby/#{$1}/shotgun_ruby"
+rescue LoadError
+  require "shotgun_ruby/shotgun_ruby"
+end
+
+module ShotgunRuby
+  class Error < StandardError; end
+  # Your code goes here...
+end

metadata

@@ -0,0 +1,51 @@
+--- !ruby/object:Gem::Specification
+name: shotgun_ruby
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Jonas Moesicke
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2023-04-14 00:00:00.000000000 Z
+dependencies: []
+description: A ruby gem adaptation of the shotgun screenshot tool for linux. Creates
+  a .pam screenshots at a given path and returns the location.
+email:
+- jmoesicke@clustermarket.com
+executables: []
+extensions:
+- ext/shotgun_ruby/extconf.rb
+extra_rdoc_files: []
+files:
+- ext/shotgun_ruby/Cargo.lock
+- ext/shotgun_ruby/Cargo.toml
+- ext/shotgun_ruby/extconf.rb
+- ext/shotgun_ruby/src/lib.rs
+- ext/shotgun_ruby/src/util.rs
+- ext/shotgun_ruby/src/xwrap.rs
+- lib/shotgun_ruby.rb
+homepage: https://github.com/jm591/shotgun_ruby
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.8
+signing_key: 
+specification_version: 4
+summary: Rust Screenshot Tool
+test_files: []