delete extension added key

kokspflanze · Oct 22, 2015 · dcee8da · dcee8da
1 parent f30aa1f
commit dcee8da
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/PServerCMS/Entity/Repository/UserExtension.php b/src/PServerCMS/Entity/Repository/UserExtension.php
@@ -9,17 +9,23 @@
 
 class UserExtension extends EntityRepository
 {
-
     /**
      * @param UserInterface $user
+     * @param string $key
      */
-    public function deleteExtension(UserInterface $user)
+    public function deleteExtension(UserInterface $user, $key = null)
     {
-        $this->createQueryBuilder('p')
+        $query = $this->createQueryBuilder('p')
             ->delete()
             ->where('p.user = :user')
-            ->setParameter('user', $user)
-            ->getQuery()
+            ->setParameter('user', $user);
+
+        if ($key) {
+            $query->where('p.key = :key')
+                ->setParameter('key', $key);
+        }
+
+        $query->getQuery()
             ->execute();
     }
 }