cmfy 0.3.0

A CLI companion app for Comfy UI
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

use cmfy::Result;
use serde::{de::DeserializeOwned, Serialize};
use std::io::Write;

pub use clio::{Input, Output};

pub trait JsonWrite {
    fn write_json(&mut self, value: &impl Serialize, pretty: bool) -> Result<()>;
}

impl JsonWrite for Output {
    fn write_json(&mut self, value: &impl Serialize, pretty: bool) -> Result<()> {
        let mut writer = self.lock();
        if pretty {
            serde_json::to_writer_pretty(&mut writer, &value)?;
        } else {
            serde_json::to_writer(&mut writer, &value)?;
        }
        writer.flush()?;
        Ok(())
    }
}

pub trait JsonRead {
    fn read_json<T: DeserializeOwned>(&mut self) -> Result<T>;
}

impl JsonRead for Input {
    fn read_json<T: DeserializeOwned>(&mut self) -> Result<T> {
        let mut read = self.lock();
        Ok(serde_json::from_reader(&mut read)?)
    }
}