<?php

/**
 * WpCafe Tools Hooks
 *
 * Registers WpCafe AI tools via the aisentic_admin_assistant_tools filter
 * when the WpCafe integration is active and connected.
 *
 * @package Aisentic\Integrations\WpCafe
 */

declare(strict_types=1);

namespace Aisentic\Integrations\WpCafe;

use Aisentic\Api\Services\Registration_Service;
use Aisentic\Integrations\Integration_Controller;
use Aisentic\Integrations\Integration_Status;
use Aisentic\Mentions\Resolvers\Wpcafe_Reservation_Resolver;
use Aisentic\Mentions\Resolvers\Wpcafe_Location_Resolver;
use Aisentic\Mentions\Resolvers\Wpcafe_Product_Resolver;
use Aisentic\Providers\Base_Service_Provider;

defined( 'ABSPATH' ) || exit;

/**
 * Registers WpCafe-specific AI tools via WordPress filter hooks.
 */
class Wpcafe_Tools_Hooks extends Base_Service_Provider {

	/**
	 * Register hooks
	 *
	 * @return void
	 */
	public function register(): void {
		parent::register();
		add_filter( 'aisentic_admin_assistant_tool_pool', [ $this, 'add_tools' ], 10, 3 );
		add_filter( 'aisentic_mention_resolvers', [ $this, 'add_mention_resolvers' ], 10, 1 );
		add_action( 'wpcafe/aisentic/register_site', [ $this, 'handle_wpcafe_register' ], 10, 3 );
	}

	/**
	 * Handle the WP Cafe registration handshake.
	 *
	 * WP Cafe fires `wpcafe/aisentic/register_site` only after the user opts in
	 * (setup wizard checkbox or dashboard banner), passing the restaurant
	 * identity it showed them. We register the site, which stores the api key
	 * and fires `aisentic/site_registered`, then mark the WpCafe integration
	 * connected so the tools and mention resolvers below start loading.
	 *
	 * WP Cafe activates this plugin in the same request, and Aisentic boots its
	 * providers on include, so this listener is attached before the action runs.
	 *
	 * @param string $name     Restaurant/account name.
	 * @param string $email    Restaurant/account email.
	 * @param string $site_url Site URL to register.
	 * @return void
	 */
	public function handle_wpcafe_register( $name, $email, $site_url ): void {
		// Skip the provider when a key already exists, so re-activating does not
		// re-register. api_key presence is the "registered" marker.
		$existing_key = aisentic_get_option( 'llmProviders.aisentic.apiKey', '' );

		if ( empty( $existing_key ) ) {
			$result = Registration_Service::register( $name, $email, $site_url );

			if ( is_wp_error( $result ) ) {
				return; // Provider error. Stay unconnected, never fatal to WP Cafe.
			}
		}

		/*
		 * Registration_Service fires `aisentic/site_registered`, not the
		 * underscore variant Integration_Controller auto-connects on, so the
		 * integration has to be marked here. Same as the Eventin handshake.
		 */
		$saved                     = get_option( Integration_Controller::OPTION_KEY, [] );
		$saved['wpcafe']           = isset( $saved['wpcafe'] ) ? $saved['wpcafe'] : [];
		$saved['wpcafe']['status'] = Integration_Status::CONNECTED;
		update_option( Integration_Controller::OPTION_KEY, $saved );
	}

	/**
	 * Get services
	 *
	 * @return array
	 */
	public function get_services(): array {
		return [];
	}

	/**
	 * Add WpCafe mention resolvers to the registry.
	 *
	 * @param array $resolvers Current list of resolvers.
	 * @return array
	 */
	public function add_mention_resolvers( array $resolvers ): array {
		if ( ! $this->is_wpcafe_connected() ) {
			return $resolvers;
		}

		$resolvers[] = new Wpcafe_Reservation_Resolver();
		$resolvers[] = new Wpcafe_Location_Resolver();
		$resolvers[] = new Wpcafe_Product_Resolver();

		return $resolvers;
	}

	/**
	 * Add WpCafe tools to the admin assistant tool pool.
	 *
	 * All WpCafe tools are added regardless of the current page so the agent
	 * can discover and use them from any admin screen via tool_search.
	 *
	 * @param array  $tools          Current list of tools.
	 * @param object $settings       Settings instance.
	 * @param string $plugin_context Active plugin context (unused; pool is page-agnostic).
	 * @return array
	 */
	public function add_tools( array $tools, $settings, string $plugin_context = '' ): array {
		if ( ! $this->is_wpcafe_connected() ) {
			return $tools;
		}

		if ( wpc_is_module_enable( 'reservation' ) ) {
			$tools[] = new Wpcafe_List_Reservations_Tool();
			$tools[] = new Wpcafe_Get_Reservation_Tool();
			$tools[] = new Wpcafe_Get_Reservation_Availability_Tool();
			$tools[] = new Wpcafe_Create_Reservation_Tool();
			$tools[] = new Wpcafe_Update_Reservation_Tool();
		}

		if ( wpc_is_module_enable( 'food_ordering' ) ) {
			$tools[] = new Wpcafe_List_Food_Orders_Tool();
			$tools[] = new Wpcafe_Get_Food_Order_Tool();
			$tools[] = new Wpcafe_Update_Order_Status_Tool();
		}

		if ( wpc_is_module_enable( 'location' ) ) {
			$tools[] = new Wpcafe_List_Locations_Tool();
			$tools[] = new Wpcafe_Get_Location_Tool();
			$tools[] = new Wpcafe_Create_Location_Tool();
			$tools[] = new Wpcafe_Update_Location_Tool();
		}

		$tools[] = new Wpcafe_List_Products_Tool();
		$tools[] = new Wpcafe_Get_Product_Tool();
		$tools[] = new Wpcafe_List_Product_Categories_Tool();
		$tools[] = new Wpcafe_Get_Dashboard_Overview_Tool();
		$tools[] = new Wpcafe_Get_Top_Sales_Tool();
		$tools[] = new Wpcafe_List_Modules_Tool();
		$tools[] = new Wpcafe_Toggle_Module_Tool();
		$tools[] = new Wpcafe_Get_Revenue_By_Period_Tool();
		$tools[] = new Wpcafe_List_Tables_Tool();
		$tools[] = new Wpcafe_Get_Tipping_Stats_Tool();
		$tools[] = new Wpcafe_List_Discounts_Tool();
		$tools[] = new Wpcafe_Get_Discount_Tool();
		$tools[] = new Wpcafe_Create_Discount_Tool();
		$tools[] = new Wpcafe_Update_Discount_Tool();

		return $tools;
	}

	/**
	 * Check whether WpCafe is active and connected via Aisentic settings.
	 *
	 * @return bool
	 */
	private function is_wpcafe_connected(): bool {
		if ( ! function_exists( 'wpcafe' ) ) {
			return false;
		}
		$saved  = get_option( Integration_Controller::OPTION_KEY, [] );
		$status = $saved['wpcafe']['status'] ?? '';
		return 'connected' === $status;
	}
}