[12.x] feat: create UsePolicy attribute (#55882)

* feat: create UsePolicy attribute

* formatting

---------

Co-authored-by: Taylor Otwell <[email protected]>

Author: calebdw

src/Illuminate/Auth/Access/Gate.php

@@ -8,6 +8,7 @@
 use Illuminate\Contracts\Auth\Access\Gate as GateContract;
 use Illuminate\Contracts\Container\Container;
 use Illuminate\Contracts\Events\Dispatcher;
+use Illuminate\Database\Eloquent\Attributes\UsePolicy;
 use Illuminate\Support\Arr;
 use Illuminate\Support\Collection;
 use Illuminate\Support\Str;
@@ -669,6 +670,12 @@ public function getPolicyFor($class)
             return $this->resolvePolicy($this->policies[$class]);
         }
 
+        $policy = $this->getPolicyFromAttribute($class);
+
+        if (! is_null($policy)) {
+            return $this->resolvePolicy($policy);
+        }
+
         foreach ($this->guessPolicyName($class) as $guessedPolicy) {
             if (class_exists($guessedPolicy)) {
                 return $this->resolvePolicy($guessedPolicy);
@@ -682,6 +689,25 @@ public function getPolicyFor($class)
         }
     }
 
+    /**
+     * Get the policy class from the class attribute.
+     *
+     * @param  class-string<*>  $class
+     * @return class-string<*>|null
+     */
+    protected function getPolicyFromAttribute(string $class): ?string
+    {
+        if (! class_exists($class)) {
+            return null;
+        }
+
+        $attributes = (new ReflectionClass($class))->getAttributes(UsePolicy::class);
+
+        return $attributes !== []
+            ? $attributes[0]->newInstance()->class
+            : null;
+    }
+
     /**
      * Guess the policy name for the given class.
      *

src/Illuminate/Database/Eloquent/Attributes/UsePolicy.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace Illuminate\Database\Eloquent\Attributes;
+
+use Attribute;
+
+#[Attribute(Attribute::TARGET_CLASS)]
+class UsePolicy
+{
+    /**
+     * Create a new attribute instance.
+     *
+     * @param  class-string<*>  $class
+     */
+    public function __construct(public string $class)
+    {
+    }
+}

tests/Integration/Auth/GatePolicyResolutionTest.php

@@ -3,6 +3,8 @@
 namespace Illuminate\Tests\Integration\Auth;
 
 use Illuminate\Auth\Access\Events\GateEvaluated;
+use Illuminate\Database\Eloquent\Attributes\UsePolicy;
+use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\Facades\Event;
 use Illuminate\Support\Facades\Gate;
 use Illuminate\Tests\Integration\Auth\Fixtures\AuthenticationTestUser;
@@ -78,4 +80,20 @@ public function testPolicyCanBeGuessedMultipleTimes()
             Gate::getPolicyFor(AuthenticationTestUser::class)
         );
     }
+
+    public function testPolicyCanBeGivenByAttribute(): void
+    {
+        Gate::guessPolicyNamesUsing(fn () => [AuthenticationTestUserPolicy::class]);
+
+        $this->assertInstanceOf(PostPolicy::class, Gate::getPolicyFor(Post::class));
+    }
+}
+
+#[UsePolicy(PostPolicy::class)]
+class Post extends Model
+{
+}
+
+class PostPolicy
+{
 }