diff options
| -rw-r--r-- | src/template.c | 37 |
1 files changed, 33 insertions, 4 deletions
diff --git a/src/template.c b/src/template.c index 6d2ffe9..14e40c2 100644 --- a/src/template.c +++ b/src/template.c @@ -27,17 +27,46 @@ void tpl_register_func(char *key, struct tplfunc_frame *frame) { tpl_pool_func_used++; } +int tpl_key_exists(char *key) { + for (size_t i = 0; i < tpl_pool_used; i++) { + if (tpl_pool[i]->key) { + if (!strcmp(tpl_pool[i]->key, key)) { + return true; + } + } + } + return false; +} + void tpl_register(char *key, char **ptr) { struct tpl_item *item = NULL; - item = calloc(1, sizeof(*item)); + int replacing = 0; + + if (tpl_key_exists(key)) { + for (size_t i = 0; i < tpl_pool_used; i++) { + if (tpl_pool[i]->key) { + if (!strcmp(tpl_pool[i]->key, key)) { + item = tpl_pool[i]; + break; + } + } + } + replacing = 1; + } else { + item = calloc(1, sizeof(*item)); + item->key = strdup(key); + } + if (!item) { SYSERROR("unable to register tpl_item for %s", key); exit(1); } - item->key = strdup(key); + item->ptr = ptr; - tpl_pool[tpl_pool_used] = item; - tpl_pool_used++; + if (!replacing) { + tpl_pool[tpl_pool_used] = item; + tpl_pool_used++; + } } void tpl_free() { |