import asyncio
import pytest
import requests
import time
import re
from Utils import config_handler
from Utils import global_variate
from Utils import websocket_handler
from _pytest import terminal
from Utils import webhook_handler

session = requests.session()
def backstage_login():
    url = 'https://backstage.willdesk.com/api/v1/BackStage/AdminUser/login'
    data = {"password":"123456","username":"willdesk"}
    res = requests.post(url=url,json=data)
    token = res.json()['data']['token']
    setattr(global_variate.Global,'backstage_token',token)
def willdesk_login():
    '''
    执行用例前获取登录态的接口，全局只执行一次，通过env参数控制是否测试环境
    '''
    env = 'test'
    email = 'test_account' if env == 'test' else 'formal_account'
    password = 'test_password' if env == 'test' else 'formal_password'
    address = 'test_address' if env == 'test' else 'formal_address'
    account = config_handler.base_config.get_value('account',email)
    password = config_handler.base_config.get_value('account',password)
    login_url = f'{config_handler.base_config.get_value("url",address)}/api/v1/customerService/loginV2' #登录的api
    token_url = f'{config_handler.base_config.get_value("url",address)}/api/v1/customerService/genToken' #登录后获取token的api
    version_url = 'https://hstest.sealapps.com/login' if env == 'test' else 'https://app.willdesk.com/login'
    headers = {
        'Content-Type' : 'application/json',
        'User-Agent' : 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0'
    }
    version_res = session.get(url=version_url,headers=headers).text
    v = re.findall(r'<script src="/(.*?)/assets',version_res)[0]
    login_data = {
        'email' : account,
        'password' : password,
        'v' : v
    }
    login_res = session.post(url=login_url,headers=headers,json=login_data).json()
    code = login_res['data']['code']
    brandName = login_res['data']['list'][0]['brandName']
    brandId = login_res['data']['list'][0]['id']
    token_data = {
        'code' : code,
        'brandId' : brandId,
        'email' : account,
        'v' : v
    }
    token_res = session.post(url=token_url,headers=headers,json=token_data).json()
    token = token_res['data']['accessToken']
    email = token_res['data']['email']
    userId = token_res['data']['id']
    name = token_res['data']['name']
    shopList = token_res['data']['shopList']
    companyId = token_res['data']['shopList'][0]['companyId']
    setattr(global_variate.Global,'access_token',token)
    setattr(global_variate.Global,'brandId',brandId)
    setattr(global_variate.Global,'brandName',brandName)
    setattr(global_variate.Global,'code',code)
    setattr(global_variate.Global,'v',v)
    setattr(global_variate.Global,'email',email)
    setattr(global_variate.Global,'userId',userId)
    setattr(global_variate.Global,'name',name)
    setattr(global_variate.Global,'shopList',shopList)
    setattr(global_variate.Global,'shopId',shopList[0]['id'])
    setattr(global_variate.Global,'shopDomain',shopList[0]['shopDomain'])
    setattr(global_variate.Global,'companyId',companyId)

def customer_login():
    env = 'test'
    address = 'test_address' if env == 'test' else 'formal_address'
    headers = {
        'Content-Type': 'application/json',
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0'
    }
    version_url = 'https://allplatformtest.sealapps.com/api/v1/shop/local/getCustomerHtmlLinks?customerDomain=https://utest.sealapps.com&shopDomain=zhs-test.myshopify.com' if env == 'test' else 'https://api.willdesk.com/api/v1/shop/local/getCustomerHtmlLinks?customerDomain=https://u.willdesk.com&shopDomain=test-2023013104.myshopify.com'
    shop_url = 'https://allplatformtest.sealapps.com/api/v1/setting/getCShopSetting?domain=zhs-test.myshopify.com' if env == 'test' else 'https://api.willdesk.com/api/v1/setting/getCShopSetting?domain=test-2023013104.myshopify.com'
    version_res = session.get(url=version_url,headers=headers).text
    customer_v = re.findall(r'utest.sealapps.com/(.*?)/assets',version_res)[0]
    shop_res = session.get(url=shop_url,headers=headers).json()
    login_url = f'{config_handler.base_config.get_value("url",address)}/api/v1/customer/login' #登录的api
    data = {
        'email' : 'autotest_customer@qq.com',
        'isVisitor' : 0,
        'location' : 'Shanghai,China',
        'phone' : '',
        'shopId' : shop_res['data']['shopInfo']['shopId'],
        'timeZone' : '8',
        'userName' : 'autotest_customer',
        'v' : customer_v
    }
    setattr(global_variate.Global,'customer_v',customer_v)
    login_res = session.post(url=login_url,json=data,headers=headers).json()
    setattr(global_variate.Global,'customerToken',login_res['data']['accessToken'])
    setattr(global_variate.Global,'customerId',login_res['data']['customerId'])
    setattr(global_variate.Global,'customerName',login_res['data']['name'])
    address = 'wss://atestws.sealapps.com/ws' if env == 'test' else 'wss://ws.willdesk.com/ws'
    print(f'{address}?platform=1&authorization={global_variate.Global.access_token}&t={int(time.time())}')
    websocket_handler.ws_willdesk = websocket_handler.ws(
        url=f'{address}?platform=1&authorization={global_variate.Global.access_token}&t={int(time.time())}',
        attr='willdesk')  # willdesk账号websocket链接
    websocket_handler.ws_client = websocket_handler.ws(
        url=f'{address}?platform=1&authorization={global_variate.Global.customerToken}&t={int(time.time())}',
        attr='customer')  # c端websocket链接

async def close_ws():
    await asyncio.gather(asyncio.create_task(websocket_handler.ws_willdesk.close()),asyncio.create_task(websocket_handler.ws_client.close()))

async def get_client_id():
    await asyncio.gather(asyncio.create_task(websocket_handler.ws_willdesk.get_client_id()),asyncio.create_task(websocket_handler.ws_client.get_client_id()))


def pytest_terminal_summary(terminalreporter, exitstatus, config):
    '''收集测试结果'''
    print("total:", terminalreporter._numcollected)
    print('passed:', len(terminalreporter.stats.get('passed', [])))
    print('failed:', len(terminalreporter.stats.get('failed', [])))
    print('error:', len(terminalreporter.stats.get('error', [])))
    print('skipped:', len(terminalreporter.stats.get('skipped', [])))
    # terminalreporter._sessionstarttime 会话开始时间
    duration = time.time() - terminalreporter._sessionstarttime
    print('total times:', round(duration,2), 'seconds')
    webhook_handler.webhook.sendAutotestReport(result=terminalreporter,second=round(duration,2))


loop = websocket_handler.loop
@pytest.fixture(scope='session',autouse=True)
def call_fixture():
    #自定义调用顺序
    willdesk_login()
    customer_login()
    backstage_login()
    loop.run_until_complete(get_client_id()) #初始化获取clientid
    yield
    loop.run_until_complete(close_ws()) #关闭服务
    loop.stop()
    loop.close()