1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use super::Image;
use crate::Capturer;
use std::ptr;
use x11::xlib::{XAllPlanes, XCloseDisplay, XDefaultRootWindow, XGetImage, XOpenDisplay, ZPixmap};

impl Capturer {
  pub fn capture_screen(&self) -> Option<Image> {
    unsafe {
      let display_ptr = XOpenDisplay(ptr::null_mut());

      if display_ptr.is_null() {
        return None;
      }

      let window_id = XDefaultRootWindow(display_ptr);
      let display_info = self.display_info;

      let ximage = XGetImage(
        display_ptr,
        window_id,
        display_info.x,
        display_info.y,
        display_info.width,
        display_info.height,
        XAllPlanes(),
        ZPixmap,
      );

      if ximage.is_null() {
        return None;
      }

      XCloseDisplay(display_ptr);

      Some(Image::new(ximage))
    }
  }
}