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_dummy() -> std::io::Result<StdIo> {
return Ok(StdIo::new());
}
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()
}