Add options for repositories (uses alpm)

Author: EXtremeExploit

CMakeLists.txt

@@ -25,6 +25,10 @@ link_directories(${GTK3_LIBRARY_DIRS})
 # Add other flags to the compiler
 add_definitions(${GTK3_CFLAGS_OTHER})
 
+# libalpm
+find_package(PkgConfig REQUIRED)
+pkg_check_modules(ALPM REQUIRED libalpm)
+
 set(CMAKE_COMPILE_WARNING_AS_ERROR ON)
 
 add_executable(pepv src/main.cpp src/pkgs.cpp src/utils.cpp src/events.cpp)
@@ -42,4 +46,4 @@ if(CMAKE_BUILD_TYPE STREQUAL "Release")
     endif()
 endif()
 
-target_link_libraries(pepv PRIVATE ${GTK3_LIBRARIES} Tracy::TracyClient)
+target_link_libraries(pepv PRIVATE ${GTK3_LIBRARIES} Tracy::TracyClient ${ALPM_LIBRARIES})

README.md

@@ -16,6 +16,7 @@
 * No uninstalling packages
 * No installing packages
 * No updating packages
+* Only support for `core`, `extra` and `multilib` repositories
 
 ## Why?
 * Because i want this program only to work as a viewer, everything regarding actually messing with your system should be in your own hands with you knowing what you are doing instead of me deciding what packages stay and go blindly

pepv.ui

@@ -147,8 +147,10 @@ void populatePkgList() {
 	static auto gShownSize     = GTK_LABEL(gtk_builder_get_object(builder, "shownSize"));
 	static auto gShownFiles    = GTK_LABEL(gtk_builder_get_object(builder, "shownFiles"));
 
-	static auto gSourceOfficial = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromOfficial"));
-	static auto gSourceAUR      = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromAUR"));
+	static auto gFromCore     = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromCore"));
+	static auto gFromExtra    = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromExtra"));
+	static auto gFromMultilib = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromMultilib"));
+	static auto gFromAUR      = GTK_CHECK_BUTTON(gtk_builder_get_object(builder, "fromAUR"));
 
 	const auto searchTerm   = gtk_entry_get_text(GTK_ENTRY(gSearchEntry));
 	const auto query        = std::string_view(searchTerm);
@@ -158,8 +160,10 @@ void populatePkgList() {
 	const auto typeExplicit   = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gTypeExplicit));
 	const auto typeDependency = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gTypeDependency));
 
-	const auto fromOfficial = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gSourceOfficial));
-	const auto fromAUR      = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gSourceAUR));
+	const auto fromCore     = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gFromCore));
+	const auto fromExtra    = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gFromExtra));
+	const auto fromMultilib = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gFromMultilib));
+	const auto fromAUR      = gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(gFromAUR));
 
 	const auto names = p->getPackagesNames();
 
@@ -193,8 +197,10 @@ void populatePkgList() {
 		totalSize += pkg.size.value_or(0);
 		totalFiles += files;
 
-		if (!fromOfficial && !pkg.isLocal) continue;
-		if (!fromAUR && pkg.isLocal) continue;
+		if (pkg.repo.has_value() && pkg.repo.value() == DBS_CORE && !fromCore) continue;
+		if (pkg.repo.has_value() && pkg.repo.value() == DBS_EXTRA && !fromExtra) continue;
+		if (pkg.repo.has_value() && pkg.repo.value() == DBS_MULTILIB && !fromMultilib) continue;
+		if (!pkg.repo.has_value() && !fromAUR) continue;
 
 		if (pkg.reason == REASON_EXPLICIT && !typeExplicit) continue;
 		if (pkg.reason == REASON_DEPEND && !typeDependency) continue;
@@ -214,6 +220,8 @@ void populatePkgList() {
 		const auto numDeps = pkg.depends.size();
 		const auto desc    = pkg.desc.value_or("");
 
+		std::string_view repo = repoIdToStr(pkg.repo.value_or((REPOS)-1));
+
 		gtk_tree_store_append(treeStore, &iter, NULL);
 
 		gtk_tree_store_set(treeStore, &iter,
@@ -223,7 +231,9 @@ void populatePkgList() {
 						   COL_LIST_NUM_DEPS, numDeps,
 						   COL_LIST_DESC, desc.c_str(),
 						   COL_LIST_SIZE, pkg.size.value_or(0),
-						   COL_LIST_FILES, files, -1);
+						   COL_LIST_FILES, files,
+						   COL_LIST_REPO, repo.data(),
+						   -1);
 		shownCount++;
 		shownSize += pkg.size.value_or(0);
 		shownFiles += files;

src/events.cpp

@@ -3,6 +3,10 @@
 #include <cstdio>
 #include <iostream>
 
+#include <alpm.h>
+#include <alpm_list.h>
+
+#include <string_view>
 #include <tracy/Tracy.hpp>
 
 void Pkgs::init() {
@@ -27,6 +31,19 @@ void Pkgs::initDescriptions() {
 	ZoneScopedN("Pkgs::initDescriptions()");
 	if (!inited) return;
 
+	std::map<REPOS, alpm_db_t*> repos;
+
+	{
+		ZoneNamedN(___tracy_pkg_pacman, "Initialize alpm and dbs", true);
+		alpm_errno_t alpmErr = ALPM_ERR_OK;
+		auto handle          = alpm_initialize("/", "/var/lib/pacman/", nullptr);
+
+		for (auto [repoId, repoName] : Repositories) {
+			alpm_db_t* db = alpm_register_syncdb(handle, repoName.data(), ALPM_DB_USAGE_SEARCH);
+			repos.emplace(repoId, db);
+		}
+	}
+
 	std::set<fs::path> tempPkgsPaths;
 	{
 		ZoneNamedN(___tracy_pkg_paths, "get package paths", true);
@@ -122,8 +139,6 @@ void Pkgs::initDescriptions() {
 					}
 					case SECTION_DESC_PACKAGER:
 						pkg.packager.emplace(currentLine);
-						if (currentLine == "Unknown Packager")
-							pkg.isLocal = true; // TODO: is this correct?
 						break;
 					case SECTION_DESC_SIZE: {
 						auto num = std::stoull(currentLine.data());
@@ -169,6 +184,18 @@ void Pkgs::initDescriptions() {
 				}
 			}
 			fclose(f);
+
+			{
+				ZoneNamedN(___tracy_pkg_pacman, "Initialize alpm and dbs", true);
+				for (auto [repoId, db] : repos) {
+					auto pkgFound = alpm_db_get_pkg(db, pkg.name.c_str());
+					if (pkgFound) {
+						pkg.repo.emplace(repoId);
+						break;
+					}
+				}
+			}
+
 			pkgPaths.insert({pkg.name, path});
 			descriptions.insert({pkg.name, pkg});
 		}

src/pkgs.cpp

@@ -4,6 +4,7 @@
 #include <set>
 #include <optional>
 #include <filesystem>
+#include <string_view>
 
 namespace fs = std::filesystem;
 
@@ -15,13 +16,26 @@ enum LIST_COLS {
 	COL_LIST_DESC,
 	COL_LIST_SIZE,
 	COL_LIST_FILES,
+	COL_LIST_REPO,
 };
 
 enum BACKUP_COLS {
 	COL_BACKUP_PATH = 0,
 	COL_BACKUP_HASH,
 };
 
+enum REPOS : uint8_t {
+	DBS_CORE,
+	DBS_EXTRA,
+	DBS_MULTILIB,
+};
+
+const std::map<REPOS, std::string_view> Repositories = {
+	{DBS_CORE, "core"},
+	{DBS_EXTRA, "extra"},
+	{DBS_MULTILIB, "multilib"},
+};
+
 enum SectionDesc {
 	SECTION_DESC_NONE,
 	SECTION_DESC_NAME,
@@ -81,7 +95,7 @@ struct PackageDescription {
 	std::optional<std::string> xdata;
 	std::set<std::string> requiredBy;
 	std::set<std::string> optRequiredBy;
-	bool isLocal = false; // AUR / self-packaged
+	std::optional<REPOS> repo;
 };
 
 class Pkgs {

src/pkgs.hpp

@@ -1,7 +1,9 @@
 #include "utils.hpp"
+#include "pkgs.hpp"
 
 #include <cmath>
 
+#include <string_view>
 #include <tracy/Tracy.hpp>
 
 std::string formattedTimestamp(const time_t t) {
@@ -21,9 +23,22 @@ std::string_view reasonToStr(PKGReason r) {
 	return reasonStr;
 }
 
+std::string_view repoIdToStr(REPOS repo) {
+	std::string_view repoStr = "Undefined";
+	switch (repo) {
+		case DBS_CORE: repoStr = "core"; break;
+		case DBS_EXTRA: repoStr = "extra"; break;
+		case DBS_MULTILIB: repoStr = "multilib"; break;
+		default:
+			repoStr = "AUR";
+			break;
+	}
+	return repoStr;
+}
+
 std::string formattedSize(const uint64_t s, bool binary) {
-	const int divider = binary ? 1024 : 1000;
-    const std::string_view byteIndicator =  binary ? "iB" : "B";
+	const int divider                    = binary ? 1024 : 1000;
+	const std::string_view byteIndicator = binary ? "iB" : "B";
 	std::stringstream ss;
 	if (s < divider) {
 		ss << s << " Bytes";
@@ -42,8 +57,8 @@ std::string formattedSize(const uint64_t s, bool binary) {
 std::string setToStr(std::set<std::string> s, const char* delimeter) {
 	ZoneScopedN("setToStr");
 	std::stringstream ss;
-	int i = 0;
-    int size = s.size();
+	int i    = 0;
+	int size = s.size();
 	for (const auto& str : s) {
 		i++;
 		if (i == size)

src/utils.cpp

@@ -5,6 +5,7 @@ extern Pkgs* p;
 
 std::string formattedTimestamp(const time_t t);
 std::string_view reasonToStr(PKGReason r);
+std::string_view repoIdToStr(REPOS repo);
 std::string formattedSize(const uint64_t s, bool binary = false);
 std::string setToStr(std::set<std::string> s, const char* delimeter = " ");
 std::string optDependsToStr(std::set<std::string> s);