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

use crate::{MasterConfig, Worker};

use rusty_yaml::Yaml;
use std::fmt::{Display, Error, Formatter};


pub struct Makefile {
    master: MasterConfig,
    workers: Vec<Worker>,
}


impl Makefile {
    pub fn new(master: MasterConfig, workers: Vec<Worker>) -> Self {
        Self { master, workers }
    }
}


impl From<Yaml> for Makefile {
    fn from(yaml: Yaml) -> Self {
        let mut workers = vec![];
        for worker in yaml.get_section("workers") {
            workers.push(Worker::from(worker));
        }

        let master = MasterConfig::from(yaml);
        Self::new(master, workers)
    }
}

impl Display for Makefile {
    fn fmt(&self, f: &mut Formatter) -> Result<(), Error> {
        write!(
            f,
            "
workerdirs := {worker_dirs}
masterdir := \"master\"


all: start

build: install_deps
\t. venv/bin/activate; buildbot create-master $(masterdir);
\tpython3 -c \"print(\\\"{master_setup}\\\")\" > $(masterdir)/master.cfg
\t{workers_setup}

start: build
\t. venv/bin/activate; buildbot stop master; buildbot reconfig master; buildbot cleanupdb master; buildbot start master;
\t-. venv/bin/activate; $(foreach dir,$(workerdirs),buildbot-worker restart $(dir);)


install_deps:
\tsudo apt-get install python3-dev -y
\tsudo apt-get install python3-pip -y
\tsudo apt-get install python3-venv -y

\tpython3 -m venv venv
\t. venv/bin/activate; python3 -m pip install -U pip; python3 -m pip install 'buildbot[bundle]';
\t. venv/bin/activate; python3 -m pip install buildbot-worker setuptools-trial

",
            master_setup = self
                .master
                .to_string()
                .replace("\n", "\\n")
                .replace("\"", "'"),
            worker_dirs = self
                .workers
                .iter()
                .map(|w| w.get_dir())
                .collect::<Vec<String>>()
                .join(" "),
            workers_setup = self
                .workers
                .iter()
                .map(|w| format!(
                    "
\tmkdir -p {worker_dir}
\t-. venv/bin/activate; buildbot-worker create-worker {worker_dir} localhost {worker_dir} pass
\tpython3 -c \"print(\\\"{script}\\\")\" > {worker_dir}/buildbot.tac",
                    worker_dir = w.get_dir(),
                    script = w.to_string().replace("\n", "\\n").replace("\"", "'")
                ))
                .collect::<Vec<String>>()
                .join("\n\n")
        )
    }
}