from pathlib import Path
from typing import List

from conf import BASE_DIR

SOCIAL_MEDIA_DOUYIN = "douyin"
SOCIAL_MEDIA_TENCENT = "tencent"
SOCIAL_MEDIA_TIKTOK = "tiktok"
SOCIAL_MEDIA_BILIBILI = "bilibili"
SOCIAL_MEDIA_KUAISHOU = "kuaishou"
SOCIAL_MEDIA_WANGYIHAO = 'wangyihao'
SOCIAL_MEDIA_XHS = 'xiaohongshu'    # 小红书最长标题是20
SOCIAL_MEDIA_TOUTIAO = 'toutiao'
SOCIAL_MEDIA_BAIJIAHAO = 'baijiahao'
SOCIAL_MEDIA_SOHU = 'sohu'  # 搜狐视频话题最多2个 标题最多100 标题不支持‍🩹


def get_supported_social_media() -> List[str]:
    return [SOCIAL_MEDIA_DOUYIN, SOCIAL_MEDIA_TENCENT, SOCIAL_MEDIA_TIKTOK, SOCIAL_MEDIA_KUAISHOU,
            SOCIAL_MEDIA_WANGYIHAO, SOCIAL_MEDIA_TOUTIAO, SOCIAL_MEDIA_BAIJIAHAO, SOCIAL_MEDIA_XHS,
            SOCIAL_MEDIA_SOHU]


def get_cli_action() -> List[str]:
    return ["upload", "login", "watch", "auto_login"]


async def set_init_script(context):
    stealth_js_path = Path(BASE_DIR / "utils/stealth.min.js")
    await context.add_init_script(path=stealth_js_path)
    return context