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
|
use anyhow::{anyhow, bail, Result};
use std::{
io::{BufRead, BufReader, BufWriter, Write},
net::{TcpStream, ToSocketAddrs},
};
use crate::Serial;
pub struct Client {
wsock: BufWriter<TcpStream>,
rsock: BufReader<TcpStream>,
}
impl Client {
pub fn new(address: impl ToSocketAddrs, secret: &str) -> Result<Self> {
let mut sock = TcpStream::connect(address)?;
writeln!(sock, "{secret}")?;
Ok(Self {
rsock: BufReader::new(sock.try_clone()?),
wsock: BufWriter::new(sock),
})
}
pub fn list(&mut self) -> Result<Vec<(i64, u64, Serial)>> {
writeln!(self.wsock, "list")?;
self.wsock.flush()?;
let mut line = String::new();
let mut out = Vec::new();
loop {
self.rsock.read_line(&mut line)?;
if line.trim().is_empty() || line.ends_with("\n\n") {
break Ok(out);
}
check_error(&line)?;
let (mtime, rest) = line.trim().split_once(":").ok_or(anyhow!("size missing"))?;
let (size, serial) = rest.split_once(":").ok_or(anyhow!("serial missing"))?;
out.push((mtime.parse()?, size.parse()?, serial.parse()?));
}
}
pub fn quit(mut self) -> Result<()> {
writeln!(self.wsock, "quit")?;
self.wsock.flush()?;
Ok(())
}
}
fn check_error(line: &str) -> Result<()> {
if let Some(message) = line.trim().strip_prefix("error") {
bail!("server error: {message}")
}
Ok(())
}
|
