keyword: Add `get_keyword()`, `create_key_value_str()`, and

`create_key_value_file()`

These functions will help in making the creation of key-value files more
generic.

There still needs to be some more work done on it, but that can be done
later.

2 files changed, 107 insertions, 0 deletions

diff --git a/keyword.c b/keyword.c
index 24db8d9..917f93a 100644
--- a/keyword.c
+++ b/keyword.c
@@ -1,10 +1,12 @@
 #define _XOPEN_SOURCE
 #include <stdint.h>
+#include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <syslog.h>
 #include <time.h>
 #include "keyword.h"
+#include "linked_list.h"
 #include "macros.h"
 #include "misc.h"
 
@@ -45,6 +47,34 @@ keyword_val get_keyword_value(const keyword *key, char *value, int *error) {
 	return val;
 }
 
+keyword_val get_keyword(const keyword *key, void *data, void *ctx, int *error) {
+	keyword_val val = {0};
+	int dummy = 0;
+	error = (error != NULL) ? error : &dummy;
+	*error = 0;
+
+	if (data != NULL && key != NULL) {
+		const int callback_ret = (key->get_callback != NULL) ? key->get_callback(ctx, data, key, &val) : 0;
+		if (!callback_ret) {
+			char *data_member = (char *)get_keyword_offset_ptr(key, data);
+
+			switch (key->type) {
+				case TYPE_INT	:
+				case TYPE_BOOL	: val.i = *(int *)data_member; break;
+				case TYPE_TIME	: val.t = *(time_t *)data_member; break;
+				case TYPE_STRING: val.str = *(char **)data_member; break;
+				case TYPE_FLOAT	: val.f = *(float *)data_member; break;
+				default		: *error = 2; break;
+			}
+		} else if (callback_ret < 0) {
+			*error = 1;
+		} else {
+			*error = -callback_ret;
+		}
+	}
+	return val;
+}
+
 int set_keyword(const keyword *key, keyword_val val, void *ret, void *ctx) {
 	const int callback_ret = (key->set_callback != NULL) ? key->set_callback(ctx, ret, key, val) : 0;
 	if (callback_ret < 0 || callback_ret > 0) {
@@ -140,3 +170,77 @@ int parse_key_value_file(void *ret, void *ctx, const keyword **keywords, char *b
 		return 8;
 	}
 }
+
+static int has_delm(const keyword *key, keyword_val val, const char *delm) {
+	const keyword_type type = key->type;
+	const char *str = (type == TYPE_STRING) ? val.str : key->time_fmt;
+	const int is_string_type = (type == TYPE_STRING || type == TYPE_TIME);
+	return is_string_type && !is_empty(str) && !is_empty(delm) && !is_empty(find_delm((char *)str, delm, 0));
+}
+
+static int key_value_strlen(const keyword *key, keyword_val val, const char *start_delm, const char *end_delm, int leading_whitespace, int trailing_whitespace) {
+	const char *quote = (has_delm(key, val, end_delm)) ? "\"" : "";
+	int len = format_len("%s%*s%*s", key->key, (int)strlen(start_delm)+leading_whitespace, start_delm, (int)strlen(quote)+trailing_whitespace, quote);
+	struct tm tm;
+	switch (key->type) {
+			case TYPE_INT	:
+			case TYPE_BOOL	: len += format_len("%i", val.i); break;
+			case TYPE_TIME	: localtime_r(&val.t, &tm); len += strftime(NULL, -1, key->time_fmt, &tm); break;
+			case TYPE_STRING: len += format_len("%s", val.str); break;
+			case TYPE_FLOAT	: len += format_len("%f", val.f); break;
+			default		: break;
+	}
+	len += format_len("%s%s", quote, end_delm);
+	return len;
+}
+
+char *create_key_value_str(const keyword *key, keyword_val val, const char *start_delm, const char *end_delm, int leading_whitespace, int trailing_whitespace) {
+	char *key_value = calloc(key_value_strlen(key, val, start_delm, end_delm, leading_whitespace, trailing_whitespace)+1, sizeof(char));
+	const char *quote = (has_delm(key, val, end_delm)) ? "\"" : "";
+	int j = sprintf(key_value, "%s%*s%*s", key->key, (int)strlen(start_delm)+leading_whitespace, start_delm, (int)strlen(quote)+trailing_whitespace, quote);
+	struct tm tm;
+	switch (key->type) {
+			case TYPE_INT	:
+			case TYPE_BOOL	: j += sprintf(key_value+j, "%i", val.i); break;
+			case TYPE_TIME	: localtime_r(&val.t, &tm); j += strftime(key_value+j, -1, key->time_fmt, &tm); break;
+			case TYPE_STRING: j += sprintf(key_value+j, "%s", val.str); break;
+			case TYPE_FLOAT	: j += sprintf(key_value+j, "%f", val.f); break;
+			default		: break;
+	}
+	j += sprintf(key_value+j, "%s%s", quote, end_delm);
+	return key_value;
+}
+
+char *create_key_value_file(void *data, void *ctx, const keyword **keywords, const char *start_delm, const char *end_delm, int leading_whitespace, int trailing_whitespace) {
+	if (data != NULL && keywords != NULL) {
+		int buf_len = 0;
+		char *key_value_buf;
+		char *tmp;
+		linked_list *key_values = NULL;
+		for (int i = 0; keywords[i] != NULL; ++i) {
+			int error;
+			keyword_val val = get_keyword(keywords[i], data, ctx, &error);
+			if (!error || error < 0) {
+				char *key_value = create_key_value_str(keywords[i], val, start_delm, end_delm, leading_whitespace, trailing_whitespace);
+				buf_len += strlen(key_value);
+				key_values = add_node(&key_values, key_value);
+				if (keywords[i]->type == TYPE_STRING && error == -2) {
+					free(val.str);
+				}
+			}
+		}
+
+		key_value_buf = calloc(buf_len+1, sizeof(char));
+		tmp = key_value_buf;
+
+		for (linked_list *node = get_head(key_values); node != NULL; node = node->next) {
+			char *key_value = (char *)node->data;
+			tmp += sprintf(tmp, "%s", key_value);
+			free(key_value);
+		}
+		cleanup_linked_list(key_values);
+		return key_value_buf;
+	} else {
+		return NULL;
+	}
+}
diff --git a/keyword.h b/keyword.h
index c39f13c..64183aa 100644
--- a/keyword.h
+++ b/keyword.h
@@ -39,8 +39,11 @@ union keyword_val {
 
 extern void *get_keyword_offset_ptr(const keyword *key, void *ptr);
 extern keyword_val get_keyword_value(const keyword *key, char *value, int *error);
+extern keyword_val get_keyword(const keyword *key, void *data, void *ctx, int *error);
 extern int set_keyword(const keyword *key, keyword_val val, void *ret, void *ctx);
 extern keyword_val parse_keyword(const keyword *key, char *key_str, char *value, int *error);
 extern int parse_keywords(const keyword **keys, char *key, char *value, void *ret, void *ctx);
 extern int parse_key_value_file(void *ret, void *ctx, const keyword **keywords, char *buf, const char *delm, parse_callback *parse_cb);
+extern char *create_key_value_str(const keyword *key, keyword_val val, const char *start_delm, const char *end_delm, int leading_whitespace, int trailing_whitespace);
+extern char *create_key_value_file(void *data, void *ctx, const keyword **keywords, const char *start_delm, const char *end_delm, int leading_whitespace, int trailing_whitespace);
 #endif