cmfy 0.1.1

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
34
35
36
37
38
39
40
41
42
43
44
45
46

use cmfy::{Error, Result};
use serde::de::DeserializeOwned;
use std::{
    ffi::OsString,
    fs::File,
    io::{stdin, Read},
    path::PathBuf,
};

#[derive(Clone, Debug, Default)]
pub struct Input(Option<PathBuf>);

impl ToString for Input {
    fn to_string(&self) -> String {
        if let Some(path) = &self.0 {
            path.display().to_string()
        } else {
            "<stdin>".into()
        }
    }
}

impl From<OsString> for Input {
    fn from(value: OsString) -> Self {
        Self(Some(PathBuf::from(value)))
    }
}

impl TryInto<Box<dyn Read>> for Input {
    type Error = Error;

    fn try_into(self) -> Result<Box<dyn Read>> {
        if let Some(path) = self.0 {
            Ok(Box::new(File::open(path)?))
        } else {
            Ok(Box::new(stdin()))
        }
    }
}

impl Input {
    pub fn read_json<T: DeserializeOwned>(self) -> Result<T> {
        let reader: Box<dyn Read> = self.try_into()?;
        Ok(serde_json::from_reader(reader)?)
    }
}