91 lines
2.5 KiB
Rust
91 lines
2.5 KiB
Rust
use crate::utils::{dirs, tmpl};
|
|
use chrono::Local;
|
|
use log::LevelFilter;
|
|
use log4rs::append::console::ConsoleAppender;
|
|
use log4rs::append::file::FileAppender;
|
|
use log4rs::config::{Appender, Config, Root};
|
|
use log4rs::encode::pattern::PatternEncoder;
|
|
use std::fs;
|
|
use std::io::Write;
|
|
use std::path::PathBuf;
|
|
use tauri::PackageInfo;
|
|
|
|
/// initialize this instance's log file
|
|
fn init_log(log_dir: &PathBuf) {
|
|
let local_time = Local::now().format("%Y-%m-%d-%H%M%S").to_string();
|
|
let log_file = format!("{}.log", local_time);
|
|
let log_file = log_dir.join(log_file);
|
|
|
|
let time_format = "{d(%Y-%m-%d %H:%M:%S)} - {m}{n}";
|
|
let stdout = ConsoleAppender::builder()
|
|
.encoder(Box::new(PatternEncoder::new(time_format)))
|
|
.build();
|
|
let tofile = FileAppender::builder()
|
|
.encoder(Box::new(PatternEncoder::new(time_format)))
|
|
.build(log_file)
|
|
.unwrap();
|
|
|
|
let config = Config::builder()
|
|
.appender(Appender::builder().build("stdout", Box::new(stdout)))
|
|
.appender(Appender::builder().build("file", Box::new(tofile)))
|
|
.build(
|
|
Root::builder()
|
|
.appenders(["stdout", "file"])
|
|
.build(LevelFilter::Info),
|
|
)
|
|
.unwrap();
|
|
|
|
log4rs::init_config(config).unwrap();
|
|
}
|
|
|
|
/// Initialize all the files from resources
|
|
fn init_config(app_dir: &PathBuf) -> std::io::Result<()> {
|
|
// target path
|
|
let clash_path = app_dir.join("config.yaml");
|
|
let verge_path = app_dir.join("verge.yaml");
|
|
let profile_path = app_dir.join("profiles.yaml");
|
|
|
|
if !clash_path.exists() {
|
|
fs::File::create(clash_path)?.write(tmpl::CLASH_CONFIG)?;
|
|
}
|
|
if !verge_path.exists() {
|
|
fs::File::create(verge_path)?.write(tmpl::VERGE_CONFIG)?;
|
|
}
|
|
if !profile_path.exists() {
|
|
fs::File::create(profile_path)?.write(tmpl::PROFILES_CONFIG)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
/// initialize app
|
|
pub fn init_app(package_info: &PackageInfo) {
|
|
// create app dir
|
|
let app_dir = dirs::app_home_dir();
|
|
let log_dir = dirs::app_logs_dir();
|
|
let profiles_dir = dirs::app_profiles_dir();
|
|
|
|
let res_dir = dirs::app_resources_dir(package_info);
|
|
|
|
if !app_dir.exists() {
|
|
fs::create_dir_all(&app_dir).unwrap();
|
|
}
|
|
if !log_dir.exists() {
|
|
fs::create_dir_all(&log_dir).unwrap();
|
|
}
|
|
if !profiles_dir.exists() {
|
|
fs::create_dir_all(&profiles_dir).unwrap();
|
|
}
|
|
|
|
init_log(&log_dir);
|
|
if let Err(err) = init_config(&app_dir) {
|
|
log::error!("{err}");
|
|
}
|
|
|
|
// copy the resource file
|
|
let mmdb_path = app_dir.join("Country.mmdb");
|
|
let mmdb_tmpl = res_dir.join("Country.mmdb");
|
|
if !mmdb_path.exists() && mmdb_tmpl.exists() {
|
|
fs::copy(mmdb_tmpl, mmdb_path).unwrap();
|
|
}
|
|
}
|