//#define _POSIX_C_SOURCE 200112
#define _GNU_SOURCE

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <signal.h>
#include <unistd.h>
#include <poll.h>
#include <errno.h>
#include <assert.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>

#include <arpa/inet.h>
#include <netinet/in.h>

#include <pthread.h>

#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
#include <luajit.h>

#include <readline/readline.h>
#include <readline/history.h>

#include <jack/jack.h>
#include <jack/midiport.h>
#include <jack/ringbuffer.h>

#include "applause.h"
#include "midi.h"

#define LUA_MODULE		"applause.lua"
#define APPLAUSE_HISTORY	".applause_history"

#define CMD_SERVER_IP   	"127.0.0.1"
#define CMD_SERVER_PORT		10000

#define MAX(X,Y)		((X) > (Y) ? (X) : (Y))

#define likely(X)		__builtin_expect((X), 1)
#define unlikely(X)		__builtin_expect((X), 0)

static jack_client_t		*client = NULL;

#define DEFAULT_NUMBER_OF_OUTPUT_PORTS	1
#define DEFAULT_BUFFER_SIZE		100	/* milliseconds */

typedef struct applause_output_port {
	jack_port_t		*jack_port;
	jack_ringbuffer_t	*buffer;
	int			buffer_sem;
	/**
	 * True if a buffer underrun occurs.
	 * FIXME: sig_atomic_t is probably the wrong type.
	 * Perhaps use the Mintomic types.
	 */
	sig_atomic_t		buffer_xrun;
} applause_output_port;

/** List of Applause output ports */
static applause_output_port *output_ports = NULL;
/** Number of Applause output ports */
static int output_ports_count = DEFAULT_NUMBER_OF_OUTPUT_PORTS;

static lua_State		*L_global = NULL;

static volatile sig_atomic_t	interrupted = 0;
static volatile sig_atomic_t	playback = 0;

typedef struct applause_midi_port {
	jack_port_t		*jack_port;
	jack_ringbuffer_t	*buffer;
} applause_midi_port;

static applause_midi_port midi_port;

static int
svsem_init(size_t value)
{
	int id;

	/*
	 * This is not in sem.h but required since
	 * sizeof(int) could be unequal sizeof(void*)
	 */
	union semun {
		int              val;    /* Value for SETVAL */
		struct semid_ds *buf;    /* Buffer for IPC_STAT, IPC_SET */
		unsigned short  *array;  /* Array for GETALL, SETALL */
		struct seminfo  *__buf;  /* Buffer for IPC_INFO
		                                           (Linux-specific) */
	} arg = {
		.val = value
	};

	id = semget(IPC_PRIVATE, 1, IPC_CREAT | 0777);
	if (id < 0)
		return -1;

	if (semctl(id, 0, SETVAL, arg) < 0)
		return -1;

	return id;
}

static inline int
svsem_op(int id, int value)
{
	struct sembuf op = {
		.sem_num = 0,
		.sem_op = value,
		.sem_flg = 0
	};
	int rc;

	/*
	 * Repeat operation when it is interrupted.
	 */
	while ((rc = semop(id, &op, 1)) < 0 &&
	       errno == EINTR);

	return rc;
}

static inline int
svsem_free(int id)
{
	return semctl(id, 0, IPC_RMID);
}

/**
 * Can be called from Lua code to check the interruption-flag,
 * ie. to detect SIGINT and CTRL+C interruptions.
 *
 * We do not simply expose the interrupted flag since LuaJIT
 * does not support volatile variables and we cannot guarantee that
 * the read is not optimized away.
 * On the other hand, this function cannot raise a Lua error
 * (and C functions via the classic C API are slow), so there is
 * another wrapper in applause.lua.
 */
int
applause_is_interrupted(void)
{
	if (likely(!interrupted))
		return 0;

	interrupted = 0;
	lua_sethook(L_global, NULL, 0, 0);
	return 1;
}

static void
signal_hook(lua_State *L, lua_Debug *ar)
{
	interrupted = 0;
	lua_sethook(L, NULL, 0, 0);

	/* Avoid luaL_error -- a C hook doesn't add an extra frame. */
	luaL_where(L, 0);
	lua_pushfstring(L, "%sinterrupted!", lua_tostring(L, -1));
	lua_error(L);
}

static inline void
applause_interrupt(void)
{
	/*
	 * This should raise a Lua error via the hook ASAP.
	 * Unfortunetely, this cannot work reliably with JIT compilation
	 * enabled (see also https://luajit.org/faq.html#ctrlc).
	 * We therefore also set an interrupted flag that can be polled
	 * efficiently in tight loops.
	 */
	interrupted = 1;
	lua_sethook(L_global, signal_hook, LUA_MASKCALL | LUA_MASKRET | LUA_MASKCOUNT, 1);
}

/**
 * Handler for SIGINT, SIGUSR1 and SIGUSR2 signals.
 *
 * SIGINT is delivered e.g. when the user presses CTRL+C.
 * SIGUSR1 and SIGUSR2 are sent by parent to child
 * clients in order to control playback.
 */
static void
signal_handler(int signum)
{
	switch (signum) {
	case SIGINT: applause_interrupt(); break;
	case SIGUSR1: playback = 1; break;
	case SIGUSR2: playback = 0; break;
	}
}

/**
 * The process callback for this JACK application is called in a
 * special realtime thread once for each audio cycle.
 */
static int
jack_process(jack_nframes_t nframes, void *arg)
{
	size_t len = sizeof(jack_default_audio_sample_t)*nframes;

	void *midi_in;
	jack_nframes_t midi_events;

	for (int i = 0; i < output_ports_count; i++) {
		applause_output_port *port = output_ports + i;
		jack_default_audio_sample_t *out;
		size_t r;

		out = (jack_default_audio_sample_t *)
			jack_port_get_buffer(port->jack_port, nframes);

		r = jack_ringbuffer_read(port->buffer, (char *)out, len);

		/*
		 * The semaphor value corresponds with the number of
		 * writable bytes in buffer, i.e. the available space.
		 * This operation should never block and is supposed to
		 * be real-time safe :-)
		 */
		if (r > 0)
			svsem_op(port->buffer_sem, r);

		/*
		 * Add silence for missing output samples.
		 * Here we're assuming that memset() is realtime-capable.
		 * It might not be on every UNIX!?
		 */
		memset((char *)out + r, 0, len - r);
		port->buffer_xrun |= len - r > 0;
	}

	/*
	 * MIDI processing.
	 * FIXME: We could try to preserve the MIDI event timing by filling
	 * out the stream with 0 samples.
	 */
	midi_in = jack_port_get_buffer(midi_port.jack_port, nframes);
	midi_events = jack_midi_get_event_count(midi_in);

	for (int i = 0; i < midi_events; i++) {
		jack_midi_event_t event;
		applause_midi_sample sample = 0;

		jack_midi_event_get(&event, midi_in, i);

		/*
		 * We don't know if event.buffer is large enough
		 * to be dereferenced as an applause_midi_sample pointer, so
		 * we manually mangle it into an integer based on
		 * buffer.size.
		 */
		for (int i = 0; i < event.size; i++)
			sample |= event.buffer[i] << (i*8);

		jack_ringbuffer_write(midi_port.buffer,
		                      (const char *)&sample, sizeof(sample));
	}

	return 0;
}

/**
 * JACK calls this shutdown_callback if the server ever shuts down or
 * decides to disconnect the client.
 */
static void
jack_shutdown(void *arg)
{
	exit(EXIT_FAILURE);
}

static int
init_audio(int buffer_size)
{
	size_t buffer_bytes;
	const char **ports;
	const char *client_name = "applause";
	const char *server_name = NULL;
	jack_options_t options = JackNullOption;
	jack_status_t status;

	/* open a client connection to the JACK server */

	client = jack_client_open(client_name, options, &status, server_name);
	if (client == NULL) {
		fprintf(stderr, "jack_client_open() failed, "
			        "status = 0x%2.0x\n", status);
		if (status & JackServerFailed) {
			fprintf(stderr, "Unable to connect to JACK server\n");
		}
		return 1;
	}
	if (status & JackServerStarted)
		fprintf(stderr, "JACK server started\n");

	if (status & JackNameNotUnique) {
		client_name = jack_get_client_name(client);
		fprintf(stderr, "unique name `%s' assigned\n", client_name);
	}

	/*
	 * Tell the JACK server to call `jack_process()' whenever
	 * there is work to be done.
	 * This will fill all output buffers and handle MIDI events.
	 */
	jack_set_process_callback(client, jack_process, NULL);

	/*
	 * Tell the JACK server to call `jack_shutdown()' if
	 * it ever shuts down, either entirely, or if it
	 * just decides to stop calling us.
	 */
	jack_on_shutdown(client, jack_shutdown, NULL);

	/*
	 * Calculate the buffer size in bytes given the `buffer_size`
	 * in milliseconds.
	 * Make sure it is at least the Jack server's buffer size,
	 * else jack_process() is very likely not able to provide
	 * enough bytes.
	 * FIXME: The Jack server's sample rate and buffer size can
	 * theoretically change at runtime but currently,
	 * the buffer size is not adapted
	 * which means it could be too small after the change.
	 */
	buffer_bytes = sizeof(jack_default_audio_sample_t)*
	               MAX(jack_get_sample_rate(client)*buffer_size/1000,
	                   jack_get_buffer_size(client));

	/*
	 * Create output ports
	 */
	free(output_ports);
	output_ports = calloc(output_ports_count, sizeof(*output_ports));

	for (int i = 0; i < output_ports_count; i++) {
		applause_output_port *port = output_ports + i;
		char name[256];

		/*
		 * NOTE: Port names must be unique.
		 * FIXME: Make the names configurable.
		 */
		snprintf(name, sizeof(name), "output_%d", i+1);

		port->jack_port = jack_port_register(client, name,
		                                     JACK_DEFAULT_AUDIO_TYPE,
		                                     JackPortIsOutput, 0);
		if (!port->jack_port) {
			fprintf(stderr, "No more JACK ports available\n");
			return 1;
		}

		/*
		 * Initialize ring buffer of samples.
		 * The semaphore is initialized with the same size
		 * since it represents the available bytes in the ring
		 * buffer.
		 */
		port->buffer = jack_ringbuffer_create(buffer_bytes);
		if (!port->buffer) {
			fprintf(stderr, "cannot create ringbuffer\n");
			return 1;
		}

		port->buffer_sem = svsem_init(buffer_bytes);
		if (port->buffer_sem < 0) {
			fprintf(stderr, "error initializing semaphore\n");
			return 1;
		}
	}

	/*
	 * Create MIDI input port
	 */
	midi_port.jack_port = jack_port_register(client, "midi_input",
	                                         JACK_DEFAULT_MIDI_TYPE,
	                                         JackPortIsInput, 0);
	if (midi_port.jack_port == NULL) {
		fprintf(stderr, "no more JACK ports available\n");
		return 1;
	}

	/*
	 * Make sure that the MIDI buffer is large enough to hold the maximum
	 * number of MIDI samples delivered in jack_process().
	 */
	midi_port.buffer = jack_ringbuffer_create(sizeof(applause_midi_sample)*
	                                          jack_get_buffer_size(client));
	if (!midi_port.buffer) {
		fprintf(stderr, "cannot create ringbuffer\n");
		return 1;
	}

	/* Tell the JACK server that we are ready to roll.  Our
	 * process() callback will start running now. */

	if (jack_activate(client)) {
		fprintf (stderr, "cannot activate client\n");
		return 1;
	}

	/* Connect the ports.  You can't do this before the client is
	 * activated, because we can't make connections to clients
	 * that aren't running.  Note the confusing (but necessary)
	 * orientation of the driver backend ports: playback ports are
	 * "input" to the backend, and capture ports are "output" from
	 * it.
	 */
	ports = jack_get_ports(client, NULL, NULL,
	                       JackPortIsPhysical | JackPortIsInput);
	if (ports == NULL) {
		fprintf(stderr, "no physical playback ports\n");
		return 0;
	}

	for (int i = 0; i < output_ports_count && ports[i]; i++) {
		if (jack_connect(client, jack_port_name(output_ports[i].jack_port),
		                 ports[i])) {
			fprintf(stderr, "Cannot connect port %s to %s\n",
			        jack_port_name(output_ports[i].jack_port),
			        ports[i]);
		}
	}

	jack_free(ports);

	return 0;
}

/**
 * Push one Jack sample into the ring buffer.
 *
 * This function should be called from the Stream:play()
 * implementation, which is faster than calling Lua functions
 * from a C implementation of Stream:play() using the Lua C API
 * (supposedly - I could not reproduce this).
 *
 * @param output_port_id The number of the output port.
 *                       The first one being 1.
 * @param sample_double The audio sample as a double (compatible
 *                      with lua_Number).
 *                      This is speed relevant since otherwise
 *                      LuaJIT tries to translate to Jack's default
 *                      float type.
 */
enum applause_audio_state
applause_push_sample(int output_port_id, double sample_double)
{
	applause_output_port *port;
	jack_default_audio_sample_t sample =
			(jack_default_audio_sample_t)sample_double;

	/*
	 * NOTE: The alternative to reporting invalid port Ids here
	 * would be exporting output_ports_count, so the Lua code can
	 * check it and assert()ing here instead.
	 */
	if (unlikely(output_port_id < 1 || output_port_id > output_ports_count))
		return APPLAUSE_AUDIO_INVALID_PORT;
	port = output_ports + output_port_id - 1;

	/*
	 * We are about to "consume" one free sample in the buffer.
	 * This can block when the buffer is full.
	 * After this operation, there is __at least__ one sample free
	 * in buffer since jack_process() will only read from the
	 * buffer.
	 */
	svsem_op(port->buffer_sem, -(int)sizeof(sample));

	jack_ringbuffer_write(port->buffer, (const char *)&sample,
	                      sizeof(sample));

	if (unlikely(port->buffer_xrun)) {
		port->buffer_xrun = 0;
		return APPLAUSE_AUDIO_XRUN;
	}

	return APPLAUSE_AUDIO_OK;
}

/**
 * Pull one MIDI sample from the ring buffer.
 *
 * This can be called from MIDIStream's tick function.
 *
 * @return A MIDI event encoded into a MIDI sample, or 0.
 */
applause_midi_sample
applause_pull_midi_sample(void)
{
	applause_midi_sample sample = 0;

	jack_ringbuffer_read(midi_port.buffer, (char *)&sample, sizeof(sample));

	return sample;
}

static int
l_Stream_fork(lua_State *L)
{
	int top = lua_gettop(L);
	pid_t child_pid;

	luaL_argcheck(L, top == 1, top, "Stream object expected");
	luaL_checktype(L, 1, LUA_TTABLE);

	/*
	 * Query the parent client's ports.
	 * We need to clone them in the child.
	 */
	//port_names = jack_get_ports(client, NULL, NULL, 0);

	/*
	 * Deactivate parent client temporarily, so forking works
	 * safely.
	 * This is possible since we can assume that fork() is only
	 * called interactively - so the parent client is not currently
	 * generating samples.
	 */
	jack_client_close(client);
	//jack_deactivate(client);

	child_pid = fork();
	if (child_pid != 0) {
		/* in the parent process */
		//jack_activate(client);
		/* FIXME: Pass on the real buffer size */
		init_audio(DEFAULT_BUFFER_SIZE);

		/* call Client:new(child_pid) */
		lua_getglobal(L, "Client");
		lua_getfield(L, -1, "new");
		/* make Client parameter to new() */
		lua_insert(L, -2);
		lua_pushinteger(L, child_pid);
		lua_call(L, 2, 1);

		return 1;
	}

	//jack_client_close(client);

	/* FIXME: Pass on the real buffer size */
	init_audio(DEFAULT_BUFFER_SIZE);

	for (;;) {
		/*
		 * Wait for start of playback (SIGUSR1)
		 */
		while (!playback)
			pause();

		/* get play() method */
		lua_getfield(L, -1, "play");
		luaL_checktype(L, -1, LUA_TFUNCTION);
		/* duplicate the object table */
		lua_pushvalue(L, 1);

		lua_call(L, 1, 0);
	}

	/* never reached */
	return 0;
}

/* stolen from luajit.c :-) */
static int
traceback(lua_State *L)
{
	if (!lua_isstring(L, 1)) { /* Non-string error object? Try metamethod. */
		if (lua_isnoneornil(L, 1) ||
		    !luaL_callmeta(L, 1, "__tostring") ||
		    !lua_isstring(L, -1))
			return 1;  /* Return non-string error object. */
		lua_remove(L, 1);  /* Replace object by result of __tostring metamethod. */
	}

	luaL_traceback(L, L, lua_tostring(L, 1), 1);

	return 1;
}

static void
do_command(lua_State *L, const char *command)
{
	int stack_top;
	char *buffer;

	/*
	 * EXPERIMENTAL: I found that performance decreases
	 * progressively when running one and the same command.
	 * This is probably because the limited space for the
	 * compiled code gets filled up rapidly and LuaJIT
	 * resorts to using the interpreter.
	 * This flashes the entire code cache for every
	 * new command and improves matters.
	 * TODO: It would probably be a good idea to increase the
	 * total size of the code cache and give some indication
	 * of its usage (e.g. in the prompt).
	 */
	luaJIT_setmode(L, 0, LUAJIT_MODE_ENGINE | LUAJIT_MODE_FLUSH);

	/* the error hanlder function for lua_pcall() */
	lua_pushcfunction(L, traceback);

	stack_top = lua_gettop(L);

	/*
	 * AFAIK, we cannot support automatic printing of values
	 * returned by expressions.
	 * The chunk must return something (using `return`).
	 * We cannot always prepend a `return` since AFAIK it is
	 * impossible to embedded statements into expressions.
	 * Therefore we support "=" as a shortcut to "return" just
	 * like the luajit shell does.
	 */
	if (*command == '=') {
		if (asprintf(&buffer, "return %s", command+1) < 0)
			buffer = NULL;
	} else {
		buffer = strdup(command);
	}
	assert(buffer != NULL);

	if (luaL_loadstring(L, buffer) || lua_pcall(L, 0, LUA_MULTRET, -2)) {
		fprintf(stderr, "Error.\n");
	}

	free(buffer);

	/*
	 * Print values left on the stack:
	 * This includes error messages left by lua_pcall()
	 */
	if (lua_gettop(L) > stack_top) {
		lua_getglobal(L, "print");
		lua_insert(L, stack_top + 1);

		if (lua_pcall(L, lua_gettop(L) - stack_top - 1, 0, 0)) {
			fprintf(stderr, "Error executing print().\n");
			/* try to continue */
		}
	}

	/* pop the traceback function */
	lua_remove(L, -1);
}

/**
 * Mutex protecting Lua state access after the command server
 * has been launched.
 */
static pthread_mutex_t lua_mutex = PTHREAD_MUTEX_INITIALIZER;

/**
 * Thread monitoring the client file desciptor.
 * This will raise interrupt the current Lua script if the remote
 * end closes its write part of the socket (just like SIGINT would).
 * That's why we need the length header at the beginning
 * of requests.
 * There seems to be no easy way to find out whether the
 * read part has been shut down (or the entire connection)
 * without doing I/O.
 * On the other hand, this is how "socat -,ignoreeof TCP:127.0.0.1:10000"
 * behaves and socat does close the connection properly when
 * interrupted in contrast to netcat.
 * So either we stay with this solution or implement a
 * stand-alone applause_cat just for sending commands (which
 * would give us more liberties, though).
 */
static void *
monitor_thread_cb(void *user_data)
{
	int fd = *(int *)user_data;

	struct pollfd pfd = {fd, POLLRDHUP};

	/*
	 * poll() is a cancellation point, so when
	 * do_command() terminates, we will cancel here.
	 */
	while (poll(&pfd, 1, -1) != 1 ||
	       !(pfd.revents & (POLLRDHUP | POLLERR | POLLHUP)));

	applause_interrupt();
	return NULL;
}

/*
 * FIXME: This should perhaps be the only interface to the
 * applause Lua state with the REPL loop implemented as a Lua
 * script. This solution would simplify the C code and avoid
 * threading. Also, then we could have multiple running
 * REPL loops.
 */
static void *
command_server(void *user_data)
{
	lua_State *L = user_data;
	int socket_fd;
	struct sockaddr_in server;

	socket_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (socket_fd < 0) {
		perror("socket");
		return NULL;
	}

	server.sin_addr.s_addr = inet_addr(CMD_SERVER_IP);
	server.sin_family = AF_INET;
	server.sin_port = htons(CMD_SERVER_PORT);

	if (bind(socket_fd, (struct sockaddr *)&server,
	    sizeof(server)) < 0) {
		perror("bind");
		close(socket_fd);
		return NULL;
	}

	if (listen(socket_fd, 1) < 0) {
		perror("listen");
		close(socket_fd);
		return NULL;
	}

	for (;;) {
		char length_buf[5 + 1];
		size_t length;
		char *command;

		pthread_t monitor_thread;

		int old_stdout, old_stderr;

		int client_fd = accept(socket_fd, NULL, NULL);

		if (client_fd < 0) {
			perror("accept");
			break;
		}

		if (read(client_fd, length_buf, sizeof(length_buf)) <
		    sizeof(length_buf)) {
			/* just close the connection and continue */
			/*
			 * FIXME: We might want to handle errors
			 * (e.g. signal interruptions)
			 */
			close(client_fd);
			continue;
		}

		/* parse message length */
		length = atoi(length_buf);
		command = malloc(length+1);
		assert(command != NULL);

		if (read(client_fd, command, length) < length) {
			/* just close the connection and continue */
			/*
			 * FIXME: We might want to handle errors
			 * (e.g. signal interruptions)
			 */
			free(command);
			close(client_fd);
			continue;
		}
		command[length] = '\0';

		pthread_mutex_lock(&lua_mutex);

		/*
		 * Start another thread monitoring client_fd for raising SIGINT
		 * in case the remote end terminates prematurely.
		 * This has the effect of ^C in SciTECO interrupting
		 * the current command (e.g. play()) at least when using
		 * socat (see monitor_thread_cb()).
		 * A monitoring thread is started instead of running do_command()
		 * in yet another thread since the monitoring thread is easier
		 * to terminate.
		 */
		if (pthread_create(&monitor_thread, NULL, monitor_thread_cb, &client_fd)) {
			perror("pthread_create");
			pthread_mutex_unlock(&lua_mutex);
			free(command);
			close(client_fd);
			continue;
		}

		/*
		 * Redirect stdout and stderr to client_fd.
		 * This way, we capture all the output that the command
		 * may have.
		 */
		old_stdout = dup(1);
		close(1);
		dup2(client_fd, 1);
		old_stderr = dup(2);
		close(2);
		dup2(client_fd, 2);

		/*
		 * FIXME: It would be nice to catch broken sockets
		 * (e.g. waiting for play() has been interrupted), in order
		 * to raise SIGINT.
		 * However it seems we'd need another thread for that running
		 * do_command(), so we can select() the fd and eventually
		 * join the thread.
		 */
		do_command(L, command);
		free(command);

		/*
		 * Terminate the monitoring thread.
		 * It will already be terminated if the remote end terminates
		 * prematurely.
		 */
		pthread_cancel(monitor_thread);
		pthread_join(monitor_thread, NULL);

		/*
		 * Restore stdout/stderr.
		 */
		close(1);
		dup2(old_stdout, 1);
		close(old_stdout);
		close(2);
		dup2(old_stderr, 2);
		close(old_stderr);

		pthread_mutex_unlock(&lua_mutex);

		close(client_fd);
	}

	close(socket_fd);
	return NULL;
}

typedef struct NativeMethod {
	const char *class_name;
	const char *method_name;
	lua_CFunction func;
} NativeMethod;

static const NativeMethod native_methods[] = {
	{"Stream",             "fork",     l_Stream_fork},
	{NULL, NULL, NULL}
};

static void
usage(const char *program)
{
	printf("%s [-h] [-o OUTPUT] [-b SIZE] [SCRIPT [ARGS]]\n"
	       "\tOUTPUT\tNumber of output ports to reserve (default: %d)\n"
	       "\tSIZE\tMinimum size of the output buffer in milliseconds (default: %dms)\n",
	       program,
	       DEFAULT_NUMBER_OF_OUTPUT_PORTS, DEFAULT_BUFFER_SIZE);
}

int
main(int argc, char **argv)
{
	int opt;
	int buffer_size = DEFAULT_BUFFER_SIZE;
	struct sigaction signal_action;

	lua_State *L;

	pthread_t command_server_thread;

	while ((opt = getopt(argc, argv, "ho:b:")) >= 0) {
		switch (opt) {
		case '?':
		case 'h': /* get help */
			usage(argv[0]);
			return 0;
		case 'o': /* output ports */
			output_ports_count = atoi(optarg);
			break;
		case 'b': /* buffer size */
			buffer_size = atoi(optarg);
			break;
		}
	}

	memset(&signal_action, 0, sizeof(signal_action));
	signal_action.sa_handler = signal_handler;
	sigaction(SIGINT, &signal_action, NULL);
	sigaction(SIGUSR1, &signal_action, NULL);
	sigaction(SIGUSR2, &signal_action, NULL);

	signal_action.sa_handler = SIG_IGN;
	sigaction(SIGPIPE, &signal_action, NULL);

	L = L_global = luaL_newstate();
	if (!L) {
		fprintf(stderr, "Error creating Lua state.\n");
		exit(EXIT_FAILURE);
	}

	luaL_openlibs(L);

	lua_pushcfunction(L, traceback);

	if (luaL_loadfile(L, LUA_MODULE) || lua_pcall(L, 0, 0, -2)) {
		fprintf(stderr, "Error loading Lua module %s: %s\n",
		        LUA_MODULE, lua_tostring(L, -1));
		exit(EXIT_FAILURE);
	}

	/* remove traceback function */
	lua_remove(L, -1);

	if (init_audio(buffer_size))
		/* error has already been printed */
		return EXIT_FAILURE;

	/*
	 * Register native C functions.
	 * This may be unefficient when registering multiple
	 * methods in the same class.
	 */
	for (const NativeMethod *method = native_methods;
	     method->class_name;
	     method++) {
		lua_getglobal(L, method->class_name);
		lua_pushcfunction(L, method->func);
		lua_setfield(L, -2, method->method_name);
		lua_pop(L, 1); /* pop class table */
	}

	/*
	 * Set global `samplerate`
	 */
	lua_pushinteger(L, (lua_Integer)jack_get_sample_rate(client));
	lua_setglobal(L, "samplerate");

	if (optind < argc) {
		/*
		 * Execute script
		 */
		lua_createtable(L, argc-optind, 0);
		for (int i = optind; i < argc; i++) {
			lua_pushinteger(L, i-optind);
			lua_pushstring(L, argv[i]);
			lua_settable(L, -3);
		}
		lua_setglobal(L, "arg");

		lua_pushcfunction(L, traceback);

		if (luaL_loadfile(L, argv[optind]) || lua_pcall(L, 0, 0, -2)) {
			fprintf(stderr, "Error running %s: %s\n",
			        argv[optind], lua_tostring(L, -1));
			exit(EXIT_FAILURE);
		}

		/* remove traceback function */
		lua_remove(L, -1);
	} else {
		/*
		 * Load the libhistory file.
		 */
		using_history();
		read_history(APPLAUSE_HISTORY);

		/*
		 * Launch the command server.
		 */
		if (pthread_create(&command_server_thread, NULL, command_server, L)) {
			perror("pthread_create");
			exit(EXIT_FAILURE);
		}

		/*
		 * Main REPL loop.
		 * Since the command server has been launched, all lua state access
		 * must be synchronized using lua_mutex.
		 */
		for (;;) {
			/*
			 * FIXME: Get global _PROMPT or _PROMPT2
			 */
			char *line = readline("> ");

			if (!line) {
				putchar('\n');
				break;
			}

			pthread_mutex_lock(&lua_mutex);
			do_command(L, line);
			pthread_mutex_unlock(&lua_mutex);

			if (*line)
				add_history(line);

			free(line);
		}

		/*
		 * Write libhistory file
		 */
		if (write_history(APPLAUSE_HISTORY))
			perror("write_history");

		/*
		 * FIXME: Shut down connection server.
		 */
	}

	/*
	 * FIXME: Clean up properly.
	 */
#if 0
	for (int i = 0; i < output_ports_count; i++)
		svsem_free(output_ports[i].buffer_sem);
	free(output_ports);
#endif

	lua_close(L);

	return 0;
}