cmfy 0.4.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use std::fmt::{Display, Formatter};

use colored::{ColoredString, Colorize};

#[derive(Debug, Clone)]
pub struct WithStatus<I, O> {
    pub inner: I,
    pub status: Status<O>,
}

#[derive(Debug, Clone)]
pub enum Status<O> {
    Completed(O),
    Pending,
    Running,
    Cancelled,
}

pub trait MarkAs {
    fn mark_as<O>(self, status: Status<O>) -> WithStatus<Self, O>
    where
        Self: Sized,
    {
        let inner = self;
        WithStatus { inner, status }
    }
}

impl<O> Display for Status<O> {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        use Status::*;
        match self {
            Completed(_) => write!(f, "completed"),
            Pending => write!(f, "pending"),
            Running => write!(f, "running"),
            Cancelled => write!(f, "cancelled"),
        }
    }
}

impl<O> Status<O> {
    pub fn map<U, F>(self, f: F) -> Status<U>
    where
        F: Fn(O) -> U,
    {
        use Status::*;
        match self {
            Completed(output) => Completed(f(output)),
            Pending => Pending,
            Running => Running,
            Cancelled => Cancelled,
        }
    }

    pub fn colored(&self) -> ColoredString {
        match self {
            Status::Completed(_) => self.to_string().green(),
            Status::Pending => self.to_string().yellow(),
            Status::Running => self.to_string().blue(),
            Status::Cancelled => self.to_string().red(),
        }
    }
}