<?php
namespace axilweb\aiimagealttextgenerator;

// If accessed directly, exit
if (!defined('ABSPATH')) exit;

/**
 * Class to track and manage alt text generation counts
 * 
 * @package Ai Image Alt Text Generator
 * @since 1.1.0
 */
class Aiimatg_Generation_Tracker {
    
    /**
     * Constructor
     * return void
     */
    public function __construct() {
        // Hook into the alt text generation process
        add_action('aiimatg_after_alt_text_generation', array($this, 'track_alt_text_generation'), 10, 2);
        
        // Add generation count to REST API responses
        add_filter('rest_prepare_attachment', array($this, 'add_generation_count_to_rest'), 10, 3);
        
        // Add generation count to AJAX responses
        add_filter('aiimatg_image_data', array($this, 'add_generation_count_to_image_data'));
    }
    
    /**
     * Track when alt text is generated for an image
     * 
     * @param int $attachment_id The attachment ID
     * @param string $alt_text The generated alt text
     * @return void
     */
    public function track_alt_text_generation($attachment_id, $alt_text) {
        if (!empty($alt_text)) {
            $this->increment_generation_count($attachment_id);
        }
    }
    
    /**
     * Increment the generation count for an image
     * 
     * @param int $attachment_id The attachment ID
     * @return void
     */
    public function increment_generation_count($attachment_id) {
        $current_count = (int) get_post_meta($attachment_id, 'aiimatg_generated_count', true);
        update_post_meta($attachment_id, 'aiimatg_generated_count', $current_count + 1);
    }
    
    /**
     * Add generation count to REST API responses
     * 
     * @param WP_REST_Response $response The response object
     * @param WP_Post $post The post object
     * @param WP_REST_Request $request The request object
     * @return WP_REST_Response Modified response
     */
    public function add_generation_count_to_rest($response, $post, $request) {
        if ($post->post_type === 'attachment' && !is_wp_error($response)) {
            $response->data['aiimatg_generated_count'] = (int) get_post_meta($post->ID, 'aiimatg_generated_count', true);
        }
        return $response;
    }
    
    /**
     * Add generation count to image data
     * 
     * @param array $image_data The image data
     * @return array Modified image data
     */
    public function add_generation_count_to_image_data($image_data) {
        if (isset($image_data['id'])) {
            $image_data['generated_count'] = (int) get_post_meta($image_data['id'], 'aiimatg_generated_count', true);
        }
        return $image_data;
    }
    
    /**
     * Get generation count for an image
     * 
     * @param int $attachment_id The attachment ID
     * @return int The generation count
     */
    public function get_generation_count($attachment_id) {
        return (int) get_post_meta($attachment_id, 'aiimatg_generated_count', true);
    }
}