/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * get_line.c
 *
 * Copyright (C) 2019 David Oberhollenzer <goliath@infraroot.at>
 */
#include "internal.h"

static void ltrim(char *buffer)
{
	size_t i = 0;

	while (isspace(buffer[i]))
		++i;

	if (i > 0)
		memmove(buffer, buffer + i, strlen(buffer + i) + 1);
}

static void rtrim(char *buffer)
{
	size_t i = strlen(buffer);

	while (i > 0 && isspace(buffer[i - 1]))
		--i;

	buffer[i] = '\0';
}

static size_t trim(char *buffer, int flags)
{
	if (flags & ISTREAM_LINE_LTRIM)
		ltrim(buffer);

	if (flags & ISTREAM_LINE_RTRIM)
		rtrim(buffer);

	return strlen(buffer);
}

int istream_get_line(istream_t *strm, char **out,
		     size_t *line_num, int flags)
{
	char *line = NULL, *new;
	size_t line_len = 0;

	for (;;) {
		bool have_line = false;
		size_t i, count, avail;
		const sqfs_u8 *ptr;
		int ret;

		ret = istream_get_buffered_data(strm, &ptr, &avail, 0);
		if (ret < 0)
			goto fail_free;
		if (ret > 0) {
			if (line_len == 0)
				goto out_eof;

			line_len = trim(line, flags);
			if (line_len > 0 ||!(flags & ISTREAM_LINE_SKIP_EMPTY))
				break;

			goto out_eof;
		}

		for (i = 0; i < avail; ++i) {
			if (ptr[i] == '\n')
				break;
		}

		if (i < avail) {
			count = i++;
			have_line = true;
		} else {
			count = i;
		}

		new = realloc(line, line_len + count + 1);
		if (new == NULL)
			goto fail_errno;

		line = new;
		memcpy(line + line_len, ptr, count);
		line_len += count;
		line[line_len] = '\0';

		istream_advance_buffer(strm, i);

		if (have_line) {
			if (line_len > 0 && line[line_len - 1] == '\r')
				line[--line_len] = '\0';

			line_len = trim(line, flags);
			if (line_len > 0 || !(flags & ISTREAM_LINE_SKIP_EMPTY))
				break;

			free(line);
			line = NULL;
			*line_num += 1;
		}
	}

	new = realloc(line, line_len + 1);
	if (new != NULL)
		line = new;

	*out = line;
	return 0;
fail_errno:
	fprintf(stderr, "%s: " PRI_SZ ": %s.\n", strm->get_filename(strm),
		*line_num, strerror(errno));
fail_free:
	free(line);
	*out = NULL;
	return -1;
out_eof:
	free(line);
	*out = NULL;
	return 1;
}