import requests
import json
import time

# 官方文档地址
# https://doc2.bitbrowser.cn/jiekou/ben-di-fu-wu-zhi-nan.html

# 此demo仅作为参考使用，以下使用的指纹参数仅是部分参数，完整参数请参考文档

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


def createBrowser(
        groupId=None
):  # 创建或者更新窗口，指纹参数 browserFingerPrint 如没有特定需求，只需要指定下内核即可，如果需要更详细的参数，请参考文档
    json_data = {
        "groupId": groupId,  # 分组id
        'name': 'google',  # 窗口名称
        'remark': '',  # 备注
        'proxyMethod': 2,  # 代理方式 2自定义 3 提取IP
        # 代理类型  ['noproxy', 'http', 'https', 'socks5', 'ssh']
        'proxyType': 'noproxy',
        'host': '',  # 代理主机
        'port': '',  # 代理端口
        'proxyUserName': '',  # 代理账号
        "browserFingerPrint": {  # 指纹对象
            'coreVersion': '124'  # 内核版本，注意，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 get_group_lists():
#         print(i)

# pass
# browser_id = createBrowser()
# openBrowser(browser_id)
#
# time.sleep(10)  # 等待10秒自动关闭窗口
#
# closeBrowser(browser_id)
#
# time.sleep(10)  # 等待10秒自动删掉窗口
#
# deleteBrowser(browser_id)

# deleteBrowser(id="43f0d0978f6a4bb7bbcb0b7786f436e9")