polygraph/main.cpp at main · kuyanov/polygraph · GitHub

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
#include <cstring>
#include <functional>
#include <iostream>

#include "clean.h"
#include "config_get.h"
#include "config_set.h"
#include "helpers.h"
#include "run.h"
#include "runner_start.h"
#include "runner_stop.h"
#include "start.h"
#include "stop.h"

template <class Options>
void InvokeWithOptions(int argc, char **argv, const std::function<void(Options)> &func) {
    Options options;
    options.Init(argc, argv);
    func(options);
}

void ParseRunnerAction(int argc, char **argv) {
    if (argc < 2 || strcmp(argv[1], "--help") == 0) {
        std::cerr << "Manage polygraph runners." << std::endl;
        std::cerr << std::endl;
        std::cerr << "Usage:  " << "polygraph runner ACTION [OPTIONS]" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Actions:" << std::endl;
        std::cerr << "  start   " << "Start runners" << std::endl;
        std::cerr << "  stop    " << "Stop runners" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Run 'polygraph runner ACTION --help' for more information about the action."
                  << std::endl;
        exit(EXIT_FAILURE);
    }
    if (strcmp(argv[1], "start") == 0) {
        InvokeWithOptions<RunnerStartOptions>(argc - 1, argv + 1, RunnerStart);
    } else if (strcmp(argv[1], "stop") == 0) {
        InvokeWithOptions<RunnerStopOptions>(argc - 1, argv + 1, RunnerStop);
    } else {
        std::cerr << "Unknown action '" << argv[1] << "'." << std::endl;
        std::cerr << "See 'polygraph runner --help'." << std::endl;
        exit(EXIT_FAILURE);
    }
}

void ParseConfigAction(int argc, char **argv) {
    if (argc < 2 || strcmp(argv[1], "--help") == 0) {
        std::cerr << "Manage polygraph configuration file." << std::endl;
        std::cerr << std::endl;
        std::cerr << "Usage:  " << "polygraph config ACTION [OPTIONS]" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Actions:" << std::endl;
        std::cerr << "  get   " << "Show current configuration" << std::endl;
        std::cerr << "  set   " << "Update configuration file" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Run 'polygraph config ACTION --help' for more information about the action."
                  << std::endl;
        exit(EXIT_FAILURE);
    }
    if (strcmp(argv[1], "get") == 0) {
        InvokeWithOptions<ConfigGetOptions>(argc - 1, argv + 1, ConfigGet);
    } else if (strcmp(argv[1], "set") == 0) {
        InvokeWithOptions<ConfigSetOptions>(argc - 1, argv + 1, ConfigSet);
    } else {
        std::cerr << "Unknown action '" << argv[1] << "'." << std::endl;
        std::cerr << "See 'polygraph config --help'." << std::endl;
        exit(EXIT_FAILURE);
    }
}

void ParseCommand(int argc, char **argv) {
    if (argc < 2 || strcmp(argv[1], "--help") == 0) {
        std::cerr
            << "Welcome to polygraph, a service for running workflows efficiently and securely."
            << std::endl;
        std::cerr << std::endl;
        std::cerr << "Usage:  " << "polygraph COMMAND [ACTION] [OPTIONS]" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Commands:" << std::endl;
        std::cerr << "  clean    " << "Cleanup workflow data" << std::endl;
        std::cerr << "  config   " << "Manage config" << std::endl;
        std::cerr << "  run      " << "Run a workflow" << std::endl;
        std::cerr << "  runner   " << "Manage runners" << std::endl;
        std::cerr << "  start    " << "Start polygraph service" << std::endl;
        std::cerr << "  stop     " << "Stop polygraph service" << std::endl;
        std::cerr << std::endl;
        std::cerr << "Run 'polygraph COMMAND --help' for more information about the command."
                  << std::endl;
        exit(EXIT_FAILURE);
    }
    if (strcmp(argv[1], "clean") == 0) {
        InvokeWithOptions<CleanOptions>(argc - 1, argv + 1, Clean);
    } else if (strcmp(argv[1], "config") == 0) {
        ParseConfigAction(argc - 1, argv + 1);
    } else if (strcmp(argv[1], "run") == 0) {
        InvokeWithOptions<RunOptions>(argc - 1, argv + 1, Run);
    } else if (strcmp(argv[1], "runner") == 0) {
        ParseRunnerAction(argc - 1, argv + 1);
    } else if (strcmp(argv[1], "start") == 0) {
        InvokeWithOptions<StartOptions>(argc - 1, argv + 1, Start);
    } else if (strcmp(argv[1], "stop") == 0) {
        InvokeWithOptions<StopOptions>(argc - 1, argv + 1, Stop);
    } else {
        std::cerr << "Unknown command '" << argv[1] << "'." << std::endl;
        std::cerr << "See 'polygraph --help'." << std::endl;
        exit(EXIT_FAILURE);
    }
}

int main(int argc, char **argv) {
    CreateDirs();
    ParseCommand(argc, argv);
}