Convert values to forType->setType with update DTO as the result

Author: roxblnfk

src/Common/SearchAttributes/SearchAttributeKey.php

@@ -0,0 +1,89 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey\BoolValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\DatetimeValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\FloatValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\IntValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\KeywordListValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\KeywordValue;
+use Temporal\Common\SearchAttributes\SearchAttributeKey\StringValue;
+
+/**
+ * @template-covariant TValue
+ * @psalm-immutable
+ */
+abstract class SearchAttributeKey
+{
+    /**
+     * @param non-empty-string $key
+     * @param TValue $value
+     */
+    final protected function __construct(
+        protected readonly string $key,
+    ) {}
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forBool(string $key): BoolValue
+    {
+        return new BoolValue($key);
+    }
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forInteger(string $key): IntValue
+    {
+        return new IntValue($key);
+    }
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forFloat(string $key): FloatValue
+    {
+        return new FloatValue($key);
+    }
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forKeyword(string $key): KeywordValue
+    {
+        return new KeywordValue($key);
+    }
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forString(string $key): StringValue
+    {
+        return new StringValue($key);
+    }
+
+    public static function forDatetime(string $key): DatetimeValue
+    {
+        return new DatetimeValue($key);
+    }
+
+    /**
+     * @param non-empty-string $key
+     */
+    public static function forKeywordList(string $key): KeywordListValue
+    {
+        return new KeywordListValue($key);
+    }
+
+    public function valueUnset(): SearchAttributeUpdate {}
+
+    protected function prepareValueSet(mixed $value): SearchAttributeUpdate {
+        return SearchAttributeUpdate::valueSet($this->key, $this->getType(), $value);
+    }
+
+    abstract protected function getType(): ValueType;
+}

src/Common/SearchAttributes/SearchAttributeKey/BoolValue.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<bool>
+ * @psalm-immutable
+ */
+final class BoolValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::Bool;
+    }
+
+    public function valueSet(bool $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet($value);
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/DatetimeValue.php

@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use DateTimeImmutable;
+use DateTimeInterface;
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<DateTimeImmutable>
+ * @psalm-immutable
+ */
+final class DatetimeValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::Datetime;
+    }
+
+    /**
+     * @param non-empty-string|DateTimeInterface $value
+     */
+    public function valueSet(string|DateTimeInterface $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet(match (true) {
+            \is_string($value) => new \DateTimeImmutable($value),
+            $value instanceof \DateTimeImmutable => $value,
+            default => \DateTimeImmutable::createFromInterface($value),
+        });
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/FloatValue.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<float>
+ * @psalm-immutable
+ */
+final class FloatValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::Float;
+    }
+
+    public function valueSet(float $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet($value);
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/IntValue.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<int>
+ * @psalm-immutable
+ */
+final class IntValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::Int;
+    }
+
+    public function valueSet(int $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet($value);
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/KeywordListValue.php

@@ -0,0 +1,34 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<list<string>>
+ * @psalm-immutable
+ */
+final class KeywordListValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::KeywordList;
+    }
+
+    /**
+     * @param iterable<string|\Stringable> $value
+     */
+    public function valueSet(array $value): SearchAttributeUpdate
+    {
+        $values = [];
+        foreach ($value as $v) {
+            $values[] = (string) $v;
+        }
+
+        return $this->prepareValueSet($values);
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/KeywordValue.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<string>
+ * @psalm-immutable
+ */
+final class KeywordValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::Keyword;
+    }
+
+    public function valueSet(string|\Stringable $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet((string) $value);
+    }
+}

src/Common/SearchAttributes/SearchAttributeKey/StringValue.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeKey;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<string>
+ * @psalm-immutable
+ */
+final class StringValue extends SearchAttributeKey
+{
+    protected function getType(): ValueType
+    {
+        return ValueType::String;
+    }
+
+    public function valueSet(string|\Stringable $value): SearchAttributeUpdate
+    {
+        return $this->prepareValueSet((string) $value);
+    }
+}

src/Common/SearchAttributes/SearchAttributeUpdate.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes;
+
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate\ValueSet;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate\ValueUnset;
+
+/**
+ * @template-extends SearchAttributeKey<string>
+ * @psalm-immutable
+ */
+abstract class SearchAttributeUpdate
+{
+    /**
+     * @param non-empty-string $key
+     */
+    protected function __construct(
+        public readonly string $key,
+        public readonly ValueType $type,
+    ) {}
+
+    public static function valueSet(string $key, ValueType $type, mixed $value): self
+    {
+        return new ValueSet($key, $type, $value);
+    }
+
+    public static function valueUnset(string $key, ValueType $type): self
+    {
+        return new ValueUnset($key, $type);
+    }
+}

src/Common/SearchAttributes/SearchAttributeUpdate/ValueSet.php

@@ -0,0 +1,27 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+
+use Temporal\Common\SearchAttributes\SearchAttributeKey;
+use Temporal\Common\SearchAttributes\SearchAttributeUpdate;
+use Temporal\Common\SearchAttributes\ValueType;
+
+/**
+ * @template-extends SearchAttributeKey<string>
+ * @psalm-immutable
+ */
+final class ValueSet extends SearchAttributeUpdate
+{
+    /**
+     * @param non-empty-string $key
+     */
+    protected function __construct(
+        string $key,
+        ValueType $type,
+        public readonly mixed $value,
+    ) {
+        parent::__construct($key, $type);
+    }
+}