From bd10ef8ec944d342ec729078903fcdb70cdc5adf Mon Sep 17 00:00:00 2001
From: Tunglies <77394545+Tunglies@users.noreply.github.com>
Date: Sat, 16 May 2026 01:08:21 +0800
Subject: [PATCH] fix: improve youtube premium unlock check

---
 .../src/cmd/media_unlock_checker/youtube.rs   | 89 ++++++++++---------
 1 file changed, 48 insertions(+), 41 deletions(-)

diff --git a/src-tauri/src/cmd/media_unlock_checker/youtube.rs b/src-tauri/src/cmd/media_unlock_checker/youtube.rs
index 01d8dfd24..0277509d4 100644
--- a/src-tauri/src/cmd/media_unlock_checker/youtube.rs
+++ b/src-tauri/src/cmd/media_unlock_checker/youtube.rs
@@ -7,55 +7,62 @@ use super::UnlockItem;
 use super::utils::{country_code_to_emoji, get_local_date_string};
 
 pub(super) async fn check_youtube_premium(client: &Client) -> UnlockItem {
-    let url = "https://www.youtube.com/premium";
+    let url = "https://www.youtube.com/premium?hl=en";
+    let item = |status: &str, region: Option<String>| UnlockItem {
+        name: "YouTube Premium".to_string(),
+        status: status.to_string(),
+        region,
+        check_time: Some(get_local_date_string()),
+    };
 
     match client.get(url).send().await {
         Ok(response) => {
+            let status_code = response.status().as_u16();
+
             if let Ok(body) = response.text().await {
-                let body_lower = body.to_lowercase();
-                let mut status = "Failed";
-                let mut region = None;
-
-                if body_lower.contains("youtube premium is not available in your country") {
-                    status = "No";
-                } else if body_lower.contains("ad-free") {
-                    match Regex::new(r#"id="country-code"[^>]*>([^<]+)<"#) {
-                        Ok(re) => {
-                            if let Some(caps) = re.captures(&body)
-                                && let Some(m) = caps.get(1)
-                            {
-                                let country_code = m.as_str().trim();
-                                let emoji = country_code_to_emoji(country_code);
-                                region = Some(format!("{emoji}{country_code}"));
-                                status = "Yes";
-                            }
-                        }
-                        Err(e) => {
-                            logging!(error, Type::Network, "Failed to compile YouTube Premium regex: {}", e);
-                        }
+                let body_lower = body.to_ascii_lowercase();
+                let region = [
+                    r#"id=["']country-code["'][^>]*>\s*([A-Za-z]{2,3})\s*<"#,
+                    r#""GL"\s*:\s*"([A-Za-z]{2})""#,
+                    r#""countryCode"\s*:\s*"([A-Za-z]{2})""#,
+                    r#""country_code"\s*:\s*"([A-Za-z]{2})""#,
+                ]
+                .iter()
+                .find_map(|pattern| match Regex::new(pattern) {
+                    Ok(re) => re
+                        .captures(&body)
+                        .and_then(|caps| caps.get(1))
+                        .map(|m| m.as_str().trim().to_ascii_uppercase()),
+                    Err(e) => {
+                        logging!(error, Type::Network, "Failed to compile YouTube Premium regex: {}", e);
+                        None
                     }
-                }
+                })
+                .map(|country_code| {
+                    let emoji = country_code_to_emoji(&country_code);
+                    format!("{emoji}{country_code}")
+                });
 
-                UnlockItem {
-                    name: "YouTube Premium".to_string(),
-                    status: status.to_string(),
-                    region,
-                    check_time: Some(get_local_date_string()),
-                }
+                let status = if body_lower.contains("youtube premium is not available in your country")
+                    || body_lower.contains("premium is not available in your country")
+                    || body_lower.contains("premium is not available in your region")
+                {
+                    "No"
+                } else if (200..300).contains(&status_code)
+                    && (body_lower.contains("youtube premium")
+                        || body_lower.contains("ad-free")
+                        || body_lower.contains(r#""browseid":"spunlimited""#))
+                {
+                    "Yes"
+                } else {
+                    "Failed"
+                };
+
+                item(status, region)
             } else {
-                UnlockItem {
-                    name: "YouTube Premium".to_string(),
-                    status: "Failed".to_string(),
-                    region: None,
-                    check_time: Some(get_local_date_string()),
-                }
+                item("Failed", None)
             }
         }
-        Err(_) => UnlockItem {
-            name: "YouTube Premium".to_string(),
-            status: "Failed".to_string(),
-            region: None,
-            check_time: Some(get_local_date_string()),
-        },
+        Err(_) => item("Failed", None),
     }
 }