<?php
/**
 * Performance tools for existing media.
 *
 * @package zeitfresser
 */

if ( ! defined( 'ABSPATH' ) ) {
    exit;
}

/**
 * Register admin page
 */
function zeitfresser_register_performance_tools_page() {
    add_theme_page(
        'Performance Tools',
        'Performance Tools',
        'manage_options',
        'zeitfresser-performance-tools',
        'zeitfresser_render_performance_tools_page'
    );
}
add_action( 'admin_menu', 'zeitfresser_register_performance_tools_page' );

/**
 * Count pending images
 */
function zeitfresser_get_pending_legacy_images_count() {
    $query = new WP_Query([
        'post_type'      => 'attachment',
        'post_status'    => 'inherit',
        'post_mime_type' => 'image',
        'fields'         => 'ids',
        'posts_per_page' => 1,
        'meta_query'     => [
            'relation' => 'OR',
            [
                'key'     => '_zeitfresser_media_optimized_version',
                'compare' => 'NOT EXISTS',
            ],
            [
                'key'     => '_zeitfresser_media_optimized_version',
                'value'   => ZEITFRESSER_IMAGE_OPTIMIZATION_VERSION,
                'compare' => '!=',
            ],
        ],
        'no_found_rows' => false,
    ]);

    return (int) $query->found_posts;
}

/**
 * Count total images
 */
function zeitfresser_get_total_images_count() {
    $query = new WP_Query([
        'post_type'      => 'attachment',
        'post_status'    => 'inherit',
        'post_mime_type' => 'image',
        'fields'         => 'ids',
        'posts_per_page' => 1,
        'no_found_rows'  => false,
    ]);

    return (int) $query->found_posts;
}

/**
 * NEW: Cleanup counters (ONLY ADDITIVE)
 */
function zeitfresser_get_total_originals_count() {
    $query = new WP_Query([
        'post_type'=>'attachment',
        'post_status'=>'inherit',
        'post_mime_type'=>'image',
        'posts_per_page'=>1,
        'fields'=>'ids',
        'meta_query'=>[
            ['key'=>'_zeitfresser_original_file','compare'=>'EXISTS']
        ],
        'no_found_rows'=>false
    ]);
    return (int) $query->found_posts;
}

function zeitfresser_get_remaining_originals_count() {
    $query = new WP_Query([
        'post_type'=>'attachment',
        'post_status'=>'inherit',
        'post_mime_type'=>'image',
        'posts_per_page'=>1,
        'fields'=>'ids',
        'meta_query'=>[
            'relation'=>'AND',
            ['key'=>'_zeitfresser_original_file','compare'=>'EXISTS'],
            ['key'=>'_zeitfresser_original_deleted','compare'=>'NOT EXISTS']
        ],
        'no_found_rows'=>false
    ]);
    return (int) $query->found_posts;
}

/**
 * DELETE ORIGINALS (UNCHANGED)
 */
function zeitfresser_delete_originals_batch( $batch_size = 10 ) {

    $deleted = 0;

    $query = new WP_Query([
        'post_type'      => 'attachment',
        'post_status'    => 'inherit',
        'post_mime_type' => 'image',
        'fields'         => 'ids',
        'posts_per_page' => $batch_size,
        'meta_query' => [
            'relation' => 'AND',
            [
                'key'     => '_zeitfresser_original_file',
                'compare' => 'EXISTS',
            ],
            [
                'key'     => '_zeitfresser_original_deleted',
                'compare' => 'NOT EXISTS',
            ],
        ],
    ]);

    foreach ( $query->posts as $attachment_id ) {

        $original = get_post_meta( $attachment_id, '_zeitfresser_original_file', true );

        if ( ! $original ) {
            continue;
        }

        $optimized_version = get_post_meta(
            $attachment_id,
            '_zeitfresser_media_optimized_version',
            true
        );

        if ( ZEITFRESSER_IMAGE_OPTIMIZATION_VERSION !== $optimized_version ) {
            continue;
        }

        $ext = strtolower( pathinfo( $original, PATHINFO_EXTENSION ) );

        if ( in_array( $ext, [ 'webp', 'avif' ], true ) ) {
            update_post_meta( $attachment_id, '_zeitfresser_original_deleted', 1 );
            continue;
        }

        if ( ! file_exists( $original ) ) {
            update_post_meta( $attachment_id, '_zeitfresser_original_deleted', 1 );
            continue;
        }

        if ( ! is_writable( $original ) ) {
            continue;
        }

        if ( unlink( $original ) ) {
            $deleted++;
            update_post_meta( $attachment_id, '_zeitfresser_original_deleted', 1 );
        }
    }

    return $deleted;
}

/**
 * Optimizer batch for manual processing.
 *
 * Manual optimization must work independently of the auto-optimize upload toggle.
 *
 * @param int $batch_size Number of images per batch.
 * @return array
 */
function zeitfresser_process_legacy_images_batch( $batch_size = 25 ) {

    $results = [
        'processed' => 0,
        'updated'   => 0,
        'skipped'   => 0,
        'errors'    => 0,
    ];

    $query = new WP_Query([
        'post_type'      => 'attachment',
        'post_status'    => 'inherit',
        'post_mime_type' => 'image',
        'fields'         => 'ids',
        'posts_per_page' => $batch_size,
        'meta_query'     => [
            'relation' => 'OR',
            [
                'key'     => '_zeitfresser_media_optimized_version',
                'compare' => 'NOT EXISTS',
            ],
            [
                'key'     => '_zeitfresser_media_optimized_version',
                'value'   => ZEITFRESSER_IMAGE_OPTIMIZATION_VERSION,
                'compare' => '!=',
            ],
        ],
    ]);

    // Force optimization for manual tool runs, regardless of upload automation setting.
    $GLOBALS['zeitfresser_force_image_optimization'] = true;

    foreach ( $query->posts as $attachment_id ) {

        $results['processed']++;

        $file = get_attached_file( $attachment_id );

        if ( empty( $file ) || ! file_exists( $file ) ) {
            update_post_meta( $attachment_id, '_zeitfresser_media_optimized_version', 'missing' );
            $results['skipped']++;
            continue;
        }

        if ( ! get_post_meta( $attachment_id, '_zeitfresser_original_file', true ) ) {
            update_post_meta( $attachment_id, '_zeitfresser_original_file', $file );
        }

        $metadata = wp_generate_attachment_metadata( $attachment_id, $file );

        if ( is_wp_error( $metadata ) || empty( $metadata ) ) {
            $results['errors']++;
            continue;
        }

        wp_update_attachment_metadata( $attachment_id, $metadata );

        update_post_meta(
            $attachment_id,
            '_zeitfresser_media_optimized_version',
            ZEITFRESSER_IMAGE_OPTIMIZATION_VERSION
        );

        $results['updated']++;
    }

    unset( $GLOBALS['zeitfresser_force_image_optimization'] );

    return $results;
}

/**
 * AJAX: Optimizer (extended output only)
 */
function zeitfresser_ajax_optimize_images() {

    if ( ! current_user_can( 'manage_options' ) ) {
        wp_send_json_error();
    }

    check_ajax_referer( 'zeitfresser_performance_tools', 'nonce' );

    $results = zeitfresser_process_legacy_images_batch( 25 );

    wp_send_json_success([
        'processed' => $results['processed'],
        'updated'   => $results['updated'],
        'pending'   => zeitfresser_get_pending_legacy_images_count(),
        'total'     => zeitfresser_get_total_images_count(),
    ]);
}
add_action( 'wp_ajax_zeitfresser_optimize_images', 'zeitfresser_ajax_optimize_images' );

/**
 * AJAX: Delete (extended ONLY)
 */
function zeitfresser_ajax_delete_originals() {

    if ( ! current_user_can( 'manage_options' ) ) {
        wp_send_json_error();
    }

    check_ajax_referer( 'zeitfresser_performance_tools', 'nonce' );

    $deleted = zeitfresser_delete_originals_batch( 10 );

    $total     = zeitfresser_get_total_originals_count();
    $remaining = zeitfresser_get_remaining_originals_count();

    wp_send_json_success([
        'deleted'       => $deleted,
        'total'         => $total,
        'remaining'     => $remaining,
        'deleted_total' => $total - $remaining,
    ]);
}
add_action( 'wp_ajax_zeitfresser_delete_originals', 'zeitfresser_ajax_delete_originals' );

/**
 * Render UI
 */
function zeitfresser_render_performance_tools_page() {

    if ( ! current_user_can( 'manage_options' ) ) {
        return;
    }

    $pending   = zeitfresser_get_pending_legacy_images_count();
    $total     = zeitfresser_get_total_images_count();
    $optimized = $total - $pending;
    $progress  = $total > 0 ? round(($optimized / $total) * 100) : 0;

    // 🔥 NEW: Cleanup counters
    $cleanup_total     = zeitfresser_get_total_originals_count();
    $cleanup_remaining = zeitfresser_get_remaining_originals_count();
    $cleanup_deleted   = $cleanup_total - $cleanup_remaining;
    $cleanup_progress  = $cleanup_total > 0 ? round(($cleanup_deleted / $cleanup_total) * 100) : 0;
?>

<div class="wrap">
    <h1>Zeitfresser Performance Tools</h1>
    
    <div class="notice notice-info" style="max-width:800px;margin-top:20px;">
        <p>
            <strong>How this tool works</strong><br><br>

            This tool helps you optimize your existing media library for better performance.<br><br>

            • Images are converted to modern formats (AVIF/WebP) for smaller file sizes.<br>
            • The original file path is safely stored before optimization.<br>
            • Once optimized, original images can be deleted to save disk space.<br><br>

            <strong>Automation:</strong><br>
            • You can enable automatic optimization on upload in the Customizer under <em>Performance Tools Settings</em>.<br>
            • Optionally, original images can also be deleted automatically after successful optimization.<br><br>

            <strong>Safety:</strong><br>
            • Images are only processed once per version.<br>
            • Original files are only deleted when safe.<br>
            • The tool can be run multiple times without side effects.<br><br>

            <em><strong>Tip:</strong> You can either automate the process via the Customizer or use this tool manually for full control.</em>
        </p>
    </div>

    <!-- OPTIMIZATION -->
    <div class="card" style="max-width:800px;padding:24px;margin-bottom:20px;">

        <h2 style="margin-top:0;">🚀 Image Optimization</h2>

        <div style="margin-bottom:20px;">
            <p><strong>Total Images:</strong> <span id="total"><?php echo $total; ?></span></p>
            <p><strong>Optimized:</strong> <span id="optimized"><?php echo $optimized; ?></span></p>
            <p><strong>Pending:</strong> <span id="remaining"><?php echo $pending; ?></span></p>
        </div>

        <div style="margin-bottom:20px;">
            <div style="background:#e0e0e0;border-radius:6px;height:12px;">
                <div id="progress-bar" style="width:<?php echo $progress; ?>%;background:#4caf50;height:100%;"></div>
            </div>
            <p><strong>Progress:</strong> <span id="progress"><?php echo $progress; ?></span>%</p>
        </div>

        <button id="start-btn" class="button button-primary">🚀 Optimize Images</button>
    </div>

    <!-- CLEANUP -->
    <div class="card" style="max-width:800px;padding:24px;">

        <h2>🧹 Original Cleanup</h2>

        <div style="margin-bottom:20px;">
            <p><strong>Total Originals:</strong> <span id="cleanup-total"><?php echo $cleanup_total; ?></span></p>
            <p><strong>Deleted:</strong> <span id="cleanup-deleted"><?php echo $cleanup_deleted; ?></span></p>
            <p><strong>Remaining:</strong> <span id="cleanup-remaining"><?php echo $cleanup_remaining; ?></span></p>
        </div>

        <div style="margin-bottom:20px;">
            <div style="background:#e0e0e0;border-radius:6px;height:12px;">
                <div id="cleanup-bar" style="width:<?php echo $cleanup_progress; ?>%;background:<?php echo $cleanup_progress === 100 ? '#4caf50' : '#ff9800'; ?>;height:100%;"></div>
            </div>
            <p><strong>Cleanup Progress:</strong> <span id="cleanup-progress"><?php echo $cleanup_progress; ?></span>%</p>
        </div>

        <button id="delete-btn" class="button">🧹 Delete Originals</button>
    </div>

    <!-- STATUS -->
    <div style="padding:12px;background:#f6f7f7;border-radius:6px;">
        <p id="status-opt">🚀 Optimizer: Idle</p>
        <p id="status-clean">🧹 Cleanup: Idle</p>
    </div>

</div>

<style>
#progress-bar,
#cleanup-bar {
    transition: width 0.35s ease, background 0.3s ease;
}
</style>

<script>
let running = false;
let deleting = false;

(function initCleanupBar() {
    const bar = document.getElementById('cleanup-bar');
    const progress = parseInt(document.getElementById('cleanup-progress').innerText, 10);

    if (!bar || isNaN(progress)) return;

    if (progress >= 100) {
        bar.style.background = '#4caf50';
    } else if (progress > 0) {
        bar.style.background = '#ff9800';
    }
})();

document.getElementById('start-btn').onclick = () => {
    running = true;

    document.getElementById('start-btn').disabled = true;
    document.getElementById('delete-btn').disabled = true;

    document.getElementById('start-btn').innerText = '⏳ Running...';
    document.getElementById('status-opt').innerText = '🚀 Optimizing images...';

    processBatch();
};

document.getElementById('delete-btn').onclick = () => {
    deleting = true;

    document.getElementById('start-btn').disabled = true;
    document.getElementById('delete-btn').disabled = true;

    document.getElementById('delete-btn').innerText = '⏳ Running...';
    document.getElementById('status-clean').innerText = '🧹 Cleaning originals...';

    deleteBatch();
};

function deleteBatch() {
    if (!deleting) return;

    fetch(ajaxurl, {
        method: 'POST',
        headers: {'Content-Type': 'application/x-www-form-urlencoded'},
        body: new URLSearchParams({
            action: 'zeitfresser_delete_originals',
            nonce: '<?php echo wp_create_nonce('zeitfresser_performance_tools'); ?>'
        })
    })
    .then(res => res.json())
    .then(data => {

        let total     = data.data.total;
        let remaining = data.data.remaining;
        let deleted   = data.data.deleted_total;
        let batch     = data.data.deleted;

        let progress = total > 0 ? Math.round((deleted / total) * 100) : 0;

        // Update Cleanup UI
        document.getElementById('cleanup-total').innerText = total;
        document.getElementById('cleanup-deleted').innerText = deleted;
        document.getElementById('cleanup-remaining').innerText = remaining;
        document.getElementById('cleanup-progress').innerText = progress;
        
        let bar = document.getElementById('cleanup-bar');

        bar.style.width = progress + '%';

        if (progress >= 100) {
            bar.style.background = '#4caf50';
        } else if (progress > 0) {
            bar.style.background = '#ff9800';
        }

        document.getElementById('status-clean').innerText = '🧹 Deleted: ' + batch;

        if (remaining > 0) {
            setTimeout(deleteBatch, 400);
        } else {
            deleting = false;

            document.getElementById('start-btn').disabled = false;
            document.getElementById('delete-btn').disabled = false;

            document.getElementById('delete-btn').innerText = '🧹 Delete Originals';

            document.getElementById('status-clean').innerText = '✔ Cleanup complete';
        }
    });
}

function processBatch() {
    if (!running) return;

    fetch(ajaxurl, {
        method: 'POST',
        headers: {'Content-Type': 'application/x-www-form-urlencoded'},
        body: new URLSearchParams({
            action: 'zeitfresser_optimize_images',
            nonce: '<?php echo wp_create_nonce('zeitfresser_performance_tools'); ?>'
        })
    })
    .then(res => res.json())
    .then(data => {

        let pending   = data.data.pending;
        let total     = data.data.total;
        let optimized = total - pending;
        let progress  = total > 0 ? Math.round((optimized / total) * 100) : 0;

        // Live update optimization counters
        document.getElementById('total').innerText = total;
        document.getElementById('optimized').innerText = optimized;
        document.getElementById('remaining').innerText = pending;

        // Live update progress UI
        document.getElementById('progress').innerText = progress;
        document.getElementById('progress-bar').style.width = progress + '%';

        document.getElementById('status-opt').innerText =
            '🚀 Processed: ' + data.data.processed +
            ' | Updated: ' + data.data.updated;

        if (pending > 0) {
            setTimeout(processBatch, 400);
        } else {
            running = false;

            document.getElementById('start-btn').disabled = false;
            document.getElementById('delete-btn').disabled = false;

            document.getElementById('start-btn').innerText = '🚀 Optimize Images';

            document.getElementById('status-opt').innerText = '✔ Optimization complete';
        }
    });
}
</script>

<?php
}