db = $wpdb; add_action('rest_api_init', array($this, 'register_endpoint')); $this->background_color = $this->hex_to_rgb('#141617'); $this->text_color = $this->hex_to_rgb('#ececec'); $this->western_font = plugin_dir_path(__FILE__) . 'assets/co1251n.ttf'; $this->japanese_font = plugin_dir_path(__FILE__) . 'assets/BIZUDPMincho-Regular.ttf'; $this->blacklisted_artists = array('Drake', 'Mac Miller'); // placeholder values } public function register_endpoint() { register_rest_route('top-albums/v1', '/last-month', array( 'methods' => 'GET', 'callback' => array($this, 'get_or_generate_image'), 'permission_callback' => '__return_true', 'args' => array( 'month' => array( 'required' => false, 'validate_callback' => array($this, 'is_valid_month'), ), 'year' => array( 'required' => false, 'validate_callback' => array($this, 'is_valid_year'), ), 'overwrite' => array( 'required' => false, 'default' => false, 'validate_callback' => array($this, 'is_valid_overwrite'), ), ), )); } public function is_valid_month($param, $request, $key) { $month = intval($param); return $month >= 1 && $month <= 12; } public function is_valid_year($param, $request, $key) { $year = intval($param); $current_year = intval(date('Y')); return $year >= 2000 && $year <= $current_year; } public function is_valid_overwrite($param, $request, $key) { // Accept boolean or string representations if (is_bool($param)) { return true; } if (is_string($param)) { $param = strtolower($param); return in_array($param, array('1', 'true', 'yes'), true); } if (is_numeric($param)) { return in_array($param, array(1, 0), true); } return false; } private function interpret_boolean($value) { if (is_bool($value)) { return $value; } if (is_string($value)) { $value = strtolower($value); return in_array($value, array('1', 'true', 'yes'), true); } if (is_numeric($value)) { return (bool)$value; } return false; } private function hex_to_rgb($hex) { $hex = str_replace('#', '', $hex); if(strlen($hex) == 3) { $r = hexdec(substr($hex,0,1).substr($hex,0,1)); $g = hexdec(substr($hex,1,1).substr($hex,1,1)); $b = hexdec(substr($hex,2,1).substr($hex,2,1)); } else { $r = hexdec(substr($hex,0,2)); $g = hexdec(substr($hex,2,2)); $b = hexdec(substr($hex,4,2)); } return array($r, $g, $b); } private function get_last_month_data() { $last_month = date('Y-m', strtotime('last day of previous month')); return explode('-', $last_month); } private function get_top_albums($year, $month) { // Prepare placeholders for blacklisted artists $placeholders = array_fill(0, count($this->blacklisted_artists), '%s'); $placeholders_str = implode(', ', $placeholders); // Merge all parameters for $wpdb->prepare $query_params = array($year, $month); $query_params = array_merge($query_params, $this->blacklisted_artists); $query = $this->db->prepare( "WITH ranked_covers AS ( SELECT album_name, author, cover_url, ROW_NUMBER() OVER (PARTITION BY album_name, author ORDER BY time DESC) as rn FROM song_scrobbles WHERE cover_url != '' AND cover_url IS NOT NULL ), album_stats AS ( SELECT album_name, author, COUNT(*) as play_count, COUNT(DISTINCT song_name) as unique_song_count FROM song_scrobbles WHERE YEAR(time) = %d AND MONTH(time) = %d AND album_name != '' AND album_name IS NOT NULL -- Exclude blacklisted artists AND author NOT IN ($placeholders_str) -- Exclude albums with '¥' symbol in album_name or author AND album_name NOT LIKE '%%¥%%' AND author NOT LIKE '%%¥%%' -- Exclude albums with Cyrillic characters in album_name or author AND album_name NOT REGEXP '[А-яЁё]' AND author NOT REGEXP '[А-яЁё]' GROUP BY album_name, author HAVING LENGTH(CONCAT(album_name, author)) <= 60 AND unique_song_count >= 1 ) SELECT a.album_name, a.author, r.cover_url, a.play_count, a.unique_song_count, CASE WHEN a.unique_song_count = 1 THEN 1.0 WHEN a.unique_song_count = 5 THEN 1.25 WHEN a.unique_song_count = 12 THEN 1.6 ELSE 1.0 + (a.unique_song_count * 0.05) END as multiplier, (a.play_count * CASE WHEN a.unique_song_count = 1 THEN 1.0 WHEN a.unique_song_count = 5 THEN 1.25 WHEN a.unique_song_count = 12 THEN 1.6 ELSE 1.0 + (a.unique_song_count * 0.05) END ) as rank_score FROM album_stats a LEFT JOIN ranked_covers r ON a.album_name = r.album_name AND a.author = r.author AND r.rn = 1 ORDER BY rank_score DESC LIMIT 25", ...$query_params ); return $this->db->get_results($query); } public function get_or_generate_image($request) { $month = $request->get_param('month'); $year = $request->get_param('year'); $overwrite = $request->get_param('overwrite'); // Retrieve 'overwrite' parameter // Convert 'overwrite' to boolean $overwrite = $this->interpret_boolean($overwrite); if ($month && $year) { $month = intval($month); $year = intval($year); // Validate that the date is not in the future $provided_date = strtotime("{$year}-{$month}-01"); $now = time(); if ($provided_date > $now) { return new WP_Error('invalid_date', 'The provided month and year cannot be in the future.', array('status' => 400)); } // Ensure the date is valid if (!checkdate($month, 1, $year)) { return new WP_Error('invalid_date', 'The provided month and year are invalid.', array('status' => 400)); } } else { list($year, $month) = $this->get_last_month_data(); } $cache_key = "top_albums_{$year}_{$month}"; // Check if we have a cached image $cached_image_id = get_option($cache_key); if ($overwrite && $cached_image_id) { // If overwrite is true and cached image exists // Delete the old attachment $deleted = wp_delete_attachment($cached_image_id, true); if ($deleted) { // Remove the cached option delete_option($cache_key); $cached_image_id = false; // Reset to indicate no cached image } else { return new WP_Error('deletion_failed', 'Failed to delete the existing image.', array('status' => 500)); } } if ($cached_image_id) { $attachment_path = get_attached_file($cached_image_id); if ($attachment_path && file_exists($attachment_path)) { // Serve the cached image directly as a response $image_data = file_get_contents($attachment_path); $mime_type = mime_content_type($attachment_path); // Send appropriate headers header('Content-Type: ' . $mime_type); header('Content-Length: ' . strlen($image_data)); header('Content-Disposition: inline; filename="' . basename($attachment_path) . '"'); // Prevent WordPress from sending additional content remove_filter('rest_pre_serve_request', 'rest_send_cors_headers'); echo $image_data; exit; } } // If no cached image, generate a new one $image_data = $this->generate_image($year, $month); // Save the generated image as an attachment $upload_dir = wp_upload_dir(); $filename = "top_albums_{$year}_{$month}.jpg"; $file_path = $upload_dir['path'] . '/' . $filename; file_put_contents($file_path, $image_data); $attachment = array( 'post_mime_type' => 'image/jpeg', 'post_title' => "Top Albums for {$year}-{$month}", 'post_content' => '', 'post_status' => 'inherit' ); $attach_id = wp_insert_attachment($attachment, $file_path); require_once(ABSPATH . 'wp-admin/includes/image.php'); $attach_data = wp_generate_attachment_metadata($attach_id, $file_path); wp_update_attachment_metadata($attach_id, $attach_data); // Save the attachment ID for future use update_option($cache_key, $attach_id); // Serve the newly generated image directly as a response $mime_type = mime_content_type($file_path); // Send appropriate headers header('Content-Type: ' . $mime_type); header('Content-Length: ' . strlen($image_data)); header('Content-Disposition: inline; filename="' . basename($file_path) . '"'); // Prevent WordPress from sending additional content remove_filter('rest_pre_serve_request', 'rest_send_cors_headers'); echo $image_data; exit; } private function generate_image($year, $month) { $albums = $this->get_top_albums($year, $month); $image_width = 2600; $image_height = 1680; $album_size = 300; $gap = 8; $left_margin = 100; $top_margin = 100; $image = imagecreatetruecolor($image_width, $image_height); $background_color = imagecolorallocate($image, $this->background_color[0], $this->background_color[1], $this->background_color[2]); imagefill($image, 0, 0, $background_color); $text_color = imagecolorallocate($image, $this->text_color[0], $this->text_color[1], $this->text_color[2]); // Add album covers (left side) for ($i = 0; $i < min(count($albums), 25); $i++) { $row = floor($i / 5); $col = $i % 5; $x = $left_margin + $col * ($album_size + $gap); $y = $top_margin + $row * ($album_size + $gap); $album_cover = @imagecreatefromjpeg($albums[$i]->cover_url); if ($album_cover !== false) { $this->apply_retro_effects($album_cover); imagecopyresampled($image, $album_cover, $x, $y, 0, 0, $album_size, $album_size, imagesx($album_cover), imagesy($album_cover)); imagedestroy($album_cover); } else { $placeholder_color = imagecolorallocate($image, 100, 100, 100); imagefilledrectangle($image, $x, $y, $x + $album_size, $y + $album_size, $placeholder_color); } } // Add album titles (right side) $text_x = $left_margin + 5 * ($album_size + $gap) + 20; $text_y = $top_margin + 18; $line_height = 35; $font_size = 20; $group_spacing = 50; for ($i = 0; $i < min(count($albums), 25); $i++) { $album_text = $albums[$i]->author . ' - ' . $albums[$i]->album_name; $font = $this->is_japanese($album_text) ? $this->japanese_font : $this->western_font; $this->add_retro_text($image, $font_size, $text_x, $text_y, $text_color, $font, $album_text); if (($i + 1) % 5 == 0 && $i < 24) { $text_y += $line_height + $group_spacing; } else { $text_y += $line_height; } } $this->apply_overall_retro_effects($image); ob_start(); imagejpeg($image); $image_data = ob_get_clean(); imagedestroy($image); return $image_data; } private function apply_retro_effects(&$image) { // Add noise $this->add_noise($image, 20); // Reduce color depth $this->reduce_color_depth($image); // Add slight blur // imagefilter($image, IMG_FILTER_GAUSSIAN_BLUR); } private function add_noise(&$image, $amount) { $width = imagesx($image); $height = imagesy($image); for ($x = 0; $x < $width; $x++) { for ($y = 0; $y < $height; $y++) { if (rand(0, 100) < $amount) { $rgb = imagecolorat($image, $x, $y); $r = ($rgb >> 16) & 0xFF; $g = ($rgb >> 8) & 0xFF; $b = $rgb & 0xFF; $noise = rand(-20, 20); $r = max(0, min(255, $r + $noise)); $g = max(0, min(255, $g + $noise)); $b = max(0, min(255, $b + $noise)); $color = imagecolorallocate($image, $r, $g, $b); imagesetpixel($image, $x, $y, $color); } } } } private function reduce_color_depth(&$image) { imagetruecolortopalette($image, false, 64); } private function add_retro_text(&$image, $font_size, $x, $y, $color, $font, $text) { // Add a slight shadow $shadow_color = imagecolorallocatealpha($image, 0, 0, 0, 75); imagettftext($image, $font_size, 0, $x + 1, $y + 1, $shadow_color, $font, $text); // Add main text with a slight glow $glow_color = imagecolorallocatealpha($image, 255, 255, 255, 75); imagettftext($image, $font_size, 0, $x - 1, $y - 1, $glow_color, $font, $text); imagettftext($image, $font_size, 0, $x, $y, $color, $font, $text); } private function apply_overall_retro_effects(&$image) { // Add scanlines $this->add_scanlines($image); // Add vignette effect $this->add_vignette($image); // Add slight color aberration $this->add_color_aberration($image); // Add subtle grain $this->add_grain($image); } private function add_scanlines(&$image) { $height = imagesy($image); $scanline_color = imagecolorallocatealpha($image, 0, 0, 0, 70); for ($y = 0; $y < $height; $y += 2) { imageline($image, 0, $y, imagesx($image), $y, $scanline_color); } } private function add_vignette(&$image) { $width = imagesx($image); $height = imagesy($image); $center_x = $width / 2; $center_y = $height / 2; $max_distance = sqrt($center_x * $center_x + $center_y * $center_y); for ($x = 0; $x < $width; $x++) { for ($y = 0; $y < $height; $y++) { $distance = sqrt(pow($x - $center_x, 2) + pow($y - $center_y, 2)); $vignette = 1 - ($distance / $max_distance); $vignette = max(0, min(1, $vignette)); // Make vignette more subtle $vignette = 1 - (1 - $vignette) * 0.25; // Adjust 0.25 to control intensity $rgb = imagecolorat($image, $x, $y); $r = ($rgb >> 16) & 0xFF; $g = ($rgb >> 8) & 0xFF; $b = $rgb & 0xFF; $r = (int)($r * $vignette); $g = (int)($g * $vignette); $b = (int)($b * $vignette); $color = imagecolorallocate($image, $r, $g, $b); imagesetpixel($image, $x, $y, $color); } } } private function add_color_aberration(&$image) { $width = imagesx($image); $height = imagesy($image); $aberration = 1; // Reduced from 2 to 1 for subtler effect $new_image = imagecreatetruecolor($width, $height); imagesavealpha($new_image, true); imagealphablending($new_image, false); $transparent = imagecolorallocatealpha($new_image, 0, 0, 0, 127); imagefilledrectangle($new_image, 0, 0, $width, $height, $transparent); imagealphablending($new_image, true); for ($x = 0; $x < $width; $x++) { for ($y = 0; $y < $height; $y++) { $rgb = imagecolorat($image, $x, $y); $r = ($rgb >> 16) & 0xFF; $g = ($rgb >> 8) & 0xFF; $b = $rgb & 0xFF; // Shift red channel slightly to the left $red_x = max(0, $x - $aberration); $red_color = imagecolorallocate($new_image, $r, 0, 0); imagesetpixel($new_image, $red_x, $y, $red_color); // Keep green channel in place $green_color = imagecolorallocate($new_image, 0, $g, 0); imagesetpixel($new_image, $x, $y, $green_color); // Shift blue channel slightly to the right $blue_x = min($width - 1, $x + $aberration); $blue_color = imagecolorallocate($new_image, 0, 0, $b); imagesetpixel($new_image, $blue_x, $y, $blue_color); } } // Merge the aberrated image back onto the original with reduced opacity imagecopymerge($image, $new_image, 0, 0, 0, 0, $width, $height, 3); // Reduced from 50 to 30 imagedestroy($new_image); } private function add_grain(&$image) { $width = imagesx($image); $height = imagesy($image); $grain_amount = 8; // Adjust this value to control grain intensity (1-10 recommended) for ($x = 0; $x < $width; $x++) { for ($y = 0; $y < $height; $y++) { if (rand(0, 100) < $grain_amount) { $rgb = imagecolorat($image, $x, $y); $r = ($rgb >> 16) & 0xFF; $g = ($rgb >> 8) & 0xFF; $b = $rgb & 0xFF; $noise = rand(-10, 10); $r = max(0, min(255, $r + $noise)); $g = max(0, min(255, $g + $noise)); $b = max(0, min(255, $b + $noise)); $color = imagecolorallocate($image, $r, $g, $b); imagesetpixel($image, $x, $y, $color); } } } } private function is_japanese($text) { return preg_match('/[\x{4E00}-\x{9FBF}\x{3040}-\x{309F}\x{30A0}-\x{30FF}]/u', $text); } } new AlbumCollageGenerator();