#!/usr/bin/python3
# coding=utf-8
# -*- coding: utf-8 -*-

from flask import Flask, request

import redis
import requests
import time
import sys
import json

from libs.proxy import Proxy

def get_websocket_info(live_stream_id, retry=0):
    param_json = {
        "operationName": "WebSocketInfoQuery",
        "variables": {
            "liveStreamId": live_stream_id
        },
        "query": "query WebSocketInfoQuery($liveStreamId: String) {\n  webSocketInfo(liveStreamId: $liveStreamId) {\n    token\n    webSocketUrls\n    __typename\n  }\n}\n"
    }

    user_agents = {
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3908.2 Safari/537.36',
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/77.0.3865.90 Safari/537.36',
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/82.0.4055.0 Safari/537.36',
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.77 Safari/537.36',
    }

    cookie_list = [
        'web_8f5a71d99d502219d7ce1ca1ffec68cc; clientid=3; client_key=65890b29; Hm_lvt_86a27b7db2c5c0ae37fee4a8a35033ee=1595212029,1597053525; userId=1089887853; kuaishou.live.bfb1s=477cb0011daca84b36b3a4676857e5a1; didv=1603703368000; kpn=GAME_ZONE; kuaishou.live.web_st=ChRrdWFpc2hvdS5saXZlLndlYi5zdBKgATKQJmTG8mk6lH2pOJeVa-184Sq95qCmEB1lP_q3lItdj6a6gnYgCIcPkt1ZvnJlE3cAkBVIiK46xUEyJ1SZe2hatA7MGwchvIhNlvumZQWfwUHwDOKWRkpNOn-4c0wYnHmgituvkv2B2XJC1lnaLBJ57zZU6iKrtgtXyekfNrf37VW3n0cpGF8LfjxvfQibwYfxewf2uOhDQzKnCA-kLosaEhqtgBlt9k0TlHg5Y4Goo9D9kyIgPsuUj-GlXMBsywOViBgpl-tt7OVb051_AZAHST9ItzAoBTAB; kuaishou.live.web_ph=8f83f1ed0c54c288352d86b3a23fe996499e; userId=1089887853',
        'kuaishou.live.bfb1s=7206d814e5c089a58c910ed8bf52ace5; clientid=3; client_key=65890b29; sid=83cf0fb9ee15fde6b6987781; did=web_c280cf49619a100d8305b6f095ed5348; userId=1509370340; didv=1600053596890; WEBLOGGER_HTTP_SEQ_ID=18928; WEBLOGGER_INCREAMENT_ID_KEY=19033; Hm_lvt_86a27b7db2c5c0ae37fee4a8a35033ee=1601391911; Hm_lpvt_86a27b7db2c5c0ae37fee4a8a35033ee=1602073621; kpn=GAME_ZONE; kuaishou.live.web_st=ChRrdWFpc2hvdS5saXZlLndlYi5zdBKgAcSZ7bO7s4CskEWV8oXNhqBeAW7qebb6Gv8_pmwLpeDLgzlIl3Nwgd2qal6895okBZxhkFFXQLIRvSJss1ALy5H2k0B9JvRZc9RNm4FzKmZ4eWPx2B3f-RLyiz6CkNpEsjcgpCQ90pQyBJgc9nWyo3vJRJS9fl53t2hFW29cvcHJktjSh4VRl-JrZ6WI3r3Et1kGxyh0wGqW3Q3xWzVMmIkaEsvpGUru20c-iIt7T0W8MQrXwiIgMbhYaFs2XBsAXpVKQoE6xmrFlpUw3VfiZaze7OM6mkEoBTAB; kuaishou.live.web_ph=6248010c3bbf0923d41c9d3c231b2903007e; userId=1509370340',
        'did=web_03d225bf290f905c1d1a5b3810af3b07; didv=1607676125725; kuaishou.live.bfb1s=9b8f70844293bed778aade6e0a8f9942; clientid=3; client_key=65890b29; kpn=GAME_ZONE; Hm_lvt_86a27b7db2c5c0ae37fee4a8a35033ee=1607676148; Hm_lpvt_86a27b7db2c5c0ae37fee4a8a35033ee=1607676148; userId=2020851346; userId=2020851346; kuaishou.live.web_st=ChRrdWFpc2hvdS5saXZlLndlYi5zdBKgAR00HvTWTmSGTYkxB2Mny84aCe3Lu8qGgMyP7Bhp7Y8lHqkC4TPLNt8H71ZesLiX4hpu9eTzOCiCmzXp7bCHQxzYt6_POWrTVT9L-jgFL4ET3JmTqrjxudwbSZQP-icUJACHgXEpf9a_BXvPS1RrBnl82Gyg1LwLXxIo0uIdwhheHrAp3u4h4C6S5lqWvZ7aNX1nDk_5Jsyq8UqsuCXLNiMaEgCrAu8bFEUPixNgRvVq1Nb0ZSIgKyTMpEkPIRE47p3L_U0hfzZ2IIBI9YfiD5jI34L2zmAoBTAB; kuaishou.live.web_ph=2f965caadcde0572313c46a7f8de50843381'
    ]

    res = redis_connection.zrangebyscore(scrape_cookie_key, 0, '+inf', start=0, num=1, withscores=True)

    if len(res) == 0:
        cookie = 'kuaishou.live.bfb1s=9b8f70844293bed778aade6e0a8f9942; clientid=3; did=web_2ba8ea2ca07df85a9e193a75d443d128; client_key=65890b29; kpn=GAME_ZONE; userId=1509370340; userId=1509370340; kuaishou.live.web_st=ChRrdWFpc2hvdS5saXZlLndlYi5zdBKgAZIqgEWU0YHOwebl3XEBuu7odxIyvSspUq1v_kvLGCVI_eURJXwii1_AN0CwuxScDBKoK2y2HIghAZLxMVzJhR0wQ4IHJ0zi3TtLsnrbNlA9VnnSZpuHK2b1M2RxEiFEKBRTK0AwnRzN7UOvkKQTfOIxtALOFXqSPOzJ4R8ScI30o_CqmgWPbg7dw0Kt5sgpGFYcZMCA1vgkFhiph-O4cE8aEjGueioax06vmORaF3eBQr3cQSIgsBvfCInpIGIqrTgvvo3648essPww4pkcAKNHOgeNHjUoBTAB; kuaishou.live.web_ph=d77227e666b43027931968af443b14bd7b28'
    else:
        cookie, times = res[0]
        print('获取cookie ' + str(int(times)) + ' ' + cookie[0:30])

    headers = {
        'user-agent': user_agents.pop(),
        'Cookie': cookie,
        'Referer': 'https://live.kuaishou.com/u/3xse' + live_stream_id
    }

    proxy = Proxy.get()
    print('获取代理 ' + proxy)

    proxies = {
        "http": "http://" + proxy,
        "https": "http://" + proxy
    }

    try:
        # https://live.kuaishou.com/live_graphql
        r = requests.post(
            "https://live.kuaishou.com/m_graphql",
            json=param_json,
            headers=headers,
            proxies=proxies,
            timeout=30
        )

        webSocketInfo = r.json()['data']['webSocketInfo']
        token = webSocketInfo.get('token')

        if token is None:
            redis_connection.zincrby(scrape_cookie_key, 1, cookie)
            Proxy.del_proxy(proxy)
            print('获取Token失败' + Proxy.proxy_info)
            # 重试10次还失败
            if retry > 10:
                return "wss://live-ws-pg.kuaishou.com/websocket", "lWQRDokMCJn2+oZyzA7dMovsp8AviGc/8NaWgfY0A4Jtp8trEaxG4Fy/49rkUtiERneEtNXYSofggh+PZUzpFHbUMCjNi/j/OxpfectTdDWWaO2VgGZbX8MBpcuQolEiGqS5fKwTqvbjV32an6gfNeIopmUI7HxwLQuobGCQQsM="

            time.sleep(2)
            retry = retry + 1
            return get_websocket_info(live_stream_id, retry)

        url = webSocketInfo['webSocketUrls'][0]

        return url, token

    except Exception as e:
        print('获取Token抛出异常：' + str(e) + '\n' + Proxy.proxy_info)
        return get_websocket_info(live_stream_id, retry)


app = Flask(__name__)

requests.packages.urllib3.disable_warnings()

redis_connection = redis.Redis(
    host='r-2ze28bdb7389a8a4.redis.rds.aliyuncs.com',
    port=6379,
    password='Zhuaduoduo2017',
    decode_responses=True
)

# cookie
scrape_cookie_key = 'barrageScraperCookieSortSet'


@app.route('/getToken', methods=['GET'])
def getToken():
    live_stream_id = request.args.get('live_stream_id')

    if live_stream_id is not None:
        url, token = get_websocket_info(live_stream_id)
        return json.dumps({
            'url': url,
            'token': token,
        })

    return ''

# def run_server(port):
#     app.run(host='0.0.0.0', port=port)


if __name__ == "__main__":
    port = int(sys.argv[1])
    app.run(host='0.0.0.0', port=port)