dict.c

// Copyright 2012 Rui Ueyama <rui314@gmail.com>
// This program is free software licensed under the MIT license.

#include <stdlib.h>
#include <string.h>
#include "dict.h"

typedef struct DictEntry {
    char *key;
    void *val;
} DictEntry;

void *make_dict(void *parent) {
    Dict *r = malloc(sizeof(Dict));
    r->list = make_list();
    r->parent = parent;
    return r;
}

void *dict_get(Dict *dict, char *key) {
    for (; dict; dict = dict->parent) {
        for (Iter *i = list_iter(dict->list); !iter_end(i);) {
            DictEntry *e = iter_next(i);
            if (!strcmp(key, e->key))
                return e->val;
        }
    }
    return NULL;
}

void dict_put(Dict *dict, char *key, void *val) {
    DictEntry *e = malloc(sizeof(DictEntry));
    e->key = key;
    e->val = val;
    list_unshift(dict->list, e);
}

void dict_remove(Dict *dict, char *key) {
    List *list = make_list();
    for (Iter *i = list_iter(dict->list); !iter_end(i);) {
        DictEntry *e = iter_next(i);
        if (strcmp(key, e->key))
            list_push(list, e);
    }
    dict->list = list;
}

bool dict_empty(Dict *dict) {
    return list_len(dict->list) == 0;
}

List *dict_keys(Dict *dict) {
    List *r = make_list();
    for (; dict; dict = dict->parent)
        for (Iter *i = list_iter(dict->list); !iter_end(i);)
            list_unshift(r, ((DictEntry *)iter_next(i))->key);
    return r;
}

List *dict_values(Dict *dict) {
    List *r = make_list();
    for (; dict; dict = dict->parent)
        for (Iter *i = list_iter(dict->list); !iter_end(i);)
            list_unshift(r, ((DictEntry *)iter_next(i))->val);
    return r;
}

void *dict_parent(Dict *dict) {
    return dict->parent;
}