gui/pkg_install_dialog: Allow license files to be automatically selec…

…ted if it is already exist Co-Authored-By: bookmist <[email protected]>
nishinji · Sep 19, 2024 · 5f2d864 · 5f2d864
1 parent 77abeaa
commit 5f2d864
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/vita3k/gui/src/pkg_install_dialog.cpp b/vita3k/gui/src/pkg_install_dialog.cpp
@@ -48,9 +48,23 @@ void draw_pkg_install_dialog(GuiState &gui, EmuEnvState &emuenv) {
     if (draw_file_dialog) {
         result = host::dialog::filesystem::open_file(pkg_path, { { "PlayStation Store Downloaded Package", { "pkg" } } });
         draw_file_dialog = false;
-        if (result == host::dialog::filesystem::Result::SUCCESS)
-            ImGui::OpenPopup("install");
-        else if (result == host::dialog::filesystem::Result::CANCEL) {
+        if (result == host::dialog::filesystem::Result::SUCCESS) {
+            fs::ifstream infile(pkg_path.native(), std::ios::binary);
+            PkgHeader pkg_header{};
+            infile.read(reinterpret_cast<char *>(&pkg_header), sizeof(PkgHeader));
+            std::string title_id_str(pkg_header.content_id);
+            std::string title_id = title_id_str.substr(7, 9);
+            const auto work_path{ emuenv.pref_path / fmt::format("ux0/license/{}/{}.rif", title_id, pkg_header.content_id) };
+            if (fs::exists(work_path)) {
+                LOG_INFO("Found license file: {}", work_path);
+                fs::ifstream binfile(work_path, std::ios::in | std::ios::binary | std::ios::ate);
+                zRIF = rif2zrif(binfile);
+                ImGui::OpenPopup("install");
+                gui::state = "install";
+            } else {
+                ImGui::OpenPopup("install");
+            }
+        } else if (result == host::dialog::filesystem::Result::CANCEL) {
             gui.file_menu.pkg_install_dialog = false;
             draw_file_dialog = true;
         } else {