From 6dcd472be67e49187edc7035ceb8d1ff8d64d3b0 Mon Sep 17 00:00:00 2001
From: Daniel Supernault <danielsupernault@gmail.com>
Date: Sun, 27 Dec 2020 14:45:31 -0700
Subject: [PATCH] Update Profile, fix status count

---
 app/Console/Commands/FixStatusCount.php       | 56 +++++++++++++++++++
 app/Http/Controllers/StatusController.php     |  1 +
 app/Jobs/StatusPipeline/StatusDelete.php      | 11 ++++
 app/Jobs/StatusPipeline/StatusEntityLexer.php | 11 ++++
 app/Profile.php                               | 16 +-----
 5 files changed, 80 insertions(+), 15 deletions(-)
 create mode 100644 app/Console/Commands/FixStatusCount.php

diff --git a/app/Console/Commands/FixStatusCount.php b/app/Console/Commands/FixStatusCount.php
new file mode 100644
index 000000000..218526b1d
--- /dev/null
+++ b/app/Console/Commands/FixStatusCount.php
@@ -0,0 +1,56 @@
+<?php
+
+namespace App\Console\Commands;
+
+use Illuminate\Console\Command;
+use App\Profile;
+
+class FixStatusCount extends Command
+{
+    /**
+     * The name and signature of the console command.
+     *
+     * @var string
+     */
+    protected $signature = 'fix:statuscount';
+
+    /**
+     * The console command description.
+     *
+     * @var string
+     */
+    protected $description = 'fix profile status count';
+
+    /**
+     * Create a new command instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        parent::__construct();
+    }
+
+    /**
+     * Execute the console command.
+     *
+     * @return int
+     */
+    public function handle()
+    {
+        Profile::whereNull('domain')
+        ->chunk(50, function($profiles) {
+            foreach($profiles as $profile) {
+                $profile->status_count = $profile->statuses()
+                ->getQuery()
+                ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+                ->whereNull('in_reply_to_id')
+                ->whereNull('reblog_of_id')
+                ->count();
+                $profile->save();
+            }
+        });
+
+        return 0;
+    }
+}
diff --git a/app/Http/Controllers/StatusController.php b/app/Http/Controllers/StatusController.php
index c2e573ea9..34aa16c40 100644
--- a/app/Http/Controllers/StatusController.php
+++ b/app/Http/Controllers/StatusController.php
@@ -210,6 +210,7 @@ class StatusController extends Controller
         }
 
         Cache::forget('_api:statuses:recent_9:' . $status->profile_id);
+        Cache::forget('profile:status_count:' . $status->profile_id);
         if ($status->profile_id == $user->profile->id || $user->is_admin == true) {
             Cache::forget('profile:status_count:'.$status->profile_id);
             StatusDelete::dispatch($status);
diff --git a/app/Jobs/StatusPipeline/StatusDelete.php b/app/Jobs/StatusPipeline/StatusDelete.php
index 346f1cf6b..ede5c67e4 100644
--- a/app/Jobs/StatusPipeline/StatusDelete.php
+++ b/app/Jobs/StatusPipeline/StatusDelete.php
@@ -56,6 +56,17 @@ class StatusDelete implements ShouldQueue
     public function handle()
     {
         $status = $this->status;
+        $profile = $this->status->profile;
+
+        $count = $profile->statuses()
+        ->getQuery()
+        ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+        ->whereNull('in_reply_to_id')
+        ->whereNull('reblog_of_id')
+        ->count();
+
+        $profile->status_count = ($count - 1);
+        $profile->save();
 
         if(config('federation.activitypub.enabled') == true) {
             $this->fanoutDelete($status);
diff --git a/app/Jobs/StatusPipeline/StatusEntityLexer.php b/app/Jobs/StatusPipeline/StatusEntityLexer.php
index ce04c6873..e173c6394 100644
--- a/app/Jobs/StatusPipeline/StatusEntityLexer.php
+++ b/app/Jobs/StatusPipeline/StatusEntityLexer.php
@@ -52,6 +52,17 @@ class StatusEntityLexer implements ShouldQueue
     public function handle()
     {
         $profile = $this->status->profile;
+
+        $count = $profile->statuses()
+        ->getQuery()
+        ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+        ->whereNull('in_reply_to_id')
+        ->whereNull('reblog_of_id')
+        ->count();
+
+        $profile->status_count = $count;
+        $profile->save();
+
         if($profile->no_autolink == false) {
             $this->parseEntities();
         }
diff --git a/app/Profile.php b/app/Profile.php
index c88b876fb..c4d07b84d 100644
--- a/app/Profile.php
+++ b/app/Profile.php
@@ -91,21 +91,7 @@ class Profile extends Model
 
     public function statusCount()
     {
-        return Cache::remember('profile:status_count:'.$this->id, now()->addMonths(1), function() {
-            $count = $this->status_count;
-            if($count) {
-                return $count;
-            }
-            $count = $this->statuses()
-                ->getQuery()
-                ->whereHas('media')
-                ->whereNull('in_reply_to_id')
-                ->whereNull('reblog_of_id')
-                ->count();
-            $this->status_count = $count;
-            $this->save();
-            return $count;
-        });
+        return $this->status_count;
     }
 
     public function following()