import asyncio
import os
from pathlib import Path
import re
import time
from httpx import AsyncClient
import requests
import json
import argparse

from bilix.sites.bilibili import api
from rich import print


def arg_parse():
    parser = argparse.ArgumentParser()
    
    ranking_group = parser.add_argument_group()
    ranking_group.title = 'by ranking'
    ranking_group.add_argument('--by-ranking', action='store_true', help='从排行榜获取 bvids')
    ranking_group.add_argument('--ranking-rid', type=int, default=0, help='目标排行 rid，0 为全站排行榜 [default: 0]')

    up_videos_group = parser.add_argument_group()
    up_videos_group.title = 'by up videos'
    up_videos_group.add_argument('--by-up_videos', action='store_true', help='从 up 主用户页获取全部的投稿的 bvids')
    up_videos_group.add_argument('--up_videos-mid', type=int, help='目标 up 主的 mid (也可以是用户页的 URL)')

    args = parser.parse_args()
    return args

def by_raning(rid: int) -> Path:
    bilibili_ranking_api = "https://api.bilibili.com/x/web-interface/ranking/v2"
    bilibili_ranking_params = {
        "rid": rid,
        "type": "all"
    }

    r = requests.get(bilibili_ranking_api, params=bilibili_ranking_params)
    r.raise_for_status()
    ranking_json = json.loads(r.text)
    assert ranking_json['code'] == 0 # 0 为成功（HTTP 200 不能信）

    ranking = ranking_json['data']['list']
    bvids = []
    for video_info in ranking:
        # print(video_info['title'], video_info['bvid'], video_info['pic'])
        bvid = video_info['bvid']
        bvids.append(bvid)

    import datetime
    today = datetime.date.today()
    os.makedirs('bvids', exist_ok=True)

    bvids_filepath = f'bvids/by-ranking/rid-{rid}/rid-{rid}-{int(time.time())}.txt'
    os.makedirs(os.path.dirname(bvids_filepath), exist_ok=True)
    with open(bvids_filepath, 'w', encoding='utf-8') as f:
        for bvid in bvids:
            f.write(f'{bvid}' + '\n')
    abs_filepath = os.path.abspath(bvids_filepath)
    print(f'已保存 {len(bvids)} 个 bvid 到 {abs_filepath}')
    return Path(abs_filepath)

async def by_up_videos(url_or_mid: str) -> Path:
    ''' 频率高了会封 '''

    if isinstance(url_or_mid, int):
        mid = str(url_or_mid)
    elif url_or_mid.startswith("http"):
        mid = re.findall(r"/(\d+)", url_or_mid)[0]
    else:
        mid = url_or_mid

    assert isinstance(mid, str)
    assert mid.isdigit(), 'mid 应是数字字符串'

    client = AsyncClient(**api.dft_client_settings)
    ps = 30 # 每页视频数，最小 1，最大 50，默认 30
    order = 'pubdate'# 默认为pubdate 最新发布：pubdate 最多播放：click 最多收藏：stow
    keyword = '' # 搜索关键词
    bv_ids = []
    pn = 1
    up_name, total_size, bv_ids_page = await api.get_up_info(client, mid, pn, ps, order, keyword)
    bv_ids += bv_ids_page
    print(f'{mid} {up_name} 共 {total_size} 个视频... （第 1 页）')
    while pn < total_size / ps:
        pn += 1
        print(f'获取第 {pn} 页 (10s...)')
        await asyncio.sleep(10)
        _, _, bv_ids_page = await api.get_up_info(client, mid, pn, ps, order, keyword)
        bv_ids += bv_ids_page

    print(mid, up_name, total_size)
    await client.aclose()
    assert len(bv_ids) == len(set(bv_ids)), '有重复的 bv_id'
    assert total_size == len(bv_ids), '视频总数不匹配'
    filepath = f'bvids/by-up_videos/mid-{mid}-{int(time.time())}.txt'
    os.makedirs(os.path.dirname(filepath), exist_ok=True)
    abs_filepath = os.path.abspath(filepath)
    with open(abs_filepath, 'w', encoding='utf-8') as f:
        for bv_id in bv_ids:
            f.write(f'{bv_id}' + '\n')
    print(f'已保存 {len(bv_ids)} 个 bvid 到 {abs_filepath}')
    return Path(abs_filepath)


async def main():
    args = arg_parse()
    if args.by_ranking:
        by_raning(args.ranking_rid)
    if args.by_up_videos:
        assert args.up_videos_mid is not None, 'up_videos_mid 不能为空'
        await by_up_videos(args.up_videos_mid)

if __name__ == '__main__':
    asyncio.run(main())