import random
import time
import requests
import werobot
from werobot.replies import ArticlesReply, Article, ImageReply, TextReply, MusicReply
robot = werobot.WeRoBot(token='token')
base_url = "https://yourdomain/api"
headers = {
    "j-auth-token": "your_token"
}

def make_request(endpoint, params=None):
    response = requests.get(f"{base_url}/{endpoint}", params=params, headers=headers)
    if response.status_code == 200:
        return response.json()
    elif response.status_code == 404:
        #print(f"未找到数据，URL: {base_url}/{endpoint}")
        return None
    else:
        #print(f"请求接口失败，状态码: {response.status_code}, URL: {base_url}/{endpoint}")
        return None
# 获取影片详情
def get_movie_details(movie_id):
    return make_request(f"movies/{movie_id}")

# 获取磁力链接
def get_movie_magnets(movie_id):
    # 获取影片详情数据
    movie_details_data = get_movie_details(movie_id)

    if movie_details_data:
        # 提取影片的 gid 和 uc
        gid = movie_details_data.get("gid", "")
        uc = movie_details_data.get("uc", "")

        # 调用磁力链接接口
        magnet_url = f"magnets/{movie_id}?gid={gid}&uc={uc}"
        movie_magnet_data = make_request(magnet_url)
        if movie_magnet_data is not None:
            link = movie_magnet_data[0]["link"]
            title = movie_magnet_data[0]["title"]
            size = movie_magnet_data[0]["size"]
            sharedate = movie_magnet_data[0]["shareDate"]
            return link, title, size, sharedate
        return "未找到"


#获取种子
@robot.text
def zz(message):
    if message.content[:2] == "种子":
        id = message.content[2:].replace(" ", "")
        magnets = get_movie_magnets(id)
        if magnets is not None:
            return f"磁力链接:{magnets[0]}\n标题:{magnets[1]}\n文件大小:{magnets[2]}\n分享日期:{magnets[3]}" 
        return f"未找到{id}"

@robot.text
def replay(msg):
    return "要发关键词哦!"
robot.config['HOST'] = '0.0.0.0'
robot.config['PORT'] = 4080

robot.run()