#include "audio.h"

int init_PulseInputDevice(PulseInputDevice* dev, int sample_rate, int channels, char* app_name, char *stream_name, char* device, pa_buffer_attr* buffer_attr) {
	if (dev->initialized) return -1;
	pa_sample_spec sample_spec = {
		.format = PA_SAMPLE_FLOAT32NE,
		.channels = channels,
		.rate = sample_rate
	};
	pa_buffer_attr new_buffer_attr = *buffer_attr;
	dev->sample_spec = sample_spec;
	dev->buffer_attr = new_buffer_attr;

	dev->app_name = strdup(app_name);
	dev->stream_name = strdup(stream_name);
	dev->device = strdup(device);

	int error;
	dev->dev = pa_simple_new(
		NULL,
		app_name,
		PA_STREAM_RECORD,
		device,
		stream_name,
		&sample_spec,
		NULL,
		&new_buffer_attr,
		&error
	);
	if (!dev->dev) return error;
	dev->initialized = 1;
	return 0;
}

int init_PulseInputDevicef(PulseInputDevice* dev, int sample_rate, int channels, char* app_name, char *stream_name, char* device, pa_buffer_attr* buffer_attr, enum pa_sample_format format) {
	if (dev->initialized) return -1;
	pa_sample_spec sample_spec = {
		.format = format,
		.channels = channels,
		.rate = sample_rate
	};
	pa_buffer_attr new_buffer_attr = *buffer_attr;
	dev->sample_spec = sample_spec;
	dev->buffer_attr = new_buffer_attr;

	dev->app_name = strdup(app_name);
	dev->stream_name = strdup(stream_name);
	dev->device = strdup(device);

	int error;
	dev->dev = pa_simple_new(
		NULL,
		app_name,
		PA_STREAM_RECORD,
		device,
		stream_name,
		&sample_spec,
		NULL,
		&new_buffer_attr,
		&error
	);
	if (!dev->dev) return error;
	dev->initialized = 1;
	return 0;
}

int read_PulseInputDevice(PulseInputDevice* dev, float* buffer, size_t size) {
	if (!dev->initialized) return -1;
	int error;
	if (pa_simple_read(dev->dev, buffer, size, &error) < 0) return error;
	return 0;
}

int read_PulseInputDevicef(PulseInputDevice* dev, void* buffer, size_t size) {
	if (!dev->initialized) return -1;
	int error;
	if (pa_simple_read(dev->dev, buffer, size, &error) < 0) return error;
	return 0;
}

void free_PulseInputDevice(PulseInputDevice* dev) {
	if (dev->dev && dev->initialized) pa_simple_free(dev->dev);
	free(dev->app_name);
	free(dev->stream_name);
	free(dev->device);
	dev->initialized = 0;
}

int init_PulseOutputDevice(PulseOutputDevice* dev, int sample_rate, int channels, char* app_name, char *stream_name, char* device, pa_buffer_attr* buffer_attr) {
	if (dev->initialized) return -1;
	pa_sample_spec sample_spec = {
		.format = PA_SAMPLE_FLOAT32NE,
		.channels = channels,
		.rate = sample_rate
	};
	pa_buffer_attr new_buffer_attr = *buffer_attr;
	dev->sample_spec = sample_spec;
	dev->buffer_attr = new_buffer_attr;

	dev->app_name = strdup(app_name);
	dev->stream_name = strdup(stream_name);
	dev->device = strdup(device);

	int error;
	dev->dev = pa_simple_new(
		NULL,
		app_name,
		PA_STREAM_PLAYBACK,
		device,
		stream_name,
		&sample_spec,
		NULL,
		&new_buffer_attr,
		&error
	);
	if (!dev->dev) return error;
	dev->initialized = 1;
	return 0;
}

int init_PulseOutputDevicef(PulseOutputDevice* dev, int sample_rate, int channels, char* app_name, char *stream_name, char* device, pa_buffer_attr* buffer_attr, enum pa_sample_format format) {
	if (dev->initialized) return -1;
	pa_sample_spec sample_spec = {
		.format = format,
		.channels = channels,
		.rate = sample_rate
	};
	pa_buffer_attr new_buffer_attr = *buffer_attr;
	dev->sample_spec = sample_spec;
	dev->buffer_attr = new_buffer_attr;

	dev->app_name = strdup(app_name);
	dev->stream_name = strdup(stream_name);
	dev->device = strdup(device);

	int error;
	dev->dev = pa_simple_new(
		NULL,
		app_name,
		PA_STREAM_PLAYBACK,
		device,
		stream_name,
		&sample_spec,
		NULL,
		&new_buffer_attr,
		&error
	);
	if (!dev->dev) return error;
	dev->initialized = 1;
	return 0;
}

int write_PulseOutputDevice(PulseOutputDevice* dev, float* buffer, size_t size) {
	if (!dev->initialized) return -1;
	int error;
	if (pa_simple_write(dev->dev, buffer, size, &error) < 0) return error;
	return 0;
}

int write_PulseOutputDevicef(PulseOutputDevice* dev, void* buffer, size_t size) {
	if (!dev->initialized) return -1;
	int error;
	if (pa_simple_write(dev->dev, buffer, size, &error) < 0) return error;
	return 0;
}

void free_PulseOutputDevice(PulseOutputDevice* dev) {
	if (dev->dev && dev->initialized) pa_simple_free(dev->dev);
	free(dev->app_name);
	free(dev->stream_name);
	free(dev->device);
	dev->initialized = 0;
}