feat(vpkpp): add an iterator over entries in a particular directory

Author: craftablescience

include/vpkpp/PackFile.h

@@ -154,6 +154,9 @@ class PackFile {
 	/// Run a callback for each entry in the pack file
 	void runForAllEntries(const EntryCallback& operation, bool includeUnbaked = true) const;
 
+	/// Run a callback for each entry in the pack file under the parent directory
+	void runForAllEntries(const std::string& parentDir, const EntryCallback& operation, bool recursive = true, bool includeUnbaked = true) const;
+
 	/// /home/user/pak01_dir.vpk
 	[[nodiscard]] std::string_view getFilepath() const;
 
@@ -186,6 +189,8 @@ class PackFile {
 
 	void runForAllEntriesInternal(const std::function<void(const std::string&, Entry&)>& operation, bool includeUnbaked = true);
 
+	void runForAllEntriesInternal(const std::string& parentDir, const std::function<void(const std::string&, Entry&)>& operation, bool recursive = true, bool includeUnbaked = true);
+
 	[[nodiscard]] std::vector<std::string> verifyEntryChecksumsUsingCRC32() const;
 
 	virtual void addEntryInternal(Entry& entry, const std::string& path, std::vector<std::byte>& buffer, EntryOptions options) = 0;

src/vpkpp/PackFile.cpp

@@ -413,7 +413,7 @@ bool PackFile::extractAll(const std::string& outputDir, bool createUnderPackFile
 	}
 	bool noneFailed = true;
 	this->runForAllEntries([this, &outputDirPath, &noneFailed](const std::string& path, const Entry& entry) {
-		std::string entryPath = path;
+		std::string entryPath = path; // NOLINT(*-unnecessary-copy-initialization)
 #ifdef _WIN32
 		::fixFilePathForWindows(entryPath);
 #endif
@@ -520,6 +520,34 @@ void PackFile::runForAllEntries(const EntryCallback& operation, bool includeUnba
 	}
 }
 
+void PackFile::runForAllEntries(const std::string& parentDir, const EntryCallback& operation, bool recursive, bool includeUnbaked) const {
+	auto dir = this->cleanEntryPath(parentDir) + '/';
+
+	std::string key;
+	for (auto [entry, end] = this->entries.equal_prefix_range(dir); entry != end; ++entry) {
+		entry.key(key);
+		if (!recursive) {
+			auto keyView = std::string_view{key}.substr(dir.length());
+			if (std::find(keyView.begin(), keyView.end(), '/') != keyView.end()) {
+				continue;
+			}
+		}
+		operation(key, entry.value());
+	}
+	if (includeUnbaked) {
+		for (auto [entry, end] = this->unbakedEntries.equal_prefix_range(dir); entry != end; ++entry) {
+			entry.key(key);
+			if (!recursive) {
+				auto keyView = std::string_view{key}.substr(dir.length());
+				if (std::find(keyView.begin(), keyView.end(), '/') != keyView.end()) {
+					continue;
+				}
+			}
+			operation(key, entry.value());
+		}
+	}
+}
+
 void PackFile::runForAllEntriesInternal(const std::function<void(const std::string&, Entry&)>& operation, bool includeUnbaked) {
 	std::string key;
 	for (auto entry = this->entries.begin(); entry != this->entries.end(); ++entry) {
@@ -534,6 +562,34 @@ void PackFile::runForAllEntriesInternal(const std::function<void(const std::stri
 	}
 }
 
+void PackFile::runForAllEntriesInternal(const std::string& parentDir, const std::function<void(const std::string&, Entry&)>& operation, bool recursive, bool includeUnbaked) {
+	auto dir = this->cleanEntryPath(parentDir) + '/';
+
+	std::string key;
+	for (auto [entry, end] = this->entries.equal_prefix_range(dir); entry != end; ++entry) {
+		entry.key(key);
+		if (!recursive) {
+			auto keyView = std::string_view{key}.substr(dir.length());
+			if (std::find(keyView.begin(), keyView.end(), '/') != keyView.end()) {
+				continue;
+			}
+		}
+		operation(key, entry.value());
+	}
+	if (includeUnbaked) {
+		for (auto [entry, end] = this->unbakedEntries.equal_prefix_range(dir); entry != end; ++entry) {
+			entry.key(key);
+			if (!recursive) {
+				auto keyView = std::string_view{key}.substr(dir.length());
+				if (std::find(keyView.begin(), keyView.end(), '/') != keyView.end()) {
+					continue;
+				}
+			}
+			operation(key, entry.value());
+		}
+	}
+}
+
 std::string_view PackFile::getFilepath() const {
 	return this->fullFilePath;
 }