novel_cli/cmd/

mod.rs

pub mod bookshelf;
pub mod build;
pub mod check;
pub mod completions;
pub mod download;
pub mod epub;
pub mod info;
pub mod read;
pub mod real_cugan;
pub mod search;
pub mod sign;
pub mod template;
pub mod transform;
pub mod unzip;
pub mod update;
pub mod zip;

use std::collections::HashMap;
use std::fs::File;
use std::io::{self, Read, Seek, Write};
use std::path::Path;
use std::sync::Arc;
use std::{cmp, env, fs, process};

use ::zip::write::SimpleFileOptions;
use ::zip::{CompressionMethod, ZipArchive, ZipWriter};
use clap::ValueEnum;
use color_eyre::eyre::Result;
use fluent_templates::Loader;
use fluent_templates::fluent_bundle::FluentValue;
use novel_api::Client;
use ratatui::buffer::Buffer;
use ratatui::layout::{Rect, Size};
use ratatui::text::Text;
use ratatui::widgets::block::Title;
use ratatui::widgets::{Block, Paragraph, StatefulWidget, Widget, Wrap};
use strum::AsRefStr;
use tokio::signal;
use tui_widgets::scrollview::{ScrollView, ScrollViewState};
use url::Url;
use walkdir::DirEntry;

use crate::{LANG_ID, LOCALES, utils};

const DEFAULT_PROXY: &str = "http://127.0.0.1:8080";

const DEFAULT_PROXY_SURGE: &str = "http://127.0.0.1:6152";

#[must_use]
#[derive(Clone, PartialEq, ValueEnum, AsRefStr)]
pub enum Source {
    #[strum(serialize = "sfacg")]
    Sfacg,
    #[strum(serialize = "ciweimao")]
    Ciweimao,
    #[strum(serialize = "ciyuanji")]
    Ciyuanji,
}

#[must_use]
#[derive(Clone, PartialEq, ValueEnum, AsRefStr)]
pub enum Format {
    Pandoc,
    Mdbook,
}

#[must_use]
#[derive(Clone, PartialEq, ValueEnum, AsRefStr)]
pub enum Convert {
    S2T,
    T2S,
    JP2T2S,
    CUSTOM,
}

#[inline]
#[must_use]
fn default_cert_path() -> String {
    novel_api::home_dir_path()
        .unwrap()
        .join(".mitmproxy")
        .join("mitmproxy-ca-cert.pem")
        .display()
        .to_string()
}

fn set_options<T, E>(client: &mut T, proxy: &Option<Url>, no_proxy: &bool, cert: &Option<E>)
where
    T: Client,
    E: AsRef<Path>,
{
    if let Some(proxy) = proxy {
        client.proxy(proxy.clone());
    }

    if *no_proxy {
        client.no_proxy();
    }

    if let Some(cert) = cert {
        client.cert(cert.as_ref().to_path_buf())
    }
}

fn handle_ctrl_c<T>(client: &Arc<T>)
where
    T: Client + Send + Sync + 'static,
{
    let client = Arc::clone(client);

    tokio::spawn(async move {
        signal::ctrl_c().await.unwrap();

        tracing::warn!("Download terminated, login data will be saved");

        client.shutdown().await.unwrap();
        process::exit(128 + libc::SIGINT);
    });
}

fn cert_help_msg() -> String {
    let args = {
        let mut map = HashMap::new();
        map.insert(
            "cert_path".into(),
            FluentValue::String(default_cert_path().into()),
        );
        map
    };

    LOCALES.lookup_with_args(&LANG_ID, "cert", &args)
}

#[derive(Default, PartialEq)]
enum Mode {
    #[default]
    Running,
    Quit,
}

pub struct ScrollableParagraph<'a> {
    title: Option<Title<'a>>,
    text: Text<'a>,
}

impl<'a> ScrollableParagraph<'a> {
    pub fn new<T>(text: T) -> Self
    where
        T: Into<Text<'a>>,
    {
        ScrollableParagraph {
            title: None,
            text: text.into(),
        }
    }

    pub fn title<T>(self, title: T) -> Self
    where
        T: Into<Title<'a>>,
    {
        ScrollableParagraph {
            title: Some(title.into()),
            ..self
        }
    }
}

impl StatefulWidget for ScrollableParagraph<'_> {
    type State = ScrollViewState;

    fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
        let mut block = Block::bordered();
        if self.title.is_some() {
            block = block.title(self.title.as_ref().unwrap().clone());
        }

        let paragraph = Paragraph::new(self.text.clone()).wrap(Wrap { trim: false });
        let mut scroll_view = ScrollView::new(Size::new(area.width - 1, area.height));
        let mut block_area = block.inner(scroll_view.buf().area);

        let scroll_height = cmp::max(
            paragraph.line_count(block_area.width) as u16
                + (scroll_view.buf().area.height - block_area.height),
            area.height,
        );

        let scroll_width = if area.height >= scroll_height {
            // 不需要滚动条
            area.width
        } else {
            area.width - 1
        };

        scroll_view = ScrollView::new(Size::new(scroll_width, scroll_height));

        let scroll_view_buf = scroll_view.buf_mut();
        block_area = block.inner(scroll_view_buf.area);

        Widget::render(block, scroll_view_buf.area, scroll_view_buf);
        Widget::render(paragraph, block_area, scroll_view_buf);
        StatefulWidget::render(scroll_view, area, buf, state);
    }
}

fn unzip<T>(path: T) -> Result<()>
where
    T: AsRef<Path>,
{
    let path = path.as_ref();

    let output_dir = env::current_dir()?.join(path.file_stem().unwrap());
    if output_dir.try_exists()? {
        tracing::warn!("The epub output directory already exists and will be deleted");
        utils::remove_file_or_dir(&output_dir)?;
    }

    let file = File::open(path)?;
    let mut archive = ZipArchive::new(file)?;

    for i in 0..archive.len() {
        let mut file = archive.by_index(i)?;
        let outpath = match file.enclosed_name() {
            Some(path) => path.to_owned(),
            None => continue,
        };
        let outpath = output_dir.join(outpath);

        if (*file.name()).ends_with('/') {
            fs::create_dir_all(&outpath)?;
        } else {
            if let Some(p) = outpath.parent()
                && !p.try_exists()?
            {
                fs::create_dir_all(p)?;
            }
            let mut outfile = fs::File::create(&outpath)?;
            io::copy(&mut file, &mut outfile)?;
        }

        #[cfg(unix)]
        {
            use std::fs::Permissions;
            use std::os::unix::fs::PermissionsExt;

            if let Some(mode) = file.unix_mode() {
                fs::set_permissions(&outpath, Permissions::from_mode(mode))?;
            }
        }
    }

    Ok(())
}

fn zip_dir<T, E>(iter: &mut dyn Iterator<Item = DirEntry>, prefix: T, writer: E) -> Result<()>
where
    T: AsRef<Path>,
    E: Write + Seek,
{
    let mut zip = ZipWriter::new(writer);
    let options = SimpleFileOptions::default()
        .compression_method(CompressionMethod::Deflated)
        .compression_level(Some(9));

    let mut buffer = Vec::new();
    for entry in iter {
        let path = entry.path();
        let name = path.strip_prefix(prefix.as_ref())?;

        if path.is_file() {
            zip.start_file(name.to_str().unwrap(), options)?;
            let mut f = File::open(path)?;

            f.read_to_end(&mut buffer)?;
            zip.write_all(&buffer)?;
            buffer.clear();
        } else if !name.as_os_str().is_empty() {
            zip.add_directory(name.to_str().unwrap(), options)?;
        }
    }
    zip.finish()?;

    Ok(())
}