#ifndef JSON_WRITER_H
#define JSON_WRITER_H

#include <string.h>
#include <stdint.h>

#define INITIAL_JSON_OUTPUT_CAPACITY 2048

typedef struct JsonBuilder
{
	char *string;
	size_t index;
	size_t capacity;
	size_t indentLevel;
} JsonBuilder;

JsonBuilder* JsonBuilder_Init()
{
	JsonBuilder *builder = malloc(sizeof(JsonBuilder));

	builder->string = malloc(INITIAL_JSON_OUTPUT_CAPACITY);
	builder->capacity = INITIAL_JSON_OUTPUT_CAPACITY;

	builder->string[0] = '\0';

	strcat(builder->string, "{\n");
	builder->indentLevel = 1;
	builder->index = 2;

	return builder;
}

void JsonBuilder_Internal_MaybeExpand(JsonBuilder *builder, size_t len)
{
	if (builder->capacity < builder->index + len)
	{
		builder->capacity = max(builder->index + len, builder->capacity * 2);
		builder->string = realloc(builder->string, builder->capacity);
	}
}

void JsonBuilder_Internal_Indent(JsonBuilder *builder)
{
	int32_t i;
	JsonBuilder_Internal_MaybeExpand(builder, builder->indentLevel);

	for (i = 0; i < builder->indentLevel; i += 1)
	{
		strcat(builder->string, "\t");
	}

	builder->index += builder->indentLevel;
}

void JsonBuilder_Internal_RemoveTrailingComma(JsonBuilder *builder)
{
	if (builder->string[builder->index - 2] == ',')
	{
		builder->index -= 2;
		builder->index += sprintf(&builder->string[builder->index], "\n");
	}
}

void JsonBuilder_AppendProperty(JsonBuilder *builder, char *propertyName, char *propertyString, uint8_t isString)
{
	JsonBuilder_Internal_Indent(builder);

	JsonBuilder_Internal_MaybeExpand(builder, strlen(propertyName) + 4);
	builder->index += sprintf(&builder->string[builder->index], "\"%s\": ", propertyName);

	if (isString)
	{
		JsonBuilder_Internal_MaybeExpand(builder, strlen(propertyString) + 4);
		builder->index += sprintf(&builder->string[builder->index], "\"%s\",\n", propertyString);
	}
	else
	{
		JsonBuilder_Internal_MaybeExpand(builder, strlen(propertyString) + 2);
		builder->index += sprintf(&builder->string[builder->index], "%s,\n", propertyString);
	}
}

void JsonBuilder_AppendStringProperty(JsonBuilder *builder, char *propertyName, char *value)
{
	JsonBuilder_AppendProperty(builder, propertyName, value, 1);
}

void JsonBuilder_AppendIntProperty(JsonBuilder *builder, char *propertyName, int32_t value)
{
	char buffer[65];
	itoa(value, buffer, 10);
	JsonBuilder_AppendProperty(builder, propertyName, buffer, 0);
}

void JsonBuilder_StartObject(JsonBuilder *builder)
{
	JsonBuilder_Internal_Indent(builder);
	JsonBuilder_Internal_MaybeExpand(builder, 2);
	builder->index += sprintf(&builder->string[builder->index], "{\n");
	builder->indentLevel += 1;
}

void JsonBuilder_EndObject(JsonBuilder *builder)
{
	JsonBuilder_Internal_RemoveTrailingComma(builder);

	builder->indentLevel -= 1;

	JsonBuilder_Internal_Indent(builder);
	JsonBuilder_Internal_MaybeExpand(builder, 3);
	builder->index += sprintf(&builder->string[builder->index], "},\n");
}

void JsonBuilder_StartArrayProperty(JsonBuilder *builder, char *propertyName)
{
	JsonBuilder_Internal_Indent(builder);
	JsonBuilder_Internal_MaybeExpand(builder, strlen(propertyName) + 6);
	builder->index += sprintf(&builder->string[builder->index], "\"%s\": [\n", propertyName);
	builder->indentLevel += 1;
}

void JsonBuilder_FinishArrayProperty(JsonBuilder *builder)
{
	JsonBuilder_Internal_RemoveTrailingComma(builder);

	builder->indentLevel -= 1;

	JsonBuilder_Internal_Indent(builder);
	JsonBuilder_Internal_MaybeExpand(builder, 3);
	builder->index += sprintf(&builder->string[builder->index], "],\n");
}

void JsonBuilder_Finish(JsonBuilder *builder)
{
	builder->indentLevel = 0;

	JsonBuilder_Internal_RemoveTrailingComma(builder);
	JsonBuilder_Internal_MaybeExpand(builder, 2);

	builder->index += sprintf(&builder->string[builder->index], "}\n");
}

void JsonBuilder_Destroy(JsonBuilder *builder)
{
	free(builder->string);
	free(builder);
}

#endif /* JSON_WRITER_H */