63 lines
2.6 KiB
C++
63 lines
2.6 KiB
C++
#include <cstring>
|
|
#include <iostream>
|
|
#include <unistd.h>
|
|
#include "str_utils.h"
|
|
|
|
const char* VERSION = "0.0.1-DEV";
|
|
|
|
bool shortHand(const char* str, char shortHand);
|
|
|
|
int main(const int argc, char** argv) {
|
|
|
|
if (argc > 1) {
|
|
for (int i = 1; i < argc; i++) {
|
|
const char* arg = argv[i];
|
|
if (strcmp(arg, "--help") == 0 || shortHand(arg, 'h')) {
|
|
std::cout << "Server Arduino Watchdog version '" << VERSION << "' by Zacharias (c) 2025" << std::endl
|
|
<< std::endl
|
|
<< "Options:" << std::endl
|
|
<< "[requires root] -short --long-option Description" << std::endl
|
|
<< "[X] --init Initializes the Server end of this system, find the arduino and creates systemd services" << std::endl
|
|
<< " -h --help Prints this help message" << std::endl
|
|
<< " -b --heartbeat Sends a heartbeat to the arduino, Intended to be send by systemd unit and timer" << std::endl
|
|
<< " -i --info Queries the Arduino Watchdog for status, like time before time-out" << std::endl
|
|
<< " -v --version Print's the version" << std::endl
|
|
;
|
|
return 0;
|
|
}
|
|
else if (strcmp(arg, "--init") == 0) {
|
|
if (getuid() != 0) {
|
|
std::cerr << "Initialization requires root!" << std::endl;
|
|
return -1;
|
|
}
|
|
std::cout << "Initialization :)" << std::endl;
|
|
// TODO: Add init system
|
|
}
|
|
else if (strcmp(arg, "--heartbeat") == 0 || shortHand(arg, 'b')) {
|
|
// TODO: Send heartbeat
|
|
}
|
|
else if (strcmp(arg, "--info") == 0 || shortHand(arg, 'i')) {
|
|
// TODO: Send info
|
|
// TODO: Print response status
|
|
}
|
|
else if (strcmp(arg, "--version") == 0 || shortHand(arg, 'v')) {
|
|
std::cout << VERSION << " by Zacharias (c) 2025" << std::endl;
|
|
}
|
|
else {
|
|
std::cerr << "Unknown option '" << arg << "'" << std::endl << "Test '" << argv[0] << " --help'" << std::endl;
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
std::cerr << "Missing arguments, try " << argv[0] << " --help" << std::endl;
|
|
return -1;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
bool shortHand(const char* str, char shortHand) {
|
|
if (str == nullptr) return false;
|
|
if (shortHand == 0) return false;
|
|
return startWith(str, '-') && contains(str, shortHand);
|
|
} |