/* sdb - MIT - Copyright 2024 - pancake */

#include <stddef.h>
#include <stdio.h>
#include <string.h>
#include "sdb_private.h"

SDB_API StrBuf* strbuf_new(void) {
	return (StrBuf*) sdb_gh_calloc (sizeof (StrBuf), 1);
}

SDB_API StrBuf* strbuf_append(StrBuf *sb, const char *str, const int nl) {
	if (!sb || !str || nl < 0) {
		return sb;
	}
	int len = strlen (str);
	if ((sb->len + len + 2) >= sb->size) {
		size_t newsize = sb->size + len + 256;
		char *b = (char *)sdb_gh_realloc (sb->buf, newsize);
		if (!b) {
			// keep the buffer owned and intact, dropping the append
			return sb;
		}
		sb->buf = b;
		sb->size = newsize;
	}
	if (sb->buf && str) {
		memcpy (sb->buf + sb->len, str, len);
		sb->len += len;
	}
#define NEWLINE_AFTER_QUERY 1
#if NEWLINE_AFTER_QUERY
	if (sb->buf && nl) {
		sb->buf[sb->len++] = '\n';
		len++;
	}
#endif
	if (sb->buf) {
		sb->buf[sb->len] = 0;
	}
	return sb;
}

SDB_API StrBuf* strbuf_appendf(StrBuf *sb, const int nl, const char *fmt, ...) {
	if (!sb || !fmt) {
		return sb;
	}
	
	va_list ap;
	va_start (ap, fmt);
	char *str = sdb_vstrdupf (fmt, ap);
	va_end (ap);
	if (!str) {
		return sb;
	}
	StrBuf *ret = strbuf_append (sb, str, nl);
	sdb_gh_free (str);
	return ret;
}

SDB_API char* strbuf_drain(StrBuf *sb) {
	if (!sb) {
		return NULL;
	}
	if (!sb->buf) {
		sdb_gh_free (sb);
		return NULL;
	}
	char *buf = sb->buf;
	sb->buf = NULL;
	sb->len = 0;
	sb->size = 0;
	sdb_gh_free (sb);
	return buf;
}

SDB_API StrBuf* strbuf_free(StrBuf *sb) {
	if (sb) {
		sdb_gh_free (sb->buf);
		sdb_gh_free (sb);
	}
	return NULL;
}