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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
extern crate serde_json;
use regex::Regex;
use serde::{Deserialize, Serialize};
pub mod io;
use io::*;
#[cfg(test)]
mod tests;
#[derive(Serialize, Deserialize, Debug)]
struct I3StatusItem {
name: String,
instance: Option<String>,
markup: String,
color: Option<String>,
full_text: String,
}
fn begin_io<IO : Io>(io: &mut IO) -> std::io::Result<()> {
let line = io.read_line()?;
io.write_line(&line)?;
assert!(line == "{\"version\":1}\n");
let line = io.read_line()?;
io.write_line(&line)?;
assert!(line == "[\n");
return Ok(());
}
pub fn begin() -> std::io::Result<StdIo> {
let mut io = StdIo::new();
begin_io(&mut io)?;
return Ok(io);
}
pub fn begin_str<'a>( input : &'a String) -> std::io::Result<StringInStdOut<'a>> {
let mut io = StringInStdOut::from_string(&input);
begin_io(&mut io)?;
return Ok(io);
}
pub fn update<IO: Io>(
io: &mut IO,
name: &str,
position: usize,
reverse: bool,
what: &str,
) -> std::io::Result<()> {
let mut line = io.read_line()?;
assert!(line != "{\"version\":1}");
assert!(line != "[");
if line.chars().next().unwrap() == ',' {
line.remove(0);
io.write_line(",")?;
}
match serde_json::from_str(&line) {
Ok(i) => {
let mut items: Vec<I3StatusItem> = i;
let w: I3StatusItem = I3StatusItem {
full_text: what.to_string(),
markup: "none".to_string(),
name: name.to_string(),
instance: None,
color: None,
};
if reverse {
items.insert(items.len() - 1 - position, w);
} else {
items.insert(position, w);
}
io.write_line(&format_json(format!("{:?}", items)))?;
}
_ => io.write_line(&line)?,
}
Ok(())
}
fn format_json(line: String) -> String {
let line = line
.replace("I3StatusItem", "")
.replace(", color: None", "")
.replace(", instance: None", "")
.replace(", full_text:", ",\"full_text\":")
.replace(", instance:", ",\"instance\":")
.replace(", color:", ",\"color\":")
.replace(", markup:", ",\"markup\":")
.replace("{ name:", "{\"name\":");
let re = Regex::new(r"Some\((?P<v>[^\)]*)\)").unwrap();
re.replace_all(&line, "$v").to_owned().to_string()
}