feat: Make ai usage counter polymorphic

Author: kofimokome

database/migrations/2026_06_16_201941_create_ai_usage_counters_table.php

@@ -12,11 +12,10 @@ public function up(): void
     {
         Schema::create('ai_usage_counters', function (Blueprint $table) {
             $table->id();
-            $table->unsignedBigInteger('user_id');
+            $table->morphs('owner');
             $table->timestamp('period_start');
             $table->integer('tokens_used');
 
-            $table->foreign('user_id')->references('id')->on('users');
             $table->timestamps();
         });
     }

src/Models/AiUsageCounter.php

@@ -20,7 +20,8 @@ class AiUsageCounter extends Model
      * @var array<int, string>
      */
     protected $fillable = [
-        'user_id',
+        'owner_id',
+        'owner_type',
         'period_start',
         'tokens_used',
     ];

src/Support/CloudEntitlements.php

@@ -62,7 +62,8 @@ public function remaining(?Model $owner, string $meter): int|float
         }
 
         $periodStart = Carbon::now()->startOfMonth();
-        $used = AiUsageCounter::where('user_id', $owner->id)
+        $used = AiUsageCounter::where('owner_id', $owner->id)
+            ->where('owner_type', get_class($owner))
             ->where('period_start', $periodStart)
             ->value('tokens_used') ?? 0;
 
@@ -101,15 +102,17 @@ public function consume(?Model $owner, string $meter, int $amount): void
 
         $periodStart = Carbon::now()->startOfMonth();
 
-        $counter = AiUsageCounter::where('user_id', $owner->id)
+        $counter = AiUsageCounter::where('owner_id', $owner->id)
+            ->where('owner_type', get_class($owner))
             ->where('period_start', $periodStart)
             ->first();
 
         if ($counter) {
             $counter->increment('tokens_used', $amount);
         } else {
             AiUsageCounter::create([
-                'user_id' => $owner->id,
+                'owner_id' => $owner->id,
+                'owner_type' => get_class($owner),
                 'period_start' => $periodStart,
                 'tokens_used' => $amount,
             ]);