Returning $default if $key is not present. And only calling ArrayHelper if needed.

Author: patel-vansh

system/Context/Context.php

@@ -52,12 +52,24 @@ public function set(array|string $key, mixed $value = null): self
     {
         if (is_array($key)) {
             foreach ($key as $k => $v) {
+                if (! $this->hasDotNotation($k)) {
+                    $this->data[$k] = $v;
+
+                    continue;
+                }
+
                 ArrayHelper::dotSet($this->data, $k, $v);
             }
 
             return $this;
         }
 
+        if (! $this->hasDotNotation($key)) {
+            $this->data[$key] = $value;
+
+            return $this;
+        }
+
         ArrayHelper::dotSet($this->data, $key, $value);
 
         return $this;
@@ -76,12 +88,24 @@ public function setHidden(#[SensitiveParameter] array|string $key, #[SensitivePa
     {
         if (is_array($key)) {
             foreach ($key as $k => $v) {
+                if (! $this->hasDotNotation($k)) {
+                    $this->hiddenData[$k] = $v;
+
+                    continue;
+                }
+
                 ArrayHelper::dotSet($this->hiddenData, $k, $v);
             }
 
             return $this;
         }
 
+        if (! $this->hasDotNotation($key)) {
+            $this->hiddenData[$key] = $value;
+
+            return $this;
+        }
+
         ArrayHelper::dotSet($this->hiddenData, $key, $value);
 
         return $this;
@@ -98,7 +122,16 @@ public function setHidden(#[SensitiveParameter] array|string $key, #[SensitivePa
      */
     public function get(string $key, mixed $default = null): mixed
     {
-        return ArrayHelper::dotSearch($key, $this->data) ?? $default;
+        if (! $this->has($key)) {
+            return $default;
+        }
+
+        // Exit early if the key is not a dot notation to avoid unnecessary processing in the common case.
+        if (! $this->hasDotNotation($key)) {
+            return $this->data[$key];
+        }
+
+        return ArrayHelper::dotSearch($key, $this->data);
     }
 
     /**
@@ -148,7 +181,16 @@ public function getAll(): array
      */
     public function getHidden(#[SensitiveParameter] string $key, #[SensitiveParameter] mixed $default = null): mixed
     {
-        return ArrayHelper::dotSearch($key, $this->hiddenData) ?? $default;
+        if (! $this->hasHidden($key)) {
+            return $default;
+        }
+
+        // Exit early if the key is not a dot notation to avoid unnecessary processing in the common case.
+        if (! $this->hasDotNotation($key)) {
+            return $this->hiddenData[$key];
+        }
+
+        return ArrayHelper::dotSearch($key, $this->hiddenData);
     }
 
     /**
@@ -197,6 +239,10 @@ public function getAllHidden(): array
      */
     public function has(string $key): bool
     {
+        if (! $this->hasDotNotation($key)) {
+            return array_key_exists($key, $this->data);
+        }
+
         return ArrayHelper::dotHas($key, $this->data);
     }
 
@@ -210,6 +256,10 @@ public function has(string $key): bool
      */
     public function hasHidden(string $key): bool
     {
+        if (! $this->hasDotNotation($key)) {
+            return array_key_exists($key, $this->hiddenData);
+        }
+
         return ArrayHelper::dotHas($key, $this->hiddenData);
     }
 
@@ -225,12 +275,24 @@ public function remove(array|string $key): self
     {
         if (is_array($key)) {
             foreach ($key as $k) {
+                if (! $this->hasDotNotation($k)) {
+                    unset($this->data[$k]);
+
+                    continue;
+                }
+
                 ArrayHelper::dotUnset($this->data, $k);
             }
 
             return $this;
         }
 
+        if (! $this->hasDotNotation($key)) {
+            unset($this->data[$key]);
+
+            return $this;
+        }
+
         ArrayHelper::dotUnset($this->data, $key);
 
         return $this;
@@ -248,12 +310,24 @@ public function removeHidden(#[SensitiveParameter] array|string $key): self
     {
         if (is_array($key)) {
             foreach ($key as $k) {
+                if (! $this->hasDotNotation($k)) {
+                    unset($this->hiddenData[$k]);
+
+                    continue;
+                }
+
                 ArrayHelper::dotUnset($this->hiddenData, $k);
             }
 
             return $this;
         }
 
+        if (! $this->hasDotNotation($key)) {
+            unset($this->hiddenData[$key]);
+
+            return $this;
+        }
+
         ArrayHelper::dotUnset($this->hiddenData, $key);
 
         return $this;
@@ -324,4 +398,9 @@ public function __unserialize(array $data): void
         $this->data       = $data['data'] ?? [];
         $this->hiddenData = [];
     }
+
+    private function hasDotNotation(string $key): bool
+    {
+        return str_contains($key, '.');
+    }
 }