#include "common.h"

#define CONFIG_FILE_LINE_MAX ((CONFIG_MAX_NAME + CONFIG_MAX_VALUE + 1024) * 2)

pthread_mutex_t config_mutex = PTHREAD_MUTEX_INITIALIZER;

// ----------------------------------------------------------------------------
// definitions

#define CONFIG_VALUE_LOADED  0x01 // has been loaded from the config
#define CONFIG_VALUE_USED    0x02 // has been accessed from the program
#define CONFIG_VALUE_CHANGED 0x04 // has been changed from the loaded value
#define CONFIG_VALUE_CHECKED 0x08 // has been checked if the value is different from the default

struct config_value {
    avl avl;                // the index - this has to be first!

    uint8_t flags;
    uint32_t hash;          // a simple hash to speed up searching
                            // we first compare hashes, and only if the hashes are equal we do string comparisons

    char *name;
    char *value;

    struct config_value *next; // config->mutex protects just this
};

struct config {
    avl avl;

    uint32_t hash;          // a simple hash to speed up searching
                            // we first compare hashes, and only if the hashes are equal we do string comparisons

    char *name;

    struct config *next;    // gloabl config_mutex protects just this

    struct config_value *values;
    avl_tree_lock values_index;

    pthread_mutex_t mutex;  // this locks only the writers, to ensure atomic updates
                            // readers are protected using the rwlock in avl_tree_lock
} *config_root = NULL;


// ----------------------------------------------------------------------------
// locking

static inline void config_global_write_lock(void) {
    pthread_mutex_lock(&config_mutex);
}

static inline void config_global_unlock(void) {
    pthread_mutex_unlock(&config_mutex);
}

static inline void config_section_write_lock(struct config *co) {
    pthread_mutex_lock(&co->mutex);
}

static inline void config_section_unlock(struct config *co) {
    pthread_mutex_unlock(&co->mutex);
}


// ----------------------------------------------------------------------------
// config name-value index

static int config_value_compare(void* a, void* b) {
    if(((struct config_value *)a)->hash < ((struct config_value *)b)->hash) return -1;
    else if(((struct config_value *)a)->hash > ((struct config_value *)b)->hash) return 1;
    else return strcmp(((struct config_value *)a)->name, ((struct config_value *)b)->name);
}

#define config_value_index_add(co, cv) (struct config_value *)avl_insert_lock(&((co)->values_index), (avl *)(cv))
#define config_value_index_del(co, cv) (struct config_value *)avl_remove_lock(&((co)->values_index), (avl *)(cv))

static struct config_value *config_value_index_find(struct config *co, const char *name, uint32_t hash) {
    struct config_value tmp;
    tmp.hash = (hash)?hash:simple_hash(name);
    tmp.name = (char *)name;

    return (struct config_value *)avl_search_lock(&(co->values_index), (avl *) &tmp);
}


// ----------------------------------------------------------------------------
// config sections index

static int config_compare(void* a, void* b) {
    if(((struct config *)a)->hash < ((struct config *)b)->hash) return -1;
    else if(((struct config *)a)->hash > ((struct config *)b)->hash) return 1;
    else return strcmp(((struct config *)a)->name, ((struct config *)b)->name);
}

avl_tree_lock config_root_index = {
        { NULL, config_compare },
        AVL_LOCK_INITIALIZER
};

#define config_index_add(cfg) (struct config *)avl_insert_lock(&config_root_index, (avl *)(cfg))
#define config_index_del(cfg) (struct config *)avl_remove_lock(&config_root_index, (avl *)(cfg))

static struct config *config_index_find(const char *name, uint32_t hash) {
    struct config tmp;
    tmp.hash = (hash)?hash:simple_hash(name);
    tmp.name = (char *)name;

    return (struct config *)avl_search_lock(&config_root_index, (avl *) &tmp);
}


// ----------------------------------------------------------------------------
// config section methods

static inline struct config *config_section_find(const char *section) {
    return config_index_find(section, 0);
}

static inline struct config *config_section_create(const char *section)
{
    debug(D_CONFIG, "Creating section '%s'.", section);

    struct config *co = callocz(1, sizeof(struct config));
    co->name = strdupz(section);
    co->hash = simple_hash(co->name);

    avl_init_lock(&co->values_index, config_value_compare);

    if(unlikely(config_index_add(co) != co))
        error("INTERNAL ERROR: indexing of section '%s', already exists.", co->name);

    config_global_write_lock();
    struct config *co2 = config_root;
    if(co2) {
        while (co2->next) co2 = co2->next;
        co2->next = co;
    }
    else config_root = co;
    config_global_unlock();

    return co;
}


// ----------------------------------------------------------------------------
// config name-value methods

static inline struct config_value *config_value_create(struct config *co, const char *name, const char *value)
{
    debug(D_CONFIG, "Creating config entry for name '%s', value '%s', in section '%s'.", name, value, co->name);

    struct config_value *cv = callocz(1, sizeof(struct config_value));
    cv->name = strdupz(name);
    cv->hash = simple_hash(cv->name);
    cv->value = strdupz(value);

    if(unlikely(config_value_index_add(co, cv) != cv))
        error("INTERNAL ERROR: indexing of config '%s' in section '%s': already exists.", cv->name, co->name);

    config_section_write_lock(co);
    struct config_value *cv2 = co->values;
    if(cv2) {
        while (cv2->next) cv2 = cv2->next;
        cv2->next = cv;
    }
    else co->values = cv;
    config_section_unlock(co);

    return cv;
}

int config_exists(const char *section, const char *name) {
    struct config_value *cv;

    debug(D_CONFIG, "request to get config in section '%s', name '%s'", section, name);

    struct config *co = config_section_find(section);
    if(!co) return 0;

    cv = config_value_index_find(co, name, 0);
    if(!cv) return 0;

    return 1;
}

int config_rename(const char *section, const char *old, const char *new) {
    struct config_value *cv, *cv2;

    debug(D_CONFIG, "request to rename config in section '%s', old name '%s', new name '%s'", section, old, new);

    struct config *co = config_section_find(section);
    if(!co) return -1;

    config_section_write_lock(co);

    cv = config_value_index_find(co, old, 0);
    if(!cv) goto cleanup;

    cv2 = config_value_index_find(co, new, 0);
    if(cv2) goto cleanup;

    if(unlikely(config_value_index_del(co, cv) != cv))
        error("INTERNAL ERROR: deletion of config '%s' from section '%s', deleted tge wrong config entry.", cv->name, co->name);

    freez(cv->name);
    cv->name = strdupz(new);
    cv->hash = simple_hash(cv->name);
    if(unlikely(config_value_index_add(co, cv) != cv))
        error("INTERNAL ERROR: indexing of config '%s' in section '%s', already exists.", cv->name, co->name);

    config_section_unlock(co);

    return 0;

cleanup:
    config_section_unlock(co);
    return -1;
}

char *config_get(const char *section, const char *name, const char *default_value)
{
    struct config_value *cv;

    debug(D_CONFIG, "request to get config in section '%s', name '%s', default_value '%s'", section, name, default_value);

    struct config *co = config_section_find(section);
    if(!co) co = config_section_create(section);

    cv = config_value_index_find(co, name, 0);
    if(!cv) {
        cv = config_value_create(co, name, default_value);
        if(!cv) return NULL;
    }
    cv->flags |= CONFIG_VALUE_USED;

    if((cv->flags & CONFIG_VALUE_LOADED) || (cv->flags & CONFIG_VALUE_CHANGED)) {
        // this is a loaded value from the config file
        // if it is different that the default, mark it
        if(!(cv->flags & CONFIG_VALUE_CHECKED)) {
            if(strcmp(cv->value, default_value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
            cv->flags |= CONFIG_VALUE_CHECKED;
        }
    }

    return(cv->value);
}

long long config_get_number(const char *section, const char *name, long long value)
{
    char buffer[100], *s;
    sprintf(buffer, "%lld", value);

    s = config_get(section, name, buffer);
    if(!s) return value;

    return strtoll(s, NULL, 0);
}

int config_get_boolean(const char *section, const char *name, int value)
{
    char *s;
    if(value) s = "yes";
    else s = "no";

    s = config_get(section, name, s);
    if(!s) return value;

    if(!strcmp(s, "yes") || !strcmp(s, "auto") || !strcmp(s, "on demand")) return 1;
    return 0;
}

int config_get_boolean_ondemand(const char *section, const char *name, int value)
{
    char *s;

    if(value == CONFIG_ONDEMAND_ONDEMAND)
        s = "auto";

    else if(value == CONFIG_ONDEMAND_NO)
        s = "no";

    else
        s = "yes";

    s = config_get(section, name, s);
    if(!s) return value;

    if(!strcmp(s, "yes"))
        return CONFIG_ONDEMAND_YES;
    else if(!strcmp(s, "no"))
        return CONFIG_ONDEMAND_NO;
    else if(!strcmp(s, "auto") || !strcmp(s, "on demand"))
        return CONFIG_ONDEMAND_ONDEMAND;

    return value;
}

const char *config_set_default(const char *section, const char *name, const char *value)
{
    struct config_value *cv;

    debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);

    struct config *co = config_section_find(section);
    if(!co) return config_set(section, name, value);

    cv = config_value_index_find(co, name, 0);
    if(!cv) return config_set(section, name, value);

    cv->flags |= CONFIG_VALUE_USED;

    if(cv->flags & CONFIG_VALUE_LOADED)
        return cv->value;

    if(strcmp(cv->value, value) != 0) {
        cv->flags |= CONFIG_VALUE_CHANGED;

        freez(cv->value);
        cv->value = strdupz(value);
    }

    return cv->value;
}

const char *config_set(const char *section, const char *name, const char *value)
{
    struct config_value *cv;

    debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);

    struct config *co = config_section_find(section);
    if(!co) co = config_section_create(section);

    cv = config_value_index_find(co, name, 0);
    if(!cv) cv = config_value_create(co, name, value);
    cv->flags |= CONFIG_VALUE_USED;

    if(strcmp(cv->value, value) != 0) {
        cv->flags |= CONFIG_VALUE_CHANGED;

        freez(cv->value);
        cv->value = strdupz(value);
    }

    return value;
}

long long config_set_number(const char *section, const char *name, long long value)
{
    char buffer[100];
    sprintf(buffer, "%lld", value);

    config_set(section, name, buffer);

    return value;
}

int config_set_boolean(const char *section, const char *name, int value)
{
    char *s;
    if(value) s = "yes";
    else s = "no";

    config_set(section, name, s);

    return value;
}


// ----------------------------------------------------------------------------
// config load/save

int load_config(char *filename, int overwrite_used)
{
    int line = 0;
    struct config *co = NULL;

    char buffer[CONFIG_FILE_LINE_MAX + 1], *s;

    if(!filename) filename = CONFIG_DIR "/" CONFIG_FILENAME;

    debug(D_CONFIG, "Opening config file '%s'", filename);

    FILE *fp = fopen(filename, "r");
    if(!fp) {
        error("Cannot open file '%s'", filename);
        return 0;
    }

    while(fgets(buffer, CONFIG_FILE_LINE_MAX, fp) != NULL) {
        buffer[CONFIG_FILE_LINE_MAX] = '\0';
        line++;

        s = trim(buffer);
        if(!s) {
            debug(D_CONFIG, "Ignoring line %d, it is empty.", line);
            continue;
        }

        int len = (int) strlen(s);
        if(*s == '[' && s[len - 1] == ']') {
            // new section
            s[len - 1] = '\0';
            s++;

            co = config_section_find(s);
            if(!co) co = config_section_create(s);

            continue;
        }

        if(!co) {
            // line outside a section
            error("Ignoring line %d ('%s'), it is outside all sections.", line, s);
            continue;
        }

        char *name = s;
        char *value = strchr(s, '=');
        if(!value) {
            error("Ignoring line %d ('%s'), there is no = in it.", line, s);
            continue;
        }
        *value = '\0';
        value++;

        name = trim(name);
        value = trim(value);

        if(!name) {
            error("Ignoring line %d, name is empty.", line);
            continue;
        }
        if(!value) {
            debug(D_CONFIG, "Ignoring line %d, value is empty.", line);
            continue;
        }

        struct config_value *cv = config_value_index_find(co, name, 0);

        if(!cv) cv = config_value_create(co, name, value);
        else {
            if(((cv->flags & CONFIG_VALUE_USED) && overwrite_used) || !(cv->flags & CONFIG_VALUE_USED)) {
                debug(D_CONFIG, "Line %d, overwriting '%s/%s'.", line, co->name, cv->name);
                freez(cv->value);
                cv->value = strdupz(value);
            }
            else
                debug(D_CONFIG, "Ignoring line %d, '%s/%s' is already present and used.", line, co->name, cv->name);
        }
        cv->flags |= CONFIG_VALUE_LOADED;
    }

    fclose(fp);

    return 1;
}

void generate_config(BUFFER *wb, int only_changed)
{
    int i, pri;
    struct config *co;
    struct config_value *cv;

    for(i = 0; i < 3 ;i++) {
        switch(i) {
            case 0:
                buffer_strcat(wb,
                    "# netdata configuration\n"
                    "#\n"
                    "# You can download the latest version of this file, using:\n"
                    "#\n"
                    "#  wget -O /etc/netdata/netdata.conf http://localhost:19999/netdata.conf\n"
                    "# or\n"
                    "#  curl -o /etc/netdata/netdata.conf http://localhost:19999/netdata.conf\n"
                    "#\n"
                    "# You can uncomment and change any of the options below.\n"
                    "# The value shown in the commented settings, is the default value.\n"
                    "#\n"
                    "\n# global netdata configuration\n");
                break;

            case 1:
                buffer_strcat(wb, "\n\n# per plugin configuration\n");
                break;

            case 2:
                buffer_strcat(wb, "\n\n# per chart configuration\n");
                break;
        }

        config_global_write_lock();
        for(co = config_root; co ; co = co->next) {
            if(!strcmp(co->name, "global") ||
                    !strcmp(co->name, "plugins")  ||
                    !strcmp(co->name, "registry") ||
                    !strcmp(co->name, "health")   ||
                    !strcmp(co->name, "backend"))
                pri = 0;
            else if(!strncmp(co->name, "plugin:", 7)) pri = 1;
            else pri = 2;

            if(i == pri) {
                int used = 0;
                int changed = 0;
                int count = 0;

                config_section_write_lock(co);
                for(cv = co->values; cv ; cv = cv->next) {
                    used += (cv->flags & CONFIG_VALUE_USED)?1:0;
                    changed += (cv->flags & CONFIG_VALUE_CHANGED)?1:0;
                    count++;
                }
                config_section_unlock(co);

                if(!count) continue;
                if(only_changed && !changed) continue;

                if(!used) {
                    buffer_sprintf(wb, "\n# node '%s' is not used.", co->name);
                }

                buffer_sprintf(wb, "\n[%s]\n", co->name);

                config_section_write_lock(co);
                for(cv = co->values; cv ; cv = cv->next) {

                    if(used && !(cv->flags & CONFIG_VALUE_USED)) {
                        buffer_sprintf(wb, "\n\t# option '%s' is not used.\n", cv->name);
                    }
                    buffer_sprintf(wb, "\t%s%s = %s\n", ((!(cv->flags & CONFIG_VALUE_CHANGED)) && (cv->flags & CONFIG_VALUE_USED))?"# ":"", cv->name, cv->value);
                }
                config_section_unlock(co);
            }
        }
        config_global_unlock();
    }
}