<?php
/**
 * Class Felix_Arntz\AI_Services\Chatbot\Chatbot
 *
 * @since 0.1.0
 * @package ai-services
 */

namespace Felix_Arntz\AI_Services\Chatbot;

use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\Dependencies\Script_Registry;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\Dependencies\Style_Registry;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\General\Contracts\With_Hooks;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\General\Current_User;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\General\Network_Env;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\General\Plugin_Env;
use Felix_Arntz\AI_Services_Dependencies\Felix_Arntz\WP_OOP_Plugin_Lib\General\Site_Env;

/**
 * Class controlling the AI-powered chatbot.
 *
 * @since 0.1.0
 */
class Chatbot implements With_Hooks {

	/**
	 * The plugin environment.
	 *
	 * @since 0.1.0
	 * @var Plugin_Env
	 */
	private $plugin_env;

	/**
	 * WordPress script registry instance.
	 *
	 * @since 0.1.0
	 * @var Script_Registry
	 */
	private $script_registry;

	/**
	 * WordPress style registry instance.
	 *
	 * @since 0.1.0
	 * @var Style_Registry
	 */
	private $style_registry;

	/**
	 * The AI instance.
	 *
	 * @since 0.1.0
	 * @var Chatbot_AI
	 */
	private $ai;

	/**
	 * Constructor.
	 *
	 * @since 0.1.0
	 *
	 * @param Plugin_Env      $plugin_env      The plugin environment.
	 * @param Site_Env        $site_env        The site environment.
	 * @param Network_Env     $network_env     The network environment.
	 * @param Current_User    $current_user    The current user instance.
	 * @param Script_Registry $script_registry The WordPress script registry instance.
	 * @param Style_Registry  $style_registry  The WordPress style registry instance.
	 */
	public function __construct(
		Plugin_Env $plugin_env,
		Site_Env $site_env,
		Network_Env $network_env,
		Current_User $current_user,
		Script_Registry $script_registry,
		Style_Registry $style_registry
	) {
		$this->plugin_env      = $plugin_env;
		$this->script_registry = $script_registry;
		$this->style_registry  = $style_registry;
		$this->ai              = new Chatbot_AI( $site_env, $network_env, $current_user );
	}

	/**
	 * Adds relevant WordPress hooks.
	 *
	 * @since 0.1.0
	 */
	public function add_hooks(): void {
		if ( doing_action( 'init' ) ) {
			$this->register_assets();
		} else {
			add_action( 'init', array( $this, 'register_assets' ) );
		}

		if ( is_admin() ) {
			add_action( 'admin_enqueue_scripts', array( $this, 'enqueue_assets' ) );
			add_action( 'admin_footer', array( $this, 'render_app_root' ) );
		} else {
			add_action( 'wp_enqueue_scripts', array( $this, 'enqueue_assets' ) );
			add_action( 'wp_footer', array( $this, 'render_app_root' ) );
		}

		add_filter(
			'ai_services_model_params',
			array( $this, 'filter_model_params' )
		);
	}

	/**
	 * Gets the model parameters to use for the chatbot.
	 *
	 * @since 0.1.0
	 *
	 * @return array<string, mixed> The model parameters, containing 'system_instruction'.
	 */
	public function get_model_params(): array {
		return array(
			'systemInstruction' => $this->ai->get_system_instruction(),
		);
	}

	/**
	 * Registers the assets needed for the chatbot.
	 *
	 * @since 0.1.0
	 */
	public function register_assets(): void {
		$this->script_registry->register(
			'ais_chatbot',
			array(
				'src'      => $this->plugin_env->url( 'build/chatbot/index.js' ),
				'manifest' => $this->plugin_env->path( 'build/chatbot/index.asset.php' ),
				'strategy' => 'defer',
			)
		);
		$this->style_registry->register(
			'ais_chatbot',
			array(
				'src'          => $this->plugin_env->url( 'build/chatbot/style-index.css' ),
				'path'         => $this->plugin_env->path( 'build/chatbot/style-index.css' ),
				'manifest'     => $this->plugin_env->path( 'build/chatbot/index.asset.php' ),
				// Don't use 'wp-components' everywhere because it is too heavy to load for just a button.
				'dependencies' => array(),
			)
		);
	}

	/**
	 * Enqueues the assets needed for the chatbot.
	 *
	 * @since 0.1.0
	 */
	public function enqueue_assets(): void {
		$this->script_registry->enqueue( 'ais_chatbot' );
		$this->style_registry->enqueue( 'ais_chatbot' );
	}

	/**
	 * Renders the chatbot app root.
	 *
	 * @since 0.1.0
	 */
	public function render_app_root(): void {
		?>
		<div id="ai-services-chatbot-root" class="chatbot-root"></div>
		<?php
	}

	/**
	 * Filters the model parameters for the REST API.
	 *
	 * @since 0.1.0
	 *
	 * @param array<string, mixed> $model_params The model parameters. Commonly supports at least the parameters
	 *                                           'feature', 'capabilities', 'generationConfig' and 'systemInstruction'.
	 * @return array<string, mixed> The filtered model parameters.
	 */
	public function filter_model_params( array $model_params ): array {
		if ( isset( $model_params['feature'] ) && 'ai-services-chatbot' === $model_params['feature'] ) {
			$model_params = array_merge( $model_params, $this->get_model_params() );
		}
		return $model_params;
	}
}