1 files changed, 120 insertions, 0 deletions
diff --git a/src/pkg_update.c b/src/pkg_update.c
new file mode 100644
index 0000000..f778438
--- /dev/null
+++ b/src/pkg_update.c
@@ -0,0 +1,120 @@
+/*
+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 <dirent.h>
+#include <string.h>
+
+#include "globs.h"
+#include "is_updated.h"
+#include "log.h"
+#include "pkgit_lua.h"
+
+bool on_pkg_update(lua_State *L, package_t *pkg) {
+	lua_getglobal(L, "repositories");
+	if (!lua_istable(L, -1)) {
+		lua_pop(L, 1);
+		return false;
+	}
+	if (!lua_try_table(L, "init.lua", pkg->name.data)) {
+		lua_pop(L, 2);
+		return false;
+	}
+	if (!lua_try_table(L, "init.lua", "targets")) {
+		lua_pop(L, 3);
+		return false;
+	}
+	if (!lua_try_table(L, "init.lua", pkg->target.data)) {
+		lua_pop(L, 4);
+		return false;
+	}
+	if (!lua_try_function(L, "init.lua", "on_update")) {
+		lua_pop(L, 4);
+		return false;
+	}
+	lua_pop(L, 4);
+	return true;
+}
+
+void pkg_update(package_t *pkg) {
+	if (is_updated(&pkg->src)) {
+		if (flags.verbose) log_info(
+			"%.*s is already up to date.",
+			str_fmt(&pkg->name)
+		);
+		return;
+	}
+	flags.force = true;
+	str_copy_cstr_into(&pkg->version, "HEAD");
+	pkg_install(pkg);
+	if (!on_pkg_update(L, pkg)) log_warn(
+		"init.lua: 'repositories.%.*s.%.*s.on_update' function failed",
+		str_fmt(&pkg->name), str_fmt(&pkg->target)
+	);
+	pkg_free(pkg);
+}
+
+bool on_all_update(lua_State *L) {
+	char* fname = "on_update";
+	char* lua_file = "init.lua";
+	lua_getglobal(L, fname);
+	if (!lua_isfunction(L, -1)) {
+		if (flags.verbose) lua_isnt_type(fname, "function");
+		lua_pop(L, 1);
+	} else if (lua_pcall(L, 0, 1, 0) != LUA_OK) {
+		if (flags.verbose) log_warn(
+			"%s: '%s' function borked: %s",
+			lua_file, fname, lua_tostring(L, -1)
+		);
+		lua_pop(L, 1);
+		return false;
+	}
+	if (!lua_isnumber(L, -1) || lua_tonumber(L, -1) != 0) {
+		if (flags.verbose) log_warn(
+			"%s: '%s' failed: %s",
+			lua_file, fname, lua_tostring(L, -1)
+		);
+		lua_pop(L, 1);
+		return false;
+	}
+	lua_pop(L, 1);
+	return true;
+}
+
+void all_update(void) {
+	struct dirent* dirent_ptr;
+	DIR* dir_ptr;
+	if ((dir_ptr = opendir(inst_dirs.src.data)) == NULL) {
+		log_error("could not open %s", str_fmt(&inst_dirs.src));
+	}
+	while ((dirent_ptr = readdir(dir_ptr)) != NULL) {
+		if (
+			strcmp(dirent_ptr->d_name, "..") == 0 ||
+			strcmp(dirent_ptr->d_name, ".") == 0
+		) continue;
+		str pkg_name = mstr(dirent_ptr->d_name);
+		if (!pkg_exists(&pkg_name)) {
+			str_free(&pkg_name);
+			continue;
+		}
+		package_t pkg = pkg_create(&pkg_name);
+		pkg_update(&pkg);
+		pkg_free(&pkg);
+		if (str_is_valid(&pkg_name)) str_free(&pkg_name);
+	}
+	closedir(dir_ptr);
+	if (!on_all_update(L))
+		log_warn("init.lua: 'on_update' function failed");
+}