<?php

namespace AIContentWriter;

defined( 'ABSPATH' ) || exit; // Exit if accessed directly.

/**
 * Class Cron
 *
 * @since 1.4.0
 * @package AIContentWriter
 */
class Cron {

	/**
	 * Cron constructor.
	 */
	public function __construct() {
		add_action( 'init', array( $this, 'schedule_cron' ) );

		// Execute cron job to generate titles, content & thumbnails.
		add_action( 'aicw_generate_title', array( $this, 'generate_titles' ) );
		add_action( 'aicw_generate_content', array( $this, 'generate_content' ) );
		add_action( 'aicw_generate_thumbnail', array( $this, 'generate_thumbnails' ) );
		add_action( 'aicw_publish_posts', array( $this, 'publish_posts' ) );
	}

	/**
	 * Schedule cron.
	 *
	 * @since 1.4.0
	 * @return void
	 */
	public function schedule_cron() {
		// Schedule the cron job for generating titles.
		if ( ! wp_next_scheduled( 'aicw_generate_title' ) ) {
			wp_schedule_event( time(), 'hourly', 'aicw_generate_title' );
		}

		// Schedule the cron job for generating content.
		if ( ! wp_next_scheduled( 'aicw_generate_content' ) ) {
			wp_schedule_event( time() + 300, 'hourly', 'aicw_generate_content' );
		}

		// Schedule the cron job for generating thumbnails.
		if ( ! wp_next_scheduled( 'aicw_generate_thumbnail' ) ) {
			wp_schedule_event( time() + 600, 'hourly', 'aicw_generate_thumbnail' );
		}

		// Schedule the cron job for publishing the generated posts.
		if ( ! wp_next_scheduled( 'aicw_publish_posts' ) ) {
			wp_schedule_event( time() + 900, 'hourly', 'aicw_publish_posts' );
		}
	}

	/**
	 * Generate titles.
	 *
	 * @since 1.4.0
	 * @return void
	 */
	public function generate_titles() {
		// Get all posts.
		$campaigns = aicw_get_campaigns(
			array(
				'posts_per_page' => -1,
				'orderby'        => 'modified',
				'order'          => 'ASC',
				'post_status'    => 'publish',
			)
		);

		// Loop through each campaign.
		foreach ( $campaigns as $campaign ) {
			// Get the content.
			$campaign_id = $campaign->ID;

			// Get the content.
			$keywords = $campaign->post_content;

			// Check if the campaign needs generating titles.
			$needs_update = aicw_campaign_needs_update( $campaign_id );

			if ( array_key_exists( 'needs_update', $needs_update ) && ! $needs_update['needs_update'] ) {
				aicw_update_last_run_time( $campaign_id );
				continue;
			}

			// Now request the OpenAI to generate the titles depending on the keywords.
			$titles = aicw_generate_titles( $keywords );

			$counter = 0;
			// Create a new post for each title.
			foreach ( $titles as $title ) {
				if ( empty( $title ) ) {
					continue;
				}

				// Check if the line starts with a number followed by a dot and space (e.g., "10. "). If not then skip the line.
				if ( ! preg_match( '/^\d+\.\s/', $title ) ) {
					continue;
				}

				// Remove leading number, dot, and space (e.g., "10. ").
				$title = preg_replace( '/^\d+\.\s*/', '', $title );

				// Remove surrounding quotes if present.
				$title = trim( $title, '"' );

				// Check if the title already exists.
				$post_exists = aicw_get_post_by_title( $title );

				if ( $post_exists ) {
					continue;
				}

				// Create a new post.
				$post_id = wp_insert_post(
					array(
						'post_type'    => 'aicw_post',
						'post_title'   => $title,
						'post_content' => '',
						'post_status'  => 'pending',
					)
				);

				if ( is_wp_error( $post_id ) ) {
					continue;
				}

				// Add the post meta.
				update_post_meta( $post_id, '_aicw_campaign_id', $campaign_id );

				// Update the counter.
				++$counter;

				// If the counter is greater than or equal to the target then break the loop.
				if ( array_key_exists( 'posts_needed', $needs_update ) && $counter >= $needs_update['posts_needed'] ) {
					break;
				}
			}

			// Update the last run time.
			aicw_update_last_run_time( $campaign_id );

			// If this line executed then break the foreach loop. This will ensure that only one campaign is processed at a time.
			break;
		}
	}

	/**
	 * Generate content.
	 *
	 * @since 1.4.0
	 * @return void
	 */
	public function generate_content() {
		// Get all posts.
		$posts = get_posts(
			array(
				'post_type'      => 'aicw_post',
				'posts_per_page' => -1,
				'orderby'        => 'modified',
				'order'          => 'ASC',
				'post_status'    => 'pending',
			)
		);

		// Loop through each post.
		foreach ( $posts as $post ) {
			// Get the content.
			$post_id = $post->ID;

			// Get the associated campaign ID.
			$campaign_id = get_post_meta( $post_id, '_aicw_campaign_id', true );

			// Check if the post needs generating content.
			$needs_update = aicw_temp_post_needs_update( $post, $campaign_id );

			if ( array_key_exists( 'content', $needs_update ) && ! $needs_update['content'] ) {
				aicw_update_last_run_time( $post_id );
				continue;
			}

			// Get the keywords.
			$keywords = get_post_field( 'post_content', $campaign_id );

			// Get the title.
			$title = get_the_title( $post_id );

			if ( empty( $title ) ) {
				continue;
			}

			// Now request the OpenAI to generate the content depending on the keywords and title.
			$content = aicw_generate_content( $title, $keywords );

			// Update the post content.
			wp_update_post(
				array(
					'ID'           => $post_id,
					'post_content' => wp_kses_post( $content ),
				)
			);

			// Update the last run time.
			aicw_update_last_run_time( $post_id );

			// If this line executed then break the foreach loop. This will ensure that only one post is processed at a time.
			break;
		}
	}

	/**
	 * Generate thumbnails.
	 *
	 * @since 1.4.0
	 * @return void
	 */
	public function generate_thumbnails() {
		// Get all posts.
		$posts = get_posts(
			array(
				'post_type'      => 'aicw_post',
				'posts_per_page' => -1,
				'orderby'        => 'modified',
				'order'          => 'ASC',
				'post_status'    => 'pending',
			)
		);

		// Loop through each post.
		foreach ( $posts as $post ) {
			// Get the content.
			$post_id = $post->ID;

			// Get the associated campaign ID.
			$campaign_id = get_post_meta( $post_id, '_aicw_campaign_id', true );

			// Check if the post needs generating thumbnail.
			$needs_update = aicw_temp_post_needs_update( $post, $campaign_id );

			if ( array_key_exists( 'thumbnail', $needs_update ) && ! $needs_update['thumbnail'] ) {
				aicw_update_last_run_time( $post_id );
				continue;
			}

			// Get the title.
			$title = get_the_title( $post_id );

			if ( empty( $title ) ) {
				continue;
			}

			// Now request the OpenAI to generate the thumbnail depending on the title and content.
			$images = ai_content_writer()->generate_images( $title );

			if ( is_wp_error( $images ) ) {
				continue;
			}

			if ( empty( $images ) ) {
				continue;
			}

			// Get a random image.
			$thumbnail = $images[ array_rand( $images ) ];

			if ( ! isset( $thumbnail['src']['original'] ) ) {
				continue;
			}

			// Get the media URL.
			$media_url = $thumbnail['src']['original'];

			// Check if the media side load function exists. If not then include the required files.
			if ( ! function_exists( 'media_sideload_image' ) ) {
				require_once ABSPATH . 'wp-admin/includes/media.php';
				require_once ABSPATH . 'wp-admin/includes/file.php';
				require_once ABSPATH . 'wp-admin/includes/image.php';
			}

			// Download the image.
			$media_id = media_sideload_image( $media_url, $post_id, $title, 'id' );

			// Set the post thumbnail.
			set_post_thumbnail( $post_id, $media_id );

			// Update the last run time.
			aicw_update_last_run_time( $post_id );

			// If this line executed then break the foreach loop. This will ensure that only one post is processed at a time.
			break;
		}
	}

	/**
	 * Publish posts.
	 * This will publish the generated posts that are in pending status and completed all the steps.
	 *
	 * @since 1.4.0
	 * @return void
	 */
	public function publish_posts() {
		// Get all posts.
		$posts = get_posts(
			array(
				'post_type'      => 'aicw_post',
				'posts_per_page' => -1,
				'orderby'        => 'modified',
				'order'          => 'ASC',
				'post_status'    => 'pending',
			)
		);

		// Loop through each post.
		foreach ( $posts as $post ) {
			if ( ! $post && ! $post instanceof \WP_Post ) {
				continue;
			}

			// Get the content.
			$post_id = $post->ID;

			if ( empty( $post_id ) ) {
				continue;
			}

			// Get the associated campaign ID.
			$campaign_id = get_post_meta( $post_id, '_aicw_campaign_id', true );

			if ( empty( $campaign_id ) ) {
				continue;
			}

			// Check if the post needs generating thumbnail.
			$needs_update = aicw_temp_post_needs_update( $post, $campaign_id );

			if ( ! empty( $needs_update ) ) {
				continue;
			}

			// Get the post type and status user selected in the campaign.
			$post_type   = get_post_meta( $campaign_id, '_aicw_post_type', true ) ?? 'post';
			$post_status = get_post_meta( $campaign_id, '_aicw_completed_post_status', true ) ?? 'publish';

			// Update the post type and status.
			wp_update_post(
				array(
					'ID'          => $post_id,
					'post_type'   => $post_type,
					'post_status' => $post_status,
				)
			);

			// If this line executed then break the foreach loop. This will ensure that only one post is processed at a time.
			break;
		}
	}
}