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
|
use std::collections::HashSet;
use std::io;
use std::sync::{Mutex, OnceLock};
use crate::protocol::render_ansi;
static RECEIVED_KITTY_GRAPHICS_IDS: OnceLock<Mutex<HashSet<u32>>> = OnceLock::new();
pub(super) fn write_encoded_frame_with_graphics(
mut writer: impl io::Write,
encoded: &[u8],
graphics: &[u8],
) -> io::Result<()> {
if graphics.is_empty() {
return writer.write_all(encoded);
}
let insertion = render_ansi::final_sync_output_end(encoded).unwrap_or(encoded.len());
writer.write_all(&encoded[..insertion])?;
record_received_kitty_graphics(graphics);
writer.write_all(b"\x1b7")?;
writer.write_all(graphics)?;
writer.write_all(b"\x1b8")?;
writer.write_all(&encoded[insertion..])
}
pub(super) fn contains_kitty_graphics_bytes(bytes: &[u8]) -> bool {
bytes.windows(3).any(|window| window == b"\x1b_G")
}
pub(super) fn record_received_kitty_graphics(bytes: &[u8]) {
let ids = kitty_graphics_image_ids(bytes);
if ids.is_empty() {
return;
}
let set = RECEIVED_KITTY_GRAPHICS_IDS.get_or_init(|| Mutex::new(HashSet::new()));
if let Ok(mut set) = set.lock() {
set.extend(ids);
}
}
pub(super) fn clear_received_kitty_graphics(mut writer: impl io::Write) -> io::Result<()> {
let Some(set) = RECEIVED_KITTY_GRAPHICS_IDS.get() else {
return Ok(());
};
let Ok(mut set) = set.lock() else {
return Ok(());
};
for id in set.drain() {
write!(writer, "\x1b_Ga=d,d=I,i={id},q=2;\x1b\\")?;
}
writer.flush()
}
pub(super) fn kitty_graphics_image_ids(bytes: &[u8]) -> Vec<u32> {
let mut ids = Vec::new();
let mut index = 0usize;
while let Some(start) = find_subslice(&bytes[index..], b"\x1b_G") {
let command_start = index + start + 3;
let Some(end) = find_subslice(&bytes[command_start..], b"\x1b\\") else {
break;
};
let command = &bytes[command_start..command_start + end];
if let Some(id) = kitty_graphics_command_image_id(command) {
ids.push(id);
}
index = command_start + end + 2;
}
ids
}
fn kitty_graphics_command_image_id(command: &[u8]) -> Option<u32> {
let header_end = command
.iter()
.position(|byte| *byte == b';')
.unwrap_or(command.len());
for part in command[..header_end].split(|byte| *byte == b',') {
let Some(value) = part.strip_prefix(b"i=") else {
continue;
};
let text = std::str::from_utf8(value).ok()?;
if let Ok(id) = text.parse::<u32>() {
return Some(id);
}
}
None
}
fn find_subslice(haystack: &[u8], needle: &[u8]) -> Option<usize> {
if needle.is_empty() || needle.len() > haystack.len() {
return None;
}
haystack
.windows(needle.len())
.position(|window| window == needle)
}
|
