/*

  pkgit - package 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 <ftw.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include "lua_state.h"
#include "remove_pkg.h"
#include "vars.h"

static int remove_installed(
	  const char *src_path, const struct stat *sb,
	  int typeflag, struct FTW *ftwbuf
) {
	(void)sb;
	(void)ftwbuf;

	if (typeflag == FTW_F) {
		const char *filename = src_path + ftwbuf->base;
		const char *ext = strrchr(filename, '.');
		if (!ext)
		ext = "";

		if (strncmp(ext, ".so", 3) == 0) {
			char dest[MAX_PATH_LEN];
			snprintf(dest, sizeof(dest), "%s/%s", get_install_dir("lib"), filename);
			if (file_exists(dest))
			remove(dest);
		} else if (access(src_path, X_OK) == 0) {
			if (
				strcmp(ext, ".sample") != 0 &&
				strcmp(filename, "bldit") != 0 &&
				strcmp(filename, "build.sh") != 0 &&
				strcmp(filename, "compile.sh") != 0
			) {
				char dest[MAX_PATH_LEN];
				snprintf(dest, sizeof(dest), "%s/%s", get_install_dir("bin"), filename);
				if (file_exists(dest))
				remove(dest);
			}
		} else if (strcmp(ext, ".h") == 0) {
			char dest[MAX_PATH_LEN];
			snprintf(dest, sizeof(dest), "%s/%s", get_install_dir("include"),
			filename);
			if (file_exists(dest))
			remove(dest);
		}
	}
	return 0;
}

static int remove_tree(
	const char *fpath, const struct stat *sb,
	int typeflag, struct FTW *ftwbuf
) {
	(void)sb;
	(void)ftwbuf;
	if (typeflag == FTW_F || typeflag == FTW_SL) {
		unlink(fpath);
	} else if (typeflag == FTW_DP) {
		rmdir(fpath);
	}
	return 0;
}

void remove_pkg(Pkg pkg) {
	if (!is_directory(pkg.src)) {
		printf("%s %s is not installed!\n", print_pkgit, pkg.name);
		return;
	}
	chdir(pkg.src);

	bool uninstall_available = false;
	if (!uninstall_available) if (repo_uninstall(pkg.name, pkg.target)) 
		uninstall_available = true;
	if (!uninstall_available) if (bldit_uninstall(pkg.target))
		uninstall_available = true;
	if (!uninstall_available) if (config_uninstall(pkg.src, pkg.target))
		uninstall_available = true;

	if (!uninstall_available) printf(
		"%s no uninstall function availible for package: %s\n",
		print_error, pkg.name
	);

	nftw(pkg.src, remove_installed, 64, FTW_PHYS);
	const char *last_slash = strrchr(pkg.src, '/');
	char target[MAX_PATH_LEN];
	if (last_slash && last_slash != pkg.src) {
		size_t parent_len = last_slash - pkg.src;
		snprintf(target, sizeof(target), "%.*s", (int)parent_len, pkg.src);
	} else {
		snprintf(target, sizeof(target), "%s", pkg.src);
	}
	nftw(pkg.src, remove_tree, 64, FTW_DEPTH | FTW_PHYS);
	printf("%s removed %s\n", print_success, pkg.name);
}