#include "postgres_fe.h"

#include "extern.h"

char *
pgtypes_alloc(long size)
{
	char *new = (char *) calloc(1L, size);

	if (!new)
	{
		errno = ENOMEM;
		return NULL;
	}

	memset(new, '\0', size);
	return (new);
}

char *
pgtypes_strdup(char *str)
{
	char *new = (char *) strdup(str);

	if (!new)
		errno = ENOMEM;
	return (new);
}

int
pgtypes_fmt_replace(union un_fmt_comb replace_val, int replace_type, char** output, int *pstr_len) {
	/* general purpose variable, set to 0 in order to fix compiler
	 * warning */
	int i = 0;
	switch(replace_type) {
		case PGTYPES_TYPE_NOTHING:
			break;
		case PGTYPES_TYPE_STRING_CONSTANT:
		case PGTYPES_TYPE_STRING_MALLOCED:
			i = strlen(replace_val.str_val);
			if (i + 1 <= *pstr_len) {
				/* copy over i + 1 bytes, that includes the
				 * tailing terminator */
				strncpy(*output, replace_val.str_val, i + 1);
				*pstr_len -= i;
				*output += i;
				if (replace_type == PGTYPES_TYPE_STRING_MALLOCED) {
					free(replace_val.str_val);
				}
				return 0;
			} else {
				return -1;
			}
			break;
		case PGTYPES_TYPE_CHAR:
			if (*pstr_len >= 2) {
				(*output)[0] = replace_val.char_val;
				(*output)[1] = '\0';
				(*pstr_len)--;
				(*output)++;
				return 0;
			} else {
				return -1;
			}
			break;
		case PGTYPES_TYPE_DOUBLE_NF:
		case PGTYPES_TYPE_INT64:
		case PGTYPES_TYPE_UINT:
		case PGTYPES_TYPE_UINT_2_LZ:
		case PGTYPES_TYPE_UINT_2_LS:
		case PGTYPES_TYPE_UINT_3_LZ:
		case PGTYPES_TYPE_UINT_4_LZ:
			{
				char* t = pgtypes_alloc(PGTYPES_FMT_NUM_MAX_DIGITS);
				if (!t) {
					return ENOMEM;
				}
				switch (replace_type) {
					case PGTYPES_TYPE_DOUBLE_NF:
						i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
								"%0.0g", replace_val.double_val);
						break;
#ifdef HAVE_INT64
					case PGTYPES_TYPE_INT64:
						i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
								INT64_FORMAT, replace_val.int64_val);
						break;
#endif
					case PGTYPES_TYPE_UINT:
							i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
									"%u", replace_val.uint_val);
						break;
					case PGTYPES_TYPE_UINT_2_LZ:
							i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
									"%02u", replace_val.uint_val);
						break;
					case PGTYPES_TYPE_UINT_2_LS:
							i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
									"%2u", replace_val.uint_val);
						break;
					case PGTYPES_TYPE_UINT_3_LZ:
							i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
									"%03u", replace_val.uint_val);
						break;
					case PGTYPES_TYPE_UINT_4_LZ:
							i = snprintf(t, PGTYPES_FMT_NUM_MAX_DIGITS,
									"%04u", replace_val.uint_val);
						break;
				}

				if (i < 0) {
					free(t);
					return -1;
				}
				i = strlen(t);
				*pstr_len -= i;
				/* if *pstr_len == 0, we don't have enough
				 * space for the terminator and the
				 * conversion fails */
				if (*pstr_len <= 0) {
					free(t);
					return -1;
				}
				strcpy(*output, t);
				*output += i;
				free(t);
			}
			break;
		default:
			break;
	}
	return 0;
}


