sci/src/cli.c

/**
 * sci - a simple ci system
   Copyright (C) 2024 Asger Gitz-Johansen

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
#include "cli.h"
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

cli_options new_options() {
    cli_options result;
    char* config_file = getenv("SCI_CONFIG_FILE");
    result.config_file.has_value = config_file != NULL;
    result.config_file.value = config_file;

    result.executors = 32;
    char* verbosity_env = getenv("SCI_VERBOSITY");
    int verbosity = 1;
    if(verbosity_env != NULL)
        verbosity = atoi(verbosity_env);
    result.verbosity = verbosity;

    result.help = false;
    result.version = false;

    char* no_color = getenv("NO_COLOR");
    bool color = true;
    if(no_color != NULL && no_color[0] != '\0')
        color = false;
    result.use_colors = color;

    char* log_file = getenv("SCI_LOG_file");
    result.log_file.has_value = log_file != NULL;
    result.log_file.value = log_file;

    char* pipeline_log_dir = getenv("SCI_PIPELINE_LOG_DIR");
    result.pipeline_log_dir.has_value = pipeline_log_dir != NULL;
    result.pipeline_log_dir.value = pipeline_log_dir;

    char* pipeline_cwd = getenv("SCI_PIPELINE_CWD");
    result.pipeline_cwd.has_value = pipeline_cwd != NULL;
    result.pipeline_cwd.value = pipeline_cwd;

    char* environment_vars = getenv("SCI_PIPELINE_ENV_VARS");
    if(environment_vars == NULL) {
        result.environment_vars.has_value = false;
        result.environment_vars.value = NULL;
    } else {
        char* tok = strtok(environment_vars, ";");
        result.environment_vars.has_value = true;
        result.environment_vars.value = create_strlist_node(tok);
		tok = strtok(NULL, ";");
        while(tok != NULL)
            add_str(tok, result.environment_vars.value);
    }
    return result;
}

void destroy_options(cli_options v) {
    if(v.config_file.has_value)
        free(v.config_file.value);
    if(v.log_file.has_value)
        free(v.log_file.value);
    if(v.pipeline_log_dir.has_value)
        free(v.pipeline_log_dir.value);
    if(v.environment_vars.has_value)
        clear_strlist(v.environment_vars.value);
}

//                                                         <max
const char* optstring = "f:L:P:w:v:Cl:e:hV";
const char* help_msg = 
    "%s %s\n"
    "Usage: [-f file] [-L dir] [-P dir] [-w count]\n\n"
    "       [-v level] [-C] [-l file] [-e ENV] [-h] [-V]\n"
    "\n"
    SCI_DESCRIPTION "\n"
    "\n"
    "OPTIONS:\n"
    "  -f file     Set sci config file\n"
    "  -L dir      Set pipeline log output directory\n"
    "  -P dir      Set pipeline working directory prefix\n"
    "  -w count    Set the amount of worker threads\n"
    "  -v level    Set verbosity level [0-4]\n"
    "  -C          Force color output, ignoring $NO_COLOR\n"
    "  -l file     Set sci's log to output to a file\n"
    "  -e ENV      Pass an env variable to pipelines\n"
    "  -h          Show this message and exit\n"
    "  -V          Show version and exit\n"
    "\n"
    "Most options can also be provided as env variables.\n"
    "See sci(1) for more details.\n"
    ;
//                                                         <max

void print_help(FILE * out, char* prog_name) {
    fprintf(out, help_msg, prog_name, SCI_VERSION);
}

cli_options parse(int argc, char** argv) {
    cli_options options = new_options();
    int opt;
    while((opt = getopt(argc, argv, optstring)) != -1) {
        switch(opt) {
            case 'f':
                options.config_file.value = strdup(optarg);
                options.config_file.has_value = true;
                break;
            case 'L':
                options.pipeline_log_dir.value = strdup(optarg);
                options.pipeline_log_dir.has_value = true;
                break;
            case 'P':
                options.pipeline_cwd.value = strdup(optarg);
                options.pipeline_cwd.has_value = true;
                break;
            case 'v':
                options.verbosity = atoi(optarg);
                break;
            case 'w':
                options.executors = atoi(optarg);
                break;
            case 'C':
                options.use_colors = true;
                break;
            case 'l':
                options.log_file.value = strdup(optarg);
                options.log_file.has_value = true;
                break;
            case 'V':
                options.version = true;
                break;
            case 'h':
                options.help = true;
                break;
            case 'e':
                if(!options.environment_vars.has_value) {
                    options.environment_vars.has_value = true;
                    options.environment_vars.value = create_strlist_node(optarg);
                } else
                    add_str(optarg, options.environment_vars.value);
                break;
            default:
                print_help(stderr, argv[0]);
                exit(EXIT_FAILURE);
        }
    }
    return options;
}