Adds DTO array transformation and JSON serialization

src/Common/Contracts/WithJsonSerialization.php

@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WordPress\AiClient\Common\Contracts;
+
+use JsonSerializable;
+
+/**
+ * Interface for objects that support JSON serialization and deserialization.
+ *
+ * @since 1.0.0
+ */
+interface WithJsonSerialization extends JsonSerializable
+{
+    /**
+     * Creates an instance from JSON data.
+     *
+     * @since 1.0.0
+     *
+     * @param array<string, mixed> $json The JSON data.
+     * @return self The created instance.
+     */
+    public static function fromJson(array $json);
+}

src/Files/DTO/File.php

@@ -5,6 +5,7 @@
 namespace WordPress\AiClient\Files\DTO;
 
 use WordPress\AiClient\Common\Contracts\WithJsonSchemaInterface;
+use WordPress\AiClient\Common\Contracts\WithJsonSerialization;
 use WordPress\AiClient\Files\Enums\FileTypeEnum;
 use WordPress\AiClient\Files\ValueObjects\MimeType;
 
@@ -16,7 +17,7 @@
  *
  * @since n.e.x.t
  */
-class File implements WithJsonSchemaInterface
+class File implements WithJsonSchemaInterface, WithJsonSerialization
 {
     /**
      * @var MimeType The MIME type of the file.
@@ -377,4 +378,45 @@
             ],
         ];
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     *
+     * @return array<string, mixed>
+     */
+    public function jsonSerialize(): array
+    {
+        $data = [
+            'fileType' => $this->fileType->value,
+            'mimeType' => $this->getMimeType(),
+        ];
+
+        if ($this->fileType->isRemote()) {
+            $data['url'] = $this->url;
+        } else {
+            $data['base64Data'] = $this->base64Data;
+        }
+
+        return $data;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): File
+    {
+        $fileType = FileTypeEnum::from((string) $json['fileType']);
+
+        if ($fileType->isRemote()) {
+            return new self((string) $json['url'], (string) $json['mimeType']);
+        } else {
+            // Create data URI from base64 data and mime type
+            $dataUri = sprintf('data:%s;base64,%s', (string) $json['mimeType'], (string) $json['base64Data']);
+            return new self($dataUri);
+        }
+    }
 }

src/Messages/DTO/Message.php

@@ -5,6 +5,7 @@
 namespace WordPress\AiClient\Messages\DTO;
 
 use WordPress\AiClient\Common\Contracts\WithJsonSchemaInterface;
+use WordPress\AiClient\Common\Contracts\WithJsonSerialization;
 use WordPress\AiClient\Messages\Enums\MessageRoleEnum;
 
 /**
@@ -15,7 +16,7 @@
  *
  * @since n.e.x.t
  */
-class Message implements WithJsonSchemaInterface
+class Message implements WithJsonSchemaInterface, WithJsonSerialization
 {
     /**
      * @var MessageRoleEnum The role of the message sender.
@@ -90,4 +91,38 @@
             'required' => ['role', 'parts'],
         ];
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     *
+     * @return array<string, mixed>
+     */
+    public function jsonSerialize(): array
+    {
+        return [
+            'role' => $this->role->value,
+            'parts' => array_map(function (MessagePart $part) {
+                return $part->jsonSerialize();
+            }, $this->parts),
+        ];
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): Message
+    {
+        $role = MessageRoleEnum::from((string) $json['role']);
+        /** @var array<array<string, mixed>> $partsData */
+        $partsData = $json['parts'];
+        $parts = array_map(function (array $partData) {
+            return MessagePart::fromJson($partData);
+        }, $partsData);
+
+        return new self($role, $parts);
+    }
 }

src/Messages/DTO/MessagePart.php

@@ -5,6 +5,7 @@
 namespace WordPress\AiClient\Messages\DTO;
 
 use WordPress\AiClient\Common\Contracts\WithJsonSchemaInterface;
+use WordPress\AiClient\Common\Contracts\WithJsonSerialization;
 use WordPress\AiClient\Files\DTO\File;
 use WordPress\AiClient\Messages\Enums\MessagePartTypeEnum;
 use WordPress\AiClient\Tools\DTO\FunctionCall;
@@ -18,7 +19,7 @@
  *
  * @since n.e.x.t
  */
-class MessagePart implements WithJsonSchemaInterface
+class MessagePart implements WithJsonSchemaInterface, WithJsonSerialization
 {
     /**
      * @var MessagePartTypeEnum The type of this message part.
@@ -202,4 +203,56 @@
             ],
         ];
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     *
+     * @return array<string, mixed>
+     */
+    public function jsonSerialize(): array
+    {
+        $data = ['type' => $this->type->value];
+
+        if ($this->type->isText() && $this->text !== null) {
+            $data['text'] = $this->text;
+        } elseif ($this->type->isFile() && $this->file !== null) {
+            $data['file'] = $this->file->jsonSerialize();
+        } elseif ($this->type->isFunctionCall() && $this->functionCall !== null) {
+            $data['functionCall'] = $this->functionCall->jsonSerialize();
+        } elseif ($this->type->isFunctionResponse() && $this->functionResponse !== null) {
+            $data['functionResponse'] = $this->functionResponse->jsonSerialize();
+        }
+
+        return $data;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): MessagePart
+    {
+        $type = MessagePartTypeEnum::from((string) $json['type']);
+
+        if ($type->isText()) {
+            return new self((string) $json['text']);
+        } elseif ($type->isFile()) {
+            /** @var array<string, mixed> $fileData */
+            $fileData = $json['file'];
+            return new self(File::fromJson($fileData));
+        } elseif ($type->isFunctionCall()) {
+            /** @var array<string, mixed> $functionCallData */
+            $functionCallData = $json['functionCall'];
+            return new self(FunctionCall::fromJson($functionCallData));
+        } elseif ($type->isFunctionResponse()) {
+            /** @var array<string, mixed> $functionResponseData */
+            $functionResponseData = $json['functionResponse'];
+            return new self(FunctionResponse::fromJson($functionResponseData));
+        }
+
+        throw new \InvalidArgumentException(sprintf('Unknown message part type: %s', $json['type']));
+    }
 }

src/Messages/DTO/ModelMessage.php

@@ -27,4 +27,20 @@ public function __construct(array $parts)
     {
         parent::__construct(MessageRoleEnum::model(), $parts);
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): ModelMessage
+    {
+        /** @var array<array<string, mixed>> $partsData */
+        $partsData = $json['parts'];
+        $parts = array_map(function (array $partData) {
+            return MessagePart::fromJson($partData);
+        }, $partsData);
+
+        return new self($parts);
+    }
 }

src/Messages/DTO/SystemMessage.php

@@ -27,4 +27,20 @@ public function __construct(array $parts)
     {
         parent::__construct(MessageRoleEnum::system(), $parts);
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): SystemMessage
+    {
+        /** @var array<array<string, mixed>> $partsData */
+        $partsData = $json['parts'];
+        $parts = array_map(function (array $partData) {
+            return MessagePart::fromJson($partData);
+        }, $partsData);
+
+        return new self($parts);
+    }
 }

src/Messages/DTO/UserMessage.php

@@ -26,4 +26,20 @@ public function __construct(array $parts)
     {
         parent::__construct(MessageRoleEnum::user(), $parts);
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): UserMessage
+    {
+        /** @var array<array<string, mixed>> $partsData */
+        $partsData = $json['parts'];
+        $parts = array_map(function (array $partData) {
+            return MessagePart::fromJson($partData);
+        }, $partsData);
+
+        return new self($parts);
+    }
 }

src/Operations/Contracts/OperationInterface.php

@@ -5,6 +5,7 @@
 namespace WordPress\AiClient\Operations\Contracts;
 
 use WordPress\AiClient\Common\Contracts\WithJsonSchemaInterface;
+use WordPress\AiClient\Common\Contracts\WithJsonSerialization;
 use WordPress\AiClient\Operations\Enums\OperationStateEnum;
 
 /**
@@ -15,7 +16,7 @@
  *
  * @since n.e.x.t
  */
-interface OperationInterface extends WithJsonSchemaInterface
+interface OperationInterface extends WithJsonSchemaInterface, WithJsonSerialization
 {
     /**
      * Gets the operation ID.

src/Operations/DTO/GenerativeAiOperation.php

@@ -132,4 +132,43 @@
             ],
         ];
     }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     *
+     * @return array<string, mixed>
+     */
+    public function jsonSerialize(): array
+    {
+        $data = [
+            'id' => $this->id,
+            'state' => $this->state->value,
+        ];
+
+        if ($this->result !== null) {
+            $data['result'] = $this->result->jsonSerialize();
+        }
+
+        return $data;
+    }
+
+    /**
+     * {@inheritDoc}
+     *
+     * @since n.e.x.t
+     */
+    public static function fromJson(array $json): GenerativeAiOperation
+    {
+        $state = OperationStateEnum::from((string) $json['state']);
+        $result = null;
+        if (isset($json['result'])) {
+            /** @var array<string, mixed> $resultData */
+            $resultData = $json['result'];
+            $result = GenerativeAiResult::fromJson($resultData);
+        }
+
+        return new self((string) $json['id'], $state, $result);
+    }
 }

src/Results/Contracts/ResultInterface.php

@@ -5,6 +5,7 @@
 namespace WordPress\AiClient\Results\Contracts;
 
 use WordPress\AiClient\Common\Contracts\WithJsonSchemaInterface;
+use WordPress\AiClient\Common\Contracts\WithJsonSerialization;
 use WordPress\AiClient\Results\DTO\TokenUsage;
 
 /**
@@ -15,7 +16,7 @@
  *
  * @since n.e.x.t
  */
-interface ResultInterface extends WithJsonSchemaInterface
+interface ResultInterface extends WithJsonSchemaInterface, WithJsonSerialization
 {
     /**
      * Gets the result ID.