From 6427906d155a5d411edf130d4408840a9612064a Mon Sep 17 00:00:00 2001
From: Anthony Correa <a@correa.co>
Date: Thu, 16 May 2024 14:25:42 -0500
Subject: [PATCH] change to dynamic image generation

---
 includes/featured-image-generator.php | 108 +++++++++++++++++++++++---
 1 file changed, 98 insertions(+), 10 deletions(-)

diff --git a/includes/featured-image-generator.php b/includes/featured-image-generator.php
index bec3263..8cf033b 100644
--- a/includes/featured-image-generator.php
+++ b/includes/featured-image-generator.php
@@ -104,8 +104,8 @@ function generate_bisected_image($color1, $color2, $logo1_path, $logo2_path) {
         }
 
         // Center logo 1
-        $logo1_x = ($width / 4) - ($new_logo1_width / 2);
-        $logo1_y = ($height / 2) - ($new_logo1_height / 2);
+        $logo1_x = (int) ($width / 4) - ($new_logo1_width / 2);
+        $logo1_y = (int) ($height / 2) - ($new_logo1_height / 2);
         imagecopyresampled($image, $logo1, $logo1_x, $logo1_y, 0, 0, $new_logo1_width, $new_logo1_height, $logo1_width, $logo1_height);
         imagedestroy($logo1);
     }
@@ -134,21 +134,22 @@ function generate_bisected_image($color1, $color2, $logo1_path, $logo2_path) {
         }
 
         // Center logo 2
-        $logo2_x = (3 * $width / 4) - ($new_logo2_width / 2);
-        $logo2_y = ($height / 2) - ($new_logo2_height / 2);
+        $logo2_x = (int) (3 * $width / 4) - ($new_logo2_width / 2);
+        $logo2_y = (int) ($height / 2) - ($new_logo2_height / 2);
         imagecopyresampled($image, $logo2, $logo2_x, $logo2_y, 0, 0, $new_logo2_width, $new_logo2_height, $logo2_width, $logo2_height);
         imagedestroy($logo2);
     }
 
+    // Start output buffering to capture the image data
+    ob_start();
+    imagepng($image); // Output the image as PNG
+    $image_data = ob_get_clean(); // Get the image data from the buffer
 
-    // Save to temp location
-    $temp_file = tempnam(sys_get_temp_dir(), 'event_image');
-    rename($temp_file, $temp_file .= '.png');
-    $temp_file .= '.png';
-    imagepng($image, $temp_file);
+    // Clean up memory
     imagedestroy($image);
 
-    return $temp_file;
+    return $image_data;
+
 }
 
 function debug_to_console($data) {
@@ -280,3 +281,90 @@ function generate_images_for_existing_events($season = '') {
         wp_reset_postdata();
     }
 }
+
+function add_image_generator_endpoint() {
+    add_rewrite_endpoint('head-to-head', EP_ROOT, true);
+}
+add_action('init', 'add_image_generator_endpoint');
+
+function handle_image_request() {
+    if (!isset($_GET['post'])) return;
+    
+    $post_id = $_GET['post'];
+    $post = get_post($post_id);
+
+    // Verify post type
+    if (!$post && $post->post_type !== 'sp_event') return;
+
+    // Get associated teams from post meta
+    $team_ids = get_post_meta($post_id, 'sp_team', false); // false to get an array of values
+
+    // Ensure we have exactly two teams
+    if (count($team_ids) < 2) return;
+
+    $team1_id = $team_ids[0];
+    $team2_id = $team_ids[1];
+
+    $team1 = get_post($team1_id);
+    $team2 = get_post($team2_id);
+    $team1_postmodified = strtotime($team1->post_modified);
+    $team2_postmodified = strtotime($team2->post_modified);
+
+    $cache_key = "team_image_{$team1_id}_{$team1_postmodified}-{$team2_id}_{$team2_postmodified}";
+    $cached_image_path = get_transient($cache_key);
+
+    if ($cached_image_path && file_exists($cached_image_path)) {
+        serve_image($cached_image_path);
+        exit;
+    }
+
+    // Get team colors and logos
+    $team1_colors = get_post_meta($team1_id, 'sp_colors', true);
+    $team2_colors = get_post_meta($team2_id, 'sp_colors', true);
+
+    $default_color = '#FFFFFF'; // Default color (black)
+    $team1_color = !empty($team1_colors['primary']) ? $team1_colors['primary'] : $default_color;
+    $team2_color = !empty($team2_colors['primary']) ? $team2_colors['primary'] : $default_color;
+
+    // Security check for hex color
+    $team1_color = preg_match('/^#[a-fA-F0-9]{6}$/', $team1_color) ? $team1_color : '#FFFFFF';
+    $team2_color = preg_match('/^#[a-fA-F0-9]{6}$/', $team2_color) ? $team2_color : '#FFFFFF';
+
+    $team1_logo_url = get_the_post_thumbnail_url($team1_id, 'full');
+    $team2_logo_url = get_the_post_thumbnail_url($team2_id, 'full');
+
+    // Check if both team colors are default and both logos are empty
+    if (($team1_color === $default_color && empty($team1_logo_url)) && ($team2_color === $default_color && empty($team2_logo_url))) {
+        return; // Do nothing if both teams have no valid color or logo
+    }
+
+    $team1_logo_thumbnail_id = get_post_thumbnail_id($team1_id, 'full');
+    $team2_logo_thumbnail_id = get_post_thumbnail_id($team2_id, 'full');
+    $team1_logo = get_attached_file($team1_logo_thumbnail_id);
+    $team2_logo = get_attached_file($team2_logo_thumbnail_id);
+
+    // Generate the image if no valid cache exists
+    $image_data = generate_bisected_image($team1_color, $team2_color, $team1_logo, $team2_logo);
+    $image_path = save_image_to_cache($image_data, $cache_key);
+    set_transient($cache_key, $image_path, DAY_IN_SECONDS * 30); // Cache for 30 days
+
+    serve_image($image_path);
+
+    exit;
+}
+add_action('template_redirect', 'handle_image_request');
+
+function serve_image($image_path) {
+    header('Content-Type: image/png');
+    readfile($image_path);
+}
+
+function save_image_to_cache($image_data, $cache_key) {
+    $upload_dir = wp_get_upload_dir();
+    $file_path = $upload_dir['path'] . '/' . $cache_key . '.png';
+
+    // Assuming $image_data is raw image data
+    file_put_contents($file_path, $image_data);
+
+    return $file_path;
+}
\ No newline at end of file