Merge pull request #2020 from pixelfed/staging

Update DeleteAccountPipeline, fixes #2016
This commit is contained in:
daniel 2020-02-15 23:56:03 -07:00 committed by GitHub
commit 6f96cafebd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 169 additions and 143 deletions

View File

@ -13,7 +13,7 @@ class UserDelete extends Command
* *
* @var string * @var string
*/ */
protected $signature = 'user:delete {id}'; protected $signature = 'user:delete {id} {--force}';
/** /**
* The console command description. * The console command description.
@ -40,12 +40,24 @@ class UserDelete extends Command
public function handle() public function handle()
{ {
$id = $this->argument('id'); $id = $this->argument('id');
$user = User::whereUsername($id)->orWhere('id', $id)->first(); $force = $this->option('force');
if(ctype_digit($id) == true) {
$user = User::find($id);
} else {
$user = User::whereUsername($id)->first();
}
if(!$user) { if(!$user) {
$this->error('Could not find any user with that username or id.'); $this->error('Could not find any user with that username or id.');
exit; exit;
} }
if($user->status == 'deleted' && $force == false) {
$this->error('Account has already been deleted.');
return;
}
if($user->is_admin == true) { if($user->is_admin == true) {
$this->error('Cannot delete an admin account from CLI.'); $this->error('Cannot delete an admin account from CLI.');
exit; exit;
@ -62,10 +74,12 @@ class UserDelete extends Command
exit; exit;
} }
$profile = $user->profile; if($user->status !== 'deleted') {
$profile->status = $user->status = 'deleted'; $profile = $user->profile;
$profile->save(); $profile->status = $user->status = 'deleted';
$user->save(); $profile->save();
$user->save();
}
DeleteAccountPipeline::dispatch($user)->onQueue('high'); DeleteAccountPipeline::dispatch($user)->onQueue('high');
} }

View File

@ -10,162 +10,170 @@ use Illuminate\Foundation\Bus\Dispatchable;
use DB; use DB;
use Illuminate\Support\Str; use Illuminate\Support\Str;
use App\{ use App\{
AccountLog, AccountLog,
Activity, Activity,
Avatar, Avatar,
Bookmark, Bookmark,
Collection, Collection,
DirectMessage, CollectionItem,
EmailVerification, Contact,
Follower, DirectMessage,
FollowRequest, EmailVerification,
Hashtag, Follower,
Like, FollowRequest,
Media, Hashtag,
Mention, HashtagFollow,
Notification, Like,
Profile, Media,
Report, Mention,
ReportComment, Notification,
ReportLog, OauthClient,
StatusHashtag, Profile,
Status, ProfileSponsor,
Story, Report,
StoryView, ReportComment,
User, ReportLog,
UserDevice, StatusHashtag,
UserFilter, Status,
UserSetting, Story,
StoryView,
User,
UserDevice,
UserFilter,
UserSetting,
}; };
class DeleteAccountPipeline implements ShouldQueue class DeleteAccountPipeline implements ShouldQueue
{ {
use Dispatchable, InteractsWithQueue, Queueable, SerializesModels; use Dispatchable, InteractsWithQueue, Queueable, SerializesModels;
protected $user; protected $user;
/** public function __construct(User $user)
* Create a new job instance. {
* $this->user = $user;
* @return void }
*/
public function __construct(User $user)
{
$this->user = $user;
}
/** public function handle()
* Execute the job. {
* $user = $this->user;
* @return void
*/
public function handle()
{
$user = $this->user;
DB::transaction(function() use ($user) {
AccountLog::chunk(200, function($logs) use ($user) {
foreach($logs as $log) {
if($log->user_id == $user->id) {
$log->forceDelete();
}
}
});
});
DB::transaction(function() use ($user) { DB::transaction(function() use ($user) {
if($user->profile) { AccountLog::chunk(200, function($logs) use ($user) {
$avatar = $user->profile->avatar; foreach($logs as $log) {
if($log->user_id == $user->id) {
$log->forceDelete();
}
}
});
});
$avatar->forceDelete(); DB::transaction(function() use ($user) {
} if($user->profile) {
$avatar = $user->profile->avatar;
$avatar->forceDelete();
}
Bookmark::whereProfileId($user->profile->id)->forceDelete(); $id = $user->profile_id;
EmailVerification::whereUserId($user->id)->forceDelete(); Bookmark::whereProfileId($user->profile_id)->forceDelete();
$id = $user->profile->id; EmailVerification::whereUserId($user->id)->forceDelete();
StatusHashtag::whereProfileId($id)->delete();
FollowRequest::whereFollowingId($id)
->orWhere('follower_id', $id)
->forceDelete();
Follower::whereProfileId($id)
->orWhere('following_id', $id)
->forceDelete();
Like::whereProfileId($id)->forceDelete();
});
StatusHashtag::whereProfileId($id)->delete(); DB::transaction(function() use ($user) {
$pid = $this->user->profile_id;
FollowRequest::whereFollowingId($id)->orWhere('follower_id', $id)->forceDelete();
Follower::whereProfileId($id)->orWhere('following_id', $id)->forceDelete(); StoryView::whereProfileId($pid)->delete();
$stories = Story::whereProfileId($pid)->get();
foreach($stories as $story) {
$path = storage_path('app/'.$story->path);
if(is_file($path)) {
unlink($path);
}
$story->forceDelete();
}
});
Like::whereProfileId($id)->forceDelete(); DB::transaction(function() use ($user) {
}); $medias = Media::whereUserId($user->id)->get();
foreach($medias as $media) {
$path = storage_path('app/'.$media->media_path);
$thumb = storage_path('app/'.$media->thumbnail_path);
if(is_file($path)) {
unlink($path);
}
if(is_file($thumb)) {
unlink($thumb);
}
$media->forceDelete();
}
});
DB::transaction(function() use ($user) { DB::transaction(function() use ($user) {
$pid = $this->user->profile_id; Mention::whereProfileId($user->profile_id)->forceDelete();
Notification::whereProfileId($user->profile_id)
->orWhere('actor_id', $user->profile_id)
->forceDelete();
});
StoryView::whereProfileId($pid)->delete(); DB::transaction(function() use ($user) {
$stories = Story::whereProfileId($pid)->get(); $collections = Collection::whereProfileId($user->profile_id)->get();
foreach($stories as $story) { foreach ($collections as $collection) {
$path = storage_path('app/'.$story->path); $collection->items()->delete();
if(is_file($path)) { $collection->delete();
unlink($path); }
} Contact::whereUserId($user->id)->delete();
$story->forceDelete(); HashtagFollow::whereUserId($user->id)->delete();
} OauthClient::whereUserId($user->id)->delete();
}); ProfileSponsor::whereProfileId($user->profile_id)->delete();
});
DB::transaction(function() use ($user) { DB::transaction(function() use ($user) {
$medias = Media::whereUserId($user->id)->get(); Status::whereProfileId($user->profile_id)->forceDelete();
foreach($medias as $media) { Report::whereUserId($user->id)->forceDelete();
$path = storage_path('app/'.$media->media_path); $this->deleteProfile($user);
$thumb = storage_path('app/'.$media->thumbnail_path); });
if(is_file($path)) { }
unlink($path);
}
if(is_file($thumb)) {
unlink($thumb);
}
$media->forceDelete();
}
});
DB::transaction(function() use ($user) { protected function deleteProfile($user) {
Mention::whereProfileId($user->profile->id)->forceDelete(); DB::transaction(function() use ($user) {
Notification::whereProfileId($user->profile->id)->orWhere('actor_id', $user->profile->id)->forceDelete(); Profile::whereUserId($user->id)->delete();
}); $this->deleteUserSettings($user);
});
}
DB::transaction(function() use ($user) { protected function deleteUserSettings($user) {
Status::whereProfileId($user->profile->id)->forceDelete();
Report::whereUserId($user->id)->forceDelete();
$this->deleteProfile($user);
});
}
protected function deleteProfile($user) { DB::transaction(function() use ($user) {
DB::transaction(function() use ($user) { UserDevice::whereUserId($user->id)->forceDelete();
Profile::whereUserId($user->id)->delete(); UserFilter::whereUserId($user->id)->forceDelete();
$this->deleteUserSettings($user); UserSetting::whereUserId($user->id)->forceDelete();
}); $this->deleteUserColumns($user);
} });
}
protected function deleteUserSettings($user) { protected function deleteUserColumns($user)
{
DB::transaction(function() use ($user) { DB::transaction(function() use ($user) {
UserDevice::whereUserId($user->id)->forceDelete(); $user->status = 'deleted';
UserFilter::whereUserId($user->id)->forceDelete(); $user->name = 'deleted';
UserSetting::whereUserId($user->id)->forceDelete(); $user->email = $user->id;
$this->deleteUserColumns($user); $user->password = '';
}); $user->remember_token = null;
} $user->is_admin = false;
$user->{'2fa_enabled'} = false;
protected function deleteUserColumns($user) $user->{'2fa_secret'} = null;
{ $user->{'2fa_backup_codes'} = null;
DB::transaction(function() use ($user) { $user->{'2fa_setup_at'} = null;
$user->status = 'deleted'; $user->save();
$user->name = 'deleted'; });
$user->email = $user->id; }
$user->password = '';
$user->remember_token = null;
$user->is_admin = false;
$user->{'2fa_enabled'} = false;
$user->{'2fa_secret'} = null;
$user->{'2fa_backup_codes'} = null;
$user->{'2fa_setup_at'} = null;
$user->save();
});
}
} }

View File

@ -19,6 +19,10 @@ class UserObserver
*/ */
public function saved(User $user) public function saved(User $user)
{ {
if($user->status == 'deleted') {
return;
}
if (empty($user->profile)) { if (empty($user->profile)) {
$profile = DB::transaction(function() use($user) { $profile = DB::transaction(function() use($user) {
$profile = new Profile(); $profile = new Profile();