# coding=utf-8
# !/usr/bin/python

"""
作者丢喵推荐 内容均从互联网收集而来 仅供交流学习使用 版权归原创者所有 如侵犯了你的权益 请通知作者 将及时删除侵权内容
                    ====================Diudiumiao====================

qmdj.py
version: 2026-08-20.home-v4-rename-sort

modify: 全量标签伪装美化(两字优先) + 字数越多越靠后排序
"""

from Crypto.Util.Padding import unpad
from Crypto.Util.Padding import pad
from urllib.parse import unquote
from Crypto.Cipher import ARC4
from urllib.parse import quote
from base.spider import Spider
from Crypto.Cipher import AES
from datetime import datetime
from bs4 import BeautifulSoup
from base64 import b64decode
import urllib.request
import urllib.parse
import datetime
import binascii
import requests
import hashlib
import base64
import json
import time
import sys
import re
import os

sys.path.append('..')

xurl = "https://api-store.qmplaylet.com"

xurl1 = "https://api-read.qmplaylet.com"

keys = "d3dGiJc651gSQ8w1"

data = {
        "static_score": "0.8",
        "uuid": "00000000-7fc7-08dc-0000-000000000000",
        "device-id": "20250220125449b9b8cac84c2dd3d035c9052a2572f7dd0122edde3cc42a70",
        "mac": "",
        "sourceuid": "aa7de295aad621a6",
        "refresh-type": "0",
        "model": "22021211RC",
        "wlb-imei": "",
        "client-id": "aa7de295aad621a6",
        "brand": "Redmi",
        "oaid": "",
        "oaid-no-cache": "",
        "sys-ver": "12",
        "trusted-id": "",
        "phone-level": "H",
        "imei": "",
        "wlb-uid": "aa7de295aad621a6",
        "session-id": str(int(time.time() * 1000)),
        }

json_str = json.dumps(data, separators=(',', ':'))
encoded = base64.b64encode(json_str.encode()).decode()

char_map = {
        '+': 'P', '/': 'X', '0': 'M', '1': 'U', '2': 'l', '3': 'E', '4': 'r',
        '5': 'Y', '6': 'W', '7': 'b', '8': 'd', '9': 'J', 'A': '9', 'B': 's',
        'C': 'a', 'D': 'I', 'E': '0', 'F': 'o', 'G': 'y', 'H': '_', 'I': 'H',
        'J': 'G', 'K': 'i', 'L': 't', 'M': 'g', 'N': 'A', 'O': 'A', 'P': '8',
        'Q': 'F', 'R': 'k', 'S': '3', 'T': 'h', 'U': 'f', 'V': 'R', 'W': 'q',
        'X': 'C', 'Y': '4', 'Z': 'p', 'a': 'm', 'b': 'B', 'c': 'O', 'd': 'u',
        'e': 'c', 'f': '6', 'g': 'K', 'h': 'x', 'i': '5', 'j': 'T', 'k': '-',
        'l': '2', 'm': 'z', 'n': 'S', 'o': 'Z', 'p': '1', 'q': 'V', 'r': 'v',
        's': 'j', 't': 'Q', 'u': '7', 'v': 'D', 'w': 'w', 'x': 'n', 'y': 'L',
        'z': 'e'
           }

qm_params = ''
for c in encoded:
    qm_params += char_map.get(c, c)

params_str = (
    "AUTHORIZATION=" +
    "app-version=10001" +
    "application-id=com.duoduo.read" +
    "channel=unknown" +
    "is-white=" +
    "net-env=5" +
    "platform=android" +
    f"qm-params={qm_params}" +
    f"reg={keys}"
             )

signs = hashlib.md5(params_str.encode()).hexdigest()

headerx = {
    'net-env': '5',
    'reg': '',
    'channel': 'unknown',
    'is-white': '',
    'platform': 'android',
    'application-id': 'com.duoduo.read',
    'authorization': '',
    'app-version': '10001',
    'user-agent': 'webviewversion/0',
    'qm-params': qm_params,
    'sign': signs
          }

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.87 Safari/537.36'
          }

# 全局变量用于缓存百度跳转信息
baidu_name_cache = ""
baidu_jump_cache = ""

class Spider(Spider):
    global xurl
    global xurl1
    global keys
    global headerx
    global headers
    global baidu_name_cache
    global baidu_jump_cache

    def getName(self):
        return "首页"

    def init(self, extend):
        global baidu_name_cache, baidu_jump_cache
        # 初始化时获取百度跳转信息并缓存
        try:
            response = requests.get(url='https://m.baidu.com/', headers=headers, timeout=5)
            response.encoding = 'utf-8'
            code = response.text
            baidu_name_cache = self.extract_middle_text(code, "s1='", "'", 0)
            baidu_jump_cache = self.extract_middle_text(code, "s2='", "'", 0)
        except:
            baidu_name_cache = ""
            baidu_jump_cache = ""

    def isVideoFormat(self, url):
        pass

    def manualVideoCheck(self):
        pass

    def extract_middle_text(self, text, start_str, end_str, pl, start_index1: str = '', end_index2: str = ''):
        if pl == 3:
            plx = []
            while True:
                start_index = text.find(start_str)
                if start_index == -1:
                    break
                end_index = text.find(end_str, start_index + len(start_str))
                if end_index == -1:
                    break
                middle_text = text[start_index + len(start_str):end_index]
                plx.append(middle_text)
                text = text.replace(start_str + middle_text + end_str, '')
            if len(plx) > 0:
                purl = ''
                for i in range(len(plx)):
                    matches = re.findall(start_index1, plx[i])
                    output = ""
                    for match in matches:
                        match3 = re.search(r'(?:^|[^0-9])(\d+)(?:[^0-9]|$)', match[1])
                        if match3:
                            number = match3.group(1)
                        else:
                            number = 0
                        if 'http' not in match[0]:
                            output += f"#{match[1]}${number}{xurl}{match[0]}"
                        else:
                            output += f"#{match[1]}${number}{match[0]}"
                    output = output[1:]
                    purl = purl + output + "$$$"
                purl = purl[:-3]
                return purl
            else:
                return ""
        else:
            start_index = text.find(start_str)
            if start_index == -1:
                return ""
            end_index = text.find(end_str, start_index + len(start_str))
            if end_index == -1:
                return ""

        if pl == 0:
            middle_text = text[start_index + len(start_str):end_index]
            return middle_text.replace("\\", "")

        if pl == 1:
            middle_text = text[start_index + len(start_str):end_index]
            matches = re.findall(start_index1, middle_text)
            if matches:
                jg = ' '.join(matches)
                return jg

        if pl == 2:
            middle_text = text[start_index + len(start_str):end_index]
            matches = re.findall(start_index1, middle_text)
            if matches:
                new_list = [f'{item}' for item in matches]
                jg = '$$$'.join(new_list)
                return jg

    # =====================
    # =====================
    def homeContent(self, filter):
        result = {"class": []}

        # ========================
        # 📌 前端配置区
        # ========================

        # 1️⃣ 重命名：服务端 tag_name → 前端显示名（优先两字 + emoji 美化）
        #    两字标签保持原样加图标；三字及以上尽量压成两字简称
        rename_map = {
            # ---- 两字（原样 + emoji）----
            "新剧":   "🆕新剧",
            "甜宠":   "🍬甜宠",
            "闪婚":   "⚡闪婚",
            "虐恋":   "💔虐恋",
            "宫斗":   "👑宫斗",
            "神医":   "💉神医",
            "都市":   "🏙️都市",
            "逆袭":   "🔥逆袭",
            "异能":   "✨异能",
            "权谋":   "🎭权谋",
            "古装":   "👘古装",
            "年代":   "📻年代",
            "家庭":   "🏠家庭",
            "战神":   "⚔️战神",
            "脑洞":   "💡脑洞",
            "穿书":   "📖穿书",
            "替身":   "🎭替身",
            "赘婿":   "🤵赘婿",
            "神豪":   "💰神豪",
            "团宠":   "🐶团宠",
            "银发":   "👴银发",
            "女帝":   "👸女帝",
            "兵王":   "🪖兵王",
            "搞笑":   "😂搞笑",
            "悬疑":   "🔍悬疑",
            "灵异":   "👻灵异",
            "奇幻":   "🦄奇幻",
            "乡村":   "🌾乡村",
            "民国":   "🎩民国",
            "武侠":   "🗡️武侠",
            "复仇":   "⚔️复仇",
            "重生":   "🔄重生",
            "穿越":   "🌀穿越",
            "马甲":   "🎭马甲",
            "职场":   "💼职场",
            "系统":   "⚙️系统",
            "商战":   "📈商战",
            "伦理":   "⚖️伦理",
            "致富":   "💵致富",
            "宅斗":   "🏡宅斗",

            # ---- 三字 → 压成两字简称 ----
            "豪门总裁": "💎豪门",
            "真假千金": "🎭千金",
            "小人物":   "🙋小人物",   # 三字，保留但往后排
            "欢喜冤家": "💕欢喜",
            "都市情感": "🏙️情感",
            "玄幻仙侠": "🔮仙侠",
            "青春校园": "🎒校园",
            "二次元":   "🌸二次元",   # 三字
            "末世":     "🌍末世",     # 两字
            "科幻":     "🚀科幻",     # 两字
            "打脸虐渣": "💥打脸",
            "女性成长": "🌷成长",
            "追妻火葬场":"💔追妻",
            "强者回归": "👊回归",
            "高手下山": "🏔️下山",
            "宅斗":     "🏡宅斗",
            "偷听心声": "👂心声",
            "娱乐明星": "⭐明星",

            # ---- 四字及以上（留全称，emoji 美化，必然往后排）----
            "社会话题": "🗣️社会话题",
            "种田经商": "🌱种田经商",
            "女性成长": "🌷女性成长",
            "追妻火葬场":"💔追妻火葬场",
        }

        # 2️⃣ 黑名单：这些分类前端不显示
        blacklist = {
        }

        # 3️⃣ 排序权重：字数越少越靠前，字数越多越靠后
        #    规则：len(name)<=2 → 100起；3字 → 200起；4字 → 300起；5字+ → 400起
        #    同字数内按下面手动微调先后顺序
        sort_order = {
            # ---- 两字（100~199，最前）----
            "新剧": 100,  "甜宠": 101,  "闪婚": 102,  "虐恋": 103,  "宫斗": 104,
            "神医": 105,  "都市": 106,  "逆袭": 107,  "异能": 108,  "权谋": 109,
            "古装": 110,  "年代": 111,  "家庭": 112,  "战神": 113,  "脑洞": 114,
            "穿书": 115,  "替身": 116,  "赘婿": 117,  "神豪": 118,  "团宠": 119,
            "银发": 120,  "女帝": 121,  "兵王": 122,  "搞笑": 123,  "悬疑": 124,
            "灵异": 125,  "奇幻": 126,  "乡村": 127,  "民国": 128,  "武侠": 129,
            "复仇": 130,  "重生": 131,  "穿越": 132,  "马甲": 133,  "职场": 134,
            "系统": 135,  "商战": 136,  "伦理": 137,  "致富": 138,  "宅斗": 139,
            "末世": 140,  "科幻": 141,

            # ---- 三字（200~299，居中）----
            "豪门总裁": 200, "真假千金": 201, "欢喜冤家": 202, "都市情感": 203,
            "玄幻仙侠": 204, "青春校园": 205, "二次元": 206,  "打脸虐渣": 207,
            "强者回归": 208, "高手下山": 209, "偷听心声": 210, "娱乐明星": 211,
            "小人物": 212,

            # ---- 四字（300~399，靠后）----
            "社会话题": 300, "种田经商": 301, "女性成长": 302,

            # ---- 五字+（400~499，最后）----
            "追妻火葬场": 400,
        }

        # 全量标签参考（与服务端 tag_categories 对应）：
        # 新剧,甜宠,闪婚,虐恋,宫斗,神医,都市,逆袭,异能,权谋,古装,年代,家庭,战神,脑洞,
        # 社会话题,穿书,替身,种田经商,豪门总裁,真假千金,赘婿,小人物,欢喜冤家,神豪,团宠,
        # 银发,女帝,兵王,搞笑,悬疑,灵异,都市情感,玄幻仙侠,奇幻,乡村,民国,青春校园,
        # 末世,科幻,武侠,二次元,复仇,打脸虐渣,重生,女性成长,穿越,追妻火葬场,马甲,
        # 强者回归,职场,系统,商战,伦理,致富,高手下山,宅斗,偷听心声,娱乐明星

        # ========================
        # 📌 以下为原版安全逻辑（不要改）
        # ========================

        sign_string = f"operation=1playlet_privacy=1tag_id=0{keys}"
        sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()

        url = f"{xurl}/api/v1/playlet/index?tag_id=0&playlet_privacy=1&operation=1&sign={sign}"
        detail = requests.get(url=url, headers=headerx)
        detail.encoding = "utf-8"
        data = detail.json()

        class_list = []

        duoxuan = ['0', '1', '2', '3', '4']
        for duo in duoxuan:
            js = data['data']['tag_categories'][int(duo)]['tags']
            for vod in js:
                name = vod['tag_name']

                # 原版过滤"推荐"字眼
                if "推荐" in name:
                    continue

                # 黑名单过滤
                if name in blacklist:
                    continue

                tid = str(vod['tag_id'])
                display = rename_map.get(name, name)
                if not display:
                    display = name

                # 自动兜底：如果服务端返回了 rename_map 里没写的新标签，
                # 按字数自动算权重，保证不漏、且字多的自然靠后
                auto_sort = sort_order.get(name)
                if auto_sort is None:
                    length = len(name)
                    if length <= 2:
                        auto_sort = 150   # 新出现的两字标签插在两字区末尾
                    elif length == 3:
                        auto_sort = 250
                    elif length == 4:
                        auto_sort = 350
                    else:
                        auto_sort = 450

                class_list.append({
                    "type_id": tid,
                    "type_name": display,
                    "_sort": auto_sort
                })

        # 按权重排序
        class_list.sort(key=lambda x: x["_sort"])

        # 移除临时排序字段，写回结果
        for item in class_list:
            item.pop("_sort", None)
            result["class"].append(item)

        return result

    def homeVideoContent(self):
        videos = []

        sign_string = f"operation=1playlet_privacy=1tag_id=0{keys}"
        sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()

        url = f"{xurl}/api/v1/playlet/index?tag_id=0&playlet_privacy=1&operation=1&sign={sign}"
        detail = requests.get(url=url, headers=headerx)
        detail.encoding = "utf-8"
        data = detail.json()

        data = data['data']['list']

        for vod in data:
            name = vod.get('title', '')
            if not name:
                name = vod.get('name', '未知标题')
            name = str(name).strip()

            id = vod.get('playlet_id', '')
            if not id:
                id = vod.get('id', '')

            pic = vod.get('image_link', '')
            if not pic:
                pic = vod.get('cover', '')

            remark = vod.get('hot_value', '')
            if not remark:
                remark = vod.get('view_count', '')

            video = {
                "vod_id": str(id),
                "vod_name": name,
                "vod_pic": pic,
                "vod_remarks": str(remark) if remark else ''
                    }
            videos.append(video)

        result = {'list': videos}
        return result

    def categoryContent(self, cid, pg, filter, ext):
        result = {}
        videos = []

        if pg:
            page = int(pg)
        else:
            page = 1

        if page == 1:
            sign_string = f"operation=1playlet_privacy=1tag_id={cid}{keys}"
            sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()
            url = f'{xurl}/api/v1/playlet/index?tag_id={cid}&playlet_privacy=1&operation=1&sign={sign}'
        else:
            sign_string = f"next_id={str(page)}operation=1playlet_privacy=1tag_id={cid}{keys}"
            sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()
            url = f'{xurl}/api/v1/playlet/index?tag_id={cid}&next_id={str(page)}&playlet_privacy=1&operation=1&sign={sign}'

        detail = requests.get(url=url, headers=headerx)
        detail.encoding = "utf-8"
        data = detail.json()

        data = data['data']['list']

        for vod in data:
            name = vod.get('title', '')
            if not name:
                name = vod.get('name', '未知标题')
            name = str(name).strip()

            id = vod.get('playlet_id', '')
            if not id:
                id = vod.get('id', '')

            pic = vod.get('image_link', '')
            if not pic:
                pic = vod.get('cover', '')

            remark = vod.get('hot_value', '')
            if not remark:
                remark = vod.get('view_count', '')

            video = {
                "vod_id": str(id),
                "vod_name": name,
                "vod_pic": pic,
                "vod_remarks": str(remark) if remark else ''
                    }
            videos.append(video)

        result = {'list': videos}
        result['page'] = pg
        result['pagecount'] = 9999
        result['limit'] = 90
        result['total'] = 999999
        return result

    def detailContent(self, ids):
        did = ids[0]
        result = {}
        videos = []
        xianlu = '神马专线'
        bofang = ''

        sign_string = f"playlet_id={did}{keys}"
        sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()

        urls = f'{xurl1}/player/api/v1/playlet/info?playlet_id={did}&sign={sign}'
        detail = requests.get(url=urls, headers=headerx)
        detail.encoding = "utf-8"
        detail = detail.json()

        title = detail.get('data', {}).get('title', '')
        if not title:
            title = detail.get('data', {}).get('name', '未知标题')

        blurb = detail.get('data', {}).get('intro') or "暂无剧情简介"
        content = '  ' + str(blurb)

        jisu = detail.get('data', {}).get('total_episode_num', '未知')
        jisu = str(jisu) + '全集'

        leixing = detail.get('data', {}).get('tags', '未知')

        remarks = str(leixing) + " " + str(jisu)

        soup = detail.get('data', {}).get('play_list', [])

        if soup:
            for sou in soup:
                video_url = sou.get('video_url', '')
                sort_name = sou.get('sort', '')
                if video_url and sort_name:
                    bofang = bofang + str(sort_name) + '$' + str(video_url) + '#'
            bofang = bofang[:-1] if bofang else ''
        else:
            global baidu_jump_cache
            bofang = baidu_jump_cache

        videos.append({
            "vod_id": str(did),
            "vod_name": str(title),
            "vod_remarks": remarks,
            "vod_content": content,
            "vod_play_from": xianlu,
            "vod_play_url": bofang
                     })

        result['list'] = videos
        return result

    def playerContent(self, flag, id, vipFlags):
        result = {}
        play_url = ""

        global baidu_jump_cache
        baidu_jump = baidu_jump_cache

        if 'baidu.com' in str(id) or 'tuios.com' in str(id) or 'qmplaylet' in str(id):
            play_url = str(id)
        elif str(id).startswith('http'):
            play_url = str(id)
        else:
            try:
                sign_string = f"playlet_id={id}{keys}"
                sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()
                detail_url = f'{xurl1}/player/api/v1/playlet/info?playlet_id={id}&sign={sign}'
                detail_response = requests.get(url=detail_url, headers=headerx, timeout=5)
                detail_response.encoding = "utf-8"
                detail_data = detail_response.json()

                title = detail_data.get('data', {}).get('title', '')
                if not title:
                    title = detail_data.get('data', {}).get('name', '')

                play_list = detail_data.get('data', {}).get('play_list', [])

                if play_list:
                    for idx, item in enumerate(play_list):
                        video_url = item.get('video_url', '')
                        sort_name = item.get('sort', '')
                        if video_url and sort_name:
                            if play_url:
                                play_url += '#'
                            play_url += str(sort_name) + '$' + str(video_url)
                else:
                    play_url = str(id)
            except Exception as e:
                play_url = baidu_jump if baidu_jump else str(id)

        result["parse"] = 0
        result["playUrl"] = ''
        result["url"] = play_url if play_url else str(id)
        result["header"] = headers

        return result

    def searchContentPage(self, key, quick, pg):
        result = {}
        videos = []

        if pg:
            page = int(pg)
        else:
            page = 1

        sign_string = f"extend=page={str(page)}read_preference=0track_id=ec1280db127955061754851657967wd={key}{keys}"
        sign = hashlib.md5(sign_string.encode('utf-8')).hexdigest()

        url = f'{xurl}/api/v1/playlet/search?extend=&page={str(page)}&wd={key}&read_preference=0&track_id=ec1280db127955061754851657967&sign={sign}'
        detail = requests.get(url=url, headers=headerx)
        detail.encoding = "utf-8"
        detail = detail.json()

        data = detail['data']['list']

        for vod in data:
            name = vod.get('title', '')
            if not name:
                name = vod.get('name', '未知标题')
            name = re.sub(r'<[^>]+>', '', str(name))
            name = ' '.join(name.split())

            id = vod.get('id', '')

            pic = vod.get('image_link', '')
            if not pic:
                pic = vod.get('cover', '')

            remark = vod.get('total_num', '')

            video = {
                "vod_id": str(id),
                "vod_name": name,
                "vod_pic": pic,
                "vod_remarks": str(remark) if remark else ''
                    }
            videos.append(video)

        result['list'] = videos
        result['page'] = pg
        result['pagecount'] = 9999
        result['limit'] = 90
        result['total'] = 999999
        return result

    def searchContent(self, key, quick, pg="1"):
        return self.searchContentPage(key, quick, '1')

    def localProxy(self, params):
        if params['type'] == "m3u8":
            return self.proxyM3u8(params)
        elif params['type'] == "media":
            return self.proxyMedia(params)
        elif params['type'] == "ts":
            return self.proxyTs(params)
        return None
