onionshot 0.3.1

Screenshot utility for Hyprland
Documentation 
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use std::{env::temp_dir, path::Path};

use chrono::{Datelike, Timelike};

use crate::{
    argparse::{ApplicationArgs, StorageMode},
    env::{ensure_screenshot_dir, screenshot_dir},
    external::{
        clipboard::copy_png,
        freeze::freeze_screen,
        grim::grim_with_geometry,
        hyprctl::{get_active_screen, get_active_window},
        notify::{
            notify_clipboard_save, notify_save_fail, notify_screenshot_save,
        },
        slurp::slurp_geometry,
    },
};

fn generate_image_name() -> String {
    let current_time = chrono::Local::now();
    format!(
        "{:04}-{:02}-{:02}-{:02}{:02}{:02}_onionshot.png",
        current_time.year(),
        current_time.month(),
        current_time.day(),
        current_time.hour(),
        current_time.minute(),
        current_time.second()
    )
}

fn save_image(from: &Path, to: &Path, mode: StorageMode) {
    match mode {
        StorageMode::FilesystemOnly => {
            if let Err(err) = std::fs::copy(from, to) {
                notify_save_fail(err);
            } else {
                notify_screenshot_save(to);
            }
        }
        StorageMode::ClipboardOnly => {
            if let Err(err) = copy_png(from) {
                notify_save_fail(err);
            } else {
                notify_clipboard_save();
            }
        }
        StorageMode::Both => {
            if let Err(err) = std::fs::copy(from, to) {
                notify_save_fail(err);
            } else {
                notify_screenshot_save(to);
            }
            if let Err(err) = copy_png(from) {
                notify_save_fail(err);
            } else {
                notify_clipboard_save();
            }
        }
    }
    _ = std::fs::remove_file(from);
}

pub fn fullscreen_shot(args: &ApplicationArgs) {
    ensure_screenshot_dir();
    let name = generate_image_name();
    let picpath = screenshot_dir().join(&name);
    let tmppath = temp_dir().join(&name);
    let geometry = get_active_screen();
    grim_with_geometry(&tmppath, geometry);
    save_image(&tmppath, &picpath, args.storage);
}

pub fn active_window_shot(args: &ApplicationArgs) {
    ensure_screenshot_dir();
    let name = generate_image_name();
    let picpath = screenshot_dir().join(&name);
    let tmppath = temp_dir().join(&name);

    let geometry = get_active_window();
    grim_with_geometry(&tmppath, geometry);
    save_image(&tmppath, &picpath, args.storage);
}

pub fn region_shot(args: &ApplicationArgs) {
    ensure_screenshot_dir();
    let name = generate_image_name();
    let picpath = screenshot_dir().join(&name);
    let tmppath = temp_dir().join(&name);

    if args.freeze {
        let f = freeze_screen();
        let Some(geometry) = slurp_geometry() else {
            return;
        };
        grim_with_geometry(&tmppath, geometry);
        drop(f);
        save_image(&tmppath, &picpath, args.storage);
    } else {
        let Some(geometry) = slurp_geometry() else {
            _ = std::fs::remove_file(&tmppath);
            return;
        };
        grim_with_geometry(&tmppath, geometry);
        save_image(&tmppath, &picpath, args.storage);
    }
}