From 4c2a5d39ab448e701b8b0cae50744bc9e4eeb9f1 Mon Sep 17 00:00:00 2001
From: Gliniak <Gliniak93@gmail.com>
Date: Mon, 9 Sep 2024 18:52:09 +0200
Subject: [PATCH] [XAM] Added support for searching publisher paths

---
 src/xenia/kernel/xam/content_manager.cc | 71 +++++++++++++++++++------
 src/xenia/kernel/xam/content_manager.h  |  4 ++
 2 files changed, 58 insertions(+), 17 deletions(-)

diff --git a/src/xenia/kernel/xam/content_manager.cc b/src/xenia/kernel/xam/content_manager.cc
index 6f0322fbb..aec68653d 100644
--- a/src/xenia/kernel/xam/content_manager.cc
+++ b/src/xenia/kernel/xam/content_manager.cc
@@ -78,14 +78,32 @@ std::filesystem::path ContentManager::ResolvePackagePath(
     const XCONTENT_AGGREGATE_DATA& data, const uint32_t disc_number) {
   // Content path:
   // content_root/title_id/content_type/data_file_name/
-  auto package_root = ResolvePackageRoot(data.content_type, data.title_id);
-  std::string final_name = xe::string_util::trim(data.file_name());
-  std::filesystem::path package_path = package_root / xe::to_path(final_name);
+  std::set<uint32_t> title_ids = {data.title_id};
 
-  if (disc_number != -1) {
-    package_path /= fmt::format("disc{:03}", disc_number);
+  if (data.content_type == XContentType::kPublisher) {
+    title_ids = FindPublisherTitleIds(data.title_id);
   }
-  return package_path;
+
+  for (const auto& title_id : title_ids) {
+    auto package_root = ResolvePackageRoot(data.content_type, title_id);
+    std::string final_name = xe::string_util::trim(data.file_name());
+    std::filesystem::path package_path = package_root / xe::to_path(final_name);
+
+    if (disc_number != -1) {
+      package_path /= fmt::format("disc{:03}", disc_number);
+    }
+
+    if (data.content_type == XContentType::kPublisher) {
+      if (!std::filesystem::exists(package_path)) {
+        continue;
+      }
+      return package_path;
+    }
+    return package_path;
+  }
+
+  assert_always("ResolvePackagePath: Reached unreachable path!");
+  return "";
 }
 
 std::filesystem::path ContentManager::ResolvePackageHeaderPath(
@@ -105,6 +123,35 @@ std::filesystem::path ContentManager::ResolvePackageHeaderPath(
          content_type_str / final_name;
 }
 
+std::set<uint32_t> ContentManager::FindPublisherTitleIds(
+    uint32_t base_title_id) const {
+  if (base_title_id == kCurrentlyRunningTitleId) {
+    base_title_id = kernel_state_->title_id();
+  }
+  std::set<uint32_t> title_ids = {base_title_id};
+
+  std::string publisher_id_regex =
+      fmt::format("^{:04X}.*", static_cast<uint16_t>(base_title_id >> 16));
+  // Get all publisher entries
+  auto publisher_entries =
+      xe::filesystem::FilterByName(xe::filesystem::ListDirectories(root_path_),
+                                   std::regex(publisher_id_regex));
+
+  for (const auto& entry : publisher_entries) {
+    std::filesystem::path path_to_publisher_dir =
+        entry.path / entry.name /
+        fmt::format("{:08X}", XContentType::kPublisher);
+
+    if (!std::filesystem::exists(path_to_publisher_dir)) {
+      continue;
+    }
+
+    title_ids.insert(xe::string_util::from_string<uint32_t>(
+        xe::path_to_utf8(entry.name), true));
+  }
+  return title_ids;
+}
+
 std::vector<XCONTENT_AGGREGATE_DATA> ContentManager::ListContent(
     uint32_t device_id, XContentType content_type, uint32_t title_id) {
   std::vector<XCONTENT_AGGREGATE_DATA> result;
@@ -116,17 +163,7 @@ std::vector<XCONTENT_AGGREGATE_DATA> ContentManager::ListContent(
   std::set<uint32_t> title_ids = {title_id};
 
   if (content_type == XContentType::kPublisher) {
-    std::string publisher_id_regex =
-        fmt::format("^{:04X}.*", static_cast<uint16_t>(title_id >> 16));
-    // Get all publisher entries
-    auto publisher_entries = xe::filesystem::FilterByName(
-        xe::filesystem::ListDirectories(root_path_),
-        std::regex(publisher_id_regex));
-
-    for (const auto& entry : publisher_entries) {
-      title_ids.insert(xe::string_util::from_string<uint32_t>(
-          xe::path_to_utf8(entry.name), true));
-    }
+    title_ids = FindPublisherTitleIds(title_id);
   }
 
   for (const uint32_t& title_id : title_ids) {
diff --git a/src/xenia/kernel/xam/content_manager.h b/src/xenia/kernel/xam/content_manager.h
index 8cad1b262..0343c1215 100644
--- a/src/xenia/kernel/xam/content_manager.h
+++ b/src/xenia/kernel/xam/content_manager.h
@@ -11,6 +11,7 @@
 #define XENIA_KERNEL_XAM_CONTENT_MANAGER_H_
 
 #include <memory>
+#include <set>
 #include <string>
 #include <unordered_map>
 #include <vector>
@@ -179,6 +180,9 @@ class ContentManager {
       const std::string_view file_name, XContentType content_type,
       uint32_t title_id = -1);
 
+  std::set<uint32_t> FindPublisherTitleIds(
+      uint32_t base_title_id = kCurrentlyRunningTitleId) const;
+
   KernelState* kernel_state_;
   std::filesystem::path root_path_;