// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/config/common/exceptions.h>
#include <unistd.h>
#include <sys/time.h>
#include <vespa/vespalib/util/signalhandler.h>
#include <vespa/vespalib/util/exceptions.h>
#include <vespa/defaults.h>
#include "config-handler.h"

#include <vespa/log/log.h>
LOG_SETUP("config-sentinel");

using namespace config;

constexpr std::chrono::milliseconds CONFIG_TIMEOUT_MS(3 * 60 * 1000);

static bool stop()
{
    return (vespalib::SignalHandler::INT.check() ||
            vespalib::SignalHandler::TERM.check());
}

int
main(int argc, char **argv)
{
    int c = getopt(argc, argv, "c:");
    if (c != 'c') {
        LOG(error, "Usage: %s -c <config-id>", argv[0]);
        EV_STOPPING("config-sentinel", "Bad arguments on command line");
        exit(EXIT_FAILURE);
    }

    const char *configId = strdup(optarg);

    const char *rootDir = getenv("ROOT");
    if (!rootDir) {
        rootDir = vespa::Defaults::vespaHome();
        LOG(warning, "ROOT is not set, using %s", rootDir);
        setenv("ROOT", rootDir, 1);
    }

    if (chdir(rootDir) == -1) {
        LOG(error, "Fatal: Cannot cd to $ROOT (%s)", rootDir);
        EV_STOPPING("config-sentinel", "Cannot cd to $ROOT");
        exit(EXIT_FAILURE);
    }

    EV_STARTED("config-sentinel");

    vespalib::SignalHandler::PIPE.ignore();
    vespalib::SignalHandler::TERM.hook();
    vespalib::SignalHandler::INT.hook();
    vespalib::SignalHandler::CHLD.hook();

    if (setenv("LC_ALL", "C", 1) != 0) {
        LOG(error, "Unable to set locale");
        exit(EXIT_FAILURE);
    }
    setlocale(LC_ALL, "C");

    sentinel::ConfigHandler handler;

    LOG(debug, "Reading configuration");
    try {
        handler.subscribe(configId, CONFIG_TIMEOUT_MS);
    } catch (ConfigTimeoutException & ex) {
        LOG(warning, "Timeout getting config, please check your setup. Will exit and restart: %s", ex.getMessage().c_str());
        EV_STOPPING("config-sentinel", ex.what());
        exit(EXIT_FAILURE);
    } catch (InvalidConfigException& ex) {
        LOG(error, "Fatal: Invalid configuration, please check your setup: %s", ex.getMessage().c_str());
        EV_STOPPING("config-sentinel", ex.what());
        exit(EXIT_FAILURE);
    } catch (ConfigRuntimeException& ex) {
        LOG(error, "Fatal: Could not get config, please check your setup: %s", ex.getMessage().c_str());
        EV_STOPPING("config-sentinel", ex.what());
        exit(EXIT_FAILURE);
    }

    struct timeval lastTv;
    gettimeofday(&lastTv, nullptr);
    while (!stop()) {
        try {
            vespalib::SignalHandler::CHLD.clear();
            handler.doWork();       // Check for child procs & commands
        } catch (InvalidConfigException& ex) {
            LOG(warning, "Configuration problem: (ignoring): %s", ex.what());
        } catch (vespalib::PortListenException& ex) {
            LOG(error, "Fatal: %s", ex.getMessage().c_str());
            EV_STOPPING("config-sentinel", ex.what());
            exit(EXIT_FAILURE);
        } catch (vespalib::FatalException& ex) {
            LOG(error, "Fatal: %s", ex.getMessage().c_str());
            EV_STOPPING("config-sentinel", ex.what());
            exit(EXIT_FAILURE);
        }
        if (vespalib::SignalHandler::CHLD.check()) {
            continue;
        }
        int maxNum = 0;
        fd_set fds;
        FD_ZERO(&fds);
        handler.updateActiveFdset(&fds, &maxNum);

        struct timeval tv;
        tv.tv_sec = 1;
        tv.tv_usec = 0;

        select(maxNum, &fds, nullptr, nullptr, &tv);

        gettimeofday(&tv, nullptr);
        double delta = tv.tv_sec - lastTv.tv_sec
                       + 1e-6 * (tv.tv_usec - lastTv.tv_usec);
        if (delta < 0.01) {
            usleep(12500); // Avoid busy looping;
        }
        lastTv = tv;
    }

    int rv = handler.terminate();

    EV_STOPPING("config-sentinel", "normal exit");
    return rv;
}