<?php

namespace Drupal\canvas_ai\Plugin\AiFunctionCall;

use Drupal\ai\Attribute\FunctionCall;
use Drupal\ai\Base\FunctionCallBase;
use Drupal\ai\Service\FunctionCalling\ExecutableFunctionCallInterface;
use Drupal\ai_agents\PluginInterfaces\AiAgentContextInterface;
use Drupal\Core\Plugin\Context\ContextDefinition;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Symfony\Component\Yaml\Yaml;

/**
 * Plugin implementation to get page data.
 */
#[FunctionCall(
  id: 'ai_agent:get_page_data',
  function_name: 'ai_agent_get_page_data',
  name: 'Get page data',
  description: 'This method allows to get page data.',
  group: 'information_tools',
  module_dependencies: ['canvas'],
  context_definitions: [
    'page_title' => new ContextDefinition(
      data_type: 'string',
      label: new TranslatableMarkup("Page Title"),
      description: new TranslatableMarkup("The title of the page."),
      required: FALSE
    ),
    'page_description' => new ContextDefinition(
      data_type: 'string',
      label: new TranslatableMarkup("Page Description"),
      description: new TranslatableMarkup("The description of the page."),
      required: FALSE,
    ),
  ],
)]
class GetPageData extends FunctionCallBase implements ExecutableFunctionCallInterface, AiAgentContextInterface {

  /**
   * The page title.
   *
   * @var string|null
   */
  protected string|null $pageTitle;

  /**
   * The page description.
   *
   * @var string|null
   */
  protected string|null $pageDescription;

  /**
   * {@inheritdoc}
   */
  public function execute(): void {
    $this->pageTitle = $this->getContextValue('page_title');
    $this->pageDescription = $this->getContextValue('page_description');
  }

  /**
   * {@inheritdoc}
   */
  public function getReadableOutput(): string {
    return Yaml::dump([
      'page_title' => $this->pageTitle,
      'page_description' => $this->pageDescription,
    ], 10, 2);
  }

}
