pixelfed/app/Console/Commands/AddUserDomainBlock.php

107 lines
2.6 KiB
PHP
Raw Permalink Normal View History

2023-12-21 10:35:47 +00:00
<?php
namespace App\Console\Commands;
use Illuminate\Console\Command;
use App\User;
2023-12-21 10:48:08 +00:00
use App\Models\DefaultDomainBlock;
2023-12-21 10:35:47 +00:00
use App\Models\UserDomainBlock;
use function Laravel\Prompts\text;
use function Laravel\Prompts\confirm;
use function Laravel\Prompts\progress;
class AddUserDomainBlock extends Command
{
/**
* The name and signature of the console command.
*
* @var string
*/
protected $signature = 'app:add-user-domain-block';
/**
* The console command description.
*
* @var string
*/
protected $description = 'Apply a domain block to all users';
/**
* Execute the console command.
*/
public function handle()
{
$domain = text('Enter domain you want to block');
2023-12-21 10:48:08 +00:00
$domain = strtolower($domain);
2023-12-21 10:35:47 +00:00
$domain = $this->validateDomain($domain);
2023-12-21 12:00:35 +00:00
if(!$domain || empty($domain)) {
$this->error('Invalid domain');
return;
}
2023-12-21 10:35:47 +00:00
$this->processBlocks($domain);
return;
}
protected function validateDomain($domain)
{
if(!strpos($domain, '.')) {
return;
}
if(str_starts_with($domain, 'https://')) {
$domain = str_replace('https://', '', $domain);
}
if(str_starts_with($domain, 'http://')) {
$domain = str_replace('http://', '', $domain);
}
2023-12-21 12:00:35 +00:00
$domain = strtolower(parse_url('https://' . $domain, PHP_URL_HOST));
2023-12-21 10:35:47 +00:00
$valid = filter_var($domain, FILTER_VALIDATE_DOMAIN, FILTER_FLAG_HOSTNAME|FILTER_NULL_ON_FAILURE);
if(!$valid) {
return;
}
if($domain === config('pixelfed.domain.app')) {
$this->error('Invalid domain');
return;
}
$confirmed = confirm('Are you sure you want to block ' . $domain . '?');
if(!$confirmed) {
return;
}
return $domain;
}
protected function processBlocks($domain)
{
2023-12-21 10:48:08 +00:00
DefaultDomainBlock::updateOrCreate([
'domain' => $domain
]);
2023-12-21 10:35:47 +00:00
progress(
label: 'Updating user domain blocks...',
steps: User::lazyById(500),
callback: fn ($user) => $this->performTask($user, $domain),
);
}
protected function performTask($user, $domain)
{
if(!$user->profile_id || $user->delete_after) {
return;
}
if($user->status != null && $user->status != 'disabled') {
return;
}
UserDomainBlock::updateOrCreate([
'profile_id' => $user->profile_id,
'domain' => $domain
]);
}
}