Add wrapping attribute and update DataProvider method to use it

Author: Jim-Webfox

src/DataProvider.php

@@ -5,16 +5,17 @@
 namespace Webfox\InertiaDataProviders;
 
 use Closure;
-use Illuminate\Contracts\Support\Arrayable;
-use Illuminate\Contracts\Support\Jsonable;
-use Inertia\DeferProp;
+use ReflectionClass;
 use Inertia\LazyProp;
 use Inertia\Response;
-use ReflectionClass;
 use ReflectionMethod;
-use ReflectionNamedType;
+use Inertia\DeferProp;
 use ReflectionProperty;
+use ReflectionNamedType;
+use Illuminate\Contracts\Support\Jsonable;
 use Symfony\Component\VarDumper\VarDumper;
+use Illuminate\Contracts\Support\Arrayable;
+use Webfox\InertiaDataProviders\WrappingAttributes\WrappingAttribute;
 use Webfox\InertiaDataProviders\AttributeNameFormatters\AttributeNameFormatter;
 
 abstract class DataProvider implements Arrayable, Jsonable
@@ -30,30 +31,41 @@ public static function collection(DataProvider|array ...$dataProviders): DataPro
 
     public function toArray(): array
     {
-        $staticData = $this->staticData instanceof Arrayable ? $this->staticData->toArray() : $this->staticData;
+        $staticData      = $this->staticData instanceof Arrayable ? $this->staticData->toArray() : $this->staticData;
         $reflectionClass = (new ReflectionClass($this));
 
         $convertedProperties = collect($reflectionClass->getProperties(ReflectionProperty::IS_PUBLIC))
-            ->filter(fn (ReflectionProperty $property) => ! $property->isStatic())
-            ->mapWithKeys(fn (ReflectionProperty $property) => [$property->getName() => $property->getValue($this)])
-            ->map(fn ($value) => $value instanceof Arrayable ? $value->toArray() : $value);
+            ->filter(fn(ReflectionProperty $property) => !$property->isStatic())
+            ->mapWithKeys(fn(ReflectionProperty $property) => [$property->getName() => $property->getValue($this)])
+            ->map(fn($value) => $value instanceof Arrayable ? $value->toArray() : $value);
 
         $convertedMethods = collect($reflectionClass->getMethods(ReflectionMethod::IS_PUBLIC))
-            ->filter(fn (ReflectionMethod $method) => ! $method->isStatic() && ! in_array($method->name, $this->excludedMethods))
+            ->filter(fn(ReflectionMethod $method) => !$method->isStatic() && !in_array($method->name, $this->excludedMethods))
             ->mapWithKeys(function (ReflectionMethod $method) {
+                $attributes = $method->getAttributes();
                 $returnType = $method->getReturnType();
+
                 if ($returnType instanceof ReflectionNamedType && in_array($returnType->getName(), [DeferProp::class, LazyProp::class, Closure::class])) {
                     return [$method->name => $method->invoke($this)];
                 }
 
-                return [$method->name => fn () => app()->call([$this, $method->name])];
+                if (count($attributes) > 0) {
+                    foreach ($attributes as $attribute) {
+                        $attributeInstance = $attribute->newInstance();
+                        if ($attributeInstance instanceof WrappingAttribute) {
+                            return [$method->name => $attributeInstance(fn() => app()->call([$this, $method->name]))];
+                        }
+                    }
+                }
+
+                return [$method->name => fn() => app()->call([$this, $method->name])];
             });
 
         return collect()
             ->merge($staticData)
             ->merge($convertedProperties)
             ->merge($convertedMethods)
-            ->mapWithKeys(fn ($value, $key) => [$this->attributeNameFormatter()($key) => $value])
+            ->mapWithKeys(fn($value, $key) => [$this->attributeNameFormatter()($key) => $value])
             ->toArray();
     }
 

src/WrappingAttributes/InertiaMerge.php

@@ -0,0 +1,16 @@
+<?php
+
+namespace Webfox\InertiaDataProviders\WrappingAttributes;
+
+
+use Attribute;
+use Inertia\Inertia;
+
+#[Attribute(Attribute::TARGET_METHOD)]
+class InertiaMerge implements WrappingAttribute
+{
+    public function __invoke($data)
+    {
+        return Inertia::Merge($data);
+    }
+}

src/WrappingAttributes/WrappingAttribute.php

@@ -0,0 +1,14 @@
+<?php
+
+namespace Webfox\InertiaDataProviders\WrappingAttributes;
+
+interface WrappingAttribute
+{
+    /**
+     * Wrap the data with the attribute.
+     *
+     * @param mixed $data
+     * @return array
+     */
+    public function __invoke($data);
+}