| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135 |
- <?php
- declare(strict_types=1);
- /**
- * This file is part of Hyperf.
- *
- * @link https://www.hyperf.io
- * @document https://hyperf.wiki
- * @contact group@hyperf.io
- * @license https://github.com/hyperf/hyperf/blob/master/LICENSE
- */
- namespace Hyperf\Database\Model;
- use function Hyperf\Tappable\tap;
- class SoftDeletingScope implements Scope
- {
- /**
- * All the extensions to be added to the builder.
- */
- protected array $extensions = ['Restore', 'RestoreOrCreate', 'WithTrashed', 'WithoutTrashed', 'OnlyTrashed'];
- /**
- * Apply the scope to a given Model query builder.
- */
- public function apply(Builder $builder, Model $model)
- {
- $builder->whereNull($model->getQualifiedDeletedAtColumn());
- }
- /**
- * Extend the query builder with the needed functions.
- */
- public function extend(Builder $builder)
- {
- foreach ($this->extensions as $extension) {
- $this->{"add{$extension}"}($builder);
- }
- $builder->onDelete(function (Builder $builder) {
- $column = $this->getDeletedAtColumn($builder);
- return $builder->update([
- $column => $builder->getModel()->freshTimestampString(),
- ]);
- });
- }
- /**
- * Get the "deleted at" column for the builder.
- *
- * @return string
- */
- protected function getDeletedAtColumn(Builder $builder)
- {
- if (count((array) $builder->getQuery()->joins) > 0) {
- return $builder->getModel()->getQualifiedDeletedAtColumn();
- }
- return $builder->getModel()->getDeletedAtColumn();
- }
- /**
- * Add the restore extension to the builder.
- */
- protected function addRestore(Builder $builder)
- {
- $builder->macro('restore', function (Builder $builder) {
- $builder->withTrashed();
- return $builder->update([$builder->getModel()->getDeletedAtColumn() => null]);
- });
- }
- /**
- * Add the restore-or-create extension to the builder.
- */
- protected function addRestoreOrCreate(Builder $builder)
- {
- $builder->macro('restoreOrCreate', function (Builder $builder, array $attributes = [], array $values = []) {
- $builder->withTrashed();
- return tap($builder->firstOrCreate($attributes, $values), function ($instance) {
- $instance->restore();
- });
- });
- }
- /**
- * Add the with-trashed extension to the builder.
- */
- protected function addWithTrashed(Builder $builder)
- {
- $builder->macro('withTrashed', function (Builder $builder, $withTrashed = true) {
- if (! $withTrashed) {
- return $builder->withoutTrashed();
- }
- return $builder->withoutGlobalScope($this);
- });
- }
- /**
- * Add the without-trashed extension to the builder.
- */
- protected function addWithoutTrashed(Builder $builder)
- {
- $builder->macro('withoutTrashed', function (Builder $builder) {
- $model = $builder->getModel();
- $builder->withoutGlobalScope($this)->whereNull(
- $model->getQualifiedDeletedAtColumn()
- );
- return $builder;
- });
- }
- /**
- * Add the only-trashed extension to the builder.
- */
- protected function addOnlyTrashed(Builder $builder)
- {
- $builder->macro('onlyTrashed', function (Builder $builder) {
- $model = $builder->getModel();
- $builder->withoutGlobalScope($this)->whereNotNull(
- $model->getQualifiedDeletedAtColumn()
- );
- return $builder;
- });
- }
- }
|