[feat] set video config

This commit is contained in:
2025-03-27 15:29:12 +08:00
parent 93cda8dc85
commit 114def0b96
4 changed files with 126 additions and 173 deletions

View File

@@ -1,6 +1,7 @@
use anyhow::Result;
use base64::engine::general_purpose::STANDARD;
use base64::Engine;
use regex::Regex;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::collections::HashMap;
@@ -82,40 +83,40 @@ pub struct VideoConfig {
impl Default for VideoConfig {
fn default() -> Self {
VideoConfig {
version: String::new(),
vendor_id: String::new(),
device_id: String::new(),
cpu_level: String::new(),
gpu_mem_level: String::new(),
gpu_level: String::new(),
knowndevice: String::new(),
defaultres: String::new(),
defaultresheight: String::new(),
refreshrate_numerator: String::new(),
refreshrate_denominator: String::new(),
fullscreen: String::new(),
coop_fullscreen: String::new(),
nowindowborder: String::new(),
mat_vsync: String::new(),
fullscreen_min_on_focus_loss: String::new(),
high_dpi: String::new(),
auto_config: String::new(),
shaderquality: String::new(),
r_texturefilteringquality: String::new(),
msaa_samples: String::new(),
r_csgo_cmaa_enable: String::new(),
videocfg_shadow_quality: String::new(),
videocfg_dynamic_shadows: String::new(),
videocfg_texture_detail: String::new(),
videocfg_particle_detail: String::new(),
videocfg_ao_detail: String::new(),
videocfg_hdr_detail: String::new(),
videocfg_fsr_detail: String::new(),
monitor_index: String::new(),
r_low_latency: String::new(),
aspectratiomode: String::new(),
}
VideoConfig {
version: "15".to_string(),
vendor_id: "0".to_string(),
device_id: "0".to_string(),
cpu_level: "3".to_string(),
gpu_mem_level: "3".to_string(),
gpu_level: "3".to_string(),
knowndevice: "0".to_string(),
defaultres: "1920".to_string(),
defaultresheight: "1080".to_string(),
refreshrate_numerator: "144".to_string(),
refreshrate_denominator: "1".to_string(),
fullscreen: "1".to_string(),
coop_fullscreen: "0".to_string(),
nowindowborder: "1".to_string(),
mat_vsync: "0".to_string(),
fullscreen_min_on_focus_loss: "1".to_string(),
high_dpi: "0".to_string(),
auto_config: "2".to_string(),
shaderquality: "0".to_string(),
r_texturefilteringquality: "3".to_string(),
msaa_samples: "2".to_string(),
r_csgo_cmaa_enable: "0".to_string(),
videocfg_shadow_quality: "0".to_string(),
videocfg_dynamic_shadows: "1".to_string(),
videocfg_texture_detail: "1".to_string(),
videocfg_particle_detail: "0".to_string(),
videocfg_ao_detail: "0".to_string(),
videocfg_hdr_detail: "3".to_string(),
videocfg_fsr_detail: "0".to_string(),
monitor_index: "0".to_string(),
r_low_latency: "1".to_string(),
aspectratiomode: "0".to_string(),
}
}
}
@@ -226,6 +227,9 @@ pub fn parse_local_users(steam_dir: &str) -> Result<Vec<LocalUser>> {
let json_data = super::parse::to_json(&data);
let kv: HashMap<String, Value> = serde_json::from_str(&json_data)?;
// 剥离顶层 UserLocalConfigStore
let kv = kv.get("UserLocalConfigStore").and_then(|v| v.as_object()).unwrap();
// 获取 friends 节点
let friends = kv.get("friends").and_then(|v| v.as_object());
if friends.is_none() {
@@ -472,108 +476,50 @@ pub fn get_cs2_video(file_path: &str) -> Result<VideoConfig> {
}
pub fn set_cs2_video(file_path: &str, data: VideoConfig) -> Result<()> {
// TODO: no kv
let mut kv = HashMap::new();
kv.insert("version".to_string(), data.version);
kv.insert("vendor_id".to_string(), data.vendor_id);
kv.insert("device_id".to_string(), data.device_id);
kv.insert("setting.cpu_level".to_string(), data.cpu_level);
kv.insert(
"setting.gpu_mem_level".to_string(),
data.gpu_mem_level,
);
kv.insert("setting.gpu_level".to_string(), data.gpu_level);
kv.insert("setting.knowndevice".to_string(), data.knowndevice);
kv.insert("setting.defaultres".to_string(), data.defaultres);
kv.insert(
"setting.defaultresheight".to_string(),
data.defaultresheight,
);
kv.insert(
"setting.refreshrate_numerator".to_string(),
data.refreshrate_numerator,
);
kv.insert(
"setting.refreshrate_denominator".to_string(),
data.refreshrate_denominator,
);
kv.insert("setting.fullscreen".to_string(), data.fullscreen);
kv.insert(
"setting.coop_fullscreen".to_string(),
data.coop_fullscreen,
);
kv.insert(
"setting.nowindowborder".to_string(),
data.nowindowborder,
);
kv.insert("setting.mat_vsync".to_string(), data.mat_vsync);
kv.insert(
"setting.fullscreen_min_on_focus_loss".to_string(),
data.fullscreen_min_on_focus_loss,
);
kv.insert("setting.high_dpi".to_string(), data.high_dpi);
kv.insert("auto_config".to_string(), data.auto_config);
kv.insert(
"setting.shaderquality".to_string(),
data.shaderquality,
);
kv.insert(
"setting.r_texturefilteringquality".to_string(),
data.r_texturefilteringquality,
);
kv.insert(
"setting.msaa_samples".to_string(),
data.msaa_samples,
);
kv.insert(
"setting.r_csgo_cmaa_enable".to_string(),
data.r_csgo_cmaa_enable,
);
kv.insert(
"setting.videocfg_shadow_quality".to_string(),
data.videocfg_shadow_quality,
);
kv.insert(
"setting.videocfg_dynamic_shadows".to_string(),
data.videocfg_dynamic_shadows,
);
kv.insert(
"setting.videocfg_texture_detail".to_string(),
data.videocfg_texture_detail,
);
kv.insert(
"setting.videocfg_particle_detail".to_string(),
data.videocfg_particle_detail,
);
kv.insert(
"setting.videocfg_ao_detail".to_string(),
data.videocfg_ao_detail,
);
kv.insert(
"setting.videocfg_hdr_detail".to_string(),
data.videocfg_hdr_detail,
);
kv.insert(
"setting.videocfg_fsr_detail".to_string(),
data.videocfg_fsr_detail,
);
kv.insert(
"setting.monitor_index".to_string(),
data.monitor_index,
);
kv.insert(
"setting.r_low_latency".to_string(),
data.r_low_latency,
);
kv.insert(
"setting.aspectratiomode".to_string(),
data.aspectratiomode,
);
// 读取文件内容
let file_content = fs::read_to_string(file_path)?;
let json_data = serde_json::to_string(&kv)?;
let vdf_data = super::parse::to_vdf(&json_data);
// println!("{}", vdf_data);
fs::write(file_path, vdf_data)?;
// 定义正则表达式匹配模式
let re = Regex::new(r#""(setting\.\w+)"\s+"\d+""#).unwrap();
// 替换字段值
let updated_content = re.replace_all(&file_content, |caps: &regex::Captures| {
let key = &caps[1]; // 捕获的键名
let value = match key {
"setting.cpu_level" => &data.cpu_level,
"setting.gpu_mem_level" => &data.gpu_mem_level,
"setting.gpu_level" => &data.gpu_level,
"setting.knowndevice" => &data.knowndevice,
"setting.defaultres" => &data.defaultres,
"setting.defaultresheight" => &data.defaultresheight,
"setting.refreshrate_numerator" => &data.refreshrate_numerator,
"setting.refreshrate_denominator" => &data.refreshrate_denominator,
"setting.fullscreen" => &data.fullscreen,
"setting.coop_fullscreen" => &data.coop_fullscreen,
"setting.nowindowborder" => &data.nowindowborder,
"setting.mat_vsync" => &data.mat_vsync,
"setting.fullscreen_min_on_focus_loss" => &data.fullscreen_min_on_focus_loss,
"setting.high_dpi" => &data.high_dpi,
"setting.shaderquality" => &data.shaderquality,
"setting.r_texturefilteringquality" => &data.r_texturefilteringquality,
"setting.msaa_samples" => &data.msaa_samples,
"setting.r_csgo_cmaa_enable" => &data.r_csgo_cmaa_enable,
"setting.videocfg_shadow_quality" => &data.videocfg_shadow_quality,
"setting.videocfg_dynamic_shadows" => &data.videocfg_dynamic_shadows,
"setting.videocfg_texture_detail" => &data.videocfg_texture_detail,
"setting.videocfg_particle_detail" => &data.videocfg_particle_detail,
"setting.videocfg_ao_detail" => &data.videocfg_ao_detail,
"setting.videocfg_hdr_detail" => &data.videocfg_hdr_detail,
"setting.videocfg_fsr_detail" => &data.videocfg_fsr_detail,
"setting.monitor_index" => &data.monitor_index,
"setting.r_low_latency" => &data.r_low_latency,
"setting.aspectratiomode" => &data.aspectratiomode,
_ => "", // 默认情况
};
format!(r#""{}" "{}""#, key, value)
});
fs::write(file_path, updated_content.as_ref())?;
Ok(())
}
@@ -599,6 +545,11 @@ mod tests {
fn test_set_cs2_video() {
let manifest_dir = env!("CARGO_MANIFEST_DIR");
let file_path = format!("{}/temp/cs2_video.txt", manifest_dir);
fs::copy(
format!("{}/src/vdf/tests/cs2_video.txt", manifest_dir),
file_path.clone(),
)
.unwrap();
let video_config = VideoConfig::default();
set_cs2_video(&file_path, video_config).unwrap();
}