Ryan/drop-app
1
0
Fork 0
mirror of https://github.com/Drop-OSS/drop-app.git synced 2025-11-17 18:21:14 +10:00
Code Issues Packages Projects Releases Wiki Activity

chore: Add extract() function

Browse Source 
Signed-off-by: quexeky <git@quexeky.dev>
This commit is contained in:
quexeky
2025-05-13 14:36:47 +10:00
parent 67b6f2aa2e
commit 8f261a5dac
5 changed files with 149 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
src-tauri/src/cloud_saves/resolver.rs
View File

@ -1,6 +1,12 @@
use std::{fs::File, io::Write, path::{Component, PathBuf}};
use std::{
fs::{create_dir_all, File},
io::{Read, Write},
path::PathBuf, thread::{sleep}, time::Duration,
};
use super::{backup_manager::BackupHandler, conditions::Condition, metadata::GameFile, placeholder::*};
use super::{
backup_manager::BackupHandler, conditions::Condition, metadata::GameFile, placeholder::*,
};
use log::warn;
use rustix::path::Arg;
use tempfile::tempfile;
@ -62,6 +68,64 @@ pub fn resolve(meta: &mut CloudSaveMetadata) -> File {
tarball.into_inner().unwrap().finish().unwrap()
}
pub fn extract(file: PathBuf) -> Result<(), BackupError> {
let tmpdir = tempfile::tempdir().unwrap();
// Reopen the file for reading
let file = File::open(file).unwrap();
let decompressor = zstd::Decoder::new(file).unwrap();
let mut f = tar::Archive::new(decompressor);
f.unpack(tmpdir.path()).unwrap();
let path = tmpdir.path();
let mut manifest = File::open(path.join("metadata")).unwrap();
let mut manifest_slice = Vec::new();
manifest.read_to_end(&mut manifest_slice).unwrap();
let manifest: CloudSaveMetadata = serde_json::from_slice(&manifest_slice).unwrap();
for file in manifest.files {
let current_path = path.join(file.id.as_ref().unwrap());
let manager = BackupManager::new();
let os = match file
.conditions
.iter()
.find_map(|p| match p {
super::conditions::Condition::Os(os) => Some(os),
_ => None,
})
.cloned()
{
Some(os) => os,
None => {
warn!(
"File {:?} could not be replaced up because it did not provide an OS",
&file
);
continue;
}
};
let handler = match manager.sources.get(&(manager.current_platform, os)) {
Some(h) => *h,
None => continue,
};
let new_path = parse_path(file.path.into(), handler, &manifest.game_version)?;
create_dir_all(&new_path.parent().unwrap()).unwrap();
println!("Current path {:?} copying to {:?}", &current_path, &new_path);
std::fs::copy(current_path, new_path).unwrap();
}
Ok(())
}
pub fn parse_path(
path: PathBuf,
backup_handler: &dyn BackupHandler,
@ -71,23 +135,25 @@ pub fn parse_path(
let mut s = PathBuf::new();
for component in path.components() {
match component.as_str().unwrap() {
ROOT => { s.push(backup_handler.root_translate(&path, game)?)},
GAME => { s.push(backup_handler.game_translate(&path, game)?)},
BASE => { s.push(backup_handler.base_translate(&path, game)?)},
HOME => { s.push(backup_handler.home_translate(&path, game)?)},
STORE_USER_ID => { s.push(backup_handler.store_user_id_translate(&path, game)?)},
OS_USER_NAME => { s.push(backup_handler.os_user_name_translate(&path, game)?)},
WIN_APP_DATA => { s.push(backup_handler.win_app_data_translate(&path, game)?)},
WIN_LOCAL_APP_DATA => { s.push(backup_handler.win_local_app_data_translate(&path, game)?)},
WIN_LOCAL_APP_DATA_LOW => { s.push(backup_handler.win_local_app_data_low_translate(&path, game)?)},
WIN_DOCUMENTS => { s.push(backup_handler.win_documents_translate(&path, game)?)},
WIN_PUBLIC => { s.push(backup_handler.win_public_translate(&path, game)?)},
WIN_PROGRAM_DATA => { s.push(backup_handler.win_program_data_translate(&path, game)?)},
WIN_DIR => { s.push(backup_handler.win_dir_translate(&path, game)?)},
XDG_DATA => { s.push(backup_handler.xdg_data_translate(&path, game)?)},
XDG_CONFIG => { s.push(backup_handler.xdg_config_translate(&path, game)?)},
SKIP => { },
_ => s.push(PathBuf::from(component.as_os_str()))
ROOT => s.push(backup_handler.root_translate(&path, game)?),
GAME => s.push(backup_handler.game_translate(&path, game)?),
BASE => s.push(backup_handler.base_translate(&path, game)?),
HOME => s.push(backup_handler.home_translate(&path, game)?),
STORE_USER_ID => s.push(backup_handler.store_user_id_translate(&path, game)?),
OS_USER_NAME => s.push(backup_handler.os_user_name_translate(&path, game)?),
WIN_APP_DATA => s.push(backup_handler.win_app_data_translate(&path, game)?),
WIN_LOCAL_APP_DATA => s.push(backup_handler.win_local_app_data_translate(&path, game)?),
WIN_LOCAL_APP_DATA_LOW => {
s.push(backup_handler.win_local_app_data_low_translate(&path, game)?)
}
WIN_DOCUMENTS => s.push(backup_handler.win_documents_translate(&path, game)?),
WIN_PUBLIC => s.push(backup_handler.win_public_translate(&path, game)?),
WIN_PROGRAM_DATA => s.push(backup_handler.win_program_data_translate(&path, game)?),
WIN_DIR => s.push(backup_handler.win_dir_translate(&path, game)?),
XDG_DATA => s.push(backup_handler.xdg_data_translate(&path, game)?),
XDG_CONFIG => s.push(backup_handler.xdg_config_translate(&path, game)?),
SKIP => s.push(backup_handler.skip_translate(&path, game)?),
_ => s.push(PathBuf::from(component.as_os_str())),
}
}
@ -97,13 +163,22 @@ pub fn parse_path(
pub fn test() {
let mut meta = CloudSaveMetadata {
files: vec![GameFile {
path: String::from("<home>/favicon.png"),
id: None,
data_type: super::metadata::DataType::File,
tags: Vec::new(),
conditions: vec![Condition::Os(Platform::Linux)],
}],
files: vec![
GameFile {
path: String::from("<home>/favicon.png"),
id: None,
data_type: super::metadata::DataType::File,
tags: Vec::new(),
conditions: vec![Condition::Os(Platform::Linux)],
},
GameFile {
path: String::from("<home>/quexeky.jpg"),
id: None,
data_type: super::metadata::DataType::File,
tags: Vec::new(),
conditions: vec![Condition::Os(Platform::Linux)],
},
],
game_version: GameVersion {
game_id: String::new(),
version_name: String::new(),
@ -121,5 +196,7 @@ pub fn test() {
},
save_id: String::from("aaaaaaa"),
};
let file = resolve(&mut meta);
//resolve(&mut meta);
extract("save".into()).unwrap();
}