/*

	pkgit - package_t it!

	Copyright (C) 2026 dacctal
	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 2 of the License, or
	(at your option) any later version.
	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>. */

#include <string.h>
#include <unistd.h>

#include "pkg.h"

#include "files.h"
#include "globs.h"
#include "log.h"
#include "pkgit_lua.h"
#include "str.h"

static str get_destdir(str *cwd, str *arg) {
	str result = {0};
	if (str_first(arg) == '.' && arg->len == 1) {
		result = str_format("%.*s/%.*s", str_fmt(&inst_dirs.src), str_fmt(cwd));
	} else {
		str name = {0};
		if (str_find_char(arg, '/') != 0) name = str_from_after_delim(arg, '/');
		else str_copy_into(&name, arg);
		result = str_format("%.*s/%.*s", str_fmt(&inst_dirs.src), str_fmt(&name));
		str_free(&name);
	}
	return result;
}

static str get_pkgsrc(package_t pkg) {
	if (pkg.is_local == 1)
		return str_format("%.*s/%.*s", str_fmt(&inst_dirs.src), str_fmt(&pkg.name));
	else
		return str_format("%.*s/%.*s/%.*s", str_fmt(&inst_dirs.src),
			str_fmt(&pkg.name), str_fmt(&pkg.version));
}

static void assign_pkg_version(package_t *pkg, str *new_arg_str) {
	if (str_find_char(new_arg_str, '@')) {
		str tmp_arg = str_from_after_delim(new_arg_str, '@');
		if (str_find_char(new_arg_str, ',')) {
			str tmp_trg = str_from_after_delim(&tmp_arg, ',');
			pkg->version = str_from_after_delim(new_arg_str, '@');
			pkg->version.len -= tmp_trg.len;
			str_free(&tmp_trg);
		} else {
			pkg->version = str_from_after_delim(new_arg_str, '@');
		}
		if (str_find_char(&pkg->version, ',')) pkg->version.len--;
		str_free(&tmp_arg);
		pkg->name.len -= pkg->version.len + 1;
	} else str_copy_cstr_into(&pkg->version, "HEAD");;
}

static void assign_pkg_target(package_t *pkg, str *new_arg_str) {
	if (str_find_char(new_arg_str, ',')) {
		str tmp_arg = str_from_after_delim(new_arg_str, ',');
		if (str_find_char(new_arg_str, '@')) {
			str tmp_ver = str_from_after_delim(&tmp_arg, '@');
			pkg->target = str_from_after_delim(new_arg_str, ',');
			pkg->target.len -= tmp_ver.len;
			str_free(&tmp_ver);
		} else {
			pkg->target = str_from_after_delim(new_arg_str, ',');
		}
		if (str_find_char(&pkg->target, '@')) pkg->target.len--;
		str_free(&tmp_arg);
		pkg->name.len -= pkg->target.len + 1;
	} else if (!flags.verbose) {
		str_copy_cstr_into(&pkg->target, "quiet");
	} else str_copy_cstr_into(&pkg->target, "quiet");
}

package_t pkg_create(str *arg) {
	package_t pkg = { .is_local = false, };

	assign_pkg_version(&pkg, arg);
	assign_pkg_target(&pkg, arg);
	if (str_find_char(arg, '@')) {
		str tmp = str_from_after_delim(arg, '@');
		arg->len -= tmp.len + 1;
		str_free(&tmp);
	} if (str_find_char(arg, ',')) {
		str tmp = str_from_after_delim(arg, ',');
		arg->len -= tmp.len + 1;
		str_free(&tmp);
	}

	char cwd[MAX_PATH_LEN];
	getcwd(cwd, MAX_PATH_LEN);
	str cwd_str = mstr(cwd);
	str dest_dir = get_destdir(&cwd_str, arg);
	bool is_installed_locally = is_directory(dest_dir.data);

	if (strncmp(arg->data, "http", 4) == 0 || strncmp(arg->data, "ssh", 3) == 0) {
		str_copy_into(&pkg.url, arg);
		pkg.name = str_from_after_delim(arg, '/');
	} else if (str_equal_cstr(arg, ".")) {
		str_copy_cstr_into(&pkg.url, "");
		pkg.name = str_from_after_delim(&cwd_str, '/');
		pkg.is_local = true;
	} else if (pkg_exists(arg)) {
		str_copy_into(&pkg.name, arg);
		pkg.url = pkg_get_url(&pkg.name);
	} else if (is_installed_locally) {
		str_copy_cstr_into(&pkg.url, "");
		pkg.name = str_from_after_delim(&dest_dir, '/');
		pkg.is_local = true;
	} else {
		log_error("'%.*s' is not a valid package", str_fmt(arg));
		str_free(&cwd_str);
		str_free(&dest_dir);
		str_free(arg);
		free_vars();
		exit(EXIT_FAILURE);
	}

	if (str_find_char(arg, ',')
		&& str_find_char(arg, '@')
	) pkg.name.len--;
	pkg.src = get_pkgsrc(pkg);

	str_free(&cwd_str);
	str_free(&dest_dir);
	if (pkg.is_local) str_free(arg);
	return pkg;
}