mirror of
https://github.com/lanyeeee/bilibili-video-downloader.git
synced 2026-09-08 17:16:52 +08:00
feat: 添加配置文件
This commit is contained in:
Generated
+1
@@ -268,6 +268,7 @@ name = "bilibili-video-downloader"
|
|||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
|
"parking_lot",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"specta",
|
"specta",
|
||||||
|
|||||||
@@ -28,6 +28,7 @@ tauri-specta = { version = "=2.0.0-rc.20", features = ["derive", "typescript"] }
|
|||||||
specta-typescript = { version = "=0.0.7" }
|
specta-typescript = { version = "=0.0.7" }
|
||||||
|
|
||||||
anyhow = { version = "1.0.98" }
|
anyhow = { version = "1.0.98" }
|
||||||
|
parking_lot = { version = "0.12.4", features = ["send_guard"] }
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
strip = true
|
strip = true
|
||||||
|
|||||||
@@ -1,5 +1,16 @@
|
|||||||
|
use parking_lot::RwLock;
|
||||||
|
|
||||||
|
use crate::config::Config;
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
#[specta::specta]
|
#[specta::specta]
|
||||||
pub fn greet(name: &str) -> String {
|
pub fn greet(name: &str) -> String {
|
||||||
format!("Hello, {}! You've been greeted from Rust!", name)
|
format!("Hello, {}! You've been greeted from Rust!", name)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tauri::command(async)]
|
||||||
|
#[specta::specta]
|
||||||
|
#[allow(clippy::needless_pass_by_value)]
|
||||||
|
pub fn get_config(config: tauri::State<RwLock<Config>>) -> Config {
|
||||||
|
config.read().clone()
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,69 @@
|
|||||||
|
use std::path::{Path, PathBuf};
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use specta::Type;
|
||||||
|
use tauri::{AppHandle, Manager};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize, Type)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct Config {
|
||||||
|
pub download_dir: PathBuf,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Config {
|
||||||
|
pub fn new(app: &AppHandle) -> anyhow::Result<Config> {
|
||||||
|
let app_data_dir = app.path().app_data_dir()?;
|
||||||
|
let config_path = app_data_dir.join("config.json");
|
||||||
|
|
||||||
|
let config = if config_path.exists() {
|
||||||
|
let config_string = std::fs::read_to_string(config_path)?;
|
||||||
|
match serde_json::from_str(&config_string) {
|
||||||
|
// 如果能够直接解析为Config,则直接返回
|
||||||
|
Ok(config) => config,
|
||||||
|
// 否则,将默认配置与文件中已有的配置合并
|
||||||
|
// 以免新版本添加了新的配置项,用户升级到新版本后,所有配置项都被重置
|
||||||
|
Err(_) => Config::merge_config(&config_string, &app_data_dir),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Config::default(&app_data_dir)
|
||||||
|
};
|
||||||
|
config.save(app)?;
|
||||||
|
Ok(config)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn save(&self, app: &AppHandle) -> anyhow::Result<()> {
|
||||||
|
let app_data_dir = app.path().app_data_dir()?;
|
||||||
|
let config_path = app_data_dir.join("config.json");
|
||||||
|
let config_string = serde_json::to_string_pretty(self)?;
|
||||||
|
std::fs::write(config_path, config_string)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn merge_config(config_string: &str, app_data_dir: &Path) -> Config {
|
||||||
|
let Ok(mut json_value) = serde_json::from_str::<serde_json::Value>(config_string) else {
|
||||||
|
return Config::default(app_data_dir);
|
||||||
|
};
|
||||||
|
let serde_json::Value::Object(ref mut map) = json_value else {
|
||||||
|
return Config::default(app_data_dir);
|
||||||
|
};
|
||||||
|
let Ok(default_config_value) = serde_json::to_value(Config::default(app_data_dir)) else {
|
||||||
|
return Config::default(app_data_dir);
|
||||||
|
};
|
||||||
|
let serde_json::Value::Object(default_map) = default_config_value else {
|
||||||
|
return Config::default(app_data_dir);
|
||||||
|
};
|
||||||
|
for (key, value) in default_map {
|
||||||
|
map.entry(key).or_insert(value);
|
||||||
|
}
|
||||||
|
let Ok(config) = serde_json::from_value(json_value) else {
|
||||||
|
return Config::default(app_data_dir);
|
||||||
|
};
|
||||||
|
config
|
||||||
|
}
|
||||||
|
|
||||||
|
fn default(app_data_dir: &Path) -> Config {
|
||||||
|
Config {
|
||||||
|
download_dir: app_data_dir.join("视频下载"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
+19
-4
@@ -1,10 +1,13 @@
|
|||||||
mod commands;
|
mod commands;
|
||||||
|
mod config;
|
||||||
mod extensions;
|
mod extensions;
|
||||||
mod utils;
|
mod utils;
|
||||||
|
|
||||||
use tauri::Wry;
|
use anyhow::Context;
|
||||||
|
use commands::*;
|
||||||
use crate::commands::*;
|
use config::Config;
|
||||||
|
use parking_lot::RwLock;
|
||||||
|
use tauri::{Manager, Wry};
|
||||||
|
|
||||||
fn generate_context() -> tauri::Context<Wry> {
|
fn generate_context() -> tauri::Context<Wry> {
|
||||||
tauri::generate_context!()
|
tauri::generate_context!()
|
||||||
@@ -13,7 +16,7 @@ fn generate_context() -> tauri::Context<Wry> {
|
|||||||
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
let builder = tauri_specta::Builder::<Wry>::new()
|
let builder = tauri_specta::Builder::<Wry>::new()
|
||||||
.commands(tauri_specta::collect_commands![greet])
|
.commands(tauri_specta::collect_commands![greet, get_config])
|
||||||
.events(tauri_specta::collect_events![]);
|
.events(tauri_specta::collect_events![]);
|
||||||
|
|
||||||
#[cfg(debug_assertions)]
|
#[cfg(debug_assertions)]
|
||||||
@@ -32,6 +35,18 @@ pub fn run() {
|
|||||||
.invoke_handler(builder.invoke_handler())
|
.invoke_handler(builder.invoke_handler())
|
||||||
.setup(move |app| {
|
.setup(move |app| {
|
||||||
builder.mount_events(app);
|
builder.mount_events(app);
|
||||||
|
|
||||||
|
let app_data_dir = app
|
||||||
|
.path()
|
||||||
|
.app_data_dir()
|
||||||
|
.context("获取app_data_dir目录失败")?;
|
||||||
|
|
||||||
|
std::fs::create_dir_all(&app_data_dir)
|
||||||
|
.context(format!("创建app_data_dir目录`{app_data_dir:?}`失败"))?;
|
||||||
|
|
||||||
|
let config = RwLock::new(Config::new(app.handle())?);
|
||||||
|
app.manage(config);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
})
|
})
|
||||||
.run(generate_context())
|
.run(generate_context())
|
||||||
|
|||||||
+4
-3
@@ -1,6 +1,6 @@
|
|||||||
<script setup lang="ts">
|
<script setup lang="ts">
|
||||||
import { ref } from 'vue'
|
import { ref } from 'vue'
|
||||||
import {commands} from "./bindings.ts";
|
import { commands } from './bindings.ts'
|
||||||
|
|
||||||
const greetMsg = ref('')
|
const greetMsg = ref('')
|
||||||
const name = ref('')
|
const name = ref('')
|
||||||
@@ -10,8 +10,9 @@ async function greet() {
|
|||||||
greetMsg.value = await commands.greet(name.value)
|
greetMsg.value = await commands.greet(name.value)
|
||||||
}
|
}
|
||||||
|
|
||||||
function test() {
|
async function test() {
|
||||||
console.log('test')
|
const result = await commands.getConfig()
|
||||||
|
console.log(result)
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
|
|||||||
+4
-1
@@ -7,6 +7,9 @@
|
|||||||
export const commands = {
|
export const commands = {
|
||||||
async greet(name: string) : Promise<string> {
|
async greet(name: string) : Promise<string> {
|
||||||
return await TAURI_INVOKE("greet", { name });
|
return await TAURI_INVOKE("greet", { name });
|
||||||
|
},
|
||||||
|
async getConfig() : Promise<Config> {
|
||||||
|
return await TAURI_INVOKE("get_config");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -20,7 +23,7 @@ async greet(name: string) : Promise<string> {
|
|||||||
|
|
||||||
/** user-defined types **/
|
/** user-defined types **/
|
||||||
|
|
||||||
|
export type Config = { downloadDir: string }
|
||||||
|
|
||||||
/** tauri-specta globals **/
|
/** tauri-specta globals **/
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user