From 409bb05adb85012a5c4c6123f4ba190bb396a8a8 Mon Sep 17 00:00:00 2001
From: mgabdev <>
Date: Wed, 13 Jan 2021 13:27:00 -0500
Subject: [PATCH] Added update_account_stats function to accounts_cli
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

• Added:
- update_account_stats function to accounts_cli
---
 lib/gabsocial/accounts_cli.rb | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/lib/gabsocial/accounts_cli.rb b/lib/gabsocial/accounts_cli.rb
index 0d7a7095..b3040fa7 100644
--- a/lib/gabsocial/accounts_cli.rb
+++ b/lib/gabsocial/accounts_cli.rb
@@ -498,6 +498,26 @@ module GabSocial
       end
     end
 
+    desc 'update_account_stats ACCT', 'Update account stats.'
+    long_desc <<-LONG_DESC
+      Update account stats (follow count, status count, following count).
+    LONG_DESC
+    def update_account_stats(acct)
+      target_account = Account.find(acct)
+      if target_account.nil?
+        say("Target account (#{acct}) was not found", :red)
+        exit(1)
+      end
+
+      AccountStat.where(account: target_account).update(
+        followers_count: target_account.local_followers_count,
+        following_count: target_account.local_following_count,
+        statuses_count: target_account.statuses.count,
+      )
+
+      say('OK', :green)
+    end
+
     private
 
     def rotate_keys_for_account(account, delay = 0)