Adding conditionable on Host interface

Author: nyamsprod

src/Domain.php

@@ -7,6 +7,8 @@
 use Iterator;
 use Stringable;
 
+use function is_bool;
+
 use const FILTER_FLAG_IPV4;
 use const FILTER_VALIDATE_IP;
 
@@ -183,4 +185,25 @@ public function isAbsolute(): bool
     {
         return '' === $this->label(0);
     }
+
+    /**
+     * Apply the callback if the given "condition" is (or resolves to) true.
+     *
+     * @param (callable($this): bool)|bool $condition
+     * @param callable($this): (self|null) $onSuccess
+     * @param ?callable($this): (self|null) $onFail
+     *
+     */
+    public function when(callable|bool $condition, callable $onSuccess, ?callable $onFail = null): self
+    {
+        if (!is_bool($condition)) {
+            $condition = $condition($this);
+        }
+
+        return match (true) {
+            $condition => $onSuccess($this),
+            null !== $onFail => $onFail($this),
+            default => $this,
+        } ?? $this;
+    }
 }

src/Host.php

@@ -15,6 +15,7 @@
  * @method bool isAbsolute() tells whether the domain is absolute or not.
  * @method static withRootLabel() returns an instance with its Root label. (see https://tools.ietf.org/html/rfc3986#section-3.2.2)
  * @method static withoutRootLabel() returns an instance without its Root label. (see https://tools.ietf.org/html/rfc3986#section-3.2.2)
+ * @method static when(callable|bool $condition, callable $onSuccess, ?callable $onFail = null) apply the callback if the given "condition" is (or resolves to) true.
  */
 interface Host extends Countable, JsonSerializable
 {

src/RegisteredName.php

@@ -16,6 +16,7 @@
 use function filter_var;
 use function implode;
 use function in_array;
+use function is_bool;
 use function ksort;
 use function preg_match;
 use function rawurldecode;
@@ -384,4 +385,25 @@ public function isAbsolute(): bool
     {
         return '' === $this->label(0);
     }
+
+    /**
+     * Apply the callback if the given "condition" is (or resolves to) true.
+     *
+     * @param (callable($this): bool)|bool $condition
+     * @param callable($this): (self|null) $onSuccess
+     * @param ?callable($this): (self|null) $onFail
+     *
+     */
+    public function when(callable|bool $condition, callable $onSuccess, ?callable $onFail = null): self
+    {
+        if (!is_bool($condition)) {
+            $condition = $condition($this);
+        }
+
+        return match (true) {
+            $condition => $onSuccess($this),
+            null !== $onFail => $onFail($this),
+            default => $this,
+        } ?? $this;
+    }
 }

src/ResolvedDomain.php

@@ -7,6 +7,7 @@
 use Stringable;
 
 use function count;
+use function is_bool;
 
 final class ResolvedDomain implements ResolvedDomainName
 {
@@ -217,7 +218,7 @@ public function withSuffix(DomainNameProvider|Host|Stringable|string|int|null $s
         $domain = $this->domain->withoutRootLabel()->slice(count($this->suffix))->append($suffix);
 
         return new self(
-            $this->domain->isAbsolute() ? $domain->withRootLabel() : $domain,
+            $domain->when($this->domain->isAbsolute(), fn (DomainName $domainName) => $domain->withRootLabel()),
             $suffix->normalize($this->domain)
         );
     }
@@ -245,7 +246,7 @@ public function withSubDomain(DomainNameProvider|Host|Stringable|string|int|null
 
         $domain = $this->registrableDomain->prepend($subDomain);
 
-        return new self($this->domain->isAbsolute() ? $domain->withRootLabel() : $domain, $this->suffix);
+        return new self($domain->when($this->domain->isAbsolute(), fn (DomainName $domainName) => $domain->withRootLabel()), $this->suffix);
     }
 
     /**
@@ -276,4 +277,25 @@ public function withSecondLevelDomain(DomainNameProvider|Host|Stringable|string|
 
         return new self($newRegistrableDomain->prepend($this->subDomain), $this->suffix);
     }
+
+    /**
+     * Apply the callback if the given "condition" is (or resolves to) true.
+     *
+     * @param (callable($this): bool)|bool $condition
+     * @param callable($this): (self|null) $onSuccess
+     * @param ?callable($this): (self|null) $onFail
+     *
+     */
+    public function when(callable|bool $condition, callable $onSuccess, ?callable $onFail = null): self
+    {
+        if (!is_bool($condition)) {
+            $condition = $condition($this);
+        }
+
+        return match (true) {
+            $condition => $onSuccess($this),
+            null !== $onFail => $onFail($this),
+            default => $this,
+        } ?? $this;
+    }
 }