import re
import json
import asyncio
import uvicorn
from contextlib import asynccontextmanager
from fastapi import FastAPI
import aiomysql
from pymysql.converters import escape_string
from playwright.async_api import async_playwright

HOST = '0.0.0.0'
PORT = 8000
CODE_RULE = r"^([0-9]{1,2}[A-Z])\-*([0-9]{4})|([0-9]{1,3})\-*([0-9]{4})$"
store = 20153
storeIdentifier = 'LEI_SHING_HONG'
auth = 'lsh'

@asynccontextmanager
async def lifespan(app: FastAPI):
    app.state.db_pool = await aiomysql.create_pool(
        host='localhost',
        port=3306,
        user='root',
        password='rxxt',
        db='workdata',
        autocommit=True,
        minsize=2,
        maxsize=12,
    )
    yield
    app.state.db_pool.close()
    await app.state.db_pool.wait_closed()

app = FastAPI(lifespan=lifespan)

@app.get('/hi')
def _():
    return {
        'version': 2.0,
        'host': HOST,
        'port': PORT
    }

def getUniqueId(s: str):
    p = re.match(r'^.+<pre.*>(.+)</pre>.*</body></html>$',s)
    if p:
        try:
            r = json.loads(p.group(1))
            if 'uniqueId' in r and r['uniqueId']:
                return r['uniqueId'] # ({'code': code,'id': r['uniqueId']})
        except Exception as e:
            print('json 格式错',e)
    return None

def getDetail(s: str):
    p = re.match(r'^.+<pre.*>(.+)</pre>.*</body></html>$',s)
    if p:
        try:
            r = json.loads(p.group(1))
            return r
        except Exception as e:
            print('json 格式错',e)
    return None

@app.get('/product/{code}')
async def getProductInfo(code: str,corp: str = 'lsh'):
    global store,storeIdentifier,auth
    auth = corp.lower()
    if auth not in ['lsh','eci']:
        auth = 'lsh'
    if auth == 'lsh':
        store = 20153
        storeIdentifier = 'LEI_SHING_HONG'
    else:
        store = 20182
        storeIdentifier = 'ECI_METRO'
    code = code.strip().upper()
    result = {'result': 101,'code': code,'info': {}}
    p = re.match(CODE_RULE,code)
    if p:
        code = f'{p.group(1)}-{p.group(2)}' if p.group(1) else f'{int(p.group(3)):03d}-{p.group(4)}'
    else:
        return result
    result['code'] = code
    async with app.state.db_pool.acquire() as conn:
        async with conn.cursor(aiomysql.DictCursor) as cursor:
            await cursor.execute(f"SELECT * FROM catdb WHERE code='{code}' AND corp='{auth}'")
            res = await cursor.fetchone()
            if res:
                result['result'] = 0
                result['info'] = json.loads(res['data'])
            else:
                res = await searchInfo(code)
                if res:
                    result['result'] = 0
                    result['info'] = res
                    query = f"INSERT INTO catdb (corp,code,data) VALUES ('{auth}','{code}','{escape_string(json.dumps(res,ensure_ascii=False))}')"
                    await cursor.execute(query)
    return result

async def searchInfo(code):
    result = None
    async with async_playwright() as playwright:
        browser = await playwright.chromium.launch(headless=False)
        context = await browser.new_context(base_url="https://parts.cat.com",storage_state=f"auth/auth-{auth}.json")
        page = await context.new_page()
        url = f"/wcs/resources/store/{store}/seo/urltoken/byProduct?token={code}"
        await page.goto(url,timeout=0)
        res = await page.content()
        id = getUniqueId(res.strip())
        if id:
            # 详情url，不包含价格信息
            url = f'/api/product/detail?productId={id}&storeIdentifier={storeIdentifier}&locale=zh_CN&partNumber={code}&storeId={store}&langId=-7'
            await page.goto(url,timeout=0)
            res = await page.content()
            result = getDetail(res.strip())
            if result:
                partsList = [{'partNumber': code,"uniqueId": f"{id}",'quantity': 1}]
                if 'merchandisingAssociations' in result and len(result['merchandisingAssociations']):
                    partsList += [{'partNumber': item['partNumber'],'quantity': 1} for item in result['merchandisingAssociations']]
                elif 'components' in result and len(result['components']):
                    partsList += [{'partNumber': item['partNumber'],'quantity': 1} for item in result['components']]
                # 价格信息url接口，POST请求，入口参数是partsList(件号数组)
                url = f'/wcs/resources/store/{store}/getDealerPriceAndAvailability?langId=-7'
                try:
                    resp = await context.request.post(url,data={'partsList': partsList},timeout=0)
                    if resp and resp.ok:
                        result['priceInfo'] = await resp.json()
                except Exception:
                    print('引用接口错误！')
                    await asyncio.sleep(1)
        await context.close()
        await browser.close()
    return result

if __name__ == "__main__":
    uvicorn.run(app, host=HOST, port=PORT,log_level="info")