Create Dockerfile

app.py

@@ -43,7 +43,7 @@ async def save_cookies(request: Request):
 
         # Validate structure
         if "data" in data and "raw_cookies_string" in data and "source_url" in data:
-            save_data["main_data"] = {
+            save_data["youtube"] = {
                 "session_data": data["data"],
                 "raw_cookies": data["raw_cookies_string"],
                 "source_url": data["source_url"]

convert_cookie_for_youtube.py

@@ -0,0 +1,103 @@
+from urllib.parse import urlparse
+import time
+
+from test import get_cookies
+
+
+def infer_base_domain(original_source_url):
+    """
+    Infers a correct base domain for YouTube/Google cookies from a
+    potentially problematic original_source_url.
+    Prioritizes common YouTube/Google domains.
+    """
+    if "youtube.com" in original_source_url:
+        return "youtube.com"
+    elif "google.com" in original_source_url:
+        return "google.com"
+    elif "youtube.com" in original_source_url:  # Catches other YouTube domains if any
+        return "youtube.com"
+    elif "googleusercontent.com" in original_source_url:  # Generic googleusercontent fallback
+        return "googleusercontent.com"
+
+    # Fallback if no specific match, but this might still give bad results
+    # if the original_source_url itself is completely unrecognizable.
+    parsed_url = urlparse(original_source_url)
+    if parsed_url.hostname:
+        return parsed_url.hostname
+    return "unknown.com"  # Should ideally not be hit for YouTube/Google cookies
+
+
+def convert_raw_cookie_string_to_netscape_fixed_domain(raw_cookie_str, original_source_url):
+    """
+    Converts a raw cookie string to Netscape format, inferring a correct
+    base domain for YouTube/Google cookies.
+    """
+    base_domain = infer_base_domain(original_source_url)
+
+    lines = [
+        "# Netscape HTTP Cookie File",
+        "# http://curl.haxx.se/rfc/cookie_spec.html",
+        "# This file was generated by script - Domain inferred for YouTube/Google compatibility."
+    ]
+
+    for raw_cookie in raw_cookie_str.split(';'):
+        cookie = raw_cookie.strip().split('=', 1)
+        if len(cookie) != 2:
+            continue
+
+        name, value = cookie[0], cookie[1]
+
+        # Determine domain field. Use a leading dot for general cookies.
+        # This handles the . prefix for subdomains correctly.
+        domain_field = f".{base_domain}"
+
+        # Heuristics for HttpOnly based on common cookie prefixes
+        # Note: True HttpOnly status isn't available from raw string, this is a guess.
+        is_httponly = name.startswith("__Secure-") or name.startswith("__Host-") or name.startswith(
+            "SID") or name.startswith("SSID") or name.startswith("LSID")
+
+        if is_httponly:
+            # Netscape format for HttpOnly is #HttpOnly_.domain
+            # Ensure we don't duplicate the leading dot if it's already there from base_domain
+            if domain_field.startswith("."):
+                domain_field = f"#HttpOnly_{domain_field}"
+            else:
+                domain_field = f"#HttpOnly_.{domain_field}"
+
+        # The 'flag' column (Include Subdomains) is generally TRUE for Netscape format.
+        include_subdomains = "TRUE"
+
+        # The 'path' column is usually '/' for general site cookies.
+        path = "/"
+
+        # The 'secure' column. Guess based on __Secure- prefix, but better if known from source.
+        secure = "TRUE" if name.startswith("__Secure-") else "FALSE"
+
+        # Expiration: A fixed, far-future date is often sufficient for yt-dlp.
+        expires = str(int(time.time()) + 180 * 24 * 3600)  # 180 days from now
+
+        # Construct the line
+        line = f"{domain_field}\t{include_subdomains}\t{path}\t{secure}\t{expires}\t{name}\t{value}"
+        lines.append(line)
+
+    return lines
+
+
+# --- Main Execution Block (adapt to your API structure) ---
+# Assuming 'cookie_entry' is received at your API endpoint
+# For testing purposes, using your provided cookie_entry_data:
+# from test import get_cookies # Uncomment if get_cookies is defined elsewhere
+
+cookie_entry_data = get_cookies()
+
+raw_str = cookie_entry_data['youtube']['raw_cookies']
+source = cookie_entry_data['youtube']['source_url']
+
+# Convert to Netscape format
+output_lines = convert_raw_cookie_string_to_netscape_fixed_domain(raw_str, source)
+
+# Save to file (this would happen on your API server)
+with open("cookies.txt", "w", encoding="utf-8") as f:
+    f.write("\n".join(output_lines))
+
+print("✅ Saved as cookies.txt in Netscape format with inferred domains.")

convert_raw_cookie_string_to_netscape.py

@@ -1,39 +0,0 @@
-from urllib.parse import urlparse
-import time
-
-import requests
-
-from test import API_ENDPOINT_GET, get_cookies
-
-
-def convert_raw_cookie_string_to_netscape(raw_cookie_str, source_url):
-    domain = urlparse(source_url).hostname
-    lines = []
-    for cookie in raw_cookie_str.split(';'):
-        name_value = cookie.strip().split('=', 1)
-        if len(name_value) != 2:
-            continue
-        name, value = name_value
-        # Netscape format: domain \t include_subdomains \t path \t secure \t expires \t name \t value
-        include_subdomains = "TRUE"
-        path = "/"
-        secure = "FALSE"
-        expires = str(int(time.time()) + 3600 * 24 * 180)  # 180 days from now
-        line = f"{domain}\t{include_subdomains}\t{path}\t{secure}\t{expires}\t{name}\t{value}"
-        lines.append(line)
-    return lines
-
-cookie_data_list = get_cookies()
-
-# Build full Netscape-format text
-output_lines = ["# Netscape HTTP Cookie File", "# Generated by script"]
-for cookie_entry in cookie_data_list:
-    raw_str = cookie_entry['raw_cookies_string']
-    source = cookie_entry['source_url']
-    output_lines.extend(convert_raw_cookie_string_to_netscape(raw_str, source))
-
-# Save to file
-with open("cookies.txt", "w") as f:
-    f.write("\n".join(output_lines))
-
-print("✅ Saved as cookies.txt in Netscape format.")

test.py

@@ -1,13 +1,7 @@
 import requests
-import json
 
 BASE_URL = "https://amit0987-cookies-recever-api.hf.space"
-API_ENDPOINT_POST = f"{BASE_URL}/save-cookies"
-API_ENDPOINT_GET = f"{BASE_URL}/cookies"
 
-HEADERS = {"Content-Type": "application/json"}
-
-# Sample test payload
 sample_data = {
     "data": {
         "session": "test_session_123_from_script",
@@ -18,43 +12,23 @@ sample_data = {
     "source_url": "https://script-test.com"
 }
 
-def post_cookies(data):
-    print(f"\n--- Sending POST to {API_ENDPOINT_POST} ---")
+def post_cookies():
     try:
-        response = requests.post(API_ENDPOINT_POST, json=data, headers=HEADERS, timeout=10)
+        response = requests.post(f"{BASE_URL}/save-cookies", json=sample_data, timeout=10)
         response.raise_for_status()
-        print("✅ POST Success:", response.status_code)
-        return response
-    except requests.exceptions.HTTPError as e:
-        print(f"❌ HTTP Error: {e.response.status_code} - {e.response.text}")
-    except requests.exceptions.RequestException as e:
-        print(f"❌ Request Failed: {e}")
-    return None
+        print("[POST] ✅", response.json())
+    except Exception as e:
+        print("[POST] ❌", e)
 
 def get_cookies():
-    print(f"\n--- Fetching with GET from {API_ENDPOINT_GET} ---")
     try:
-        response = requests.get(API_ENDPOINT_GET, timeout=10)
+        response = requests.get(f"{BASE_URL}/cookies", timeout=10)
         response.raise_for_status()
-        print("✅ GET Success:", response.status_code)
-        try:
-            return response.json()
-        except json.JSONDecodeError:
-            print("⚠️ Failed to decode JSON from GET response.")
-    except requests.exceptions.RequestException as e:
-        print(f"❌ Request Failed: {e}")
-    return None
-
-def main():
-    post_resp = post_cookies(sample_data)
-    if post_resp:
-        print("🔍 POST Response:", post_resp.text)
-
-    stored_data = get_cookies()
-    if stored_data:
-        print("📦 Stored Cookies:", json.dumps(stored_data, indent=2))
-    else:
-        print("⚠️ No cookies retrieved or data might be empty.")
+        print("[GET] ✅", response.json())
+        return response.json()
+    except Exception as e:
+        print("[GET] ❌", e)
 
 if __name__ == "__main__":
-    main()
+    # post_cookies()
+    get_cookies()