Add PluginInterface::IsBlueprintPlugin()

loot · Aug 9, 2024 · ac366ac · ac366ac
1 parent e67a141
commit ac366ac
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 0 deletions.
diff --git a/include/loot/plugin_interface.h b/include/loot/plugin_interface.h
@@ -111,6 +111,12 @@ class PluginInterface {
    */
   virtual bool IsUpdatePlugin() const = 0;
 
+  /**
+   * Check if the plugin is a blueprint plugin.
+   * @return True if plugin is a blueprint plugin, false otherwise.
+   */
+  virtual bool IsBlueprintPlugin() const = 0;
+
   /**
    * Check if the plugin is or would be valid as a light plugin.
    * @return True if the plugin is a valid light plugin or would be a valid

diff --git a/src/api/plugin.cpp b/src/api/plugin.cpp
@@ -324,6 +324,15 @@ bool Plugin::IsUpdatePlugin() const {
   return isUpdatePlugin;
 }
 
+bool Plugin::IsBlueprintPlugin() const {
+  bool isBlueprintPlugin = false;
+  const auto ret =
+      esp_plugin_is_blueprint_plugin(esPlugin.get(), &isBlueprintPlugin);
+  HandleEspluginError("check if \"" + name_ + "\" is a blueprint plugin", ret);
+
+  return isBlueprintPlugin;
+}
+
 bool Plugin::IsValidAsLightPlugin() const {
   bool isValid = false;
   const auto ret =

diff --git a/src/api/plugin.h b/src/api/plugin.h
@@ -71,6 +71,7 @@ class Plugin final : public PluginSortingInterface {
   bool IsLightPlugin() const override;
   bool IsMediumPlugin() const override;
   bool IsUpdatePlugin() const override;
+  bool IsBlueprintPlugin() const override;
 
   bool IsValidAsLightPlugin() const override;
   bool IsValidAsMediumPlugin() const override;

diff --git a/src/tests/api/internals/plugin_test.h b/src/tests/api/internals/plugin_test.h
@@ -199,6 +199,7 @@ class OtherPluginType final : public PluginSortingInterface {
   bool IsLightPlugin() const override { return false; }
   bool IsMediumPlugin() const override { return false; }
   bool IsUpdatePlugin() const override { return false; }
+  bool IsBlueprintPlugin() const override { return false; }
   bool IsValidAsLightPlugin() const override { return false; }
   bool IsValidAsMediumPlugin() const override { return false; }
   bool IsValidAsUpdatePlugin() const override { return false; }
@@ -389,6 +390,22 @@ TEST_P(PluginTest,
   EXPECT_EQ(GetParam() == GameType::starfield, plugin2.IsUpdatePlugin());
 }
 
+TEST_P(PluginTest, isBlueprintPluginShouldOnlyBeTrueForAStarfieldBlueprintPlugin) {
+  auto bytes = ReadFile(dataPath / blankMasterDependentEsp);
+  bytes[9] = 0x8;
+  WriteFile(dataPath / blankMasterDependentEsp, bytes);
+
+  Plugin plugin1(
+      game_.GetType(), game_.GetCache(), game_.DataPath() / blankEsp, true);
+  Plugin plugin2(game_.GetType(),
+                 game_.GetCache(),
+                 game_.DataPath() / blankMasterDependentEsp,
+                 true);
+
+  EXPECT_FALSE(plugin1.IsBlueprintPlugin());
+  EXPECT_EQ(GetParam() == GameType::starfield, plugin2.IsBlueprintPlugin());
+}
+
 TEST_P(PluginTest, loadingAPluginWithMastersShouldReadThemCorrectly) {
   Plugin plugin(game_.GetType(),
                 game_.GetCache(),

diff --git a/src/tests/api/internals/sorting/plugin_graph_test.h b/src/tests/api/internals/sorting/plugin_graph_test.h
@@ -64,6 +64,8 @@ class TestPlugin : public PluginSortingInterface {
 
   bool IsUpdatePlugin() const override { return false; }
 
+  bool IsBlueprintPlugin() const override { return false; }
+
   bool IsValidAsLightPlugin() const override { return false; }
 
   bool IsValidAsMediumPlugin() const override { return false; }