Allow use different badges for each group in groups.xml

2 files changed, 18 insertions, 3 deletions

diff --git a/src/resources/db/groupdb.cpp b/src/resources/db/groupdb.cpp
index c65406841..bd78e117c 100644
--- a/src/resources/db/groupdb.cpp
+++ b/src/resources/db/groupdb.cpp
@@ -96,6 +96,9 @@ void GroupDb::loadXmlFile(const std::string &fileName,
             mGroups[id].longName = XML::langProperty(node,
                 "longName",
                 "");
+            mGroups[id].badge = XML::langProperty(node,
+                "badge",
+                paths.getStringValue("gmbadge"));
             mGroups[id].showBadge = XML::getBoolProperty(node,
                 "showBadge",
                 false);
@@ -142,6 +145,17 @@ bool GroupDb::getShowBadge(const int id)
     return (*it).second.showBadge;
 }
 
+const std::string &GroupDb::getBadge(const int id)
+{
+    GroupInfos::const_iterator it = mGroups.find(id);
+    if (it == mGroups.end())
+    {
+        reportAlways("Unknown group id requested: %d", id);
+        return mEmptyGroup.badge;
+    }
+    return (*it).second.badge;
+}
+
 #ifdef UNITTESTS
 GroupDb::GroupInfos &GroupDb::getGroups()
 {
diff --git a/src/resources/db/groupdb.h b/src/resources/db/groupdb.h
index 4e687dea9..3db3b1921 100644
--- a/src/resources/db/groupdb.h
+++ b/src/resources/db/groupdb.h
@@ -38,9 +38,10 @@ namespace GroupDb
 
     void loadXmlFile(const std::string &fileName,
                      const SkipError skipError);
-    const std::string &getName(const int id);
-    const std::string &getLongName(const int id);
-    bool getShowBadge(const int id);
+    const std::string &getName(const int id) A_WARN_UNUSED;
+    const std::string &getLongName(const int id) A_WARN_UNUSED;
+    bool getShowBadge(const int id) A_WARN_UNUSED;
+    const std::string &getBadge(const int id) A_WARN_UNUSED;
 
     typedef std::map<int, GroupInfo> GroupInfos;