113 lines
2.9 KiB
Rust
113 lines
2.9 KiB
Rust
use anyhow::Result;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::utils::filenamify::filenamify;
|
|
|
|
/// NFO 文件使用的时间类型
|
|
#[derive(Serialize, Deserialize, Default, Clone, Copy)]
|
|
#[serde(rename_all = "lowercase")]
|
|
pub enum NFOTimeType {
|
|
#[default]
|
|
FavTime,
|
|
PubTime,
|
|
}
|
|
|
|
/// 并发下载相关的配置
|
|
#[derive(Serialize, Deserialize, Clone)]
|
|
pub struct ConcurrentLimit {
|
|
pub video: usize,
|
|
pub page: usize,
|
|
pub rate_limit: Option<RateLimit>,
|
|
#[serde(default)]
|
|
pub download: ConcurrentDownloadLimit,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Clone)]
|
|
pub struct ConcurrentDownloadLimit {
|
|
pub enable: bool,
|
|
pub concurrency: usize,
|
|
pub threshold: u64,
|
|
}
|
|
|
|
impl Default for ConcurrentDownloadLimit {
|
|
fn default() -> Self {
|
|
Self {
|
|
enable: true,
|
|
concurrency: 4,
|
|
threshold: 20 * (1 << 20), // 20 MB
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Clone)]
|
|
pub struct RateLimit {
|
|
pub limit: usize,
|
|
pub duration: u64,
|
|
}
|
|
|
|
impl Default for ConcurrentLimit {
|
|
fn default() -> Self {
|
|
Self {
|
|
video: 3,
|
|
page: 2,
|
|
// 默认的限速配置，每 250ms 允许请求 4 次
|
|
rate_limit: Some(RateLimit {
|
|
limit: 4,
|
|
duration: 250,
|
|
}),
|
|
download: ConcurrentDownloadLimit::default(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Clone, Default)]
|
|
pub struct SkipOption {
|
|
pub no_poster: bool,
|
|
pub no_video_nfo: bool,
|
|
pub no_upper: bool,
|
|
pub no_danmaku: bool,
|
|
pub no_subtitle: bool,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Clone)]
|
|
#[serde(untagged)]
|
|
pub enum Trigger {
|
|
Interval(u64),
|
|
Cron(String),
|
|
}
|
|
|
|
impl Default for Trigger {
|
|
fn default() -> Self {
|
|
Trigger::Interval(1200)
|
|
}
|
|
}
|
|
|
|
pub trait PathSafeTemplate {
|
|
fn path_safe_register(&mut self, name: &'static str, template: impl Into<String>) -> Result<()>;
|
|
fn path_safe_render(&self, name: &'static str, data: &serde_json::Value) -> Result<String>;
|
|
}
|
|
|
|
/// 通过将模板字符串中的分隔符替换为自定义的字符串，使得模板字符串中的分隔符得以保留
|
|
impl PathSafeTemplate for handlebars::Handlebars<'_> {
|
|
fn path_safe_register(&mut self, name: &'static str, template: impl Into<String>) -> Result<()> {
|
|
let template = template.into();
|
|
Ok(self.register_template_string(name, template.replace(std::path::MAIN_SEPARATOR_STR, "__SEP__"))?)
|
|
}
|
|
|
|
fn path_safe_render(&self, name: &'static str, data: &serde_json::Value) -> Result<String> {
|
|
Ok(filenamify(&self.render(name, data)?).replace("__SEP__", std::path::MAIN_SEPARATOR_STR))
|
|
}
|
|
}
|
|
|
|
pub fn default_favorite_path() -> String {
|
|
"收藏夹/{{name}}".to_owned()
|
|
}
|
|
|
|
pub fn default_collection_path() -> String {
|
|
"合集/{{name}}".to_owned()
|
|
}
|
|
|
|
pub fn default_submission_path() -> String {
|
|
"投稿/{{name}}".to_owned()
|
|
}
|