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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use std::process::exit;
use crate::alfred::{Alfred, AlfredEnv};
use crate::workflow_item::WorkflowItem;
use serde::{Deserialize, Serialize};
use crate::icon::BuiltinIcon;

// Alfred workflow object
#[derive(Serialize, Deserialize)]
pub struct AlfredWorkflow {
    #[serde(skip_serializing)]
    alfred: Alfred,
    items: Vec<WorkflowItem>,
}

impl AlfredWorkflow {
    pub fn init() -> AlfredWorkflow {
        AlfredWorkflow {
            alfred: Alfred::init(),
            items: Vec::new(),
        }
    }

    pub fn get_workflow_env(key: &str) -> String {
        std::env::var(key).unwrap_or_default()
    }

    pub fn send_feedback(&self) {
        print!("{}", serde_json::to_string(self).unwrap())
    }

    pub fn add_item(mut self, item: WorkflowItem) -> AlfredWorkflow {
        self.items.push(item);
        self
    }

    // pub fn add_item2(&mut self, item: WorkflowItem) {
    //     let vec = &mut self.items;
    //     vec.push(item);
    // }

    pub fn fatal_error(&mut self, msg: &str, detail: &str) {
        let error = WorkflowItem::new(msg)
            .subtitle(detail)
            .icon(BuiltinIcon::ERROR.get_icon());
        self.items = vec![error];
        self.send_feedback();
        exit(0)
    }

    pub fn get_config(&self, name: &str) -> String {
        std::env::var(name).unwrap_or_default()
    }
}

impl AlfredEnv for AlfredWorkflow {
    fn get_preference_path(&self) -> String {
        self.alfred.get_preference_path()
    }

    fn get_preference_hash_path(&self) -> String {
        self.alfred.get_preference_hash_path()
    }

    fn get_theme(&self) -> String {
        self.alfred.get_theme()
    }

    fn get_version(&self) -> String {
        self.alfred.get_version()
    }

    fn get_version_build(&self) -> String {
        self.alfred.get_version_build()
    }

    fn get_workflow_bundle_id(&self) -> String {
        self.alfred.get_workflow_bundle_id()
    }

    fn get_workflow_cache_path(&self) -> String {
        self.alfred.get_workflow_cache_path()
    }

    fn get_workflow_data_path(&self) -> String {
        self.alfred.get_workflow_data_path()
    }

    fn get_workflow_name(&self) -> String {
        self.alfred.get_workflow_name()
    }

    fn is_debug_mode(&self) -> bool {
        self.alfred.is_debug_mode()
    }

    fn get_workflow_uuid(&self) -> String {
        self.alfred.get_workflow_uuid()
    }

    fn get_workflow_version(&self) -> String {
        self.alfred.get_workflow_version()
    }

    fn get_theme_background(&self) -> String {
        self.alfred.get_theme_background()
    }

    fn get_theme_selection_background(&self) -> String {
        self.alfred.get_theme_selection_background()
    }

    fn get_theme_subtext(&self) -> String {
        self.alfred.get_theme_subtext()
    }
}