hectic-lab/util.nix
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: hel: c89...

Browse Source
This commit is contained in:
yukkop
2025-04-19 23:04:29 +00:00
parent 62abe7b846
commit a9d64fe059
4 changed files with 100 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
package/c/hel/Makefile
View File

@@ -14,4 +14,4 @@ LDFLAGS += $(shell $(HECTIC_CONFIG) --libs)
SHLIB_LINK := $(LIBS) SHLIB_LINK := $(LIBS)
PGXS := $(shell $(PG_CONFIG) --pgxs) PGXS := $(shell $(PG_CONFIG) --pgxs)
include $(PGXS) include $(PGXS)

142
package/c/hel/hel.c
View File

@@ -11,13 +11,18 @@ PG_MODULE_MAGIC;
/* Helper function to get a JSON value by key path */ /* Helper function to get a JSON value by key path */
static Json *json_get_by_path(Arena *arena, const Json *context, const char *key_path) { static Json *json_get_by_path(Arena *arena, const Json *context, const char *key_path) {
char *path_copy;
char *token;
Json *current;
if (!context || !key_path || !*key_path) { if (!context || !key_path || !*key_path) {
return NULL; return NULL;
} }
char *path_copy = arena_strdup(arena, key_path); path_copy = arena_strdup(arena, key_path);
char *token = strtok(path_copy, "."); token = strtok(path_copy, ".");
Json *current = (Json*)context; current = (Json*)context;
while (token && current) { while (token && current) {
current = json_get_object_item(current, token); current = json_get_object_item(current, token);
@@ -67,12 +72,15 @@ static char *render_text_node(Arena *arena, const TemplateNode *node) {
/* Render an interpolation node */ /* Render an interpolation node */
static char *render_interpolation_node(Arena *arena, const TemplateNode *node, const Json *context) { static char *render_interpolation_node(Arena *arena, const TemplateNode *node, const Json *context) {
const char *key;
Json *value;
if (!node || node->type != TEMPLATE_NODE_INTERPOLATE || !context) { if (!node || node->type != TEMPLATE_NODE_INTERPOLATE || !context) {
return ""; return "";
} }
const char *key = node->value.interpolate.key; key = node->value.interpolate.key;
Json *value = json_get_by_path(arena, context, key); value = json_get_by_path(arena, context, key);
if (!value) { if (!value) {
return ""; return "";
@@ -83,40 +91,60 @@ static char *render_interpolation_node(Arena *arena, const TemplateNode *node, c
/* Render a section node (for loop) */ /* Render a section node (for loop) */
static char *render_section_node(Arena *arena, const TemplateNode *node, const Json *context) { static char *render_section_node(Arena *arena, const TemplateNode *node, const Json *context) {
const char *collection_key;
const char *iterator_name;
TemplateNode *body;
Json *collection;
size_t buffer_size;
char *buffer;
size_t buffer_pos;
Json *item;
const char *empty_json;
Json *iter_context;
Json *item_json;
char *rendered_body;
size_t rendered_len;
if (!node || node->type != TEMPLATE_NODE_SECTION || !context) { if (!node || node->type != TEMPLATE_NODE_SECTION || !context) {
return ""; return "";
} }
const char *collection_key = node->value.section.collection; collection_key = node->value.section.collection;
const char *iterator_name = node->value.section.iterator; iterator_name = node->value.section.iterator;
TemplateNode *body = node->value.section.body; body = node->value.section.body;
Json *collection = json_get_by_path(arena, context, collection_key); collection = json_get_by_path(arena, context, collection_key);
if (!collection || collection->type != JSON_ARRAY) { if (!collection || collection->type != JSON_ARRAY) {
return ""; return "";
} }
size_t buffer_size = 1024; buffer_size = 1024;
char *buffer = arena_alloc(arena, buffer_size); buffer = arena_alloc(arena, buffer_size);
size_t buffer_pos = 0; buffer_pos = 0;
Json *item = collection->value.child; item = collection->value.child;
while (item) { while (item) {
const char *empty_json = "{}"; empty_json = "{}";
Json *iter_context = json_parse(arena, &empty_json); iter_context = json_parse(arena, &empty_json);
if (!iter_context) { if (!iter_context) {
return ""; return "";
} }
Json *item_json = arena_alloc(arena, sizeof(Json)); item_json = arena_alloc(arena, sizeof(Json));
memcpy(item_json, item, sizeof(Json)); memcpy(item_json, item, sizeof(Json));
item_json->key = arena_strdup(arena, iterator_name); item_json->key = arena_strdup(arena, iterator_name);
item_json->next = NULL; item_json->next = NULL;
char *rendered_body = render_template_node(arena, body, iter_context); rendered_body = render_template_node(arena, body, iter_context);
size_t rendered_len = strlen(rendered_body); rendered_len = strlen(rendered_body);
if (buffer_pos + rendered_len + 1 > buffer_size) { if (buffer_pos + rendered_len + 1 > buffer_size) {
buffer_size = (buffer_pos + rendered_len + 1) * 2; buffer_size = (buffer_pos + rendered_len + 1) * 2;
buffer = arena_realloc(arena, buffer, buffer_size / 2, buffer_size); buffer = arena_realloc(arena, buffer, buffer_size / 2, buffer_size);
@@ -134,26 +162,38 @@ static char *render_section_node(Arena *arena, const TemplateNode *node, const J
/* Render an include node */ /* Render an include node */
static char *render_include_node(Arena *arena, const TemplateNode *node, const Json *context) { static char *render_include_node(Arena *arena, const TemplateNode *node, const Json *context) {
const char *include_key;
Json *include_value;
char *buffer;
size_t buffer_pos;
Json *include_item;
Json *template_json;
Json *content_json;
Json *context_json;
if (!node || node->type != TEMPLATE_NODE_INCLUDE || !context) { if (!node || node->type != TEMPLATE_NODE_INCLUDE || !context) {
return ""; return "";
} }
include_key = node->value.include.key;
const char *include_key = node->value.include.key; include_value = json_get_by_path(arena, context, include_key);
Json *include_value = json_get_by_path(arena, context, include_key);
if (!include_value || include_value->type != JSON_ARRAY) { if (!include_value || include_value->type != JSON_ARRAY) {
return ""; return "";
} }
buffer = arena_alloc(arena, 1024);
buffer_pos = 0;
char *buffer = arena_alloc(arena, 1024);
size_t buffer_pos = 0;
Json *include_item = include_value->value.child; include_item = include_value->value.child;
while (include_item) { while (include_item) {
if (include_item->type == JSON_OBJECT) { if (include_item->type == JSON_OBJECT) {
Json *template_json = json_get_object_item(include_item, "template"); template_json = json_get_object_item(include_item, "template");
Json *content_json = json_get_object_item(include_item, "content"); content_json = json_get_object_item(include_item, "content");
Json *context_json = json_get_object_item(include_item, "context"); context_json = json_get_object_item(include_item, "context");
if (template_json && template_json->type == JSON_STRING) { if (template_json && template_json->type == JSON_STRING) {
const char *template_str = template_json->value.string; const char *template_str = template_json->value.string;
@@ -183,17 +223,22 @@ static char *render_include_node(Arena *arena, const TemplateNode *node, const J
/* Render a template node tree recursively */ /* Render a template node tree recursively */
static char *render_template_node(Arena *arena, const TemplateNode *node, const Json *context) { static char *render_template_node(Arena *arena, const TemplateNode *node, const Json *context) {
if (!node) {
return "";
}
size_t buffer_size = 4096; size_t buffer_size = 4096;
char *output = arena_alloc(arena, buffer_size); char *output = arena_alloc(arena, buffer_size);
size_t output_pos = 0; size_t output_pos = 0;
size_t rendered_len;
const TemplateNode *current;
char *rendered;
if (!node) {
return "";
}
current = node;
const TemplateNode *current = node;
while (current) { while (current) {
char *rendered = NULL; rendered = NULL;
switch (current->type) { switch (current->type) {
case TEMPLATE_NODE_TEXT: case TEMPLATE_NODE_TEXT:
@@ -222,7 +267,7 @@ static char *render_template_node(Arena *arena, const TemplateNode *node, const
break; break;
} }
size_t rendered_len = strlen(rendered); rendered_len = strlen(rendered);
if (output_pos + rendered_len + 1 > buffer_size) { if (output_pos + rendered_len + 1 > buffer_size) {
buffer_size = (output_pos + rendered_len + 1) * 2; buffer_size = (output_pos + rendered_len + 1) * 2;
output = arena_realloc(arena, output, buffer_size / 2, buffer_size); output = arena_realloc(arena, output, buffer_size / 2, buffer_size);
@@ -271,10 +316,21 @@ Datum render(PG_FUNCTION_ARGS)
/* Initialize arena for memory management */ /* Initialize arena for memory management */
Arena arena = arena_init(MEM_MiB); Arena arena = arena_init(MEM_MiB);
TemplateNode root_node;
TemplateResult template_result;
TemplateConfig config;
Json *context;
const char *template_ptr;
char *result_str;
text *result;
/* Parse the JSON context */ /* Parse the JSON context */
const char *json_ptr = context_str; const char *json_ptr = context_str;
Json *context = json_parse(&arena, &json_ptr); context = json_parse(&arena, &json_ptr);
if (!context) { if (!context) {
arena_free(&arena); arena_free(&arena);
@@ -283,9 +339,9 @@ Datum render(PG_FUNCTION_ARGS)
} }
/* Parse the template text */ /* Parse the template text */
const char *template_ptr = template_str; template_ptr = template_str;
TemplateConfig config = template_default_config(); config = template_default_config();
TemplateResult template_result = template_parse(&arena, &template_ptr, &config); template_result = template_parse(&arena, &template_ptr, &config);
if (IS_RESULT_ERROR(template_result)) { if (IS_RESULT_ERROR(template_result)) {
arena_free(&arena); arena_free(&arena);
@@ -295,11 +351,11 @@ Datum render(PG_FUNCTION_ARGS)
} }
/* Render the template */ /* Render the template */
TemplateNode root_node = RESULT_SOME_VALUE(template_result); root_node = RESULT_SOME_VALUE(template_result);
char *result_str = render_template_node(&arena, &root_node, context); result_str = render_template_node(&arena, &root_node, context);
/* Prepare return value */ /* Prepare return value */
text *result = cstring_to_text(result_str); result = cstring_to_text(result_str);
arena_free(&arena); arena_free(&arena);

BIN
package/c/hel/hel.o Normal file
View File

Binary file not shown.

BIN
package/c/hel/hel.so Executable file
View File

Binary file not shown.