from textual.app import App
from textual.app import ComposeResult
from textual.binding import Binding
from textual.containers import HorizontalScroll
from textual.containers import ScrollableContainer
from textual.widgets import Button
from textual.widgets import Footer
from textual.widgets import Header
from textual.widgets import Input
from textual.widgets import Label
from textual.widgets import Log

from source import Batch
from source import Settings
from source import XHS


def example():
    """通过代码设置参数，适合二次开发"""
    # 测试链接
    error_demo = "https://github.com/JoeanAmier/XHS_Downloader"
    image_demo = "https://www.xiaohongshu.com/explore/63b275a30000000019020185"
    video_demo = "https://www.xiaohongshu.com/explore/64edb460000000001f03cadc"
    # 实例对象
    path = ""  # 作品下载储存根路径，默认值：当前路径
    folder = "Download"  # 作品下载文件夹名称（自动创建），默认值：Download
    cookie = ""  # 小红书网页版 Cookie
    proxies = None  # 网络代理
    timeout = 5  # 网络请求超时限制，默认值：10
    chunk = 1024 * 1024  # 下载文件时，每次从服务器获取的数据块大小，单位字节
    xhs = XHS(
        path=path,
        folder=folder,
        cookie=cookie,
        proxies=proxies,
        timeout=timeout,
        chunk=chunk, )  # 使用自定义参数
    # xhs = XHS()  # 使用默认参数
    download = True  # 是否下载作品文件
    # 返回作品详细信息，包括下载地址
    print(xhs.extract(error_demo))  # 获取数据失败时返回空字典
    print(xhs.extract(image_demo, download=download))
    print(xhs.extract(video_demo, download=download))


class XHSDownloader(App):
    CSS_PATH = "static/XHS_Downloader.tcss"
    BINDINGS = [
        Binding(key="q", action="quit", description="退出程序"),
        ("d", "toggle_dark", "切换主题"),
    ]
    APP = XHS(**Settings().run())
    Batch = Batch()

    def compose(self) -> ComposeResult:
        yield Header()
        yield ScrollableContainer(Label("请输入小红书图文/视频作品链接："),
                                  Input(placeholder="URL"),
                                  HorizontalScroll(Button("下载无水印图片/视频", id="solo"),
                                                   Button("读取 xhs.txt 文件并批量下载作品", id="batch"),
                                                   Button("清空输入框", id="reset"), ))
        yield Log(auto_scroll=True)
        yield Footer()

    def on_mount(self) -> None:
        self.title = "小红书作品采集工具"

    def on_button_pressed(self, event: Button.Pressed) -> None:
        if event.button.id == "solo":
            self.solo()
        elif event.button.id == "batch":
            self.batch()
        elif event.button.id == "reset":
            self.query_one(Input).value = ""

    def solo(self):
        url = self.query_one(Input).value
        log = self.query_one(Log)
        log.write_line(f"当前作品链接: {url}")
        self.APP.extract(url, True, log)

    def batch(self):
        urls = self.Batch.read_txt()
        log = self.query_one(Log)
        if not urls:
            log.write_line("未检测到 xhs.txt 文件 或者 该文件为空！")
        for url in urls:
            log.write_line(f"当前作品链接: {url}")
            self.APP.extract(url, True, log)


if __name__ == '__main__':
    # example()
    app = XHSDownloader()
    app.run()