feat(type): Notnull type optimizations

docs/component/type.md

@@ -40,7 +40,7 @@
 - [non_empty_dict](./../../src/Psl/Type/non_empty_dict.php#L18)
 - [non_empty_string](./../../src/Psl/Type/non_empty_string.php#L14)
 - [non_empty_vec](./../../src/Psl/Type/non_empty_vec.php#L16)
-- [nonnull](./../../src/Psl/Type/nonnull.php#L16)
+- [nonnull](./../../src/Psl/Type/nonnull.php#L18)
 - [null](./../../src/Psl/Type/null.php#L14)
 - [nullable](./../../src/Psl/Type/nullable.php#L16)
 - [num](./../../src/Psl/Type/num.php#L14)

src/Psl/Type/Internal/NonNullType.php

@@ -18,19 +18,29 @@
 final readonly class NonNullType extends Type\Type
 {
     /**
-     * @psalm-assert-if-true mixed $value
+     * @template T of mixed
+     *
+     * @param T|null $value
+     *
+     * @psalm-assert-if-true T $value
      *
      * @ara-assert-if-true nonnull $value
+     *
+     * @return ($value is null ? false : true)
      */
     public function matches(mixed $value): bool
     {
         return null !== $value;
     }
 
     /**
+     * @template T of mixed
+     *
+     * @param T|null $value
+     *
      * @ara-return nonnull
      *
-     * @return mixed
+     * @return ($value is null ? never : T)
      */
     public function coerce(mixed $value): mixed
     {
@@ -42,13 +52,17 @@ public function coerce(mixed $value): mixed
     }
 
     /**
+     * @template T
+     *
+     * @param T|null $value
+     *
      * @ara-assert nonnull $value
      *
-     * @psalm-assert mixed $value
+     * @psalm-assert T $value
      *
      * @ara-return nonnull
      *
-     * @return mixed
+     * @return ($value is null ? never : T)
      */
     public function assert(mixed $value): mixed
     {

src/Psl/Type/nonnull.php

@@ -4,14 +4,16 @@
 
 namespace Psl\Type;
 
+use Psl\Type\Internal\NonNullType;
+
 /**
  * @psalm-pure
  *
  * @psalm-suppress ImpureStaticVariable - The $instance is always the same and is considered pure.
  *
  * @ara-return TypeInterface<nonnull>
  *
- * @return TypeInterface<mixed>
+ * @return NonNullType
  */
 function nonnull(): TypeInterface
 {

tests/static-analysis/Type/nonnull.php

@@ -0,0 +1,64 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Psl\Tests\StaticAnalysis\Type;
+
+use Psl\Type;
+
+/**
+ * @throws Type\Exception\AssertException
+ */
+function returns_non_null_assertion(?string $state): string
+{
+    return Type\nonnull()->assert($state);
+}
+
+/**
+ * @throws Type\Exception\AssertException
+ */
+function returns_non_null_assertion_asserted(?string $state): string
+{
+    Type\nonnull()->assert($state);
+
+    return $state;
+}
+
+/**
+ * @throws Type\Exception\CoercionException
+ */
+function returns_non_null_coercion(?string $state): string
+{
+    return Type\nonnull()->coerce($state);
+}
+
+/**
+ * @return true
+ */
+function returns_truthy_match(string $state): bool
+{
+    return Type\nonnull()->matches($state);
+}
+
+/**
+ * @return false
+ */
+function returns_falsy_match(null $state = null): bool
+{
+    return Type\nonnull()->matches($state);
+}
+
+/**
+ * @throws Type\Exception\CoercionException
+ *
+ * Wrapping it in container types still leeds to mixed (including null) return type.
+ * This won't be solvable until psalm supports a TNotNull type.
+ *
+ * @return array<'mightBeNull', mixed>
+ */
+function returns_mixed_in_shape(mixed $data): array
+{
+    return Type\shape([
+        'mightBeNull' => Type\nonnull(),
+    ])->coerce($data);
+}