libhv/main.cpp.tmpl

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "h.h"
#ifdef _WIN32
#include "win32_getopt.h"
#endif

void pexit(int code) {
    printf("Exit with code %d.\n", code);
    exit(code);
}

typedef struct arg_s {
    string  cmd;
    string  confile;    // -c
    string  logfile;
    int     port;       // -p
    int     daemon;     // -d
} arg_t;
arg_t g_arg;

const char* options = "hvc:p:d";
const char* options_descr = "\
Options:\n\
-h                      Print help\n\
-v                      Print version\n\
-c confile              Configure file\n\
-p port                 Listen port\n\
-d                      Daemon\n\
";

string version() {
    return asprintf("%s version %s", g_arg.cmd.c_str(), get_compile_version());
}

string usage() {
    return asprintf("Usage: %s [%s]", g_arg.cmd.c_str(), options);
}

void print_help() {
    puts(usage().c_str());
    puts(options_descr);
    puts(version().c_str());
}

bool parse_command_line(int argc, char** argv) {
    int opt = -1;
    while ((opt = getopt(argc, argv, options)) != -1) {
        switch (opt) {
            case 'h': {
                print_help();
                pexit(0);
            }
            break;
            case 'v': {
                puts(version().c_str());
                pexit(0);
            }
            break;
            case 'c': {
                g_arg.confile = optarg;
            }
            break;
            case 'p': {
                g_arg.port = atoi(optarg);
            }
            break;
            case 'd': {
                g_arg.daemon = 1;
            }
            break;
            default: {
                puts("Unrecognized option!");
                pexit(-10);
            }
            break;
        }
    }

    return true;
}

int main(int argc, char** argv) {
    string strCmd((const char*)argv[0]);
    g_arg.cmd = filename(strCmd);
    //g_arg.confile = g_arg.cmd + ".conf";
    g_arg.logfile = g_arg.cmd + ".log";
    g_arg.port = 0;
    g_arg.daemon = 0;

    hlog_set_file(g_arg.logfile.c_str());
    hlog_set_level(LOG_LEVEL_INFO);
    hlogi("%s", version().c_str());

    if (!parse_command_line(argc, argv)) {
        printf("Parse command line error, please use -h to get help.\n");
        pexit(-10);
    }

    IniParser ini;
    if (!g_arg.confile.empty()) {
        int iRet = ini.LoadFromFile(g_arg.confile.c_str());
        if (iRet != 0) {
            printf("Load confile [%s] failed: %d\n", g_arg.confile.c_str(), iRet);
            pexit(-20);
        }
    }

    if (g_arg.port == 0) {
        g_arg.port = atoi(ini.GetValue("port").c_str());
    }
    if (g_arg.port == 0) {
        puts("Please config listen port!");
        pexit(-30);
    }

#ifdef __unix__
    if (g_arg.daemon) {
        daemon(1, 1);
    }
#endif

    pexit(0);
}