forklaunch 1.20.1

Launch faster with forklaunch
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

use action::ActionCommand;
use anyhow::Result;
use clap::{ArgMatches, Command};

use crate::{CliCommand, core::command::command};

mod action;

// ── Top-level command ─────────────────────────────────────────────────────────

#[derive(Debug)]
pub(crate) struct WorkerCommand {
    pause: ActionCommand,
    resume: ActionCommand,
    restart: ActionCommand,
}

impl WorkerCommand {
    pub(crate) fn new() -> Self {
        Self {
            pause: ActionCommand::new("pause", "Pause a running worker", "pause", "paused"),
            resume: ActionCommand::new("resume", "Resume a paused worker", "resume", "resumed"),
            restart: ActionCommand::new(
                "restart",
                "Restart a worker (force a new deployment)",
                "restart",
                "restarted",
            ),
        }
    }
}

impl CliCommand for WorkerCommand {
    fn command(&self) -> Command {
        command("worker", "Pause, resume, or restart a deployed worker")
            .subcommand_required(true)
            .subcommand(self.pause.command())
            .subcommand(self.resume.command())
            .subcommand(self.restart.command())
    }

    fn handler(&self, matches: &ArgMatches) -> Result<()> {
        match matches.subcommand() {
            Some(("pause", sub_matches)) => self.pause.handler(sub_matches),
            Some(("resume", sub_matches)) => self.resume.handler(sub_matches),
            Some(("restart", sub_matches)) => self.restart.handler(sub_matches),
            _ => unreachable!(),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn worker_cmd() -> Command {
        WorkerCommand::new().command().version("0.0.0-test")
    }

    #[test]
    fn command_definition_is_valid() {
        worker_cmd().debug_assert();
    }

    #[test]
    fn requires_a_subcommand() {
        assert!(worker_cmd().try_get_matches_from(["worker"]).is_err());
    }

    #[test]
    fn pause_requires_worker_id() {
        assert!(worker_cmd().try_get_matches_from(["worker", "pause"]).is_err());
        assert!(
            worker_cmd()
                .try_get_matches_from(["worker", "pause", "worker-1"])
                .is_ok()
        );
    }

    #[test]
    fn resume_requires_worker_id() {
        assert!(
            worker_cmd()
                .try_get_matches_from(["worker", "resume", "worker-1"])
                .is_ok()
        );
    }

    #[test]
    fn restart_requires_worker_id() {
        assert!(
            worker_cmd()
                .try_get_matches_from(["worker", "restart", "worker-1"])
                .is_ok()
        );
    }
}