fix persistDeviceCode usage

thephpleague · Sep 25, 2024 · e95d664 · e95d664
1 parent 1d82315
commit e95d664
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 2 deletions.
diff --git a/examples/src/Repositories/DeviceCodeRepository.php b/examples/src/Repositories/DeviceCodeRepository.php
@@ -36,6 +36,28 @@ public function persistDeviceCode(DeviceCodeEntityInterface $deviceCodeEntity):
         // Some logic to persist a new device code to a database
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function persistUser(DeviceCodeEntityInterface $deviceCodeEntity): void
+    {
+        $user = $deviceCodeEntity->getUserIdentifier();
+        $approved = $deviceCodeEntity->getUserApproved();
+
+        // Some logic to persist user ID and approval status of the given device code entity to a database
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function persistLastPolledAt(DeviceCodeEntityInterface $deviceCodeEntity): void
+    {
+        $lastPolledAt = $deviceCodeEntity->getLastPolledAt();
+        $approved = $deviceCodeEntity->getUserApproved();
+
+        // Some logic to persist "last polled at" datetime of the given device code entity to a database
+    }
+
     /**
      * {@inheritdoc}
      */

diff --git a/src/Grant/DeviceCodeGrant.php b/src/Grant/DeviceCodeGrant.php
@@ -124,7 +124,7 @@ public function completeDeviceAuthorizationRequest(string $deviceCode, string $u
         $deviceCode->setUserIdentifier($userId);
         $deviceCode->setUserApproved($userApproved);
 
-        $this->deviceCodeRepository->persistDeviceCode($deviceCode);
+        $this->deviceCodeRepository->persistUser($deviceCode);
     }
 
     /**
@@ -141,7 +141,7 @@ public function respondToAccessTokenRequest(
         $deviceCodeEntity = $this->validateDeviceCode($request, $client);
 
         $deviceCodeEntity->setLastPolledAt(new DateTimeImmutable());
-        $this->deviceCodeRepository->persistDeviceCode($deviceCodeEntity);
+        $this->deviceCodeRepository->persistLastPolledAt($deviceCodeEntity);
 
         // If device code has no user associated, respond with pending
         if (is_null($deviceCodeEntity->getUserIdentifier())) {

diff --git a/src/Repositories/DeviceCodeRepositoryInterface.php b/src/Repositories/DeviceCodeRepositoryInterface.php
@@ -29,6 +29,16 @@ public function getNewDeviceCode(): DeviceCodeEntityInterface;
      */
     public function persistDeviceCode(DeviceCodeEntityInterface $deviceCodeEntity): void;
 
+    /**
+     * Persists user ID and approval status of the given device code entity to permanent storage.
+     */
+    public function persistUser(DeviceCodeEntityInterface $deviceCodeEntity): void;
+
+    /**
+     * Persists "last polled at" datetime of the given device code entity to permanent storage.
+     */
+    public function persistLastPolledAt(DeviceCodeEntityInterface $deviceCodeEntity): void;
+
     /**
      * Get a device code entity.
      */