tag

lyrasoft · Nov 21, 2023 · 1784b13 · 1784b13
1 parent 5634632
commit 1784b13
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 19 deletions.
diff --git a/src/Entity/TagMap.php b/src/Entity/TagMap.php
@@ -11,6 +11,8 @@
 use Windwalker\ORM\EntityTrait;
 use Windwalker\ORM\Metadata\EntityMetadata;
 
+use function Windwalker\unwrap_enum;
+
 /**
  * The TagMap class.
  */
@@ -64,9 +66,9 @@ public function getType(): string
         return $this->type;
     }
 
-    public function setType(string $type): static
+    public function setType(string|\BackedEnum $type): static
     {
-        $this->type = $type;
+        $this->type = (string) unwrap_enum($type);
 
         return $this;
     }

diff --git a/src/Services/TagService.php b/src/Services/TagService.php
@@ -10,6 +10,7 @@
 use Windwalker\Database\Driver\StatementInterface;
 use Windwalker\ORM\EntityMapper;
 use Windwalker\ORM\ORM;
+use Windwalker\ORM\SelectorQuery;
 use Windwalker\Utilities\Str;
 
 /**
@@ -51,35 +52,31 @@ public function createTagsIfNew(iterable $tagIds): array
     }
 
     /**
-     * flushTags
-     *
-     * @param  string    $type
-     * @param  mixed     $targetId
-     * @param  iterable  $tagIds
+     * @param  string|\BackedEnum  $type
+     * @param  mixed               $targetId
+     * @param  iterable            $tagIds
      *
      * @return  iterable<TagMap>
      *
      * @throws ReflectionException
      */
-    public function flushTagMapsFromInput(string $type, mixed $targetId, iterable $tagIds): iterable
+    public function flushTagMapsFromInput(string|\BackedEnum $type, mixed $targetId, iterable $tagIds): iterable
     {
         $tagIds = $this->createTagsIfNew($tagIds);
 
         return $this->flushTagMaps($type, $targetId, $tagIds);
     }
 
     /**
-     * flushTags
-     *
-     * @param  string    $type
-     * @param  mixed     $targetId
-     * @param  iterable  $tagIds
+     * @param  string|\BackedEnum  $type
+     * @param  mixed               $targetId
+     * @param  iterable            $tagIds
      *
      * @return  iterable<TagMap>
      *
      * @throws ReflectionException
      */
-    public function flushTagMaps(string $type, mixed $targetId, iterable $tagIds): iterable
+    public function flushTagMaps(string|\BackedEnum $type, mixed $targetId, iterable $tagIds): iterable
     {
         /** @var EntityMapper<TagMap> $tagMapMapper */
         $tagMapMapper = $this->orm->mapper(TagMap::class);
@@ -101,36 +98,54 @@ public function flushTagMaps(string $type, mixed $targetId, iterable $tagIds): i
      * @param  string  $type
      * @param  mixed   $targetId
      *
-     * @return  array<StatementInterface>
+     * @return void
      *
      * @throws ReflectionException
      */
-    public function clearMapsOfTarget(string $type, mixed $targetId): array
+    public function clearMapsOfTarget(string $type, mixed $targetId): void
     {
         /** @var EntityMapper<TagMap> $tagMapMapper */
         $tagMapMapper = $this->orm->mapper(TagMap::class);
 
-        return $tagMapMapper->deleteWhere(
+        $tagMapMapper->deleteWhere(
             [
                 'type' => $type,
                 'target_id' => $targetId,
             ]
         );
     }
 
-    public function clearMapsOfTag(string $type, mixed $tagId): array
+    public function clearMapsOfTag(string $type, mixed $tagId): void
     {
         /** @var EntityMapper<TagMap> $tagMapMapper */
         $tagMapMapper = $this->orm->mapper(TagMap::class);
 
-        return $tagMapMapper->deleteWhere(
+        $tagMapMapper->deleteWhere(
             [
                 'type' => $type,
                 'tag_id' => $tagId,
             ]
         );
     }
 
+    public function getTagListQuery(string|\BackedEnum $type, mixed $targetIds = null): SelectorQuery
+    {
+        $query = $this->orm->from(Tag::class)
+            ->leftJoin(TagMap::class)
+            ->where('tag_map.type', $type);
+
+        if (is_iterable($targetIds)) {
+            $targetIds = iterator_to_array($targetIds);
+        }
+
+        if ($targetIds !== null && $targetIds !== []) {
+            $query->where('tag_map.target_id', $targetIds);
+        }
+
+        return $query->groupByJoins()
+            ->setDefaultItemClass(Tag::class);
+    }
+
     /**
      * @return string
      */