/*
 * Copyright 2005-2021 Gentoo Foundation
 * Distributed under the terms of the GNU General Public License v2
 *
 * Copyright 2005-2010 Ned Ludd        - <solar@gentoo.org>
 * Copyright 2005-2014 Mike Frysinger  - <vapier@gentoo.org>
 */

#include "main.h"
#include "applets.h"

#include <string.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "basename.h"
#include "safe_io.h"
#include "scandirat.h"
#include "xpak.h"

#define QXPAK_FLAGS "lxcd:O" COMMON_FLAGS
static struct option const qxpak_long_opts[] = {
	{"list",      no_argument, NULL, 'l'},
	{"extract",   no_argument, NULL, 'x'},
	{"create",    no_argument, NULL, 'c'},
	{"dir",        a_argument, NULL, 'd'},
	{"stdout",    no_argument, NULL, 'O'},
	COMMON_LONG_OPTS
};
static const char * const qxpak_opts_help[] = {
	"List the contents of an archive",
	"Extract the contents of an archive",
	"Create an archive of a directory/files",
	"Change to specified directory",
	"Write files to stdout",
	COMMON_OPTS_HELP
};
#define qxpak_usage(ret) usage(ret, QXPAK_FLAGS, qxpak_long_opts, qxpak_opts_help, NULL, lookup_applet_idx("qxpak"))

static char xpak_stdout;

struct qxpak_cb {
	int dir_fd;
	int argc;
	char **argv;
	bool extract;
};

static void
_xpak_callback(
	void *ctx,
	char *pathname,
	int pathname_len,
	int data_offset,
	int data_len,
	char *data)
{
	FILE *out;
	struct qxpak_cb *xctx = (struct qxpak_cb *)ctx;

	/* see if there is a match when there is a selection */
	if (xctx->argc > 0) {
		int i;
		for (i = 0; i < xctx->argc; i++) {
			if (xctx->argv[i] && !strcmp(pathname, xctx->argv[i])) {
				xctx->argv[i] = NULL;
				break;
			}
		}
		if (i == xctx->argc)
			return;
	}

	if (verbose == 0 + (xctx->extract ? 1 : 0))
		printf("%.*s\n", pathname_len, pathname);
	else if (verbose == 1 + (xctx->extract ? 1 : 0))
		printf("%.*s: %d byte%s\n",
				pathname_len, pathname,
				data_len, (data_len > 1 ? "s" : ""));
	else if (verbose != 0)
		printf("%.*s: %d byte%s @ offset byte %d\n",
				pathname_len, pathname,
				data_len, (data_len > 1 ? "s" : ""), data_offset);

	if (!xctx->extract)
		return;

	if (!xpak_stdout) {
		int fd = openat(xctx->dir_fd, pathname,
				O_WRONLY | O_CLOEXEC | O_CREAT | O_TRUNC, 0644);
		if (fd < 0)
			return;
		out = fdopen(fd, "w");
		if (!out)
			return;
	} else
		out = stdout;

	fwrite(data + data_offset, 1, data_len, out);

	if (!xpak_stdout)
		fclose(out);
}

int qxpak_main(int argc, char **argv)
{
	enum { XPAK_ACT_NONE, XPAK_ACT_LIST, XPAK_ACT_EXTRACT, XPAK_ACT_CREATE };
	int i, ret;
	char *xpak;
	char action = XPAK_ACT_NONE;
	struct qxpak_cb cbctx;

	xpak_stdout = 0;
	cbctx.dir_fd = AT_FDCWD;
	cbctx.extract = false;

	while ((i = GETOPT_LONG(QXPAK, qxpak, "")) != -1) {
		switch (i) {
		COMMON_GETOPTS_CASES(qxpak)
		case 'l': action = XPAK_ACT_LIST; break;
		case 'x': action = XPAK_ACT_EXTRACT; break;
		case 'c': action = XPAK_ACT_CREATE; break;
		case 'O': xpak_stdout = 1; break;
		case 'd':
			if (cbctx.dir_fd != AT_FDCWD)
				err("Only use -d once");
			cbctx.dir_fd = open(optarg, O_RDONLY|O_CLOEXEC|O_PATH);
			if (cbctx.dir_fd < 0)
				errp("Could not open directory %s", optarg);
			break;
		}
	}
	if (optind == argc || action == XPAK_ACT_NONE)
		qxpak_usage(EXIT_FAILURE);

	xpak = argv[optind++];
	argc -= optind;
	argv += optind;
	cbctx.argc = argc;
	cbctx.argv = argv;

	switch (action) {
	case XPAK_ACT_LIST:
		ret = xpak_list(xpak, &cbctx, &_xpak_callback);
		break;
	case XPAK_ACT_EXTRACT:
		cbctx.extract = true;
		ret = xpak_extract(xpak, &cbctx, &_xpak_callback);
		break;
	case XPAK_ACT_CREATE:
		ret = xpak_create(cbctx.dir_fd, xpak, argc, argv, 0, verbose);
		break;
	default:
		ret = -1;
	}
	ret = ret < 0 ? EXIT_FAILURE : 0;

	if (cbctx.dir_fd != AT_FDCWD)
		close(cbctx.dir_fd);

	/* warn about non-matched args */
	if (argc > 0 && (action == XPAK_ACT_LIST || action == XPAK_ACT_EXTRACT))
		for (i = 0; i < argc; ++i)
			if (argv[i])
				warn("Could not locate '%s' in archive", argv[i]);

	return ret;
}