forked from ground/ground
132 lines
4.4 KiB
C
132 lines
4.4 KiB
C
#include "hashmap.h"
|
|
#include <groundext.h>
|
|
#include <groundvm.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <uthash.h>
|
|
#include <stdio.h>
|
|
|
|
GroundValue hashmapStructSet(GroundScope* scope, List args) {
|
|
char* key = args.values[0].data.stringVal;
|
|
GroundValue value = args.values[1];
|
|
|
|
GroundVariable* ptrField = groundFindVariable(scope, "private_ptr");
|
|
if (ptrField == NULL) {
|
|
ERROR("A field called \"private_ptr\" was not found", "FieldNotFound");
|
|
}
|
|
HashmapItem* keys = (HashmapItem*)ptrField->value.data.intVal;
|
|
|
|
HashmapItem* newItem = malloc(sizeof(HashmapItem));
|
|
newItem->key = strdup(key);
|
|
newItem->value = value;
|
|
|
|
HASH_ADD_KEYPTR(hh, keys, newItem->key, strlen(newItem->key), newItem);
|
|
|
|
ptrField->value.data.intVal = (long long)keys;
|
|
|
|
return groundCreateValue(INT, 0);
|
|
}
|
|
|
|
GroundValue hashmapStructGet(GroundScope* scope, List args) {
|
|
char* query = args.values[0].data.stringVal;
|
|
|
|
GroundVariable* ptrField = groundFindVariable(scope, "private_ptr");
|
|
if (ptrField == NULL) {
|
|
ERROR("A field called \"private_ptr\" was not found", "FieldNotFound");
|
|
}
|
|
HashmapItem* keys = (HashmapItem*)ptrField->value.data.intVal;
|
|
|
|
HashmapItem* out;
|
|
|
|
HASH_FIND_STR(keys, query, out);
|
|
|
|
if (out == NULL) {
|
|
ERROR("Key was not found in hashmap (tip: use hashmap.getOr if you want to specify a fallback value)", "KeyNotFound");
|
|
}
|
|
|
|
return out->value;
|
|
}
|
|
|
|
GroundValue hashmapStructGetOr(GroundScope* scope, List args) {
|
|
char* query = args.values[0].data.stringVal;
|
|
GroundValue fallback = args.values[1];
|
|
|
|
GroundVariable* ptrField = groundFindVariable(scope, "private_ptr");
|
|
if (ptrField == NULL) {
|
|
ERROR("A field called \"private_ptr\" was not found", "FieldNotFound");
|
|
}
|
|
HashmapItem* keys = (HashmapItem*)ptrField->value.data.intVal;
|
|
|
|
HashmapItem* out;
|
|
|
|
HASH_FIND_STR(keys, query, out);
|
|
|
|
if (out == NULL)
|
|
return fallback;
|
|
|
|
return out->value;
|
|
}
|
|
|
|
GroundValue hashmapStructRemove(GroundScope* scope, List args) {
|
|
char* query = args.values[0].data.stringVal;
|
|
|
|
GroundVariable* ptrField = groundFindVariable(scope, "private_ptr");
|
|
if (ptrField == NULL) {
|
|
ERROR("A field called \"private_ptr\" was not found", "FieldNotFound");
|
|
}
|
|
HashmapItem* keys = (HashmapItem*)ptrField->value.data.intVal;
|
|
|
|
HashmapItem* out;
|
|
|
|
HASH_FIND_STR(keys, query, out);
|
|
|
|
if (out == NULL) {
|
|
ERROR("Key was not found in hashmap (tip: use hashmap.removeIfPresent if you want to avoid errors)", "KeyNotFound");
|
|
}
|
|
|
|
HASH_DEL(keys, out);
|
|
|
|
free(out->key);
|
|
free(out);
|
|
|
|
return groundCreateValue(INT, 0);
|
|
}
|
|
|
|
GroundValue hashmapStructRemoveIfPresent(GroundScope* scope, List args) {
|
|
char* query = args.values[0].data.stringVal;
|
|
|
|
GroundVariable* ptrField = groundFindVariable(scope, "private_ptr");
|
|
if (ptrField == NULL) {
|
|
ERROR("A field called \"private_ptr\" was not found", "FieldNotFound");
|
|
}
|
|
HashmapItem* keys = (HashmapItem*)ptrField->value.data.intVal;
|
|
|
|
HashmapItem* out;
|
|
|
|
HASH_FIND_STR(keys, query, out);
|
|
|
|
if (out == NULL) {
|
|
return groundCreateValue(BOOL, false);
|
|
}
|
|
|
|
HASH_DEL(keys, out);
|
|
|
|
free(out->key);
|
|
free(out);
|
|
|
|
return groundCreateValue(BOOL, true);
|
|
}
|
|
|
|
GroundValue createHashmapStruct() {
|
|
GroundStruct hashmapStruct = groundCreateStruct();
|
|
|
|
groundAddFieldToStruct(&hashmapStruct, "private_ptr", groundCreateValue(INT, 0));
|
|
|
|
groundAddFunctionToStruct(&hashmapStruct, "set", hashmapStructSet, INT, 2, STRING, "key", ANY, "value"); // set a key in the hashmap
|
|
groundAddFunctionToStruct(&hashmapStruct, "get", hashmapStructGet, INT, 1, STRING, "key"); // get a key in the hashmap, throws KeyNotFound if the key does not exist
|
|
groundAddFunctionToStruct(&hashmapStruct, "getOr", hashmapStructGetOr, INT, 2, STRING, "key", ANY, "default"); // get a key in the hashmap, or fallback to a default value
|
|
groundAddFunctionToStruct(&hashmapStruct, "remove", hashmapStructRemove, INT, 1, STRING, "key"); // remove a key from the hashmap, throws KeyNotFound if the key does not exist
|
|
groundAddFunctionToStruct(&hashmapStruct, "removeIfPresent", hashmapStructRemoveIfPresent, BOOL, 1, STRING, "key"); // remove a key from the hashmap, returns true if the key was removed or false if it didn't exist
|
|
|
|
return groundCreateValue(STRUCTVAL, hashmapStruct);
|
|
} |