added plugin interface to add/remove/list categories to/from/of mods

4 files changed, 73 insertions, 5 deletions

diff --git a/src/categories.cpp b/src/categories.cpp
index 0b201dcf..400cc74b 100644
--- a/src/categories.cpp
+++ b/src/categories.cpp
@@ -166,22 +166,32 @@ void CategoryFactory::saveCategories()
 unsigned int CategoryFactory::countCategories(std::tr1::function<bool (const Category &category)> filter)

 {

   unsigned int result = 0;

-  for (auto iter = m_Categories.begin(); iter != m_Categories.end(); ++iter) {

-    if (filter(*iter)) {

+  for (const Category &cat : m_Categories) {

+    if (filter(cat)) {

       ++result;

     }

   }

   return result;

 }

 

+int CategoryFactory::addCategory(const QString &name, const std::vector<int> &nexusIDs, int parentID)

+{

+  int id = 1;

+  while (m_IDMap.find(id) != m_IDMap.end()) {

+    ++id;

+  }

+  addCategory(id, name, nexusIDs, parentID);

+

+  saveCategories();

+  return id;

+}

 

 void CategoryFactory::addCategory(int id, const QString &name, const std::vector<int> &nexusIDs, int parentID)

 {

   int index = m_Categories.size();

   m_Categories.push_back(Category(index, id, name, nexusIDs, parentID));

-  for (std::vector<int>::const_iterator iter = nexusIDs.begin();

-       iter != nexusIDs.end(); ++iter) {

-    m_NexusMap[*iter] = index;

+  for (int nexusID : nexusIDs) {

+    m_NexusMap[nexusID] = index;

   }

   m_IDMap[id] = index;

 }

@@ -305,6 +315,18 @@ int CategoryFactory::getCategoryIndex(int ID) const
 }

 

 

+int CategoryFactory::getCategoryID(const QString &name) const

+{

+  auto iter = std::find_if(m_Categories.begin(), m_Categories.end(), [name] (const Category &cat) -> bool {

+    return cat.m_Name == name;

+  });

+  if (iter != m_Categories.end()) {

+    return iter->m_ID;

+  } else {

+    return -1;

+  }

+}

+

 

 unsigned int CategoryFactory::resolveNexusID(int nexusID) const

 {

diff --git a/src/categories.h b/src/categories.h
index 96e436e5..f2a5de39 100644
--- a/src/categories.h
+++ b/src/categories.h
@@ -80,6 +80,8 @@ public:
    **/

   void saveCategories();

 

+  int addCategory(const QString &name, const std::vector<int> &nexusIDs, int parentID);

+

   /**

    * @brief retrieve the number of available categories

    *

@@ -143,6 +145,12 @@ public:
   int getCategoryID(unsigned int index) const;

 

   /**

+   * @brief look up the id of a category by its name

+   * @note O(n)

+   */

+  int getCategoryID(const QString &name) const;

+

+  /**

    * @brief look up the index of a category by its id

    *

    * @param id index of the category to look up

diff --git a/src/modinfo.cpp b/src/modinfo.cpp
index 3fa42da9..526ec62b 100644
--- a/src/modinfo.cpp
+++ b/src/modinfo.cpp
@@ -303,6 +303,40 @@ void ModInfo::setVersion(const VersionInfo &version)
   m_Version = version;
 }
 
+void ModInfo::addCategory(const QString &categoryName)
+{
+  int id = CategoryFactory::instance().getCategoryID(categoryName);
+  if (id == -1) {
+    id = CategoryFactory::instance().addCategory(categoryName, std::vector<int>(), 0);
+  }
+  setCategory(id, true);
+}
+
+bool ModInfo::removeCategory(const QString &categoryName)
+{
+  int id = CategoryFactory::instance().getCategoryID(categoryName);
+  if (id == -1) {
+    return false;
+  }
+  if (!categorySet(id)) {
+    return false;
+  }
+  setCategory(id, false);
+  return true;
+}
+
+QStringList ModInfo::categories()
+{
+  QStringList result;
+
+  CategoryFactory &catFac = CategoryFactory::instance();
+  for (int id : m_Categories) {
+    result.append(catFac.getCategoryName(catFac.getCategoryIndex(id)));
+  }
+
+  return result;
+}
+
 bool ModInfo::hasFlag(ModInfo::EFlag flag) const
 {
   std::vector<EFlag> flags = getFlags();
diff --git a/src/modinfo.h b/src/modinfo.h
index 8c8d2baf..ba248bdb 100644
--- a/src/modinfo.h
+++ b/src/modinfo.h
@@ -300,6 +300,10 @@ public:
    */

   virtual void addNexusCategory(int categoryID) = 0;

 

+  virtual void addCategory(const QString &categoryName) override;

+  virtual bool removeCategory(const QString &categoryName) override;

+  virtual QStringList categories() override;

+

   /**

    * update the endorsement state for the mod. This only changes the

    * buffered state, it does not sync with Nexus