1 files changed, 33 insertions, 0 deletions
diff --git a/common/mastodonpostsdatabase.cpp b/common/mastodonpostsdatabase.cpp
index 08fc777..ee04327 100644
--- a/common/mastodonpostsdatabase.cpp
+++ b/common/mastodonpostsdatabase.cpp
@@ -22,6 +22,9 @@ static const char *DB_NAME = "mastodon.db";
 static const char *ACCOUNT_NAME_KEY = "account_name";
 static const char *URL_KEY = "url";
 static const char *BOOSTED_BY_KEY = "boosted_by";
+static const char *REPLIES_COUNT_KEY = "replies_count";
+static const char *FAVOURITES_COUNT_KEY = "favourites_count";
+static const char *REBLOGS_COUNT_KEY = "reblogs_count";
 static const char *INSTANCE_URL_KEY = "instance_url";
 
 MastodonPostsDatabase::MastodonPostsDatabase()
@@ -43,6 +46,9 @@ void MastodonPostsDatabase::addMastodonPost(
         const QList<QPair<QString, SocialPostImage::ImageType> > &images,
         const QString &url,
         const QString &boostedBy,
+        int repliesCount,
+        int favouritesCount,
+        int reblogsCount,
         const QString &instanceUrl,
         int account)
 {
@@ -50,6 +56,9 @@ void MastodonPostsDatabase::addMastodonPost(
     extra.insert(ACCOUNT_NAME_KEY, accountName);
     extra.insert(URL_KEY, url);
     extra.insert(BOOSTED_BY_KEY, boostedBy);
+    extra.insert(REPLIES_COUNT_KEY, repliesCount);
+    extra.insert(FAVOURITES_COUNT_KEY, favouritesCount);
+    extra.insert(REBLOGS_COUNT_KEY, reblogsCount);
     extra.insert(INSTANCE_URL_KEY, instanceUrl);
     addPost(identifier, name, body, timestamp, icon, images, extra, account);
 }
@@ -78,6 +87,30 @@ QString MastodonPostsDatabase::boostedBy(const SocialPost::ConstPtr &post)
     return post->extra().value(BOOSTED_BY_KEY).toString();
 }
 
+int MastodonPostsDatabase::repliesCount(const SocialPost::ConstPtr &post)
+{
+    if (post.isNull()) {
+        return 0;
+    }
+    return post->extra().value(REPLIES_COUNT_KEY).toInt();
+}
+
+int MastodonPostsDatabase::favouritesCount(const SocialPost::ConstPtr &post)
+{
+    if (post.isNull()) {
+        return 0;
+    }
+    return post->extra().value(FAVOURITES_COUNT_KEY).toInt();
+}
+
+int MastodonPostsDatabase::reblogsCount(const SocialPost::ConstPtr &post)
+{
+    if (post.isNull()) {
+        return 0;
+    }
+    return post->extra().value(REBLOGS_COUNT_KEY).toInt();
+}
+
 QString MastodonPostsDatabase::instanceUrl(const SocialPost::ConstPtr &post)
 {
     if (post.isNull()) {