from app.api.tiktok_scraper import Scraper as Tiktok_Scraper
from app.core.scraper_config import ScraperConfig

tiktok_scraper = Tiktok_Scraper(quiet=True)


class Tiktok(ScraperConfig):
    def __init__(self, url):
        super().__init__()
        self.url = url
        self.set_sauce(url)

    async def download_or_extract(self):
        media = await tiktok_scraper.hybrid_parsing(self.url)
        if not media or "status" not in media or media["status"] == "failed":
            return
        if "video_data" in media:
            self.link = media["video_data"]["nwm_video_url_HQ"]
            self.thumb = media["cover_data"]["dynamic_cover"]["url_list"][0]
            self.video = self.success = True
        if "image_data" in media:
            self.link = media["image_data"]["no_watermark_image_list"]
            self.group = self.success = True