console.c

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdint.h>
#include <stdbool.h>

#include <glib.h>

#include "types.h"
#include "console.h"

int console_outfd = 1;

static GString *tstamp(void)
{
	char stamp[sizeof "HH:MM:SS "];

	time_t now = time(NULL);
	struct tm *tm = localtime(&now);
	strftime(stamp, sizeof stamp, "%H:%M:%S ", tm);

	return g_string_new(stamp);
}

static inline void log_vput(GString *prefix, char *fmt, va_list ap)
{
	g_string_append_vprintf(prefix, fmt, ap);
	g_string_append_c(prefix, '\n');

	struct buffer buf = { prefix->len, (unsigned char *) prefix->str };

	while (buf.len > 0)
	{
		ssize_t wrote = write(console_outfd, buf.data, buf.len);
		if (wrote < 0 && errno == EINTR)
			continue;
		else if (wrote < 0)
			break; /* give up outputting anything */
		ADVANCE_BUFFER(buf, wrote);
	}

	g_string_free(prefix, true);
}

void log_print(char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
	log_vput(tstamp(), fmt, ap);
	va_end(ap);
}

void log_die(char *fmt, ...)
{
	GString *msg = tstamp();
	g_string_append_printf(msg, "[DIED] ");

	va_list ap;
	va_start(ap, fmt);
	log_vput(msg, fmt, ap);
	va_end(ap);

	exit(1);
}