from auto_media_publisher.uploaders.uploader_douyin import UploaderDouyin
from auto_media_publisher.uploaders.uploader_kuaishou import UploaderKuaishou
from auto_media_publisher.uploaders.uploader_shipinhao import UploaderShipinhao
from auto_media_publisher.uploaders.uploader_tiktok import UploaderTiktok
from auto_media_publisher.uploaders.uploader_bilibili import UploaderBilibili
from auto_media_publisher.uploaders.uploader_baijiahao import UploaderBaijiahao
from auto_media_publisher.uploaders.uploader_xiaohongshu import UploaderXiaohongshu
from auto_media_publisher.uploaders.uploader_twitterX import UploaderTwitterX

platform_map = {
    "douyin": UploaderDouyin,
    "kuaishou": UploaderKuaishou,
    "shipinhao": UploaderShipinhao,
    "xiaohongshu": UploaderXiaohongshu,
    "bilibili": UploaderBilibili,
    "baijiahao": UploaderBaijiahao,
    "tiktok": UploaderTiktok,
    "twitterX": UploaderTwitterX,
}


async def create_media_uploader(user:str,platform:str,proxy:str):
    if platform in platform_map:
        UploaderClass = platform_map[platform]
        return UploaderClass.create_sub_class(user,platform,proxy)
    else:
        return None