trnovel 0.18.2

#![allow(clippy::needless_update)]

use app::App;
use clap::{Parser, Subcommand};
use ratatui_kit::{ElementExt, RunOptions, element};
use std::{env, ffi::OsString, fmt::Debug, path::PathBuf};

pub mod app;
pub mod browser_assist;
pub mod cache;
pub mod components;
pub mod config;
pub mod doctor;
pub mod errors;
pub mod file_list;
pub mod gen_fontmap;
pub mod hooks;
pub mod import;
pub mod keymap;
pub mod login;
mod migration;
pub mod novel;
pub mod pages;
pub mod paths;
pub mod state;
pub mod theme;
#[cfg(feature = "tts")]
pub mod tts;
pub mod utils;

pub use cache::*;
pub use errors::Result;

use crate::app::AppProps;

pub async fn run() -> Result<()> {
    try_run(env::args()).await
}

pub async fn try_run<I, A>(args: I) -> Result<()>
where
    I: IntoIterator<Item = A> + Debug,
    A: Into<OsString> + Clone,
{
    #[cfg(feature = "tts")]
    let (listening, listening_task) = tts::Handle::new();
    let result = tokio::select! {
        result = try_run_inner(args, #[cfg(feature = "tts")] listening.clone()) => result,
        signal = tokio::signal::ctrl_c() => {
            signal?;
            Ok(())
        }
    };
    #[cfg(feature = "tts")]
    {
        listening.shutdown().await;
        let _ = listening_task.await;
    }
    parse_book_source::shutdown_render_pool().await;
    result
}

async fn try_run_inner<I, A>(args: I, #[cfg(feature = "tts")] listening: tts::Handle) -> Result<()>
where
    I: IntoIterator<Item = A> + Debug,
    A: Into<OsString> + Clone,
{
    let trnovel = TRNovel::parse_from(args);
    migration::startup().await;

    #[cfg(feature = "tts")]
    if let Some(Commands::Tts { command }) = &trnovel.subcommand {
        tts::cli::run(command, trnovel.tts_program.as_deref()).await?;
        return Ok(());
    }

    if let Some(Commands::Clear) = trnovel.subcommand {
        migration::clear(&paths::AppPaths::user_default()?)?;
        return Ok(());
    }

    // 书源体检:非 TUI,跑全流程后打印 ✓/✗ 列表并退出。
    if let Some(Commands::Doctor { path }) = &trnovel.subcommand {
        doctor::run(path).await;
        return Ok(());
    }

    // 导入书源:非 TUI,把书源 JSON(文件/URL)写入 ~/.trnovel 后退出。
    if let Some(Commands::Import { source }) = &trnovel.subcommand {
        import::run(source).await;
        return Ok(());
    }

    // 生成字体反爬映射表:非 TUI,字形匹配后写出 {码点:真字} JSON 退出。
    if let Some(Commands::GenFontmap {
        font,
        out,
        base_font,
    }) = &trnovel.subcommand
    {
        gen_fontmap::run(font, out, base_font.as_deref()).await;
        return Ok(());
    }

    #[cfg(feature = "tts")]
    listening.set_path(trnovel.tts_program.clone());
    let props = AppProps {
        trnovel,
        #[cfg(feature = "tts")]
        listening,
    };

    // 鼠标捕获:组件侧的 use_pointer/use_click/use_hover 都要求入口启用;
    // 按住 Shift 时多数终端仍可走本地选择。
    element!(App(..props))
        .fullscreen_with(RunOptions { mouse: true })
        .await?;

    Ok(())
}

#[derive(Parser, Debug, Clone, Hash, PartialEq, Eq)]
#[command(
    author,
    version,
    about = r#"
  _______ _____  _   _                _ 
 |__   __|  __ \| \ | |              | |
    | |  | |__) |  \| | _____   _____| |
    | |  |  _  /| . ` |/ _ \ \ / / _ \ |
    | |  | | \ \| |\  | (_) \ V /  __/ |
    |_|  |_|  \_\_| \_|\___/ \_/ \___|_|
                                            
  终端小说阅读器 (Terminal reader for novel)
  ==========================================

  TRNovel 是一个终端小说阅读器，支持以下功能。
    - 本地小说
    - 网络小说
    - 历史记录
    - 主题设置

  GitHub: https://github.com/yexiyue/trnovel
  
  如果您觉得还不错，请考虑给项目点个 star，谢谢！
"#
)]
pub struct TRNovel {
    /// 指定独立听书程序路径；无效时直接报错，不改用其他程序。
    #[cfg(feature = "tts")]
    #[arg(long, global = true)]
    pub tts_program: Option<PathBuf>,
    #[command(subcommand)]
    pub subcommand: Option<Commands>,
}

#[derive(Debug, Subcommand, Clone, Hash, PartialEq, Eq)]
pub enum Commands {
    /// 不进入 TUI，查询和快速配置听书
    #[cfg(feature = "tts")]
    Tts {
        #[command(subcommand)]
        command: tts::cli::TtsCommand,
    },
    /// 快速模式，接着上一次阅读的位置继续阅读
    #[command(short_flag = 'q')]
    Quick,

    /// 清空历史记录和小说缓存
    #[command(short_flag = 'c')]
    Clear,

    /// 网络模式，使用网络小说源
    #[command(short_flag = 'n')]
    Network,

    /// 本地小说
    #[command(short_flag = 'l')]
    Local {
        /// 小说文件夹路径
        path: Option<PathBuf>,
    },

    /// 历史记录模式，查看阅读记录
    #[command(short_flag = 'H')]
    History,

    /// 体检书源：全流程验证书源 JSON,逐项报告 ✓/✗(用于校验 AI 生成的书源)
    #[command(short_flag = 'd')]
    Doctor {
        /// 书源 JSON 文件路径
        path: PathBuf,
    },

    /// 导入书源：把书源 JSON(本地文件或 URL)写入 ~/.trnovel/data/,使其在网络小说里可用
    #[command(short_flag = 'i')]
    Import {
        /// 书源 JSON 文件路径或 URL
        source: String,
    },

    /// 生成字体反爬映射表：对加密字体做字形匹配,输出 {码点:真字} JSON(可内联进书源 fontMap)
    GenFontmap {
        /// 加密字体的 URL 或本地路径(woff2/ttf/otf)
        font: String,
        /// 输出映射表 JSON 文件路径
        #[arg(short, long)]
        out: PathBuf,
        /// 基准中文字体路径(缺省自动下载思源黑体 Noto Sans CJK SC)
        #[arg(short, long)]
        base_font: Option<PathBuf>,
    },
}