import re
import os
from urllib.parse import urlparse, urlunparse

input_file = "linkss.txt"
output_file = "linkss.txt"

if not os.path.exists(input_file):
    print(f"❌ فایل {input_file} پیدا نشد!")
    exit()

with open(input_file, "r", encoding="utf-8") as f:
    lines = f.readlines()

cleaned_lines = []
for line in lines:
    parts = line.strip().split(";")
    if not parts:
        continue

    # پاک‌سازی و جایگزینی همه‌ی لینک‌هایی که در خط وجود دارند
    new_parts = []
    for part in parts:
        part = part.strip()  # تمیز کردن اضافی
        if part.startswith("http"):
            # حذف query string و fragment
            parsed = urlparse(part)
            path = parsed.path  # path کامل

            # تشخیص و جایگزینی base URL بر اساس وجود /stream_movies/
            if '/stream_movies/' in path:
                # جایگزینی با dl02
                new_base = "https://dl02.hostikadeh.ir/zr"
            else:
                # جایگزینی با dl01
                new_base = "https://dl01.hostikadeh.ir"

            # اگر path با .mp4 تمام می‌شود، پاک‌سازی zarfilm از انتهای filename
            if path.lower().endswith('.mp4'):
                cleaned_path = re.sub(r'[-._]zarfilm\.mp4$', '.mp4', path, flags=re.IGNORECASE)
                new_url = new_base + cleaned_path
            else:
                new_url = new_base + path

            new_parts.append(new_url)
        else:
            # اگر part یک فایل .mp4 است، پاک‌سازی zarfilm از انتها
            if part.lower().endswith('.mp4'):
                cleaned_filename = re.sub(r'[-._]zarfilm\.mp4$', '.mp4', part, flags=re.IGNORECASE)
                new_parts.append(cleaned_filename)
            else:
                new_parts.append(part)

    cleaned_lines.append(";".join(new_parts))

with open(output_file, "w", encoding="utf-8") as f:
    f.write("\n".join(cleaned_lines) + "\n")  # اضافه کردن newline نهایی برای تمیزی

print(f"✅✅✅")