mirror of https://github.com/pixelfed/pixelfed
Add app:hashtag-related-generate command
parent
aa166ab11a
commit
176b4ed793
@ -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!');
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue