import json
import requests
from peewee import fn

from models.ips import Ips
from models.xstart import Xstart

url = "http://127.0.0.1:54345"
headers = {'Content-Type': 'application/json'}
tg_url = "https://web.telegram.org/a/"


def createBrowser(
        groupId=None,
        host=None,
        port=None,
        proxyUserName=None,
        proxyPassword=None
):  # 创建或者更新窗口，指纹参数 browserFingerPrint 如没有特定需求，只需要指定下内核即可，如果需要更详细的参数，请参考文档
    json_data = {
        "groupId": groupId,  # 分组id
        'name': 'google',  # 窗口名称
        'remark': '',  # 备注
        'proxyMethod': 1,  # 代理方式 2自定义 3 提取IP
        # 代理类型  ['noproxy', 'http', 'https', 'socks5', 'ssh']
        'proxyType': 'socks5',
        'host': host,  # 代理主机EE
        'port': port,  # 代理端口
        'proxyUserName': proxyUserName,  # 代理账号
        'proxyPassword': proxyPassword,  # 代理账号
        "browserFingerPrint": {  # 指纹对象
            'coreVersion': '140'  # 内核版本，注意，win7/win8/winserver 2012 已经不支持112及以上内核了，无法打开
        }
    }

    res = requests.post(f"{url}/browser/update",
                        data=json.dumps(json_data), headers=headers).json()
    browserId = res['data']['id']

    return browserId


def updateBrowser():  # 更新窗口，支持批量更新和按需更新，ids 传入数组，单独更新只传一个id即可，只传入需要修改的字段即可，比如修改备注，具体字段请参考文档，browserFingerPrint指纹对象不修改，则无需传入
    json_data = {'ids': ['93672cf112a044f08b653cab691216f0'],
                 'remark': '我是一个备注', 'browserFingerPrint': {}}
    res = requests.post(f"{url}/browser/update/partial",
                        data=json.dumps(json_data), headers=headers).json()
    print(res)


def openBrowser(id):  # 直接指定ID打开窗口，也可以使用 createBrowser 方法返回的ID
    json_data = {"id": f'{id}', "args": [
        # "--disable-application-cache",
        # "--disable-cache",
        # "--disable-gpu-shader-disk-cache",
        # "--media-cache-size=1",
        # "--disk-cache-size=1",
        # "--incognito"
    ]}
    res = requests.post(f"{url}/browser/open",
                        data=json.dumps(json_data), headers=headers).json()

    return res["data"]["http"].split(":")[1]


def closeBrowser(id):  # 关闭窗口
    json_data = {'id': f'{id}'}
    res = requests.post(f"{url}/browser/close",
                        data=json.dumps(json_data), headers=headers)
    return res.json()


def deleteBrowser(id):  # 删除窗口
    json_data = {'id': f'{id}'}
    print(requests.post(f"{url}/browser/delete",
                        data=json.dumps(json_data), headers=headers).json())


def query_bit_browser(page, page_size):
    data = {"page": page, "pageSize": page_size, 'sort': 'asc'}

    res = requests.post(f'{url}/browser/list', data=json.dumps(data), headers=headers)

    return res.json()["data"]["list"]


def update_proxy_Browser(
        id,
        host, port, proxyUserName, proxyPassword
):
    json_data = {
        "ids": [id],
        # "ipCheckService": "ip123in",
        "proxyMethod": 2,
        "proxyType": "socks5",
        "host": host,
        "port": port,
        "proxyUserName": proxyUserName,
        "proxyPassword": proxyPassword
    }

    res = requests.post(f'{url}/browser/proxy/update', data=json.dumps(json_data), headers=headers)
    return res.json()


def get_group_lists_Browser():
    json_data = {
        "page": 0,
        "pageSize": 100,
        "all": True
    }

    res = requests.post(f'{url}/group/list', data=json.dumps(json_data), headers=headers)
    return res.json()["data"]["list"]


def get_browser_lists_Browser(id):
    json_data = {
        "groupId": id,
        "page": 0,
        "pageSize": 100
    }

    res = requests.post(f'{url}/browser/list', data=json.dumps(json_data), headers=headers)
    return res.json()["data"]["list"]


def get_group_lists():  # 获取全部分组的信息
    # url  = "/group/list"

    json_data = {
        "page": 0,
        "pageSize": 10,
        "all": True
    }

    res = requests.post(f'{url}/group/list', data=json.dumps(json_data), headers=headers)

    data = {}

    for i in res.json()["data"]["list"]:
        data[i["groupName"]] = i["id"]

    return data


def group_add(groupName):
    json_data = {
        "groupName": groupName,
        "sortNum": 0
    }

    res = requests.post(f'{url}/group/add', data=json.dumps(json_data), headers=headers)

    return res.json()

if __name__ == '__main__':
    for i in Xstart.select().where(
            Xstart.x_id.is_null()
    ):
        ips_info = Ips.select().where(Ips.start == 1,Ips.country == "法国").order_by(fn.Rand()).get()

        update_proxy_Browser(
            id=i.bit_id,
            host=ips_info.host,
            port=ips_info.port,
            proxyUserName=ips_info.username,
            proxyPassword=ips_info.password
        )