<?php

namespace Drupal\ai\OperationType\Chat;

use Drupal\Component\Serialization\Json;
use Drupal\ai\OperationType\Chat\Tools\ToolsFunctionOutput;
use Drupal\ai\OperationType\GenericType\FileBaseInterface;
use Drupal\ai\OperationType\GenericType\ImageFile;
use Drupal\ai\Traits\File\FileMimeTypeTrait;
use Drupal\file\Entity\File;

/**
 * One chat messages for chat input.
 */
class ChatMessage {

  use FileMimeTypeTrait;

  /**
   * The role of the message.
   *
   * @var string
   */
  private string $role;

  /**
   * The text.
   *
   * @var string
   */
  private string $text;

  /**
   * The files in an array.
   *
   * @var \Drupal\ai\OperationType\GenericType\FileBaseInterface[]
   */
  private array $files;

  /**
   * Remote file identifiers (e.g. IDs from a remote AI provider API).
   *
   * @var string[]
   */
  private array $remoteFiles = [];

  /**
   * The tools.
   *
   * @var \Drupal\ai\OperationType\Chat\Tools\ToolsFunctionOutputInterface[]|null
   */
  private ?array $tools = NULL;

  /**
   * The tool id if any.
   *
   * @var string|null
   */
  private ?string $toolId = NULL;

  /**
   * The message timestamp.
   *
   * @var int|null
   */
  private ?int $timestamp = NULL;

  /**
   * The constructor.
   *
   * @param string $role
   *   The role of the message.
   * @param string $text
   *   The text.
   * @param \Drupal\ai\OperationType\GenericType\FileBaseInterface[] $images
   *   The files.
   * @param int|null $timestamp
   *   The message timestamp.
   */
  public function __construct(string $role = "", string $text = "", array $images = [], ?int $timestamp = NULL) {
    $this->role = $role;
    $this->text = $text;
    $this->files = $images;
    $this->timestamp = $timestamp;
  }

  /**
   * Get the role of the text.
   *
   * @return string
   *   The role.
   */
  public function getRole(): string {
    return $this->role;
  }

  /**
   * Get the text.
   *
   * @return string
   *   The text.
   */
  public function getText(): string {
    return $this->text;
  }

  /**
   * Set the role of the message.
   *
   * @param string $role
   *   The role.
   */
  public function setRole(string $role): void {
    $this->role = $role;
  }

  /**
   * Set the text.
   *
   * @param string $text
   *   The text.
   */
  public function setText(string $text): void {
    $this->text = $text;
  }

  /**
   * Get the files.
   *
   * @return \Drupal\ai\OperationType\GenericType\FileBaseInterface[]
   *   The files.
   */
  public function getFiles(): array {
    return $this->files;
  }

  /**
   * Get the images.
   *
   * @return \Drupal\ai\OperationType\GenericType\ImageFile[]
   *   The images.
   */
  public function getImages(): array {
    // As part of the BC we return only images here.
    return array_filter($this->files, fn($file) => $file instanceof ImageFile);
  }

  /**
   * Get remote file identifiers.
   *
   * @return string[]
   *   The remote file IDs.
   */
  public function getRemoteFiles(): array {
    return $this->remoteFiles;
  }

  /**
   * Add a remote file identifier.
   *
   * @param string $remote_file_id
   *   The remote file ID.
   */
  public function addRemoteFile(string $remote_file_id): void {
    $this->remoteFiles[] = $remote_file_id;
  }

  /**
   * Remove a remote file identifier.
   *
   * @param string $remote_file_id
   *   The remote file ID to remove.
   */
  public function removeRemoteFile(string $remote_file_id): void {
    $this->remoteFiles = array_filter($this->remoteFiles, fn($id) => $id !== $remote_file_id);
  }

  /**
   * Set the file.
   *
   * @param \Drupal\ai\OperationType\GenericType\FileBaseInterface $file
   *   The file.
   */
  public function setFile(FileBaseInterface $file): void {
    $this->files[] = $file;
  }

  /**
   * Set the image.
   *
   * @param \Drupal\ai\OperationType\GenericType\ImageFile $image
   *   The image.
   */
  public function setImage(ImageFile $image): void {
    $this->files[] = $image;
  }

  /**
   * Get the tools.
   *
   * @return \Drupal\ai\OperationType\Chat\Tools\ToolsFunctionOutputInterface[]|null
   *   The tools.
   */
  public function getTools(): ?array {
    return $this->tools;
  }

  /**
   * Set the tools.
   *
   * @param \Drupal\ai\OperationType\Chat\Tools\ToolsFunctionOutputInterface[] $tools
   *   The tools.
   */
  public function setTools(array $tools): void {
    $this->tools = $tools;
  }

  /**
   * Get rendered tools output array.
   *
   * @return array
   *   The rendered array.
   */
  public function getRenderedTools(): array {
    $output = [];
    if ($this->tools) {
      foreach ($this->tools as $tool) {
        $output[] = $tool->getOutputRenderArray();
      }
    }
    return $output;
  }

  /**
   * Get the tool id.
   *
   * @return string|null
   *   The tool id.
   */
  public function getToolsId(): string|null {
    return $this->toolId;
  }

  /**
   * Set the tool id.
   *
   * @param string $tool_id
   *   The tool id.
   */
  public function setToolsId(string $tool_id): void {
    $this->toolId = $tool_id;
  }

  /**
   * Sets the image from a binary string.
   *
   * @param string $binary
   *   The binary string.
   * @param string $mime_type
   *   The mime type.
   */
  public function setImageFromBinary(string $binary, string $mime_type): void {
    $this->files[] = new ImageFile($binary, $mime_type);
  }

  /**
   * Sets the image from an url.
   *
   * @param string $url
   *   The url.
   */
  public function setImageFromUrl(string $url): void {
    // Get mime type from the uri.
    $mime_type = $this->getFileMimeTypeGuesser()->guessMimeType($url);
    $filename = basename($url);
    $this->files[] = new ImageFile(file_get_contents($url), $mime_type, $filename);
  }

  /**
   * Set the image from a Drupal uri.
   *
   * @param string $uri
   *   The uri.
   */
  public function setImageFromUri(string $uri): void {
    // Get mime type from the uri.
    $mime_type = $this->getFileMimeTypeGuesser()->guessMimeType($uri);
    $filename = basename($uri);
    $this->files[] = new ImageFile(file_get_contents($uri), $mime_type, $filename);
  }

  /**
   * Sets the image from a Drupal file.
   *
   * @param \Drupal\file\Entity\File $file
   *   The file.
   */
  public function setImageFromFile(File $file): void {
    $this->files[] = new ImageFile(file_get_contents($file->getFileUri()), $file->getMimeType(), $file->getFilename());
  }

  /**
   * Create an array of the message.
   *
   * @return array
   *   The array of the message.
   */
  public function toArray(): array {
    $images = [];
    foreach ($this->files as $image) {
      $images[] = $image->toArray();
    }
    return [
      'role' => $this->role,
      'text' => $this->text,
      // @todo find out if this can be changed to 'files'
      'images' => $images,
      'remote_files' => $this->remoteFiles,
      'tools' => $this->tools ? $this->getRenderedTools() : NULL,
      'tool_id' => $this->toolId,
      'timestamp' => $this->timestamp,
    ];
  }

  /**
   * Create an instance from an array.
   *
   * @param array $data
   *   The data to create the instance from.
   *
   * @return static
   *   The created instance.
   */
  public static function fromArray(array $data): static {
    $instance = new static($data['role'] ?? '', $data['text'] ?? '', []);
    if (isset($data['images'])) {
      foreach ($data['images'] as $imageData) {
        $instance->setImage(ImageFile::fromArray($imageData));
      }
    }
    if (isset($data['remote_files']) && is_array($data['remote_files'])) {
      foreach ($data['remote_files'] as $remote_file_id) {
        $instance->addRemoteFile($remote_file_id);
      }
    }
    if (isset($data['tools'])) {
      $tools = [];
      $function_call_manager = \Drupal::service('plugin.manager.ai.function_calls');
      foreach ($data['tools'] as $tool_data) {
        // Get the real actual plugin.
        $tool = $function_call_manager->getFunctionCallFromFunctionName($tool_data['function']['name']);
        // Set a new ToolsFunctionOutput.
        $input = $tool->normalize();
        $tools[] = new ToolsFunctionOutput($input, $tool_data['id'], Json::decode($tool_data['function']['arguments']));
      }
      // Now we set it all.
      $instance->setTools($tools);
    }
    if (isset($data['tool_id'])) {
      $instance->setToolsId($data['tool_id']);
    }
    if (isset($data['text'])) {
      $instance->setText($data['text']);
    }
    if (isset($data['role'])) {
      $instance->setRole($data['role']);
    }
    if (isset($data['timestamp'])) {
      $instance->setTimestamp($data['timestamp']);
    }
    // @todo Files.
    return $instance;
  }

  /**
   * Gets message timestamp.
   *
   * @return int|null
   *   The timestamp when message was created.
   */
  public function getTimestamp(): ?int {
    return $this->timestamp;
  }

  /**
   * Sets message timestamp.
   *
   * @param int|null $timestamp
   *   The timestamp when message was created.
   */
  public function setTimestamp(?int $timestamp): void {
    $this->timestamp = $timestamp;
  }

}
