From 176b4ed793b0690c05fa0103a77c0089c55bb594 Mon Sep 17 00:00:00 2001
From: Daniel Supernault <danielsupernault@gmail.com>
Date: Fri, 17 Nov 2023 22:21:55 -0700
Subject: [PATCH] Add app:hashtag-related-generate command

---
 .../Commands/HashtagRelatedGenerate.php       | 83 +++++++++++++++++++
 app/Models/HashtagRelated.php                 |  2 +
 app/Services/HashtagRelatedService.php        | 12 +--
 3 files changed, 92 insertions(+), 5 deletions(-)
 create mode 100644 app/Console/Commands/HashtagRelatedGenerate.php

diff --git a/app/Console/Commands/HashtagRelatedGenerate.php b/app/Console/Commands/HashtagRelatedGenerate.php
new file mode 100644
index 000000000..0613495c9
--- /dev/null
+++ b/app/Console/Commands/HashtagRelatedGenerate.php
@@ -0,0 +1,83 @@
+<?php
+
+namespace App\Console\Commands;
+
+use Illuminate\Console\Command;
+use App\Hashtag;
+use App\StatusHashtag;
+use App\Models\HashtagRelated;
+use App\Services\HashtagRelatedService;
+use Illuminate\Contracts\Console\PromptsForMissingInput;
+use function Laravel\Prompts\multiselect;
+
+class HashtagRelatedGenerate extends Command implements PromptsForMissingInput
+{
+    /**
+     * The name and signature of the console command.
+     *
+     * @var string
+     */
+    protected $signature = 'app:hashtag-related-generate {tag}';
+
+    /**
+     * The console command description.
+     *
+     * @var string
+     */
+    protected $description = 'Command description';
+
+    /**
+     * Prompt for missing input arguments using the returned questions.
+     *
+     * @return array
+     */
+    protected function promptForMissingArgumentsUsing()
+    {
+        return [
+            'tag' => 'Which hashtag should we generate related tags for?',
+        ];
+    }
+
+    /**
+     * Execute the console command.
+     */
+    public function handle()
+    {
+        $tag = $this->argument('tag');
+        $hashtag = Hashtag::whereName($tag)->orWhere('slug', $tag)->first();
+        if(!$hashtag) {
+            $this->error('Hashtag not found, aborting...');
+            exit;
+        }
+
+        $this->info('Looking up #' . $tag . '...');
+
+        $tags = StatusHashtag::whereHashtagId($hashtag->id)->count();
+        if(!$tags || $tags < 100) {
+            $this->error('Not enough posts found to generate related hashtags!');
+            exit;
+        }
+
+        $this->info('Found ' . $tags . ' posts that use that hashtag');
+        $related = collect(HashtagRelatedService::fetchRelatedTags($tag));
+
+        $selected = multiselect(
+            label: 'Which tags do you want to generate?',
+            options: $related->pluck('name'),
+            required: true,
+        );
+
+        $filtered = $related->filter(fn($i) => in_array($i['name'], $selected))->all();
+        $agg_score = $related->filter(fn($i) => in_array($i['name'], $selected))->sum('related_count');
+
+        HashtagRelated::updateOrCreate([
+            'hashtag_id' => $hashtag->id,
+        ], [
+            'related_tags' => array_values($filtered),
+            'agg_score' => $agg_score,
+            'last_calculated_at' => now()
+        ]);
+
+        $this->info('Finished!');
+    }
+}
diff --git a/app/Models/HashtagRelated.php b/app/Models/HashtagRelated.php
index 42722a205..bbaa1c06c 100644
--- a/app/Models/HashtagRelated.php
+++ b/app/Models/HashtagRelated.php
@@ -9,6 +9,8 @@ class HashtagRelated extends Model
 {
     use HasFactory;
 
+    protected $guarded = [];
+
     /**
      * The attributes that should be mutated to dates and other custom formats.
      *
diff --git a/app/Services/HashtagRelatedService.php b/app/Services/HashtagRelatedService.php
index 53a387b45..b96483987 100644
--- a/app/Services/HashtagRelatedService.php
+++ b/app/Services/HashtagRelatedService.php
@@ -8,10 +8,13 @@ use App\Models\HashtagRelated;
 
 class HashtagRelatedService
 {
-
     public static function get($id)
     {
-        return HashtagRelated::whereHashtagId($id)->first();
+        $tag = HashtagRelated::whereHashtagId($id)->first();
+        if(!$tag) {
+            return [];
+        }
+        return $tag->related_tags;
     }
 
     public static function fetchRelatedTags($tag)
@@ -20,15 +23,14 @@ class HashtagRelatedService
             ->select('h2.name', DB::raw('COUNT(*) as related_count'))
             ->join('status_hashtags as hs2', function ($join) {
                 $join->on('status_hashtags.status_id', '=', 'hs2.status_id')
-                     ->whereRaw('status_hashtags.hashtag_id != hs2.hashtag_id')
-                     ->where('status_hashtags.created_at', '>', now()->subMonths(3));
+                     ->whereRaw('status_hashtags.hashtag_id != hs2.hashtag_id');
             })
             ->join('hashtags as h1', 'status_hashtags.hashtag_id', '=', 'h1.id')
             ->join('hashtags as h2', 'hs2.hashtag_id', '=', 'h2.id')
             ->where('h1.name', '=', $tag)
             ->groupBy('h2.name')
             ->orderBy('related_count', 'desc')
-            ->limit(10)
+            ->limit(30)
             ->get();
 
         return $res;