mirror of
https://github.com/Drop-OSS/drop-app.git
synced 2025-11-13 00:02:41 +10:00
Compare commits
6 Commits
bigpicture
...
0f48f3fb44
| Author | SHA1 | Date | |
|---|---|---|---|
| 0f48f3fb44 | |||
| 974666efe2 | |||
| 9e1bf9852f | |||
| 5d22b883d5 | |||
| 62a2561539 | |||
| 59f040bc8b |
2
.gitignore
vendored
2
.gitignore
vendored
@ -30,3 +30,5 @@ src-tauri/perf*
|
|||||||
|
|
||||||
/*.AppImage
|
/*.AppImage
|
||||||
/squashfs-root
|
/squashfs-root
|
||||||
|
|
||||||
|
/target/
|
||||||
|
|||||||
8290
Cargo.lock
generated
Normal file
8290
Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
14
Cargo.toml
Normal file
14
Cargo.toml
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
[workspace]
|
||||||
|
members = [
|
||||||
|
"client",
|
||||||
|
"database",
|
||||||
|
"src-tauri",
|
||||||
|
"process",
|
||||||
|
"remote",
|
||||||
|
"utils",
|
||||||
|
"cloud_saves",
|
||||||
|
"download_manager",
|
||||||
|
"games",
|
||||||
|
]
|
||||||
|
|
||||||
|
resolver = "3"
|
||||||
4862
client/Cargo.lock
generated
Normal file
4862
client/Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
12
client/Cargo.toml
Normal file
12
client/Cargo.toml
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
[package]
|
||||||
|
name = "client"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
bitcode = "0.6.7"
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
log = "0.4.28"
|
||||||
|
serde = { version = "1.0.228", features = ["derive"] }
|
||||||
|
tauri = "2.8.5"
|
||||||
|
tauri-plugin-autostart = "2.5.0"
|
||||||
12
client/src/app_status.rs
Normal file
12
client/src/app_status.rs
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
use serde::Serialize;
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Serialize, Eq, PartialEq)]
|
||||||
|
pub enum AppStatus {
|
||||||
|
NotConfigured,
|
||||||
|
Offline,
|
||||||
|
ServerError,
|
||||||
|
SignedOut,
|
||||||
|
SignedIn,
|
||||||
|
SignedInNeedsReauth,
|
||||||
|
ServerUnavailable,
|
||||||
|
}
|
||||||
26
client/src/autostart.rs
Normal file
26
client/src/autostart.rs
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
use database::borrow_db_checked;
|
||||||
|
use log::debug;
|
||||||
|
use tauri::AppHandle;
|
||||||
|
use tauri_plugin_autostart::ManagerExt;
|
||||||
|
|
||||||
|
// New function to sync state on startup
|
||||||
|
pub fn sync_autostart_on_startup(app: &AppHandle) -> Result<(), String> {
|
||||||
|
let db_handle = borrow_db_checked();
|
||||||
|
let should_be_enabled = db_handle.settings.autostart;
|
||||||
|
drop(db_handle);
|
||||||
|
|
||||||
|
let manager = app.autolaunch();
|
||||||
|
let current_state = manager.is_enabled().map_err(|e| e.to_string())?;
|
||||||
|
|
||||||
|
if current_state != should_be_enabled {
|
||||||
|
if should_be_enabled {
|
||||||
|
manager.enable().map_err(|e| e.to_string())?;
|
||||||
|
debug!("synced autostart: enabled");
|
||||||
|
} else {
|
||||||
|
manager.disable().map_err(|e| e.to_string())?;
|
||||||
|
debug!("synced autostart: disabled");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
52
client/src/compat.rs
Normal file
52
client/src/compat.rs
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
use std::{
|
||||||
|
ffi::OsStr,
|
||||||
|
path::PathBuf,
|
||||||
|
process::{Command, Stdio},
|
||||||
|
sync::LazyLock,
|
||||||
|
};
|
||||||
|
|
||||||
|
use log::info;
|
||||||
|
|
||||||
|
pub static COMPAT_INFO: LazyLock<Option<CompatInfo>> = LazyLock::new(create_new_compat_info);
|
||||||
|
|
||||||
|
pub static UMU_LAUNCHER_EXECUTABLE: LazyLock<Option<PathBuf>> = LazyLock::new(|| {
|
||||||
|
let x = get_umu_executable();
|
||||||
|
info!("{:?}", &x);
|
||||||
|
x
|
||||||
|
});
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct CompatInfo {
|
||||||
|
pub umu_installed: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn create_new_compat_info() -> Option<CompatInfo> {
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
return None;
|
||||||
|
|
||||||
|
let has_umu_installed = UMU_LAUNCHER_EXECUTABLE.is_some();
|
||||||
|
Some(CompatInfo {
|
||||||
|
umu_installed: has_umu_installed,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const UMU_BASE_LAUNCHER_EXECUTABLE: &str = "umu-run";
|
||||||
|
const UMU_INSTALL_DIRS: [&str; 4] = ["/app/share", "/use/local/share", "/usr/share", "/opt"];
|
||||||
|
|
||||||
|
fn get_umu_executable() -> Option<PathBuf> {
|
||||||
|
if check_executable_exists(UMU_BASE_LAUNCHER_EXECUTABLE) {
|
||||||
|
return Some(PathBuf::from(UMU_BASE_LAUNCHER_EXECUTABLE));
|
||||||
|
}
|
||||||
|
|
||||||
|
for dir in UMU_INSTALL_DIRS {
|
||||||
|
let p = PathBuf::from(dir).join(UMU_BASE_LAUNCHER_EXECUTABLE);
|
||||||
|
if check_executable_exists(&p) {
|
||||||
|
return Some(p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
fn check_executable_exists<P: AsRef<OsStr>>(exec: P) -> bool {
|
||||||
|
let has_umu_installed = Command::new(exec).stdout(Stdio::null()).output();
|
||||||
|
has_umu_installed.is_ok()
|
||||||
|
}
|
||||||
4
client/src/lib.rs
Normal file
4
client/src/lib.rs
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
pub mod app_status;
|
||||||
|
pub mod autostart;
|
||||||
|
pub mod compat;
|
||||||
|
pub mod user;
|
||||||
12
client/src/user.rs
Normal file
12
client/src/user.rs
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
use bitcode::{Decode, Encode};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
#[derive(Clone, Serialize, Deserialize, Encode, Decode)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct User {
|
||||||
|
id: String,
|
||||||
|
username: String,
|
||||||
|
admin: bool,
|
||||||
|
display_name: String,
|
||||||
|
profile_picture_object_id: String,
|
||||||
|
}
|
||||||
19
cloud_saves/Cargo.toml
Normal file
19
cloud_saves/Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "cloud_saves"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
dirs = "6.0.0"
|
||||||
|
log = "0.4.28"
|
||||||
|
regex = "1.11.3"
|
||||||
|
rustix = "1.1.2"
|
||||||
|
serde = "1.0.228"
|
||||||
|
serde_json = "1.0.145"
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
tar = "0.4.44"
|
||||||
|
tempfile = "3.23.0"
|
||||||
|
uuid = "1.18.1"
|
||||||
|
whoami = "1.6.1"
|
||||||
|
zstd = "0.13.3"
|
||||||
234
cloud_saves/src/backup_manager.rs
Normal file
234
cloud_saves/src/backup_manager.rs
Normal file
@ -0,0 +1,234 @@
|
|||||||
|
use std::{collections::HashMap, path::PathBuf, str::FromStr};
|
||||||
|
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
use database::platform::Platform;
|
||||||
|
use database::{GameVersion, db::DATA_ROOT_DIR};
|
||||||
|
use log::warn;
|
||||||
|
|
||||||
|
use crate::error::BackupError;
|
||||||
|
|
||||||
|
use super::path::CommonPath;
|
||||||
|
|
||||||
|
pub struct BackupManager<'a> {
|
||||||
|
pub current_platform: Platform,
|
||||||
|
pub sources: HashMap<(Platform, Platform), &'a (dyn BackupHandler + Sync + Send)>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for BackupManager<'_> {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BackupManager<'_> {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
BackupManager {
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
current_platform: Platform::Windows,
|
||||||
|
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
current_platform: Platform::MacOs,
|
||||||
|
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
current_platform: Platform::Linux,
|
||||||
|
|
||||||
|
sources: HashMap::from([
|
||||||
|
// Current platform to target platform
|
||||||
|
(
|
||||||
|
(Platform::Windows, Platform::Windows),
|
||||||
|
&WindowsBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
||||||
|
),
|
||||||
|
(
|
||||||
|
(Platform::Linux, Platform::Linux),
|
||||||
|
&LinuxBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
||||||
|
),
|
||||||
|
(
|
||||||
|
(Platform::MacOs, Platform::MacOs),
|
||||||
|
&MacBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
||||||
|
),
|
||||||
|
]),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait BackupHandler: Send + Sync {
|
||||||
|
fn root_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(DATA_ROOT_DIR.join("games"))
|
||||||
|
}
|
||||||
|
fn game_translate(&self, _path: &PathBuf, game: &GameVersion) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(PathBuf::from_str(&game.game_id).unwrap())
|
||||||
|
}
|
||||||
|
fn base_translate(&self, path: &PathBuf, game: &GameVersion) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(self
|
||||||
|
.root_translate(path, game)?
|
||||||
|
.join(self.game_translate(path, game)?))
|
||||||
|
}
|
||||||
|
fn home_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
||||||
|
let c = CommonPath::Home.get().ok_or(BackupError::NotFound);
|
||||||
|
println!("{:?}", c);
|
||||||
|
c
|
||||||
|
}
|
||||||
|
fn store_user_id_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
PathBuf::from_str(&game.game_id).map_err(|_| BackupError::ParseError)
|
||||||
|
}
|
||||||
|
fn os_user_name_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(PathBuf::from_str(&whoami::username()).unwrap())
|
||||||
|
}
|
||||||
|
fn win_app_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winAppData>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_local_app_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winLocalAppData>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_local_app_data_low_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winLocalAppDataLow>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_documents_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winDocuments>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_public_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winPublic>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_program_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winProgramData>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn win_dir_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected Windows Reference in Backup <winDir>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn xdg_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected XDG Reference in Backup <xdgData>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn xdg_config_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
warn!("Unexpected XDG Reference in Backup <xdgConfig>");
|
||||||
|
Err(BackupError::InvalidSystem)
|
||||||
|
}
|
||||||
|
fn skip_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(PathBuf::new())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct LinuxBackupManager {}
|
||||||
|
impl BackupHandler for LinuxBackupManager {
|
||||||
|
fn xdg_config_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::Data.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
fn xdg_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::Config.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub struct WindowsBackupManager {}
|
||||||
|
impl BackupHandler for WindowsBackupManager {
|
||||||
|
fn win_app_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::Config.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
fn win_local_app_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::DataLocal.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
fn win_local_app_data_low_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::DataLocalLow
|
||||||
|
.get()
|
||||||
|
.ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
fn win_dir_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(PathBuf::from_str("C:/Windows").unwrap())
|
||||||
|
}
|
||||||
|
fn win_documents_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::Document.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
fn win_program_data_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
Ok(PathBuf::from_str("C:/ProgramData").unwrap())
|
||||||
|
}
|
||||||
|
fn win_public_translate(
|
||||||
|
&self,
|
||||||
|
_path: &PathBuf,
|
||||||
|
_game: &GameVersion,
|
||||||
|
) -> Result<PathBuf, BackupError> {
|
||||||
|
CommonPath::Public.get().ok_or(BackupError::NotFound)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pub struct MacBackupManager {}
|
||||||
|
impl BackupHandler for MacBackupManager {}
|
||||||
@ -1,6 +1,7 @@
|
|||||||
use crate::process::process_manager::Platform;
|
use database::platform::Platform;
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
#[derive(Clone, Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
||||||
pub enum Condition {
|
pub enum Condition {
|
||||||
Os(Platform)
|
Os(Platform),
|
||||||
|
Other
|
||||||
}
|
}
|
||||||
27
cloud_saves/src/error.rs
Normal file
27
cloud_saves/src/error.rs
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
use std::fmt::Display;
|
||||||
|
|
||||||
|
use serde_with::SerializeDisplay;
|
||||||
|
|
||||||
|
#[derive(Debug, SerializeDisplay, Clone, Copy)]
|
||||||
|
|
||||||
|
pub enum BackupError {
|
||||||
|
InvalidSystem,
|
||||||
|
|
||||||
|
NotFound,
|
||||||
|
|
||||||
|
ParseError,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Display for BackupError {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
let s = match self {
|
||||||
|
BackupError::InvalidSystem => "Attempted to generate path for invalid system",
|
||||||
|
|
||||||
|
BackupError::NotFound => "Could not generate or find path",
|
||||||
|
|
||||||
|
BackupError::ParseError => "Failed to parse path",
|
||||||
|
};
|
||||||
|
|
||||||
|
write!(f, "{}", s)
|
||||||
|
}
|
||||||
|
}
|
||||||
8
cloud_saves/src/lib.rs
Normal file
8
cloud_saves/src/lib.rs
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
pub mod backup_manager;
|
||||||
|
pub mod conditions;
|
||||||
|
pub mod error;
|
||||||
|
pub mod metadata;
|
||||||
|
pub mod normalise;
|
||||||
|
pub mod path;
|
||||||
|
pub mod placeholder;
|
||||||
|
pub mod resolver;
|
||||||
@ -1,7 +1,6 @@
|
|||||||
use crate::database::db::GameVersion;
|
use database::GameVersion;
|
||||||
|
|
||||||
use super::conditions::{Condition};
|
|
||||||
|
|
||||||
|
use super::conditions::Condition;
|
||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
#[derive(Clone, Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
|
||||||
pub struct CloudSaveMetadata {
|
pub struct CloudSaveMetadata {
|
||||||
@ -16,15 +15,17 @@ pub struct GameFile {
|
|||||||
pub id: Option<String>,
|
pub id: Option<String>,
|
||||||
pub data_type: DataType,
|
pub data_type: DataType,
|
||||||
pub tags: Vec<Tag>,
|
pub tags: Vec<Tag>,
|
||||||
pub conditions: Vec<Condition>
|
pub conditions: Vec<Condition>,
|
||||||
}
|
}
|
||||||
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, serde::Serialize, serde::Deserialize)]
|
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, serde::Serialize, serde::Deserialize)]
|
||||||
pub enum DataType {
|
pub enum DataType {
|
||||||
Registry,
|
Registry,
|
||||||
File,
|
File,
|
||||||
Other
|
Other,
|
||||||
}
|
}
|
||||||
#[derive(Clone, Debug, Default, PartialEq, Eq, PartialOrd, Ord, serde::Serialize, serde::Deserialize)]
|
#[derive(
|
||||||
|
Clone, Debug, Default, PartialEq, Eq, PartialOrd, Ord, serde::Serialize, serde::Deserialize,
|
||||||
|
)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub enum Tag {
|
pub enum Tag {
|
||||||
Config,
|
Config,
|
||||||
@ -1,11 +1,10 @@
|
|||||||
use std::sync::LazyLock;
|
use std::sync::LazyLock;
|
||||||
|
|
||||||
|
use database::platform::Platform;
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use crate::process::process_manager::Platform;
|
|
||||||
|
|
||||||
use super::placeholder::*;
|
use super::placeholder::*;
|
||||||
|
|
||||||
|
|
||||||
pub fn normalize(path: &str, os: Platform) -> String {
|
pub fn normalize(path: &str, os: Platform) -> String {
|
||||||
let mut path = path.trim().trim_end_matches(['/', '\\']).replace('\\', "/");
|
let mut path = path.trim().trim_end_matches(['/', '\\']).replace('\\', "/");
|
||||||
|
|
||||||
@ -14,18 +13,25 @@ pub fn normalize(path: &str, os: Platform) -> String {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static CONSECUTIVE_SLASHES: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"/{2,}").unwrap());
|
static CONSECUTIVE_SLASHES: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"/{2,}").unwrap());
|
||||||
static UNNECESSARY_DOUBLE_STAR_1: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"([^/*])\*{2,}").unwrap());
|
static UNNECESSARY_DOUBLE_STAR_1: LazyLock<Regex> =
|
||||||
static UNNECESSARY_DOUBLE_STAR_2: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"\*{2,}([^/*])").unwrap());
|
LazyLock::new(|| Regex::new(r"([^/*])\*{2,}").unwrap());
|
||||||
|
static UNNECESSARY_DOUBLE_STAR_2: LazyLock<Regex> =
|
||||||
|
LazyLock::new(|| Regex::new(r"\*{2,}([^/*])").unwrap());
|
||||||
static ENDING_WILDCARD: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\*)+$").unwrap());
|
static ENDING_WILDCARD: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\*)+$").unwrap());
|
||||||
static ENDING_DOT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\.)$").unwrap());
|
static ENDING_DOT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\.)$").unwrap());
|
||||||
static INTERMEDIATE_DOT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\./)").unwrap());
|
static INTERMEDIATE_DOT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\./)").unwrap());
|
||||||
static BLANK_SEGMENT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\s+/)").unwrap());
|
static BLANK_SEGMENT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(/\s+/)").unwrap());
|
||||||
static APP_DATA: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%appdata%").unwrap());
|
static APP_DATA: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%appdata%").unwrap());
|
||||||
static APP_DATA_ROAMING: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%userprofile%/AppData/Roaming").unwrap());
|
static APP_DATA_ROAMING: LazyLock<Regex> =
|
||||||
static APP_DATA_LOCAL: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%localappdata%").unwrap());
|
LazyLock::new(|| Regex::new(r"(?i)%userprofile%/AppData/Roaming").unwrap());
|
||||||
static APP_DATA_LOCAL_2: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%userprofile%/AppData/Local/").unwrap());
|
static APP_DATA_LOCAL: LazyLock<Regex> =
|
||||||
static USER_PROFILE: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%userprofile%").unwrap());
|
LazyLock::new(|| Regex::new(r"(?i)%localappdata%").unwrap());
|
||||||
static DOCUMENTS: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(?i)%userprofile%/Documents").unwrap());
|
static APP_DATA_LOCAL_2: LazyLock<Regex> =
|
||||||
|
LazyLock::new(|| Regex::new(r"(?i)%userprofile%/AppData/Local/").unwrap());
|
||||||
|
static USER_PROFILE: LazyLock<Regex> =
|
||||||
|
LazyLock::new(|| Regex::new(r"(?i)%userprofile%").unwrap());
|
||||||
|
static DOCUMENTS: LazyLock<Regex> =
|
||||||
|
LazyLock::new(|| Regex::new(r"(?i)%userprofile%/Documents").unwrap());
|
||||||
|
|
||||||
for (pattern, replacement) in [
|
for (pattern, replacement) in [
|
||||||
(&CONSECUTIVE_SLASHES, "/"),
|
(&CONSECUTIVE_SLASHES, "/"),
|
||||||
@ -66,7 +72,9 @@ pub fn normalize(path: &str, os: Platform) -> String {
|
|||||||
|
|
||||||
fn too_broad(path: &str) -> bool {
|
fn too_broad(path: &str) -> bool {
|
||||||
println!("Path: {}", path);
|
println!("Path: {}", path);
|
||||||
use {BASE, HOME, ROOT, STORE_USER_ID, WIN_APP_DATA, WIN_DIR, WIN_DOCUMENTS, XDG_CONFIG, XDG_DATA};
|
use {
|
||||||
|
BASE, HOME, ROOT, STORE_USER_ID, WIN_APP_DATA, WIN_DIR, WIN_DOCUMENTS, XDG_CONFIG, XDG_DATA,
|
||||||
|
};
|
||||||
|
|
||||||
let path_lower = path.to_lowercase();
|
let path_lower = path.to_lowercase();
|
||||||
|
|
||||||
@ -77,7 +85,9 @@ fn too_broad(path: &str) -> bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for item in AVOID_WILDCARDS {
|
for item in AVOID_WILDCARDS {
|
||||||
if path.starts_with(&format!("{}/*", item)) || path.starts_with(&format!("{}/{}", item, STORE_USER_ID)) {
|
if path.starts_with(&format!("{}/*", item))
|
||||||
|
|| path.starts_with(&format!("{}/{}", item, STORE_USER_ID))
|
||||||
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -125,7 +135,6 @@ fn too_broad(path: &str) -> bool {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Drive letters:
|
// Drive letters:
|
||||||
let drives: Regex = Regex::new(r"^[a-zA-Z]:$").unwrap();
|
let drives: Regex = Regex::new(r"^[a-zA-Z]:$").unwrap();
|
||||||
if drives.is_match(path) {
|
if drives.is_match(path) {
|
||||||
@ -13,12 +13,12 @@ pub enum CommonPath {
|
|||||||
|
|
||||||
impl CommonPath {
|
impl CommonPath {
|
||||||
pub fn get(&self) -> Option<PathBuf> {
|
pub fn get(&self) -> Option<PathBuf> {
|
||||||
static CONFIG: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::config_dir());
|
static CONFIG: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::config_dir);
|
||||||
static DATA: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::data_dir());
|
static DATA: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::data_dir);
|
||||||
static DATA_LOCAL: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::data_local_dir());
|
static DATA_LOCAL: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::data_local_dir);
|
||||||
static DOCUMENT: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::document_dir());
|
static DOCUMENT: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::document_dir);
|
||||||
static HOME: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::home_dir());
|
static HOME: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::home_dir);
|
||||||
static PUBLIC: LazyLock<Option<PathBuf>> = LazyLock::new(|| dirs::public_dir());
|
static PUBLIC: LazyLock<Option<PathBuf>> = LazyLock::new(dirs::public_dir);
|
||||||
|
|
||||||
#[cfg(windows)]
|
#[cfg(windows)]
|
||||||
static DATA_LOCAL_LOW: LazyLock<Option<PathBuf>> = LazyLock::new(|| {
|
static DATA_LOCAL_LOW: LazyLock<Option<PathBuf>> = LazyLock::new(|| {
|
||||||
@ -48,4 +48,4 @@ pub const XDG_DATA: &str = "<xdgData>"; // %WINDIR% on Windows
|
|||||||
pub const XDG_CONFIG: &str = "<xdgConfig>"; // $XDG_DATA_HOME on Linux
|
pub const XDG_CONFIG: &str = "<xdgConfig>"; // $XDG_DATA_HOME on Linux
|
||||||
pub const SKIP: &str = "<skip>"; // $XDG_CONFIG_HOME on Linux
|
pub const SKIP: &str = "<skip>"; // $XDG_CONFIG_HOME on Linux
|
||||||
|
|
||||||
pub static OS_USERNAME: LazyLock<String> = LazyLock::new(|| whoami::username());
|
pub static OS_USERNAME: LazyLock<String> = LazyLock::new(whoami::username);
|
||||||
@ -1,22 +1,17 @@
|
|||||||
use std::{
|
use std::{
|
||||||
fs::{self, create_dir_all, File},
|
fs::{self, File, create_dir_all},
|
||||||
io::{self, ErrorKind, Read, Write},
|
io::{self, Read, Write},
|
||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
thread::sleep,
|
|
||||||
time::Duration,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{
|
use crate::error::BackupError;
|
||||||
backup_manager::BackupHandler, conditions::Condition, metadata::GameFile, placeholder::*,
|
|
||||||
};
|
use super::{backup_manager::BackupHandler, placeholder::*};
|
||||||
|
use database::GameVersion;
|
||||||
use log::{debug, warn};
|
use log::{debug, warn};
|
||||||
use rustix::path::Arg;
|
use rustix::path::Arg;
|
||||||
use tempfile::tempfile;
|
use tempfile::tempfile;
|
||||||
|
|
||||||
use crate::{
|
|
||||||
database::db::GameVersion, error::backup_error::BackupError, process::process_manager::Platform,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{backup_manager::BackupManager, metadata::CloudSaveMetadata, normalise::normalize};
|
use super::{backup_manager::BackupManager, metadata::CloudSaveMetadata, normalise::normalize};
|
||||||
|
|
||||||
pub fn resolve(meta: &mut CloudSaveMetadata) -> File {
|
pub fn resolve(meta: &mut CloudSaveMetadata) -> File {
|
||||||
@ -31,7 +26,7 @@ pub fn resolve(meta: &mut CloudSaveMetadata) -> File {
|
|||||||
.iter()
|
.iter()
|
||||||
.find_map(|p| match p {
|
.find_map(|p| match p {
|
||||||
super::conditions::Condition::Os(os) => Some(os),
|
super::conditions::Condition::Os(os) => Some(os),
|
||||||
_ => None,
|
_ => None
|
||||||
})
|
})
|
||||||
.cloned()
|
.cloned()
|
||||||
{
|
{
|
||||||
@ -64,7 +59,7 @@ pub fn resolve(meta: &mut CloudSaveMetadata) -> File {
|
|||||||
let binding = serde_json::to_string(meta).unwrap();
|
let binding = serde_json::to_string(meta).unwrap();
|
||||||
let serialized = binding.as_bytes();
|
let serialized = binding.as_bytes();
|
||||||
let mut file = tempfile().unwrap();
|
let mut file = tempfile().unwrap();
|
||||||
file.write(serialized).unwrap();
|
file.write_all(serialized).unwrap();
|
||||||
tarball.append_file("metadata", &mut file).unwrap();
|
tarball.append_file("metadata", &mut file).unwrap();
|
||||||
tarball.into_inner().unwrap().finish().unwrap()
|
tarball.into_inner().unwrap().finish().unwrap()
|
||||||
}
|
}
|
||||||
@ -97,7 +92,7 @@ pub fn extract(file: PathBuf) -> Result<(), BackupError> {
|
|||||||
.iter()
|
.iter()
|
||||||
.find_map(|p| match p {
|
.find_map(|p| match p {
|
||||||
super::conditions::Condition::Os(os) => Some(os),
|
super::conditions::Condition::Os(os) => Some(os),
|
||||||
_ => None,
|
_ => None
|
||||||
})
|
})
|
||||||
.cloned()
|
.cloned()
|
||||||
{
|
{
|
||||||
@ -116,7 +111,7 @@ pub fn extract(file: PathBuf) -> Result<(), BackupError> {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let new_path = parse_path(file.path.into(), handler, &manifest.game_version)?;
|
let new_path = parse_path(file.path.into(), handler, &manifest.game_version)?;
|
||||||
create_dir_all(&new_path.parent().unwrap()).unwrap();
|
create_dir_all(new_path.parent().unwrap()).unwrap();
|
||||||
|
|
||||||
println!(
|
println!(
|
||||||
"Current path {:?} copying to {:?}",
|
"Current path {:?} copying to {:?}",
|
||||||
@ -133,23 +128,22 @@ pub fn copy_item<P: AsRef<Path>>(src: P, dest: P) -> io::Result<()> {
|
|||||||
let src_path = src.as_ref();
|
let src_path = src.as_ref();
|
||||||
let dest_path = dest.as_ref();
|
let dest_path = dest.as_ref();
|
||||||
|
|
||||||
let metadata = fs::metadata(&src_path)?;
|
let metadata = fs::metadata(src_path)?;
|
||||||
|
|
||||||
if metadata.is_file() {
|
if metadata.is_file() {
|
||||||
// Ensure the parent directory of the destination exists for a file copy
|
// Ensure the parent directory of the destination exists for a file copy
|
||||||
if let Some(parent) = dest_path.parent() {
|
if let Some(parent) = dest_path.parent() {
|
||||||
fs::create_dir_all(parent)?;
|
fs::create_dir_all(parent)?;
|
||||||
}
|
}
|
||||||
fs::copy(&src_path, &dest_path)?;
|
fs::copy(src_path, dest_path)?;
|
||||||
} else if metadata.is_dir() {
|
} else if metadata.is_dir() {
|
||||||
// For directories, we call the recursive helper function.
|
// For directories, we call the recursive helper function.
|
||||||
// The destination for the recursive copy is the `dest_path` itself.
|
// The destination for the recursive copy is the `dest_path` itself.
|
||||||
copy_dir_recursive(&src_path, &dest_path)?;
|
copy_dir_recursive(src_path, dest_path)?;
|
||||||
} else {
|
} else {
|
||||||
// Handle other file types like symlinks if necessary,
|
// Handle other file types like symlinks if necessary,
|
||||||
// for now, return an error or skip.
|
// for now, return an error or skip.
|
||||||
return Err(io::Error::new(
|
return Err(io::Error::other(
|
||||||
io::ErrorKind::Other,
|
|
||||||
format!("Source {:?} is neither a file nor a directory", src_path),
|
format!("Source {:?} is neither a file nor a directory", src_path),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
@ -158,7 +152,7 @@ pub fn copy_item<P: AsRef<Path>>(src: P, dest: P) -> io::Result<()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn copy_dir_recursive(src: &Path, dest: &Path) -> io::Result<()> {
|
fn copy_dir_recursive(src: &Path, dest: &Path) -> io::Result<()> {
|
||||||
fs::create_dir_all(&dest)?;
|
fs::create_dir_all(dest)?;
|
||||||
|
|
||||||
for entry in fs::read_dir(src)? {
|
for entry in fs::read_dir(src)? {
|
||||||
let entry = entry?;
|
let entry = entry?;
|
||||||
@ -220,43 +214,3 @@ pub fn parse_path(
|
|||||||
println!("Final line: {:?}", &s);
|
println!("Final line: {:?}", &s);
|
||||||
Ok(s)
|
Ok(s)
|
||||||
}
|
}
|
||||||
|
|
||||||
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)],
|
|
||||||
},
|
|
||||||
GameFile {
|
|
||||||
path: String::from("<home>/Documents/Pixel Art"),
|
|
||||||
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(),
|
|
||||||
platform: Platform::Linux,
|
|
||||||
launch_command: String::new(),
|
|
||||||
launch_args: Vec::new(),
|
|
||||||
launch_command_template: String::new(),
|
|
||||||
setup_command: String::new(),
|
|
||||||
setup_args: Vec::new(),
|
|
||||||
setup_command_template: String::new(),
|
|
||||||
only_setup: true,
|
|
||||||
version_index: 0,
|
|
||||||
delta: false,
|
|
||||||
umu_id_override: None,
|
|
||||||
},
|
|
||||||
save_id: String::from("aaaaaaa"),
|
|
||||||
};
|
|
||||||
//resolve(&mut meta);
|
|
||||||
|
|
||||||
extract("save".into()).unwrap();
|
|
||||||
}
|
|
||||||
0
cloud_saves/src/strict_path.rs
Normal file
0
cloud_saves/src/strict_path.rs
Normal file
15
database/Cargo.toml
Normal file
15
database/Cargo.toml
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
[package]
|
||||||
|
name = "database"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
chrono = "0.4.42"
|
||||||
|
dirs = "6.0.0"
|
||||||
|
log = "0.4.28"
|
||||||
|
native_model = { version = "0.6.4", features = ["rmp_serde_1_3"], git = "https://github.com/Drop-OSS/native_model.git"}
|
||||||
|
rustbreak = "2.0.0"
|
||||||
|
serde = "1.0.228"
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
url = "2.5.7"
|
||||||
|
whoami = "1.6.1"
|
||||||
45
database/src/db.rs
Normal file
45
database/src/db.rs
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
use std::{
|
||||||
|
path::PathBuf,
|
||||||
|
sync::{Arc, LazyLock},
|
||||||
|
};
|
||||||
|
|
||||||
|
use rustbreak::{DeSerError, DeSerializer};
|
||||||
|
use serde::{Serialize, de::DeserializeOwned};
|
||||||
|
|
||||||
|
use crate::interface::{DatabaseImpls, DatabaseInterface};
|
||||||
|
|
||||||
|
pub static DB: LazyLock<DatabaseInterface> = LazyLock::new(DatabaseInterface::set_up_database);
|
||||||
|
|
||||||
|
#[cfg(not(debug_assertions))]
|
||||||
|
static DATA_ROOT_PREFIX: &str = "drop";
|
||||||
|
#[cfg(debug_assertions)]
|
||||||
|
static DATA_ROOT_PREFIX: &str = "drop-debug";
|
||||||
|
|
||||||
|
pub static DATA_ROOT_DIR: LazyLock<Arc<PathBuf>> = LazyLock::new(|| {
|
||||||
|
Arc::new(
|
||||||
|
dirs::data_dir()
|
||||||
|
.expect("Failed to get data dir")
|
||||||
|
.join(DATA_ROOT_PREFIX),
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
// Custom JSON serializer to support everything we need
|
||||||
|
#[derive(Debug, Default, Clone)]
|
||||||
|
pub struct DropDatabaseSerializer;
|
||||||
|
|
||||||
|
impl<T: native_model::Model + Serialize + DeserializeOwned> DeSerializer<T>
|
||||||
|
for DropDatabaseSerializer
|
||||||
|
{
|
||||||
|
fn serialize(&self, val: &T) -> rustbreak::error::DeSerResult<Vec<u8>> {
|
||||||
|
native_model::encode(val).map_err(|e| DeSerError::Internal(e.to_string()))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn deserialize<R: std::io::Read>(&self, mut s: R) -> rustbreak::error::DeSerResult<T> {
|
||||||
|
let mut buf = Vec::new();
|
||||||
|
s.read_to_end(&mut buf)
|
||||||
|
.map_err(|e| rustbreak::error::DeSerError::Other(e.into()))?;
|
||||||
|
let (val, _version) =
|
||||||
|
native_model::decode(buf).map_err(|e| DeSerError::Internal(e.to_string()))?;
|
||||||
|
Ok(val)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -3,53 +3,18 @@ use std::{
|
|||||||
mem::ManuallyDrop,
|
mem::ManuallyDrop,
|
||||||
ops::{Deref, DerefMut},
|
ops::{Deref, DerefMut},
|
||||||
path::PathBuf,
|
path::PathBuf,
|
||||||
sync::{Arc, LazyLock, RwLockReadGuard, RwLockWriteGuard},
|
sync::{RwLockReadGuard, RwLockWriteGuard},
|
||||||
};
|
};
|
||||||
|
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
use log::{debug, error, info, warn};
|
use log::{debug, error, info, warn};
|
||||||
use rustbreak::{DeSerError, DeSerializer, PathDatabase, RustbreakError};
|
use rustbreak::{PathDatabase, RustbreakError};
|
||||||
use serde::{Serialize, de::DeserializeOwned};
|
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
use crate::DB;
|
use crate::{
|
||||||
|
db::{DATA_ROOT_DIR, DB, DropDatabaseSerializer},
|
||||||
use super::models::data::Database;
|
models::data::Database,
|
||||||
|
};
|
||||||
#[cfg(not(debug_assertions))]
|
|
||||||
static DATA_ROOT_PREFIX: &'static str = "drop";
|
|
||||||
#[cfg(debug_assertions)]
|
|
||||||
static DATA_ROOT_PREFIX: &str = "drop-debug";
|
|
||||||
|
|
||||||
pub static DATA_ROOT_DIR: LazyLock<Arc<PathBuf>> = LazyLock::new(|| {
|
|
||||||
Arc::new(
|
|
||||||
dirs::data_dir()
|
|
||||||
.expect("Failed to get data dir")
|
|
||||||
.join(DATA_ROOT_PREFIX),
|
|
||||||
)
|
|
||||||
});
|
|
||||||
|
|
||||||
// Custom JSON serializer to support everything we need
|
|
||||||
#[derive(Debug, Default, Clone)]
|
|
||||||
pub struct DropDatabaseSerializer;
|
|
||||||
|
|
||||||
impl<T: native_model::Model + Serialize + DeserializeOwned> DeSerializer<T>
|
|
||||||
for DropDatabaseSerializer
|
|
||||||
{
|
|
||||||
fn serialize(&self, val: &T) -> rustbreak::error::DeSerResult<Vec<u8>> {
|
|
||||||
native_model::encode(val)
|
|
||||||
.map_err(|e| DeSerError::Internal(e.to_string()))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn deserialize<R: std::io::Read>(&self, mut s: R) -> rustbreak::error::DeSerResult<T> {
|
|
||||||
let mut buf = Vec::new();
|
|
||||||
s.read_to_end(&mut buf)
|
|
||||||
.map_err(|e| rustbreak::error::DeSerError::Other(e.into()))?;
|
|
||||||
let (val, _version) = native_model::decode(buf)
|
|
||||||
.map_err(|e| DeSerError::Internal(e.to_string()))?;
|
|
||||||
Ok(val)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub type DatabaseInterface =
|
pub type DatabaseInterface =
|
||||||
rustbreak::Database<Database, rustbreak::backend::PathBackend, DropDatabaseSerializer>;
|
rustbreak::Database<Database, rustbreak::backend::PathBackend, DropDatabaseSerializer>;
|
||||||
14
database/src/lib.rs
Normal file
14
database/src/lib.rs
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
#![feature(nonpoison_rwlock)]
|
||||||
|
|
||||||
|
pub mod db;
|
||||||
|
pub mod debug;
|
||||||
|
pub mod interface;
|
||||||
|
pub mod models;
|
||||||
|
pub mod platform;
|
||||||
|
|
||||||
|
pub use db::DB;
|
||||||
|
pub use interface::{borrow_db_checked, borrow_db_mut_checked};
|
||||||
|
pub use models::data::{
|
||||||
|
ApplicationTransientStatus, Database, DatabaseApplications, DatabaseAuth, DownloadType,
|
||||||
|
DownloadableMetadata, GameDownloadStatus, GameVersion, Settings,
|
||||||
|
};
|
||||||
@ -37,17 +37,18 @@ pub mod data {
|
|||||||
}
|
}
|
||||||
|
|
||||||
mod v1 {
|
mod v1 {
|
||||||
use crate::process::process_manager::Platform;
|
|
||||||
use serde_with::serde_as;
|
use serde_with::serde_as;
|
||||||
use std::{collections::HashMap, path::PathBuf};
|
use std::{collections::HashMap, path::PathBuf};
|
||||||
|
|
||||||
|
use crate::platform::Platform;
|
||||||
|
|
||||||
use super::{Deserialize, Serialize, native_model};
|
use super::{Deserialize, Serialize, native_model};
|
||||||
|
|
||||||
fn default_template() -> String {
|
fn default_template() -> String {
|
||||||
"{}".to_owned()
|
"{}".to_owned()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Clone, Debug)]
|
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
#[native_model(id = 2, version = 1, with = native_model::rmp_serde_1_3::RmpSerde)]
|
#[native_model(id = 2, version = 1, with = native_model::rmp_serde_1_3::RmpSerde)]
|
||||||
pub struct GameVersion {
|
pub struct GameVersion {
|
||||||
@ -190,9 +191,7 @@ pub mod data {
|
|||||||
|
|
||||||
use serde_with::serde_as;
|
use serde_with::serde_as;
|
||||||
|
|
||||||
use super::{
|
use super::{Deserialize, Serialize, native_model, v1};
|
||||||
Deserialize, Serialize, native_model, v1,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[native_model(id = 1, version = 2, with = native_model::rmp_serde_1_3::RmpSerde, from = v1::Database)]
|
#[native_model(id = 1, version = 2, with = native_model::rmp_serde_1_3::RmpSerde, from = v1::Database)]
|
||||||
#[derive(Serialize, Deserialize, Clone, Default)]
|
#[derive(Serialize, Deserialize, Clone, Default)]
|
||||||
@ -281,7 +280,8 @@ pub mod data {
|
|||||||
pub installed_game_version: HashMap<String, v1::DownloadableMetadata>,
|
pub installed_game_version: HashMap<String, v1::DownloadableMetadata>,
|
||||||
|
|
||||||
#[serde(skip)]
|
#[serde(skip)]
|
||||||
pub transient_statuses: HashMap<v1::DownloadableMetadata, v1::ApplicationTransientStatus>,
|
pub transient_statuses:
|
||||||
|
HashMap<v1::DownloadableMetadata, v1::ApplicationTransientStatus>,
|
||||||
}
|
}
|
||||||
impl From<v1::DatabaseApplications> for DatabaseApplications {
|
impl From<v1::DatabaseApplications> for DatabaseApplications {
|
||||||
fn from(value: v1::DatabaseApplications) -> Self {
|
fn from(value: v1::DatabaseApplications) -> Self {
|
||||||
@ -302,10 +302,7 @@ pub mod data {
|
|||||||
mod v3 {
|
mod v3 {
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use super::{
|
use super::{Deserialize, Serialize, native_model, v1, v2};
|
||||||
Deserialize, Serialize,
|
|
||||||
native_model, v2, v1,
|
|
||||||
};
|
|
||||||
#[native_model(id = 1, version = 3, with = native_model::rmp_serde_1_3::RmpSerde, from = v2::Database)]
|
#[native_model(id = 1, version = 3, with = native_model::rmp_serde_1_3::RmpSerde, from = v2::Database)]
|
||||||
#[derive(Serialize, Deserialize, Clone, Default)]
|
#[derive(Serialize, Deserialize, Clone, Default)]
|
||||||
pub struct Database {
|
pub struct Database {
|
||||||
@ -357,6 +354,20 @@ pub mod data {
|
|||||||
compat_info: None,
|
compat_info: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
impl DatabaseAuth {
|
||||||
|
pub fn new(
|
||||||
|
private: String,
|
||||||
|
cert: String,
|
||||||
|
client_id: String,
|
||||||
|
web_token: Option<String>,
|
||||||
|
) -> Self {
|
||||||
|
Self {
|
||||||
|
private,
|
||||||
|
cert,
|
||||||
|
client_id,
|
||||||
|
web_token,
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
46
database/src/platform.rs
Normal file
46
database/src/platform.rs
Normal file
@ -0,0 +1,46 @@
|
|||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
#[derive(Eq, Hash, PartialEq, Serialize, Deserialize, Clone, Copy, Debug)]
|
||||||
|
pub enum Platform {
|
||||||
|
Windows,
|
||||||
|
Linux,
|
||||||
|
MacOs,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Platform {
|
||||||
|
#[cfg(target_os = "windows")]
|
||||||
|
pub const HOST: Platform = Self::Windows;
|
||||||
|
#[cfg(target_os = "macos")]
|
||||||
|
pub const HOST: Platform = Self::MacOs;
|
||||||
|
#[cfg(target_os = "linux")]
|
||||||
|
pub const HOST: Platform = Self::Linux;
|
||||||
|
|
||||||
|
pub fn is_case_sensitive(&self) -> bool {
|
||||||
|
match self {
|
||||||
|
Self::Windows | Self::MacOs => false,
|
||||||
|
Self::Linux => true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<&str> for Platform {
|
||||||
|
fn from(value: &str) -> Self {
|
||||||
|
match value.to_lowercase().trim() {
|
||||||
|
"windows" => Self::Windows,
|
||||||
|
"linux" => Self::Linux,
|
||||||
|
"mac" | "macos" => Self::MacOs,
|
||||||
|
_ => unimplemented!(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<whoami::Platform> for Platform {
|
||||||
|
fn from(value: whoami::Platform) -> Self {
|
||||||
|
match value {
|
||||||
|
whoami::Platform::Windows => Platform::Windows,
|
||||||
|
whoami::Platform::Linux => Platform::Linux,
|
||||||
|
whoami::Platform::MacOS => Platform::MacOs,
|
||||||
|
platform => unimplemented!("Playform {} is not supported", platform),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
17
download_manager/Cargo.toml
Normal file
17
download_manager/Cargo.toml
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
[package]
|
||||||
|
name = "download_manager"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
atomic-instant-full = "0.1.0"
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
humansize = "2.1.3"
|
||||||
|
log = "0.4.28"
|
||||||
|
parking_lot = "0.12.5"
|
||||||
|
remote = { version = "0.1.0", path = "../remote" }
|
||||||
|
serde = "1.0.228"
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
tauri = "2.8.5"
|
||||||
|
throttle_my_fn = "0.2.6"
|
||||||
|
utils = { version = "0.1.0", path = "../utils" }
|
||||||
@ -7,11 +7,15 @@ use std::{
|
|||||||
thread::{JoinHandle, spawn},
|
thread::{JoinHandle, spawn},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use database::DownloadableMetadata;
|
||||||
use log::{debug, error, info, warn};
|
use log::{debug, error, info, warn};
|
||||||
use tauri::{AppHandle, Emitter};
|
use tauri::AppHandle;
|
||||||
|
use utils::{app_emit, lock, send};
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
app_emit, database::models::data::DownloadableMetadata, download_manager::download_manager_frontend::DownloadStatus, error::application_download_error::ApplicationDownloadError, games::library::{QueueUpdateEvent, QueueUpdateEventQueueData, StatsUpdateEvent}, lock, send
|
download_manager_frontend::DownloadStatus,
|
||||||
|
error::ApplicationDownloadError,
|
||||||
|
frontend_updates::{QueueUpdateEvent, QueueUpdateEventQueueData, StatsUpdateEvent},
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
@ -288,7 +292,10 @@ impl DownloadManagerBuilder {
|
|||||||
|
|
||||||
if validate_result {
|
if validate_result {
|
||||||
download_agent.on_complete(&app_handle);
|
download_agent.on_complete(&app_handle);
|
||||||
send!(sender, DownloadManagerSignal::Completed(download_agent.metadata()));
|
send!(
|
||||||
|
sender,
|
||||||
|
DownloadManagerSignal::Completed(download_agent.metadata())
|
||||||
|
);
|
||||||
send!(sender, DownloadManagerSignal::UpdateUIQueue);
|
send!(sender, DownloadManagerSignal::UpdateUIQueue);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -369,7 +376,7 @@ impl DownloadManagerBuilder {
|
|||||||
fn push_ui_stats_update(&self, kbs: usize, time: usize) {
|
fn push_ui_stats_update(&self, kbs: usize, time: usize) {
|
||||||
let event_data = StatsUpdateEvent { speed: kbs, time };
|
let event_data = StatsUpdateEvent { speed: kbs, time };
|
||||||
|
|
||||||
app_emit!(self.app_handle, "update_stats", event_data);
|
app_emit!(&self.app_handle, "update_stats", event_data);
|
||||||
}
|
}
|
||||||
fn push_ui_queue_update(&self) {
|
fn push_ui_queue_update(&self) {
|
||||||
let queue = &self.download_queue.read();
|
let queue = &self.download_queue.read();
|
||||||
@ -388,6 +395,6 @@ impl DownloadManagerBuilder {
|
|||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
let event_data = QueueUpdateEvent { queue: queue_objs };
|
let event_data = QueueUpdateEvent { queue: queue_objs };
|
||||||
app_emit!(self.app_handle, "update_queue", event_data);
|
app_emit!(&self.app_handle, "update_queue", event_data);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -9,13 +9,12 @@ use std::{
|
|||||||
thread::JoinHandle,
|
thread::JoinHandle,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use database::DownloadableMetadata;
|
||||||
use log::{debug, info};
|
use log::{debug, info};
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
|
use utils::{lock, send};
|
||||||
|
|
||||||
use crate::{
|
use crate::error::ApplicationDownloadError;
|
||||||
database::models::data::DownloadableMetadata,
|
|
||||||
error::application_download_error::ApplicationDownloadError, lock, send,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
download_manager_builder::{CurrentProgressObject, DownloadAgent},
|
download_manager_builder::{CurrentProgressObject, DownloadAgent},
|
||||||
@ -80,6 +79,7 @@ pub enum DownloadStatus {
|
|||||||
/// The actual download queue may be accessed through the .`edit()` function,
|
/// The actual download queue may be accessed through the .`edit()` function,
|
||||||
/// which provides raw access to the underlying queue.
|
/// which provides raw access to the underlying queue.
|
||||||
/// THIS EDITING IS BLOCKING!!!
|
/// THIS EDITING IS BLOCKING!!!
|
||||||
|
#[derive(Debug)]
|
||||||
pub struct DownloadManager {
|
pub struct DownloadManager {
|
||||||
terminator: Mutex<Option<JoinHandle<Result<(), ()>>>>,
|
terminator: Mutex<Option<JoinHandle<Result<(), ()>>>>,
|
||||||
download_queue: Queue,
|
download_queue: Queue,
|
||||||
@ -124,8 +124,11 @@ impl DownloadManager {
|
|||||||
}
|
}
|
||||||
pub fn rearrange_string(&self, meta: &DownloadableMetadata, new_index: usize) {
|
pub fn rearrange_string(&self, meta: &DownloadableMetadata, new_index: usize) {
|
||||||
let mut queue = self.edit();
|
let mut queue = self.edit();
|
||||||
let current_index = get_index_from_id(&mut queue, meta).expect("Failed to get meta index from id");
|
let current_index =
|
||||||
let to_move = queue.remove(current_index).expect("Failed to remove meta at index from queue");
|
get_index_from_id(&mut queue, meta).expect("Failed to get meta index from id");
|
||||||
|
let to_move = queue
|
||||||
|
.remove(current_index)
|
||||||
|
.expect("Failed to remove meta at index from queue");
|
||||||
queue.insert(new_index, to_move);
|
queue.insert(new_index, to_move);
|
||||||
send!(self.command_sender, DownloadManagerSignal::UpdateUIQueue);
|
send!(self.command_sender, DownloadManagerSignal::UpdateUIQueue);
|
||||||
}
|
}
|
||||||
@ -1,11 +1,9 @@
|
|||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use database::DownloadableMetadata;
|
||||||
use tauri::AppHandle;
|
use tauri::AppHandle;
|
||||||
|
|
||||||
use crate::{
|
use crate::error::ApplicationDownloadError;
|
||||||
database::models::data::DownloadableMetadata,
|
|
||||||
error::application_download_error::ApplicationDownloadError,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
download_manager_frontend::DownloadStatus,
|
download_manager_frontend::DownloadStatus,
|
||||||
@ -1,12 +1,36 @@
|
|||||||
|
use humansize::{BINARY, format_size};
|
||||||
use std::{
|
use std::{
|
||||||
fmt::{Display, Formatter},
|
fmt::{Display, Formatter},
|
||||||
io, sync::Arc,
|
io,
|
||||||
|
sync::{Arc, mpsc::SendError},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use remote::error::RemoteAccessError;
|
||||||
use serde_with::SerializeDisplay;
|
use serde_with::SerializeDisplay;
|
||||||
use humansize::{format_size, BINARY};
|
|
||||||
|
|
||||||
use super::remote_access_error::RemoteAccessError;
|
#[derive(SerializeDisplay)]
|
||||||
|
pub enum DownloadManagerError<T> {
|
||||||
|
IOError(io::Error),
|
||||||
|
SignalError(SendError<T>),
|
||||||
|
}
|
||||||
|
impl<T> Display for DownloadManagerError<T> {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
match self {
|
||||||
|
DownloadManagerError::IOError(error) => write!(f, "{error}"),
|
||||||
|
DownloadManagerError::SignalError(send_error) => write!(f, "{send_error}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<T> From<SendError<T>> for DownloadManagerError<T> {
|
||||||
|
fn from(value: SendError<T>) -> Self {
|
||||||
|
DownloadManagerError::SignalError(value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<T> From<io::Error> for DownloadManagerError<T> {
|
||||||
|
fn from(value: io::Error) -> Self {
|
||||||
|
DownloadManagerError::IOError(value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// TODO: Rename / separate from downloads
|
// TODO: Rename / separate from downloads
|
||||||
#[derive(Debug, SerializeDisplay)]
|
#[derive(Debug, SerializeDisplay)]
|
||||||
@ -24,7 +48,9 @@ pub enum ApplicationDownloadError {
|
|||||||
impl Display for ApplicationDownloadError {
|
impl Display for ApplicationDownloadError {
|
||||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||||
match self {
|
match self {
|
||||||
ApplicationDownloadError::NotInitialized => write!(f, "Download not initalized, did something go wrong?"),
|
ApplicationDownloadError::NotInitialized => {
|
||||||
|
write!(f, "Download not initalized, did something go wrong?")
|
||||||
|
}
|
||||||
ApplicationDownloadError::DiskFull(required, available) => write!(
|
ApplicationDownloadError::DiskFull(required, available) => write!(
|
||||||
f,
|
f,
|
||||||
"Game requires {}, {} remaining left on disk.",
|
"Game requires {}, {} remaining left on disk.",
|
||||||
@ -40,10 +66,9 @@ impl Display for ApplicationDownloadError {
|
|||||||
write!(f, "checksum failed to validate for download")
|
write!(f, "checksum failed to validate for download")
|
||||||
}
|
}
|
||||||
ApplicationDownloadError::IoError(error) => write!(f, "io error: {error}"),
|
ApplicationDownloadError::IoError(error) => write!(f, "io error: {error}"),
|
||||||
ApplicationDownloadError::DownloadError(error) => write!(
|
ApplicationDownloadError::DownloadError(error) => {
|
||||||
f,
|
write!(f, "Download failed with error {error:?}")
|
||||||
"Download failed with error {error:?}"
|
}
|
||||||
),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
24
download_manager/src/frontend_updates.rs
Normal file
24
download_manager/src/frontend_updates.rs
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
use database::DownloadableMetadata;
|
||||||
|
use serde::Serialize;
|
||||||
|
|
||||||
|
use crate::download_manager_frontend::DownloadStatus;
|
||||||
|
|
||||||
|
#[derive(Serialize, Clone)]
|
||||||
|
pub struct QueueUpdateEventQueueData {
|
||||||
|
pub meta: DownloadableMetadata,
|
||||||
|
pub status: DownloadStatus,
|
||||||
|
pub progress: f64,
|
||||||
|
pub current: usize,
|
||||||
|
pub max: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Clone)]
|
||||||
|
pub struct QueueUpdateEvent {
|
||||||
|
pub queue: Vec<QueueUpdateEventQueueData>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Clone)]
|
||||||
|
pub struct StatsUpdateEvent {
|
||||||
|
pub speed: usize,
|
||||||
|
pub time: usize,
|
||||||
|
}
|
||||||
44
download_manager/src/lib.rs
Normal file
44
download_manager/src/lib.rs
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
#![feature(duration_millis_float)]
|
||||||
|
#![feature(nonpoison_mutex)]
|
||||||
|
#![feature(sync_nonpoison)]
|
||||||
|
|
||||||
|
use std::{ops::Deref, sync::OnceLock};
|
||||||
|
|
||||||
|
use tauri::AppHandle;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
download_manager_builder::DownloadManagerBuilder, download_manager_frontend::DownloadManager,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub mod download_manager_builder;
|
||||||
|
pub mod download_manager_frontend;
|
||||||
|
pub mod downloadable;
|
||||||
|
pub mod error;
|
||||||
|
pub mod frontend_updates;
|
||||||
|
pub mod util;
|
||||||
|
|
||||||
|
pub static DOWNLOAD_MANAGER: DownloadManagerWrapper = DownloadManagerWrapper::new();
|
||||||
|
|
||||||
|
pub struct DownloadManagerWrapper(OnceLock<DownloadManager>);
|
||||||
|
impl DownloadManagerWrapper {
|
||||||
|
const fn new() -> Self {
|
||||||
|
DownloadManagerWrapper(OnceLock::new())
|
||||||
|
}
|
||||||
|
pub fn init(app_handle: AppHandle) {
|
||||||
|
DOWNLOAD_MANAGER
|
||||||
|
.0
|
||||||
|
.set(DownloadManagerBuilder::build(app_handle))
|
||||||
|
.expect("Failed to initialise download manager");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Deref for DownloadManagerWrapper {
|
||||||
|
type Target = DownloadManager;
|
||||||
|
|
||||||
|
fn deref(&self) -> &Self::Target {
|
||||||
|
match self.0.get() {
|
||||||
|
Some(download_manager) => download_manager,
|
||||||
|
None => unreachable!("Download manager should always be initialised"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,6 +1,6 @@
|
|||||||
use std::sync::{
|
use std::sync::{
|
||||||
atomic::{AtomicBool, Ordering},
|
|
||||||
Arc,
|
Arc,
|
||||||
|
atomic::{AtomicBool, Ordering},
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(PartialEq, Eq, PartialOrd, Ord)]
|
#[derive(PartialEq, Eq, PartialOrd, Ord)]
|
||||||
@ -22,7 +22,11 @@ impl From<DownloadThreadControlFlag> for bool {
|
|||||||
/// false => Stop
|
/// false => Stop
|
||||||
impl From<bool> for DownloadThreadControlFlag {
|
impl From<bool> for DownloadThreadControlFlag {
|
||||||
fn from(value: bool) -> Self {
|
fn from(value: bool) -> Self {
|
||||||
if value { DownloadThreadControlFlag::Go } else { DownloadThreadControlFlag::Stop }
|
if value {
|
||||||
|
DownloadThreadControlFlag::Go
|
||||||
|
} else {
|
||||||
|
DownloadThreadControlFlag::Stop
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -9,12 +9,13 @@ use std::{
|
|||||||
|
|
||||||
use atomic_instant_full::AtomicInstant;
|
use atomic_instant_full::AtomicInstant;
|
||||||
use throttle_my_fn::throttle;
|
use throttle_my_fn::throttle;
|
||||||
|
use utils::{lock, send};
|
||||||
|
|
||||||
use crate::{download_manager::download_manager_frontend::DownloadManagerSignal, lock, send};
|
use crate::download_manager_frontend::DownloadManagerSignal;
|
||||||
|
|
||||||
use super::rolling_progress_updates::RollingProgressWindow;
|
use super::rolling_progress_updates::RollingProgressWindow;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct ProgressObject {
|
pub struct ProgressObject {
|
||||||
max: Arc<Mutex<usize>>,
|
max: Arc<Mutex<usize>>,
|
||||||
progress_instances: Arc<Mutex<Vec<Arc<AtomicUsize>>>>,
|
progress_instances: Arc<Mutex<Vec<Arc<AtomicUsize>>>>,
|
||||||
@ -116,7 +117,9 @@ pub fn calculate_update(progress: &ProgressObject) {
|
|||||||
let last_update_time = progress
|
let last_update_time = progress
|
||||||
.last_update_time
|
.last_update_time
|
||||||
.swap(Instant::now(), Ordering::SeqCst);
|
.swap(Instant::now(), Ordering::SeqCst);
|
||||||
let time_since_last_update = Instant::now().duration_since(last_update_time).as_millis_f64();
|
let time_since_last_update = Instant::now()
|
||||||
|
.duration_since(last_update_time)
|
||||||
|
.as_millis_f64();
|
||||||
|
|
||||||
let current_bytes_downloaded = progress.sum();
|
let current_bytes_downloaded = progress.sum();
|
||||||
let max = progress.get_max();
|
let max = progress.get_max();
|
||||||
@ -124,7 +127,8 @@ pub fn calculate_update(progress: &ProgressObject) {
|
|||||||
.bytes_last_update
|
.bytes_last_update
|
||||||
.swap(current_bytes_downloaded, Ordering::Acquire);
|
.swap(current_bytes_downloaded, Ordering::Acquire);
|
||||||
|
|
||||||
let bytes_since_last_update = current_bytes_downloaded.saturating_sub(bytes_at_last_update) as f64;
|
let bytes_since_last_update =
|
||||||
|
current_bytes_downloaded.saturating_sub(bytes_at_last_update) as f64;
|
||||||
|
|
||||||
let kilobytes_per_second = bytes_since_last_update / time_since_last_update;
|
let kilobytes_per_second = bytes_since_last_update / time_since_last_update;
|
||||||
|
|
||||||
@ -3,9 +3,10 @@ use std::{
|
|||||||
sync::{Arc, Mutex, MutexGuard},
|
sync::{Arc, Mutex, MutexGuard},
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{database::models::data::DownloadableMetadata, lock};
|
use database::DownloadableMetadata;
|
||||||
|
use utils::lock;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct Queue {
|
pub struct Queue {
|
||||||
inner: Arc<Mutex<VecDeque<DownloadableMetadata>>>,
|
inner: Arc<Mutex<VecDeque<DownloadableMetadata>>>,
|
||||||
}
|
}
|
||||||
@ -3,11 +3,17 @@ use std::sync::{
|
|||||||
atomic::{AtomicUsize, Ordering},
|
atomic::{AtomicUsize, Ordering},
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct RollingProgressWindow<const S: usize> {
|
pub struct RollingProgressWindow<const S: usize> {
|
||||||
window: Arc<[AtomicUsize; S]>,
|
window: Arc<[AtomicUsize; S]>,
|
||||||
current: Arc<AtomicUsize>,
|
current: Arc<AtomicUsize>,
|
||||||
}
|
}
|
||||||
|
impl<const S: usize> Default for RollingProgressWindow<S> {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<const S: usize> RollingProgressWindow<S> {
|
impl<const S: usize> RollingProgressWindow<S> {
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
Self {
|
Self {
|
||||||
26
games/Cargo.toml
Normal file
26
games/Cargo.toml
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
[package]
|
||||||
|
name = "games"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
atomic-instant-full = "0.1.0"
|
||||||
|
bitcode = "0.6.7"
|
||||||
|
boxcar = "0.2.14"
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
download_manager = { version = "0.1.0", path = "../download_manager" }
|
||||||
|
hex = "0.4.3"
|
||||||
|
log = "0.4.28"
|
||||||
|
md5 = "0.8.0"
|
||||||
|
rayon = "1.11.0"
|
||||||
|
remote = { version = "0.1.0", path = "../remote" }
|
||||||
|
reqwest = "0.12.23"
|
||||||
|
rustix = "1.1.2"
|
||||||
|
serde = { version = "1.0.228", features = ["derive"] }
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
sysinfo = "0.37.2"
|
||||||
|
tauri = "2.8.5"
|
||||||
|
throttle_my_fn = "0.2.6"
|
||||||
|
utils = { version = "0.1.0", path = "../utils" }
|
||||||
|
native_model = { version = "0.6.4", features = ["rmp_serde_1_3"], git = "https://github.com/Drop-OSS/native_model.git"}
|
||||||
|
serde_json = "1.0.145"
|
||||||
@ -1,7 +1,7 @@
|
|||||||
use bitcode::{Decode, Encode};
|
use bitcode::{Decode, Encode};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
use crate::games::library::Game;
|
use crate::library::Game;
|
||||||
|
|
||||||
pub type Collections = Vec<Collection>;
|
pub type Collections = Vec<Collection>;
|
||||||
|
|
||||||
@ -1,2 +1 @@
|
|||||||
pub mod collection;
|
pub mod collection;
|
||||||
pub mod commands;
|
|
||||||
@ -1,28 +1,20 @@
|
|||||||
use crate::auth::generate_authorization_header;
|
use database::{
|
||||||
use crate::database::db::{borrow_db_checked, borrow_db_mut_checked};
|
ApplicationTransientStatus, DownloadType, DownloadableMetadata, borrow_db_checked,
|
||||||
use crate::database::models::data::{
|
borrow_db_mut_checked,
|
||||||
ApplicationTransientStatus, DownloadType, DownloadableMetadata,
|
|
||||||
};
|
};
|
||||||
use crate::download_manager::download_manager_frontend::{DownloadManagerSignal, DownloadStatus};
|
use download_manager::download_manager_frontend::{DownloadManagerSignal, DownloadStatus};
|
||||||
use crate::download_manager::downloadable::Downloadable;
|
use download_manager::downloadable::Downloadable;
|
||||||
use crate::download_manager::util::download_thread_control_flag::{
|
use download_manager::error::ApplicationDownloadError;
|
||||||
|
use download_manager::util::download_thread_control_flag::{
|
||||||
DownloadThreadControl, DownloadThreadControlFlag,
|
DownloadThreadControl, DownloadThreadControlFlag,
|
||||||
};
|
};
|
||||||
use crate::download_manager::util::progress_object::{ProgressHandle, ProgressObject};
|
use download_manager::util::progress_object::{ProgressHandle, ProgressObject};
|
||||||
use crate::error::application_download_error::ApplicationDownloadError;
|
|
||||||
use crate::error::remote_access_error::RemoteAccessError;
|
|
||||||
use crate::games::downloads::manifest::{
|
|
||||||
DownloadBucket, DownloadContext, DownloadDrop, DropManifest, DropValidateContext, ManifestBody,
|
|
||||||
};
|
|
||||||
use crate::games::downloads::validate::validate_game_chunk;
|
|
||||||
use crate::games::library::{on_game_complete, push_game_update, set_partially_installed};
|
|
||||||
use crate::games::state::GameStatusManager;
|
|
||||||
use crate::process::utils::get_disk_available;
|
|
||||||
use crate::remote::requests::generate_url;
|
|
||||||
use crate::remote::utils::{DROP_CLIENT_ASYNC, DROP_CLIENT_SYNC};
|
|
||||||
use crate::{app_emit, lock, send};
|
|
||||||
use log::{debug, error, info, warn};
|
use log::{debug, error, info, warn};
|
||||||
use rayon::ThreadPoolBuilder;
|
use rayon::ThreadPoolBuilder;
|
||||||
|
use remote::auth::generate_authorization_header;
|
||||||
|
use remote::error::RemoteAccessError;
|
||||||
|
use remote::requests::generate_url;
|
||||||
|
use remote::utils::{DROP_CLIENT_ASYNC, DROP_CLIENT_SYNC};
|
||||||
use std::collections::{HashMap, HashSet};
|
use std::collections::{HashMap, HashSet};
|
||||||
use std::fs::{OpenOptions, create_dir_all};
|
use std::fs::{OpenOptions, create_dir_all};
|
||||||
use std::io;
|
use std::io;
|
||||||
@ -30,11 +22,20 @@ use std::path::{Path, PathBuf};
|
|||||||
use std::sync::mpsc::Sender;
|
use std::sync::mpsc::Sender;
|
||||||
use std::sync::{Arc, Mutex};
|
use std::sync::{Arc, Mutex};
|
||||||
use std::time::Instant;
|
use std::time::Instant;
|
||||||
use tauri::{AppHandle, Emitter};
|
use tauri::AppHandle;
|
||||||
|
use utils::{app_emit, lock, send};
|
||||||
|
|
||||||
#[cfg(target_os = "linux")]
|
#[cfg(target_os = "linux")]
|
||||||
use rustix::fs::{FallocateFlags, fallocate};
|
use rustix::fs::{FallocateFlags, fallocate};
|
||||||
|
|
||||||
|
use crate::downloads::manifest::{
|
||||||
|
DownloadBucket, DownloadContext, DownloadDrop, DropManifest, DropValidateContext, ManifestBody,
|
||||||
|
};
|
||||||
|
use crate::downloads::utils::get_disk_available;
|
||||||
|
use crate::downloads::validate::validate_game_chunk;
|
||||||
|
use crate::library::{on_game_complete, push_game_update, set_partially_installed};
|
||||||
|
use crate::state::GameStatusManager;
|
||||||
|
|
||||||
use super::download_logic::download_game_bucket;
|
use super::download_logic::download_game_bucket;
|
||||||
use super::drop_data::DropData;
|
use super::drop_data::DropData;
|
||||||
|
|
||||||
@ -103,8 +104,7 @@ impl GameDownloadAgent {
|
|||||||
|
|
||||||
result.ensure_manifest_exists().await?;
|
result.ensure_manifest_exists().await?;
|
||||||
|
|
||||||
let required_space = lock!(result
|
let required_space = lock!(result.manifest)
|
||||||
.manifest)
|
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.values()
|
.values()
|
||||||
@ -453,9 +453,13 @@ impl GameDownloadAgent {
|
|||||||
|
|
||||||
let sender = self.sender.clone();
|
let sender = self.sender.clone();
|
||||||
|
|
||||||
let download_context = download_contexts
|
let download_context =
|
||||||
.get(&bucket.version)
|
download_contexts.get(&bucket.version).unwrap_or_else(|| {
|
||||||
.unwrap_or_else(|| panic!("Could not get bucket version {}. Corrupted state.", bucket.version));
|
panic!(
|
||||||
|
"Could not get bucket version {}. Corrupted state.",
|
||||||
|
bucket.version
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
scope.spawn(move |_| {
|
scope.spawn(move |_| {
|
||||||
// 3 attempts
|
// 3 attempts
|
||||||
@ -693,7 +697,10 @@ impl Downloadable for GameDownloadAgent {
|
|||||||
Ok(_) => {}
|
Ok(_) => {}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
error!("could not mark game as complete: {e}");
|
error!("could not mark game as complete: {e}");
|
||||||
send!(self.sender, DownloadManagerSignal::Error(ApplicationDownloadError::DownloadError(e)));
|
send!(
|
||||||
|
self.sender,
|
||||||
|
DownloadManagerSignal::Error(ApplicationDownloadError::DownloadError(e))
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -1,18 +1,3 @@
|
|||||||
use crate::download_manager::util::download_thread_control_flag::{
|
|
||||||
DownloadThreadControl, DownloadThreadControlFlag,
|
|
||||||
};
|
|
||||||
use crate::download_manager::util::progress_object::ProgressHandle;
|
|
||||||
use crate::error::application_download_error::ApplicationDownloadError;
|
|
||||||
use crate::error::drop_server_error::DropServerError;
|
|
||||||
use crate::error::remote_access_error::RemoteAccessError;
|
|
||||||
use crate::games::downloads::manifest::{ChunkBody, DownloadBucket, DownloadContext, DownloadDrop};
|
|
||||||
use crate::remote::auth::generate_authorization_header;
|
|
||||||
use crate::remote::requests::generate_url;
|
|
||||||
use crate::remote::utils::DROP_CLIENT_SYNC;
|
|
||||||
use log::{debug, info, warn};
|
|
||||||
use md5::{Context, Digest};
|
|
||||||
use reqwest::blocking::Response;
|
|
||||||
|
|
||||||
use std::fs::{Permissions, set_permissions};
|
use std::fs::{Permissions, set_permissions};
|
||||||
use std::io::Read;
|
use std::io::Read;
|
||||||
#[cfg(unix)]
|
#[cfg(unix)]
|
||||||
@ -25,6 +10,21 @@ use std::{
|
|||||||
path::PathBuf,
|
path::PathBuf,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use download_manager::error::ApplicationDownloadError;
|
||||||
|
use download_manager::util::download_thread_control_flag::{
|
||||||
|
DownloadThreadControl, DownloadThreadControlFlag,
|
||||||
|
};
|
||||||
|
use download_manager::util::progress_object::ProgressHandle;
|
||||||
|
use log::{debug, info, warn};
|
||||||
|
use md5::{Context, Digest};
|
||||||
|
use remote::auth::generate_authorization_header;
|
||||||
|
use remote::error::{DropServerError, RemoteAccessError};
|
||||||
|
use remote::requests::generate_url;
|
||||||
|
use remote::utils::DROP_CLIENT_SYNC;
|
||||||
|
use reqwest::blocking::Response;
|
||||||
|
|
||||||
|
use crate::downloads::manifest::{ChunkBody, DownloadBucket, DownloadContext, DownloadDrop};
|
||||||
|
|
||||||
static MAX_PACKET_LENGTH: usize = 4096 * 4;
|
static MAX_PACKET_LENGTH: usize = 4096 * 4;
|
||||||
static BUMP_SIZE: usize = 4096 * 16;
|
static BUMP_SIZE: usize = 4096 * 16;
|
||||||
|
|
||||||
@ -49,7 +49,7 @@ impl DropWriter<File> {
|
|||||||
|
|
||||||
fn finish(mut self) -> io::Result<Digest> {
|
fn finish(mut self) -> io::Result<Digest> {
|
||||||
self.flush()?;
|
self.flush()?;
|
||||||
Ok(self.hasher.compute())
|
Ok(self.hasher.finalize())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Write automatically pushes to file and hasher
|
// Write automatically pushes to file and hasher
|
||||||
@ -118,7 +118,10 @@ impl<'a> DropDownloadPipeline<'a, Response, File> {
|
|||||||
let mut last_bump = 0;
|
let mut last_bump = 0;
|
||||||
loop {
|
loop {
|
||||||
let size = MAX_PACKET_LENGTH.min(remaining);
|
let size = MAX_PACKET_LENGTH.min(remaining);
|
||||||
let size = self.source.read(&mut copy_buffer[0..size]).inspect_err(|_| {
|
let size = self
|
||||||
|
.source
|
||||||
|
.read(&mut copy_buffer[0..size])
|
||||||
|
.inspect_err(|_| {
|
||||||
info!("got error from {}", drop.filename);
|
info!("got error from {}", drop.filename);
|
||||||
})?;
|
})?;
|
||||||
remaining -= size;
|
remaining -= size;
|
||||||
@ -1,11 +1,13 @@
|
|||||||
use std::{
|
use std::{
|
||||||
collections::HashMap, fs::File, io::{self, Read, Write}, path::{Path, PathBuf}
|
collections::HashMap,
|
||||||
|
fs::File,
|
||||||
|
io::{self, Read, Write},
|
||||||
|
path::{Path, PathBuf},
|
||||||
};
|
};
|
||||||
|
|
||||||
use log::error;
|
use log::error;
|
||||||
use native_model::{Decode, Encode};
|
use native_model::{Decode, Encode};
|
||||||
|
use utils::lock;
|
||||||
use crate::lock;
|
|
||||||
|
|
||||||
pub type DropData = v1::DropData;
|
pub type DropData = v1::DropData;
|
||||||
|
|
||||||
@ -78,7 +80,10 @@ impl DropData {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn set_contexts(&self, completed_contexts: &[(String, bool)]) {
|
pub fn set_contexts(&self, completed_contexts: &[(String, bool)]) {
|
||||||
*lock!(self.contexts) = completed_contexts.iter().map(|s| (s.0.clone(), s.1)).collect();
|
*lock!(self.contexts) = completed_contexts
|
||||||
|
.iter()
|
||||||
|
.map(|s| (s.0.clone(), s.1))
|
||||||
|
.collect();
|
||||||
}
|
}
|
||||||
pub fn set_context(&self, context: String, state: bool) {
|
pub fn set_context(&self, context: String, state: bool) {
|
||||||
lock!(self.contexts).entry(context).insert_entry(state);
|
lock!(self.contexts).entry(context).insert_entry(state);
|
||||||
29
games/src/downloads/error.rs
Normal file
29
games/src/downloads/error.rs
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
use std::fmt::Display;
|
||||||
|
|
||||||
|
use serde_with::SerializeDisplay;
|
||||||
|
|
||||||
|
#[derive(SerializeDisplay)]
|
||||||
|
pub enum LibraryError {
|
||||||
|
MetaNotFound(String),
|
||||||
|
VersionNotFound(String),
|
||||||
|
}
|
||||||
|
impl Display for LibraryError {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
write!(
|
||||||
|
f,
|
||||||
|
"{}",
|
||||||
|
match self {
|
||||||
|
LibraryError::MetaNotFound(id) => {
|
||||||
|
format!(
|
||||||
|
"Could not locate any installed version of game ID {id} in the database"
|
||||||
|
)
|
||||||
|
}
|
||||||
|
LibraryError::VersionNotFound(game_id) => {
|
||||||
|
format!(
|
||||||
|
"Could not locate any installed version for game id {game_id} in the database"
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,6 +1,7 @@
|
|||||||
pub mod commands;
|
|
||||||
pub mod download_agent;
|
pub mod download_agent;
|
||||||
mod download_logic;
|
mod download_logic;
|
||||||
pub mod drop_data;
|
pub mod drop_data;
|
||||||
|
pub mod error;
|
||||||
mod manifest;
|
mod manifest;
|
||||||
|
pub mod utils;
|
||||||
pub mod validate;
|
pub mod validate;
|
||||||
@ -1,10 +1,8 @@
|
|||||||
use std::{path::PathBuf, sync::Arc};
|
use std::{io, path::PathBuf, sync::Arc};
|
||||||
|
|
||||||
use futures_lite::io;
|
use download_manager::error::ApplicationDownloadError;
|
||||||
use sysinfo::{Disk, DiskRefreshKind, Disks};
|
use sysinfo::{Disk, DiskRefreshKind, Disks};
|
||||||
|
|
||||||
use crate::error::application_download_error::ApplicationDownloadError;
|
|
||||||
|
|
||||||
pub fn get_disk_available(mount_point: PathBuf) -> Result<u64, ApplicationDownloadError> {
|
pub fn get_disk_available(mount_point: PathBuf) -> Result<u64, ApplicationDownloadError> {
|
||||||
let disks = Disks::new_with_refreshed_list_specifics(DiskRefreshKind::nothing().with_storage());
|
let disks = Disks::new_with_refreshed_list_specifics(DiskRefreshKind::nothing().with_storage());
|
||||||
|
|
||||||
@ -21,7 +19,7 @@ pub fn get_disk_available(mount_point: PathBuf) -> Result<u64, ApplicationDownlo
|
|||||||
return Ok(disk.available_space());
|
return Ok(disk.available_space());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Err(ApplicationDownloadError::IoError(Arc::new(io::Error::other(
|
Err(ApplicationDownloadError::IoError(Arc::new(
|
||||||
"could not find disk of path",
|
io::Error::other("could not find disk of path"),
|
||||||
))))
|
)))
|
||||||
}
|
}
|
||||||
@ -3,17 +3,17 @@ use std::{
|
|||||||
io::{self, BufWriter, Read, Seek, SeekFrom, Write},
|
io::{self, BufWriter, Read, Seek, SeekFrom, Write},
|
||||||
};
|
};
|
||||||
|
|
||||||
use log::debug;
|
use download_manager::{
|
||||||
use md5::Context;
|
error::ApplicationDownloadError,
|
||||||
|
util::{
|
||||||
use crate::{
|
|
||||||
download_manager::util::{
|
|
||||||
download_thread_control_flag::{DownloadThreadControl, DownloadThreadControlFlag},
|
download_thread_control_flag::{DownloadThreadControl, DownloadThreadControlFlag},
|
||||||
progress_object::ProgressHandle,
|
progress_object::ProgressHandle,
|
||||||
},
|
},
|
||||||
error::application_download_error::ApplicationDownloadError,
|
|
||||||
games::downloads::manifest::DropValidateContext,
|
|
||||||
};
|
};
|
||||||
|
use log::debug;
|
||||||
|
use md5::Context;
|
||||||
|
|
||||||
|
use crate::downloads::manifest::DropValidateContext;
|
||||||
|
|
||||||
pub fn validate_game_chunk(
|
pub fn validate_game_chunk(
|
||||||
ctx: &DropValidateContext,
|
ctx: &DropValidateContext,
|
||||||
@ -22,7 +22,10 @@ pub fn validate_game_chunk(
|
|||||||
) -> Result<bool, ApplicationDownloadError> {
|
) -> Result<bool, ApplicationDownloadError> {
|
||||||
debug!(
|
debug!(
|
||||||
"Starting chunk validation {}, {}, {} #{}",
|
"Starting chunk validation {}, {}, {} #{}",
|
||||||
ctx.path.display(), ctx.index, ctx.offset, ctx.checksum
|
ctx.path.display(),
|
||||||
|
ctx.index,
|
||||||
|
ctx.offset,
|
||||||
|
ctx.checksum
|
||||||
);
|
);
|
||||||
// If we're paused
|
// If we're paused
|
||||||
if control_flag.get() == DownloadThreadControlFlag::Stop {
|
if control_flag.get() == DownloadThreadControlFlag::Stop {
|
||||||
@ -42,13 +45,12 @@ pub fn validate_game_chunk(
|
|||||||
|
|
||||||
let mut hasher = md5::Context::new();
|
let mut hasher = md5::Context::new();
|
||||||
|
|
||||||
let completed =
|
let completed = validate_copy(&mut source, &mut hasher, ctx.length, control_flag, progress)?;
|
||||||
validate_copy(&mut source, &mut hasher, ctx.length, control_flag, progress)?;
|
|
||||||
if !completed {
|
if !completed {
|
||||||
return Ok(false);
|
return Ok(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
let res = hex::encode(hasher.compute().0);
|
let res = hex::encode(hasher.finalize().0);
|
||||||
if res != ctx.checksum {
|
if res != ctx.checksum {
|
||||||
return Ok(false);
|
return Ok(false);
|
||||||
}
|
}
|
||||||
@ -1,5 +1,7 @@
|
|||||||
|
#![feature(iterator_try_collect)]
|
||||||
|
|
||||||
pub mod collections;
|
pub mod collections;
|
||||||
pub mod commands;
|
|
||||||
pub mod downloads;
|
pub mod downloads;
|
||||||
pub mod library;
|
pub mod library;
|
||||||
|
pub mod scan;
|
||||||
pub mod state;
|
pub mod state;
|
||||||
300
games/src/library.rs
Normal file
300
games/src/library.rs
Normal file
@ -0,0 +1,300 @@
|
|||||||
|
use bitcode::{Decode, Encode};
|
||||||
|
use database::{
|
||||||
|
ApplicationTransientStatus, Database, DownloadableMetadata, GameDownloadStatus, GameVersion,
|
||||||
|
borrow_db_checked, borrow_db_mut_checked,
|
||||||
|
};
|
||||||
|
use log::{debug, error, warn};
|
||||||
|
use remote::{
|
||||||
|
auth::generate_authorization_header, error::RemoteAccessError, requests::generate_url,
|
||||||
|
utils::DROP_CLIENT_SYNC,
|
||||||
|
};
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use std::fs::remove_dir_all;
|
||||||
|
use std::thread::spawn;
|
||||||
|
use tauri::AppHandle;
|
||||||
|
use utils::app_emit;
|
||||||
|
|
||||||
|
use crate::state::{GameStatusManager, GameStatusWithTransient};
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, Debug)]
|
||||||
|
pub struct FetchGameStruct {
|
||||||
|
game: Game,
|
||||||
|
status: GameStatusWithTransient,
|
||||||
|
version: Option<GameVersion>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FetchGameStruct {
|
||||||
|
pub fn new(game: Game, status: GameStatusWithTransient, version: Option<GameVersion>) -> Self {
|
||||||
|
Self {
|
||||||
|
game,
|
||||||
|
status,
|
||||||
|
version,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize, Clone, Debug, Default, Encode, Decode)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct Game {
|
||||||
|
id: String,
|
||||||
|
m_name: String,
|
||||||
|
m_short_description: String,
|
||||||
|
m_description: String,
|
||||||
|
// mDevelopers
|
||||||
|
// mPublishers
|
||||||
|
m_icon_object_id: String,
|
||||||
|
m_banner_object_id: String,
|
||||||
|
m_cover_object_id: String,
|
||||||
|
m_image_library_object_ids: Vec<String>,
|
||||||
|
m_image_carousel_object_ids: Vec<String>,
|
||||||
|
}
|
||||||
|
impl Game {
|
||||||
|
pub fn id(&self) -> &String {
|
||||||
|
&self.id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#[derive(serde::Serialize, Clone)]
|
||||||
|
pub struct GameUpdateEvent {
|
||||||
|
pub game_id: String,
|
||||||
|
pub status: (
|
||||||
|
Option<GameDownloadStatus>,
|
||||||
|
Option<ApplicationTransientStatus>,
|
||||||
|
),
|
||||||
|
pub version: Option<GameVersion>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Called by:
|
||||||
|
* - on_cancel, when cancelled, for obvious reasons
|
||||||
|
* - when downloading, so if drop unexpectedly quits, we can resume the download. hidden by the "Downloading..." transient state, though
|
||||||
|
* - when scanning, to import the game
|
||||||
|
*/
|
||||||
|
pub fn set_partially_installed(
|
||||||
|
meta: &DownloadableMetadata,
|
||||||
|
install_dir: String,
|
||||||
|
app_handle: Option<&AppHandle>,
|
||||||
|
) {
|
||||||
|
set_partially_installed_db(&mut borrow_db_mut_checked(), meta, install_dir, app_handle);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_partially_installed_db(
|
||||||
|
db_lock: &mut Database,
|
||||||
|
meta: &DownloadableMetadata,
|
||||||
|
install_dir: String,
|
||||||
|
app_handle: Option<&AppHandle>,
|
||||||
|
) {
|
||||||
|
db_lock.applications.transient_statuses.remove(meta);
|
||||||
|
db_lock.applications.game_statuses.insert(
|
||||||
|
meta.id.clone(),
|
||||||
|
GameDownloadStatus::PartiallyInstalled {
|
||||||
|
version_name: meta.version.as_ref().unwrap().clone(),
|
||||||
|
install_dir,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
db_lock
|
||||||
|
.applications
|
||||||
|
.installed_game_version
|
||||||
|
.insert(meta.id.clone(), meta.clone());
|
||||||
|
|
||||||
|
if let Some(app_handle) = app_handle {
|
||||||
|
push_game_update(
|
||||||
|
app_handle,
|
||||||
|
&meta.id,
|
||||||
|
None,
|
||||||
|
GameStatusManager::fetch_state(&meta.id, db_lock),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn uninstall_game_logic(meta: DownloadableMetadata, app_handle: &AppHandle) {
|
||||||
|
debug!("triggered uninstall for agent");
|
||||||
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.transient_statuses
|
||||||
|
.insert(meta.clone(), ApplicationTransientStatus::Uninstalling {});
|
||||||
|
|
||||||
|
push_game_update(
|
||||||
|
app_handle,
|
||||||
|
&meta.id,
|
||||||
|
None,
|
||||||
|
GameStatusManager::fetch_state(&meta.id, &db_handle),
|
||||||
|
);
|
||||||
|
|
||||||
|
let previous_state = db_handle.applications.game_statuses.get(&meta.id).cloned();
|
||||||
|
|
||||||
|
let previous_state = if let Some(state) = previous_state {
|
||||||
|
state
|
||||||
|
} else {
|
||||||
|
warn!("uninstall job doesn't have previous state, failing silently");
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some((_, install_dir)) = match previous_state {
|
||||||
|
GameDownloadStatus::Installed {
|
||||||
|
version_name,
|
||||||
|
install_dir,
|
||||||
|
} => Some((version_name, install_dir)),
|
||||||
|
GameDownloadStatus::SetupRequired {
|
||||||
|
version_name,
|
||||||
|
install_dir,
|
||||||
|
} => Some((version_name, install_dir)),
|
||||||
|
GameDownloadStatus::PartiallyInstalled {
|
||||||
|
version_name,
|
||||||
|
install_dir,
|
||||||
|
} => Some((version_name, install_dir)),
|
||||||
|
_ => None,
|
||||||
|
} {
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.transient_statuses
|
||||||
|
.insert(meta.clone(), ApplicationTransientStatus::Uninstalling {});
|
||||||
|
|
||||||
|
drop(db_handle);
|
||||||
|
|
||||||
|
let app_handle = app_handle.clone();
|
||||||
|
spawn(move || {
|
||||||
|
if let Err(e) = remove_dir_all(install_dir) {
|
||||||
|
error!("{e}");
|
||||||
|
} else {
|
||||||
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
|
db_handle.applications.transient_statuses.remove(&meta);
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.installed_game_version
|
||||||
|
.remove(&meta.id);
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.game_statuses
|
||||||
|
.insert(meta.id.clone(), GameDownloadStatus::Remote {});
|
||||||
|
let _ = db_handle.applications.transient_statuses.remove(&meta);
|
||||||
|
|
||||||
|
push_game_update(
|
||||||
|
&app_handle,
|
||||||
|
&meta.id,
|
||||||
|
None,
|
||||||
|
GameStatusManager::fetch_state(&meta.id, &db_handle),
|
||||||
|
);
|
||||||
|
|
||||||
|
debug!("uninstalled game id {}", &meta.id);
|
||||||
|
app_emit!(&app_handle, "update_library", ());
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
warn!("invalid previous state for uninstall, failing silently.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_current_meta(game_id: &String) -> Option<DownloadableMetadata> {
|
||||||
|
borrow_db_checked()
|
||||||
|
.applications
|
||||||
|
.installed_game_version
|
||||||
|
.get(game_id)
|
||||||
|
.cloned()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn on_game_complete(
|
||||||
|
meta: &DownloadableMetadata,
|
||||||
|
install_dir: String,
|
||||||
|
app_handle: &AppHandle,
|
||||||
|
) -> Result<(), RemoteAccessError> {
|
||||||
|
// Fetch game version information from remote
|
||||||
|
if meta.version.is_none() {
|
||||||
|
return Err(RemoteAccessError::GameNotFound(meta.id.clone()));
|
||||||
|
}
|
||||||
|
|
||||||
|
let client = DROP_CLIENT_SYNC.clone();
|
||||||
|
let response = generate_url(
|
||||||
|
&["/api/v1/client/game/version"],
|
||||||
|
&[
|
||||||
|
("id", &meta.id),
|
||||||
|
("version", meta.version.as_ref().unwrap()),
|
||||||
|
],
|
||||||
|
)?;
|
||||||
|
let response = client
|
||||||
|
.get(response)
|
||||||
|
.header("Authorization", generate_authorization_header())
|
||||||
|
.send()?;
|
||||||
|
|
||||||
|
let game_version: GameVersion = response.json()?;
|
||||||
|
|
||||||
|
let mut handle = borrow_db_mut_checked();
|
||||||
|
handle
|
||||||
|
.applications
|
||||||
|
.game_versions
|
||||||
|
.entry(meta.id.clone())
|
||||||
|
.or_default()
|
||||||
|
.insert(meta.version.clone().unwrap(), game_version.clone());
|
||||||
|
handle
|
||||||
|
.applications
|
||||||
|
.installed_game_version
|
||||||
|
.insert(meta.id.clone(), meta.clone());
|
||||||
|
|
||||||
|
drop(handle);
|
||||||
|
|
||||||
|
let status = if game_version.setup_command.is_empty() {
|
||||||
|
GameDownloadStatus::Installed {
|
||||||
|
version_name: meta.version.clone().unwrap(),
|
||||||
|
install_dir,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
GameDownloadStatus::SetupRequired {
|
||||||
|
version_name: meta.version.clone().unwrap(),
|
||||||
|
install_dir,
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.game_statuses
|
||||||
|
.insert(meta.id.clone(), status.clone());
|
||||||
|
drop(db_handle);
|
||||||
|
app_emit!(
|
||||||
|
app_handle,
|
||||||
|
&format!("update_game/{}", meta.id),
|
||||||
|
GameUpdateEvent {
|
||||||
|
game_id: meta.id.clone(),
|
||||||
|
status: (Some(status), None),
|
||||||
|
version: Some(game_version),
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn push_game_update(
|
||||||
|
app_handle: &AppHandle,
|
||||||
|
game_id: &String,
|
||||||
|
version: Option<GameVersion>,
|
||||||
|
status: GameStatusWithTransient,
|
||||||
|
) {
|
||||||
|
if let Some(GameDownloadStatus::Installed { .. } | GameDownloadStatus::SetupRequired { .. }) =
|
||||||
|
&status.0
|
||||||
|
&& version.is_none()
|
||||||
|
{
|
||||||
|
panic!("pushed game for installed game that doesn't have version information");
|
||||||
|
}
|
||||||
|
|
||||||
|
app_emit!(
|
||||||
|
app_handle,
|
||||||
|
&format!("update_game/{game_id}"),
|
||||||
|
GameUpdateEvent {
|
||||||
|
game_id: game_id.clone(),
|
||||||
|
status,
|
||||||
|
version,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct FrontendGameOptions {
|
||||||
|
launch_string: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FrontendGameOptions {
|
||||||
|
pub fn launch_string(&self) -> &String {
|
||||||
|
&self.launch_string
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,16 +1,11 @@
|
|||||||
use std::fs;
|
use std::fs;
|
||||||
|
|
||||||
|
use database::{DownloadType, DownloadableMetadata, borrow_db_mut_checked};
|
||||||
use log::warn;
|
use log::warn;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
database::{
|
downloads::drop_data::{DROP_DATA_PATH, DropData},
|
||||||
db::borrow_db_mut_checked,
|
|
||||||
models::data::{DownloadType, DownloadableMetadata},
|
|
||||||
},
|
|
||||||
games::{
|
|
||||||
downloads::drop_data::{DropData, DROP_DATA_PATH},
|
|
||||||
library::set_partially_installed_db,
|
library::set_partially_installed_db,
|
||||||
},
|
|
||||||
};
|
};
|
||||||
|
|
||||||
pub fn scan_install_dirs() {
|
pub fn scan_install_dirs() {
|
||||||
@ -1,4 +1,4 @@
|
|||||||
use crate::database::models::data::{
|
use database::models::data::{
|
||||||
ApplicationTransientStatus, Database, DownloadType, DownloadableMetadata, GameDownloadStatus,
|
ApplicationTransientStatus, Database, DownloadType, DownloadableMetadata, GameDownloadStatus,
|
||||||
};
|
};
|
||||||
|
|
||||||
19
process/Cargo.toml
Normal file
19
process/Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "process"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
chrono = "0.4.42"
|
||||||
|
client = { version = "0.1.0", path = "../client" }
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
dynfmt = "0.1.5"
|
||||||
|
games = { version = "0.1.0", path = "../games" }
|
||||||
|
log = "0.4.28"
|
||||||
|
page_size = "0.6.0"
|
||||||
|
serde = "1.0.228"
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
shared_child = "1.1.1"
|
||||||
|
tauri = "2.8.5"
|
||||||
|
tauri-plugin-opener = "2.5.0"
|
||||||
|
utils = { version = "0.1.0", path = "../utils" }
|
||||||
@ -12,7 +12,8 @@ pub enum ProcessError {
|
|||||||
FormatError(String), // String errors supremacy
|
FormatError(String), // String errors supremacy
|
||||||
InvalidPlatform,
|
InvalidPlatform,
|
||||||
OpenerError(tauri_plugin_opener::Error),
|
OpenerError(tauri_plugin_opener::Error),
|
||||||
InvalidArguments(String)
|
InvalidArguments(String),
|
||||||
|
FailedLaunch(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Display for ProcessError {
|
impl Display for ProcessError {
|
||||||
@ -26,7 +27,12 @@ impl Display for ProcessError {
|
|||||||
ProcessError::InvalidPlatform => "This game cannot be played on the current platform",
|
ProcessError::InvalidPlatform => "This game cannot be played on the current platform",
|
||||||
ProcessError::FormatError(error) => &format!("Could not format template: {error:?}"),
|
ProcessError::FormatError(error) => &format!("Could not format template: {error:?}"),
|
||||||
ProcessError::OpenerError(error) => &format!("Could not open directory: {error:?}"),
|
ProcessError::OpenerError(error) => &format!("Could not open directory: {error:?}"),
|
||||||
ProcessError::InvalidArguments(arguments) => &format!("Invalid arguments in command {arguments}"),
|
ProcessError::InvalidArguments(arguments) => {
|
||||||
|
&format!("Invalid arguments in command {arguments}")
|
||||||
|
}
|
||||||
|
ProcessError::FailedLaunch(game_id) => {
|
||||||
|
&format!("Drop detected that the game {game_id} may have failed to launch properly")
|
||||||
|
}
|
||||||
};
|
};
|
||||||
write!(f, "{s}")
|
write!(f, "{s}")
|
||||||
}
|
}
|
||||||
@ -8,7 +8,12 @@ pub struct DropFormatArgs {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl DropFormatArgs {
|
impl DropFormatArgs {
|
||||||
pub fn new(launch_string: String, working_dir: &String, executable_name: &String, absolute_executable_name: String) -> Self {
|
pub fn new(
|
||||||
|
launch_string: String,
|
||||||
|
working_dir: &String,
|
||||||
|
executable_name: &String,
|
||||||
|
absolute_executable_name: String,
|
||||||
|
) -> Self {
|
||||||
let mut positional = Vec::new();
|
let mut positional = Vec::new();
|
||||||
let mut map: HashMap<&'static str, String> = HashMap::new();
|
let mut map: HashMap<&'static str, String> = HashMap::new();
|
||||||
|
|
||||||
41
process/src/lib.rs
Normal file
41
process/src/lib.rs
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
#![feature(nonpoison_mutex)]
|
||||||
|
#![feature(sync_nonpoison)]
|
||||||
|
|
||||||
|
use std::{
|
||||||
|
ops::Deref,
|
||||||
|
sync::{OnceLock, nonpoison::Mutex},
|
||||||
|
};
|
||||||
|
|
||||||
|
use tauri::AppHandle;
|
||||||
|
|
||||||
|
use crate::process_manager::ProcessManager;
|
||||||
|
|
||||||
|
pub static PROCESS_MANAGER: ProcessManagerWrapper = ProcessManagerWrapper::new();
|
||||||
|
|
||||||
|
pub mod error;
|
||||||
|
pub mod format;
|
||||||
|
pub mod process_handlers;
|
||||||
|
pub mod process_manager;
|
||||||
|
|
||||||
|
pub struct ProcessManagerWrapper(OnceLock<Mutex<ProcessManager<'static>>>);
|
||||||
|
impl ProcessManagerWrapper {
|
||||||
|
const fn new() -> Self {
|
||||||
|
ProcessManagerWrapper(OnceLock::new())
|
||||||
|
}
|
||||||
|
pub fn init(app_handle: AppHandle) {
|
||||||
|
PROCESS_MANAGER
|
||||||
|
.0
|
||||||
|
.set(Mutex::new(ProcessManager::new(app_handle)))
|
||||||
|
.unwrap_or_else(|_| panic!("Failed to initialise Process Manager")); // Using panic! here because we can't implement Debug
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl Deref for ProcessManagerWrapper {
|
||||||
|
type Target = Mutex<ProcessManager<'static>>;
|
||||||
|
|
||||||
|
fn deref(&self) -> &Self::Target {
|
||||||
|
match self.0.get() {
|
||||||
|
Some(process_manager) => process_manager,
|
||||||
|
None => unreachable!("Download manager should always be initialised"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,18 +1,8 @@
|
|||||||
use std::{
|
use client::compat::{COMPAT_INFO, UMU_LAUNCHER_EXECUTABLE};
|
||||||
ffi::OsStr,
|
use database::{Database, DownloadableMetadata, GameVersion, platform::Platform};
|
||||||
path::PathBuf,
|
use log::debug;
|
||||||
process::{Command, Stdio},
|
|
||||||
sync::LazyLock,
|
|
||||||
};
|
|
||||||
|
|
||||||
use log::{debug, info};
|
use crate::{error::ProcessError, process_manager::ProcessHandler};
|
||||||
|
|
||||||
use crate::{
|
|
||||||
AppState,
|
|
||||||
database::models::data::{Database, DownloadableMetadata, GameVersion},
|
|
||||||
error::process_error::ProcessError,
|
|
||||||
process::process_manager::{Platform, ProcessHandler},
|
|
||||||
};
|
|
||||||
|
|
||||||
pub struct NativeGameLauncher;
|
pub struct NativeGameLauncher;
|
||||||
impl ProcessHandler for NativeGameLauncher {
|
impl ProcessHandler for NativeGameLauncher {
|
||||||
@ -27,36 +17,11 @@ impl ProcessHandler for NativeGameLauncher {
|
|||||||
Ok(format!("\"{}\" {}", launch_command, args.join(" ")))
|
Ok(format!("\"{}\" {}", launch_command, args.join(" ")))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn valid_for_platform(&self, _db: &Database, _state: &AppState, _target: &Platform) -> bool {
|
fn valid_for_platform(&self, _db: &Database, _target: &Platform) -> bool {
|
||||||
true
|
true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub static UMU_LAUNCHER_EXECUTABLE: LazyLock<Option<PathBuf>> = LazyLock::new(|| {
|
|
||||||
let x = get_umu_executable();
|
|
||||||
info!("{:?}", &x);
|
|
||||||
x
|
|
||||||
});
|
|
||||||
const UMU_BASE_LAUNCHER_EXECUTABLE: &str = "umu-run";
|
|
||||||
const UMU_INSTALL_DIRS: [&str; 4] = ["/app/share", "/use/local/share", "/usr/share", "/opt"];
|
|
||||||
|
|
||||||
fn get_umu_executable() -> Option<PathBuf> {
|
|
||||||
if check_executable_exists(UMU_BASE_LAUNCHER_EXECUTABLE) {
|
|
||||||
return Some(PathBuf::from(UMU_BASE_LAUNCHER_EXECUTABLE));
|
|
||||||
}
|
|
||||||
|
|
||||||
for dir in UMU_INSTALL_DIRS {
|
|
||||||
let p = PathBuf::from(dir).join(UMU_BASE_LAUNCHER_EXECUTABLE);
|
|
||||||
if check_executable_exists(&p) {
|
|
||||||
return Some(p);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
None
|
|
||||||
}
|
|
||||||
fn check_executable_exists<P: AsRef<OsStr>>(exec: P) -> bool {
|
|
||||||
let has_umu_installed = Command::new(exec).stdout(Stdio::null()).output();
|
|
||||||
has_umu_installed.is_ok()
|
|
||||||
}
|
|
||||||
pub struct UMULauncher;
|
pub struct UMULauncher;
|
||||||
impl ProcessHandler for UMULauncher {
|
impl ProcessHandler for UMULauncher {
|
||||||
fn create_launch_process(
|
fn create_launch_process(
|
||||||
@ -80,14 +45,16 @@ impl ProcessHandler for UMULauncher {
|
|||||||
};
|
};
|
||||||
Ok(format!(
|
Ok(format!(
|
||||||
"GAMEID={game_id} {umu:?} \"{launch}\" {args}",
|
"GAMEID={game_id} {umu:?} \"{launch}\" {args}",
|
||||||
umu = UMU_LAUNCHER_EXECUTABLE.as_ref().expect("Failed to get UMU_LAUNCHER_EXECUTABLE as ref"),
|
umu = UMU_LAUNCHER_EXECUTABLE
|
||||||
|
.as_ref()
|
||||||
|
.expect("Failed to get UMU_LAUNCHER_EXECUTABLE as ref"),
|
||||||
launch = launch_command,
|
launch = launch_command,
|
||||||
args = args.join(" ")
|
args = args.join(" ")
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn valid_for_platform(&self, _db: &Database, state: &AppState, _target: &Platform) -> bool {
|
fn valid_for_platform(&self, _db: &Database, _target: &Platform) -> bool {
|
||||||
let Some(ref compat_info) = state.compat_info else {
|
let Some(compat_info) = &*COMPAT_INFO else {
|
||||||
return false;
|
return false;
|
||||||
};
|
};
|
||||||
compat_info.umu_installed
|
compat_info.umu_installed
|
||||||
@ -120,14 +87,19 @@ impl ProcessHandler for AsahiMuvmLauncher {
|
|||||||
.next()
|
.next()
|
||||||
.ok_or(ProcessError::InvalidArguments(umu_string.clone()))?
|
.ok_or(ProcessError::InvalidArguments(umu_string.clone()))?
|
||||||
.trim();
|
.trim();
|
||||||
let cmd = format!("umu-run{}", args_cmd.next().ok_or(ProcessError::InvalidArguments(umu_string.clone()))?);
|
let cmd = format!(
|
||||||
|
"umu-run{}",
|
||||||
|
args_cmd
|
||||||
|
.next()
|
||||||
|
.ok_or(ProcessError::InvalidArguments(umu_string.clone()))?
|
||||||
|
);
|
||||||
|
|
||||||
Ok(format!("{args} muvm -- {cmd}"))
|
Ok(format!("{args} muvm -- {cmd}"))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[allow(unreachable_code)]
|
#[allow(unreachable_code)]
|
||||||
#[allow(unused_variables)]
|
#[allow(unused_variables)]
|
||||||
fn valid_for_platform(&self, _db: &Database, state: &AppState, _target: &Platform) -> bool {
|
fn valid_for_platform(&self, _db: &Database, _target: &Platform) -> bool {
|
||||||
#[cfg(not(target_os = "linux"))]
|
#[cfg(not(target_os = "linux"))]
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
@ -139,7 +111,7 @@ impl ProcessHandler for AsahiMuvmLauncher {
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
let Some(ref compat_info) = state.compat_info else {
|
let Some(compat_info) = &*COMPAT_INFO else {
|
||||||
return false;
|
return false;
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -1,39 +1,31 @@
|
|||||||
use std::{
|
use std::{
|
||||||
collections::HashMap,
|
collections::HashMap,
|
||||||
fs::{OpenOptions, create_dir_all},
|
fs::{OpenOptions, create_dir_all},
|
||||||
io::{self},
|
io,
|
||||||
path::PathBuf,
|
path::PathBuf,
|
||||||
process::{Command, ExitStatus},
|
process::{Command, ExitStatus},
|
||||||
str::FromStr,
|
str::FromStr,
|
||||||
sync::{Arc, Mutex},
|
sync::Arc,
|
||||||
thread::spawn,
|
thread::spawn,
|
||||||
time::{Duration, SystemTime},
|
time::{Duration, SystemTime},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use database::{
|
||||||
|
ApplicationTransientStatus, Database, DownloadType, DownloadableMetadata, GameDownloadStatus,
|
||||||
|
GameVersion, borrow_db_checked, borrow_db_mut_checked, db::DATA_ROOT_DIR, platform::Platform,
|
||||||
|
};
|
||||||
use dynfmt::Format;
|
use dynfmt::Format;
|
||||||
use dynfmt::SimpleCurlyFormat;
|
use dynfmt::SimpleCurlyFormat;
|
||||||
|
use games::{library::push_game_update, state::GameStatusManager};
|
||||||
use log::{debug, info, warn};
|
use log::{debug, info, warn};
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use shared_child::SharedChild;
|
use shared_child::SharedChild;
|
||||||
use tauri::{AppHandle, Emitter, Manager};
|
use tauri::AppHandle;
|
||||||
use tauri_plugin_opener::OpenerExt;
|
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
AppState,
|
PROCESS_MANAGER,
|
||||||
database::{
|
error::ProcessError,
|
||||||
db::{DATA_ROOT_DIR, borrow_db_checked, borrow_db_mut_checked},
|
|
||||||
models::data::{
|
|
||||||
ApplicationTransientStatus, Database, DownloadType, DownloadableMetadata,
|
|
||||||
GameDownloadStatus, GameVersion,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
error::process_error::ProcessError,
|
|
||||||
games::{library::push_game_update, state::GameStatusManager},
|
|
||||||
process::{
|
|
||||||
format::DropFormatArgs,
|
format::DropFormatArgs,
|
||||||
process_handlers::{AsahiMuvmLauncher, NativeGameLauncher, UMULauncher},
|
process_handlers::{AsahiMuvmLauncher, NativeGameLauncher, UMULauncher},
|
||||||
},
|
|
||||||
lock,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
pub struct RunningProcess {
|
pub struct RunningProcess {
|
||||||
@ -46,11 +38,11 @@ pub struct ProcessManager<'a> {
|
|||||||
current_platform: Platform,
|
current_platform: Platform,
|
||||||
log_output_dir: PathBuf,
|
log_output_dir: PathBuf,
|
||||||
processes: HashMap<String, RunningProcess>,
|
processes: HashMap<String, RunningProcess>,
|
||||||
app_handle: AppHandle,
|
|
||||||
game_launchers: Vec<(
|
game_launchers: Vec<(
|
||||||
(Platform, Platform),
|
(Platform, Platform),
|
||||||
&'a (dyn ProcessHandler + Sync + Send + 'static),
|
&'a (dyn ProcessHandler + Sync + Send + 'static),
|
||||||
)>,
|
)>,
|
||||||
|
app_handle: AppHandle,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ProcessManager<'_> {
|
impl ProcessManager<'_> {
|
||||||
@ -67,7 +59,6 @@ impl ProcessManager<'_> {
|
|||||||
#[cfg(target_os = "linux")]
|
#[cfg(target_os = "linux")]
|
||||||
current_platform: Platform::Linux,
|
current_platform: Platform::Linux,
|
||||||
|
|
||||||
app_handle,
|
|
||||||
processes: HashMap::new(),
|
processes: HashMap::new(),
|
||||||
log_output_dir,
|
log_output_dir,
|
||||||
game_launchers: vec![
|
game_launchers: vec![
|
||||||
@ -93,6 +84,7 @@ impl ProcessManager<'_> {
|
|||||||
&UMULauncher {} as &(dyn ProcessHandler + Sync + Send + 'static),
|
&UMULauncher {} as &(dyn ProcessHandler + Sync + Send + 'static),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
|
app_handle,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -111,25 +103,20 @@ impl ProcessManager<'_> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_log_dir(&self, game_id: String) -> PathBuf {
|
pub fn get_log_dir(&self, game_id: String) -> PathBuf {
|
||||||
self.log_output_dir.join(game_id)
|
self.log_output_dir.join(game_id)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn open_process_logs(&mut self, game_id: String) -> Result<(), ProcessError> {
|
fn on_process_finish(
|
||||||
let dir = self.get_log_dir(game_id);
|
&mut self,
|
||||||
self.app_handle
|
game_id: String,
|
||||||
.opener()
|
result: Result<ExitStatus, std::io::Error>,
|
||||||
.open_path(dir.display().to_string(), None::<&str>)
|
) -> Result<(), ProcessError> {
|
||||||
.map_err(ProcessError::OpenerError)?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn on_process_finish(&mut self, game_id: String, result: Result<ExitStatus, std::io::Error>) {
|
|
||||||
if !self.processes.contains_key(&game_id) {
|
if !self.processes.contains_key(&game_id) {
|
||||||
warn!(
|
warn!(
|
||||||
"process on_finish was called, but game_id is no longer valid. finished with result: {result:?}"
|
"process on_finish was called, but game_id is no longer valid. finished with result: {result:?}"
|
||||||
);
|
);
|
||||||
return;
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
debug!("process for {:?} exited with {:?}", &game_id, result);
|
debug!("process for {:?} exited with {:?}", &game_id, result);
|
||||||
@ -138,7 +125,7 @@ impl ProcessManager<'_> {
|
|||||||
Some(process) => process,
|
Some(process) => process,
|
||||||
None => {
|
None => {
|
||||||
info!("Attempted to stop process {game_id} which didn't exist");
|
info!("Attempted to stop process {game_id} which didn't exist");
|
||||||
return;
|
return Ok(());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -176,7 +163,8 @@ impl ProcessManager<'_> {
|
|||||||
&& (elapsed.as_secs() <= 2 || result.map_or(true, |r| !r.success()))
|
&& (elapsed.as_secs() <= 2 || result.map_or(true, |r| !r.success()))
|
||||||
{
|
{
|
||||||
warn!("drop detected that the game {game_id} may have failed to launch properly");
|
warn!("drop detected that the game {game_id} may have failed to launch properly");
|
||||||
let _ = self.app_handle.emit("launch_external_error", &game_id);
|
return Err(ProcessError::FailedLaunch(game_id));
|
||||||
|
// let _ = self.app_handle.emit("launch_external_error", &game_id);
|
||||||
}
|
}
|
||||||
|
|
||||||
let version_data = match db_handle.applications.game_versions.get(&game_id) {
|
let version_data = match db_handle.applications.game_versions.get(&game_id) {
|
||||||
@ -193,12 +181,12 @@ impl ProcessManager<'_> {
|
|||||||
Some(version_data.clone()),
|
Some(version_data.clone()),
|
||||||
status,
|
status,
|
||||||
);
|
);
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn fetch_process_handler(
|
fn fetch_process_handler(
|
||||||
&self,
|
&self,
|
||||||
db_lock: &Database,
|
db_lock: &Database,
|
||||||
state: &AppState,
|
|
||||||
target_platform: &Platform,
|
target_platform: &Platform,
|
||||||
) -> Result<&(dyn ProcessHandler + Send + Sync), ProcessError> {
|
) -> Result<&(dyn ProcessHandler + Send + Sync), ProcessError> {
|
||||||
Ok(self
|
Ok(self
|
||||||
@ -208,23 +196,20 @@ impl ProcessManager<'_> {
|
|||||||
let (e_current, e_target) = e.0;
|
let (e_current, e_target) = e.0;
|
||||||
e_current == self.current_platform
|
e_current == self.current_platform
|
||||||
&& e_target == *target_platform
|
&& e_target == *target_platform
|
||||||
&& e.1.valid_for_platform(db_lock, state, target_platform)
|
&& e.1.valid_for_platform(db_lock, target_platform)
|
||||||
})
|
})
|
||||||
.ok_or(ProcessError::InvalidPlatform)?
|
.ok_or(ProcessError::InvalidPlatform)?
|
||||||
.1)
|
.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn valid_platform(&self, platform: &Platform, state: &AppState) -> bool {
|
pub fn valid_platform(&self, platform: &Platform) -> bool {
|
||||||
let db_lock = borrow_db_checked();
|
let db_lock = borrow_db_checked();
|
||||||
let process_handler = self.fetch_process_handler(&db_lock, state, platform);
|
let process_handler = self.fetch_process_handler(&db_lock, platform);
|
||||||
process_handler.is_ok()
|
process_handler.is_ok()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn launch_process(
|
/// Must be called through spawn as it is currently blocking
|
||||||
&mut self,
|
pub fn launch_process(&mut self, game_id: String) -> Result<(), ProcessError> {
|
||||||
game_id: String,
|
|
||||||
state: &AppState,
|
|
||||||
) -> Result<(), ProcessError> {
|
|
||||||
if self.processes.contains_key(&game_id) {
|
if self.processes.contains_key(&game_id) {
|
||||||
return Err(ProcessError::AlreadyRunning);
|
return Err(ProcessError::AlreadyRunning);
|
||||||
}
|
}
|
||||||
@ -306,7 +291,7 @@ impl ProcessManager<'_> {
|
|||||||
|
|
||||||
let target_platform = game_version.platform;
|
let target_platform = game_version.platform;
|
||||||
|
|
||||||
let process_handler = self.fetch_process_handler(&db_lock, state, &target_platform)?;
|
let process_handler = self.fetch_process_handler(&db_lock, &target_platform)?;
|
||||||
|
|
||||||
let (launch, args) = match game_status {
|
let (launch, args) = match game_status {
|
||||||
GameDownloadStatus::Installed {
|
GameDownloadStatus::Installed {
|
||||||
@ -388,27 +373,8 @@ impl ProcessManager<'_> {
|
|||||||
);
|
);
|
||||||
|
|
||||||
let wait_thread_handle = launch_process_handle.clone();
|
let wait_thread_handle = launch_process_handle.clone();
|
||||||
let wait_thread_apphandle = self.app_handle.clone();
|
|
||||||
let wait_thread_game_id = meta.clone();
|
let wait_thread_game_id = meta.clone();
|
||||||
|
|
||||||
spawn(move || {
|
|
||||||
let result: Result<ExitStatus, std::io::Error> = launch_process_handle.wait();
|
|
||||||
|
|
||||||
let app_state = wait_thread_apphandle.state::<Mutex<AppState>>();
|
|
||||||
let app_state_handle = lock!(app_state);
|
|
||||||
|
|
||||||
let mut process_manager_handle = app_state_handle
|
|
||||||
.process_manager
|
|
||||||
.lock()
|
|
||||||
.expect("Failed to lock onto process manager");
|
|
||||||
process_manager_handle.on_process_finish(wait_thread_game_id.id, result);
|
|
||||||
|
|
||||||
// As everything goes out of scope, they should get dropped
|
|
||||||
// But just to explicit about it
|
|
||||||
drop(process_manager_handle);
|
|
||||||
drop(app_state_handle);
|
|
||||||
});
|
|
||||||
|
|
||||||
self.processes.insert(
|
self.processes.insert(
|
||||||
meta.id,
|
meta.id,
|
||||||
RunningProcess {
|
RunningProcess {
|
||||||
@ -417,55 +383,17 @@ impl ProcessManager<'_> {
|
|||||||
manually_killed: false,
|
manually_killed: false,
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
|
spawn(move || {
|
||||||
|
let result: Result<ExitStatus, std::io::Error> = launch_process_handle.wait();
|
||||||
|
|
||||||
|
PROCESS_MANAGER
|
||||||
|
.lock()
|
||||||
|
.on_process_finish(wait_thread_game_id.id, result)
|
||||||
|
});
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Eq, Hash, PartialEq, Serialize, Deserialize, Clone, Copy, Debug)]
|
|
||||||
pub enum Platform {
|
|
||||||
Windows,
|
|
||||||
Linux,
|
|
||||||
MacOs,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Platform {
|
|
||||||
#[cfg(target_os = "windows")]
|
|
||||||
pub const HOST: Platform = Self::Windows;
|
|
||||||
#[cfg(target_os = "macos")]
|
|
||||||
pub const HOST: Platform = Self::MacOs;
|
|
||||||
#[cfg(target_os = "linux")]
|
|
||||||
pub const HOST: Platform = Self::Linux;
|
|
||||||
|
|
||||||
pub fn is_case_sensitive(&self) -> bool {
|
|
||||||
match self {
|
|
||||||
Self::Windows | Self::MacOs => false,
|
|
||||||
Self::Linux => true,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<&str> for Platform {
|
|
||||||
fn from(value: &str) -> Self {
|
|
||||||
match value.to_lowercase().trim() {
|
|
||||||
"windows" => Self::Windows,
|
|
||||||
"linux" => Self::Linux,
|
|
||||||
"mac" | "macos" => Self::MacOs,
|
|
||||||
_ => unimplemented!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<whoami::Platform> for Platform {
|
|
||||||
fn from(value: whoami::Platform) -> Self {
|
|
||||||
match value {
|
|
||||||
whoami::Platform::Windows => Platform::Windows,
|
|
||||||
whoami::Platform::Linux => Platform::Linux,
|
|
||||||
whoami::Platform::MacOS => Platform::MacOs,
|
|
||||||
_ => unimplemented!(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait ProcessHandler: Send + 'static {
|
pub trait ProcessHandler: Send + 'static {
|
||||||
fn create_launch_process(
|
fn create_launch_process(
|
||||||
&self,
|
&self,
|
||||||
@ -476,5 +404,5 @@ pub trait ProcessHandler: Send + 'static {
|
|||||||
current_dir: &str,
|
current_dir: &str,
|
||||||
) -> Result<String, ProcessError>;
|
) -> Result<String, ProcessError>;
|
||||||
|
|
||||||
fn valid_for_platform(&self, db: &Database, state: &AppState, target: &Platform) -> bool;
|
fn valid_for_platform(&self, db: &Database, target: &Platform) -> bool;
|
||||||
}
|
}
|
||||||
23
remote/Cargo.toml
Normal file
23
remote/Cargo.toml
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
[package]
|
||||||
|
name = "remote"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
bitcode = "0.6.7"
|
||||||
|
chrono = "0.4.42"
|
||||||
|
client = { version = "0.1.0", path = "../client" }
|
||||||
|
database = { version = "0.1.0", path = "../database" }
|
||||||
|
droplet-rs = "0.7.3"
|
||||||
|
gethostname = "1.0.2"
|
||||||
|
hex = "0.4.3"
|
||||||
|
http = "1.3.1"
|
||||||
|
log = "0.4.28"
|
||||||
|
md5 = "0.8.0"
|
||||||
|
reqwest = "0.12.23"
|
||||||
|
reqwest-websocket = "0.5.1"
|
||||||
|
serde = "1.0.228"
|
||||||
|
serde_with = "3.15.0"
|
||||||
|
tauri = "2.8.5"
|
||||||
|
url = "2.5.7"
|
||||||
|
utils = { version = "0.1.0", path = "../utils" }
|
||||||
@ -1,18 +1,18 @@
|
|||||||
use std::{collections::HashMap, env, sync::Mutex};
|
use std::{collections::HashMap, env};
|
||||||
|
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
|
use client::{app_status::AppStatus, user::User};
|
||||||
|
use database::{DatabaseAuth, interface::borrow_db_checked};
|
||||||
use droplet_rs::ssl::sign_nonce;
|
use droplet_rs::ssl::sign_nonce;
|
||||||
use gethostname::gethostname;
|
use gethostname::gethostname;
|
||||||
use log::{debug, error, warn};
|
use log::{error, warn};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use tauri::{AppHandle, Emitter, Manager};
|
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
app_emit, database::{
|
error::{DropServerError, RemoteAccessError},
|
||||||
db::{borrow_db_checked, borrow_db_mut_checked},
|
requests::make_authenticated_get,
|
||||||
models::data::DatabaseAuth,
|
utils::DROP_CLIENT_SYNC,
|
||||||
}, error::{drop_server_error::DropServerError, remote_access_error::RemoteAccessError}, lock, remote::{cache::clear_cached_object, requests::make_authenticated_get, utils::{DROP_CLIENT_ASYNC, DROP_CLIENT_SYNC}}, AppState, AppStatus, User
|
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
@ -35,19 +35,31 @@ struct InitiateRequestBody {
|
|||||||
|
|
||||||
#[derive(Serialize)]
|
#[derive(Serialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
struct HandshakeRequestBody {
|
pub struct HandshakeRequestBody {
|
||||||
client_id: String,
|
client_id: String,
|
||||||
token: String,
|
token: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl HandshakeRequestBody {
|
||||||
|
pub fn new(client_id: String, token: String) -> Self {
|
||||||
|
Self { client_id, token }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
struct HandshakeResponse {
|
pub struct HandshakeResponse {
|
||||||
private: String,
|
private: String,
|
||||||
certificate: String,
|
certificate: String,
|
||||||
id: String,
|
id: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl From<HandshakeResponse> for DatabaseAuth {
|
||||||
|
fn from(value: HandshakeResponse) -> Self {
|
||||||
|
DatabaseAuth::new(value.private, value.certificate, value.id, None)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn generate_authorization_header() -> String {
|
pub fn generate_authorization_header() -> String {
|
||||||
let certs = {
|
let certs = {
|
||||||
let db = borrow_db_checked();
|
let db = borrow_db_checked();
|
||||||
@ -81,94 +93,6 @@ pub async fn fetch_user() -> Result<User, RemoteAccessError> {
|
|||||||
.map_err(std::convert::Into::into)
|
.map_err(std::convert::Into::into)
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn recieve_handshake_logic(app: &AppHandle, path: String) -> Result<(), RemoteAccessError> {
|
|
||||||
let path_chunks: Vec<&str> = path.split('/').collect();
|
|
||||||
if path_chunks.len() != 3 {
|
|
||||||
app_emit!(app, "auth/failed", ());
|
|
||||||
return Err(RemoteAccessError::HandshakeFailed(
|
|
||||||
"failed to parse token".to_string(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
let base_url = {
|
|
||||||
let handle = borrow_db_checked();
|
|
||||||
Url::parse(handle.base_url.as_str())?
|
|
||||||
};
|
|
||||||
|
|
||||||
let client_id = path_chunks
|
|
||||||
.get(1)
|
|
||||||
.expect("Failed to get client id from path chunks");
|
|
||||||
let token = path_chunks
|
|
||||||
.get(2)
|
|
||||||
.expect("Failed to get token from path chunks");
|
|
||||||
let body = HandshakeRequestBody {
|
|
||||||
client_id: (client_id).to_string(),
|
|
||||||
token: (token).to_string(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let endpoint = base_url.join("/api/v1/client/auth/handshake")?;
|
|
||||||
let client = DROP_CLIENT_ASYNC.clone();
|
|
||||||
let response = client.post(endpoint).json(&body).send().await?;
|
|
||||||
debug!("handshake responsded with {}", response.status().as_u16());
|
|
||||||
if !response.status().is_success() {
|
|
||||||
return Err(RemoteAccessError::InvalidResponse(response.json().await?));
|
|
||||||
}
|
|
||||||
let response_struct: HandshakeResponse = response.json().await?;
|
|
||||||
|
|
||||||
{
|
|
||||||
let mut handle = borrow_db_mut_checked();
|
|
||||||
handle.auth = Some(DatabaseAuth {
|
|
||||||
private: response_struct.private,
|
|
||||||
cert: response_struct.certificate,
|
|
||||||
client_id: response_struct.id,
|
|
||||||
web_token: None,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
let web_token = {
|
|
||||||
let header = generate_authorization_header();
|
|
||||||
let token = client
|
|
||||||
.post(base_url.join("/api/v1/client/user/webtoken")?)
|
|
||||||
.header("Authorization", header)
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
token.text().await?
|
|
||||||
};
|
|
||||||
let mut handle = borrow_db_mut_checked();
|
|
||||||
handle.auth.as_mut().unwrap().web_token = Some(web_token);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn recieve_handshake(app: AppHandle, path: String) {
|
|
||||||
// Tell the app we're processing
|
|
||||||
app_emit!(app, "auth/processing", ());
|
|
||||||
|
|
||||||
let handshake_result = recieve_handshake_logic(&app, path).await;
|
|
||||||
if let Err(e) = handshake_result {
|
|
||||||
warn!("error with authentication: {e}");
|
|
||||||
app_emit!(app, "auth/failed", e.to_string());
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let app_state = app.state::<Mutex<AppState>>();
|
|
||||||
|
|
||||||
let (app_status, user) = setup().await;
|
|
||||||
|
|
||||||
let mut state_lock = lock!(app_state);
|
|
||||||
|
|
||||||
state_lock.status = app_status;
|
|
||||||
state_lock.user = user;
|
|
||||||
|
|
||||||
let _ = clear_cached_object("collections");
|
|
||||||
let _ = clear_cached_object("library");
|
|
||||||
|
|
||||||
drop(state_lock);
|
|
||||||
|
|
||||||
app_emit!(app, "auth/finished", ());
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn auth_initiate_logic(mode: String) -> Result<String, RemoteAccessError> {
|
pub fn auth_initiate_logic(mode: String) -> Result<String, RemoteAccessError> {
|
||||||
let base_url = {
|
let base_url = {
|
||||||
let db_lock = borrow_db_checked();
|
let db_lock = borrow_db_checked();
|
||||||
@ -5,18 +5,19 @@ use std::{
|
|||||||
time::SystemTime,
|
time::SystemTime,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{
|
|
||||||
database::{db::borrow_db_checked, models::data::Database},
|
|
||||||
error::{cache_error::CacheError, remote_access_error::RemoteAccessError},
|
|
||||||
};
|
|
||||||
use bitcode::{Decode, DecodeOwned, Encode};
|
use bitcode::{Decode, DecodeOwned, Encode};
|
||||||
use http::{header::{CONTENT_TYPE}, response::Builder as ResponseBuilder, Response};
|
use database::{Database, borrow_db_checked};
|
||||||
|
use http::{Response, header::CONTENT_TYPE, response::Builder as ResponseBuilder};
|
||||||
|
|
||||||
|
use crate::error::{CacheError, RemoteAccessError};
|
||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! offline {
|
macro_rules! offline {
|
||||||
($var:expr, $func1:expr, $func2:expr, $( $arg:expr ),* ) => {
|
($var:expr, $func1:expr, $func2:expr, $( $arg:expr ),* ) => {
|
||||||
|
|
||||||
async move { if $crate::borrow_db_checked().settings.force_offline || $crate::lock!($var).status == $crate::AppStatus::Offline {
|
async move {
|
||||||
|
if ::database::borrow_db_checked().settings.force_offline
|
||||||
|
|| $var.lock().status == ::client::app_status::AppStatus::Offline {
|
||||||
$func2( $( $arg ), *).await
|
$func2( $( $arg ), *).await
|
||||||
} else {
|
} else {
|
||||||
$func1( $( $arg ), *).await
|
$func1( $( $arg ), *).await
|
||||||
@ -82,10 +83,7 @@ pub fn get_cached_object_db<D: DecodeOwned>(
|
|||||||
pub fn clear_cached_object(key: &str) -> Result<(), RemoteAccessError> {
|
pub fn clear_cached_object(key: &str) -> Result<(), RemoteAccessError> {
|
||||||
clear_cached_object_db(key, &borrow_db_checked())
|
clear_cached_object_db(key, &borrow_db_checked())
|
||||||
}
|
}
|
||||||
pub fn clear_cached_object_db(
|
pub fn clear_cached_object_db(key: &str, db: &Database) -> Result<(), RemoteAccessError> {
|
||||||
key: &str,
|
|
||||||
db: &Database,
|
|
||||||
) -> Result<(), RemoteAccessError> {
|
|
||||||
delete_sync(&db.cache_dir, key).map_err(RemoteAccessError::Cache)?;
|
delete_sync(&db.cache_dir, key).map_err(RemoteAccessError::Cache)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@ -119,14 +117,15 @@ impl TryFrom<Response<Vec<u8>>> for ObjectCache {
|
|||||||
body: value.body().clone(),
|
body: value.body().clone(),
|
||||||
expiry: get_sys_time_in_secs() + 60 * 60 * 24,
|
expiry: get_sys_time_in_secs() + 60 * 60 * 24,
|
||||||
})
|
})
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
impl TryFrom<ObjectCache> for Response<Vec<u8>> {
|
impl TryFrom<ObjectCache> for Response<Vec<u8>> {
|
||||||
type Error = CacheError;
|
type Error = CacheError;
|
||||||
fn try_from(value: ObjectCache) -> Result<Self, Self::Error> {
|
fn try_from(value: ObjectCache) -> Result<Self, Self::Error> {
|
||||||
let resp_builder = ResponseBuilder::new().header(CONTENT_TYPE, value.content_type);
|
let resp_builder = ResponseBuilder::new().header(CONTENT_TYPE, value.content_type);
|
||||||
resp_builder.body(value.body).map_err(CacheError::ConstructionError)
|
resp_builder
|
||||||
|
.body(value.body)
|
||||||
|
.map_err(CacheError::ConstructionError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
impl TryFrom<&ObjectCache> for Response<Vec<u8>> {
|
impl TryFrom<&ObjectCache> for Response<Vec<u8>> {
|
||||||
@ -134,6 +133,8 @@ impl TryFrom<&ObjectCache> for Response<Vec<u8>> {
|
|||||||
|
|
||||||
fn try_from(value: &ObjectCache) -> Result<Self, Self::Error> {
|
fn try_from(value: &ObjectCache) -> Result<Self, Self::Error> {
|
||||||
let resp_builder = ResponseBuilder::new().header(CONTENT_TYPE, value.content_type.clone());
|
let resp_builder = ResponseBuilder::new().header(CONTENT_TYPE, value.content_type.clone());
|
||||||
resp_builder.body(value.body.clone()).map_err(CacheError::ConstructionError)
|
resp_builder
|
||||||
|
.body(value.body.clone())
|
||||||
|
.map_err(CacheError::ConstructionError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -4,11 +4,20 @@ use std::{
|
|||||||
sync::Arc,
|
sync::Arc,
|
||||||
};
|
};
|
||||||
|
|
||||||
use http::StatusCode;
|
use http::{HeaderName, StatusCode, header::ToStrError};
|
||||||
use serde_with::SerializeDisplay;
|
use serde_with::SerializeDisplay;
|
||||||
use url::ParseError;
|
use url::ParseError;
|
||||||
|
|
||||||
use super::drop_server_error::DropServerError;
|
use serde::Deserialize;
|
||||||
|
|
||||||
|
#[derive(Deserialize, Debug, Clone)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct DropServerError {
|
||||||
|
pub status_code: usize,
|
||||||
|
pub status_message: String,
|
||||||
|
// pub message: String,
|
||||||
|
// pub url: String,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, SerializeDisplay)]
|
#[derive(Debug, SerializeDisplay)]
|
||||||
pub enum RemoteAccessError {
|
pub enum RemoteAccessError {
|
||||||
@ -104,3 +113,31 @@ impl From<ParseError> for RemoteAccessError {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
impl std::error::Error for RemoteAccessError {}
|
impl std::error::Error for RemoteAccessError {}
|
||||||
|
|
||||||
|
#[derive(Debug, SerializeDisplay)]
|
||||||
|
pub enum CacheError {
|
||||||
|
HeaderNotFound(HeaderName),
|
||||||
|
ParseError(ToStrError),
|
||||||
|
Remote(RemoteAccessError),
|
||||||
|
ConstructionError(http::Error),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Display for CacheError {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
let s = match self {
|
||||||
|
CacheError::HeaderNotFound(header_name) => {
|
||||||
|
format!("Could not find header {header_name} in cache")
|
||||||
|
}
|
||||||
|
CacheError::ParseError(to_str_error) => {
|
||||||
|
format!("Could not parse cache with error {to_str_error}")
|
||||||
|
}
|
||||||
|
CacheError::Remote(remote_access_error) => {
|
||||||
|
format!("Cache got remote access error: {remote_access_error}")
|
||||||
|
}
|
||||||
|
CacheError::ConstructionError(error) => {
|
||||||
|
format!("Could not construct cache body with error {error}")
|
||||||
|
}
|
||||||
|
};
|
||||||
|
write!(f, "{s}")
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,8 +1,9 @@
|
|||||||
use http::{header::CONTENT_TYPE, response::Builder as ResponseBuilder, Response};
|
use database::{DB, interface::DatabaseImpls};
|
||||||
|
use http::{Response, header::CONTENT_TYPE, response::Builder as ResponseBuilder};
|
||||||
use log::{debug, warn};
|
use log::{debug, warn};
|
||||||
use tauri::UriSchemeResponder;
|
use tauri::UriSchemeResponder;
|
||||||
|
|
||||||
use crate::{database::db::DatabaseImpls, error::cache_error::CacheError, remote::utils::DROP_CLIENT_ASYNC, DB};
|
use crate::{error::CacheError, utils::DROP_CLIENT_ASYNC};
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
auth::generate_authorization_header,
|
auth::generate_authorization_header,
|
||||||
@ -14,13 +15,19 @@ pub async fn fetch_object_wrapper(request: http::Request<Vec<u8>>, responder: Ur
|
|||||||
Ok(r) => responder.respond(r),
|
Ok(r) => responder.respond(r),
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
warn!("Cache error: {e}");
|
warn!("Cache error: {e}");
|
||||||
responder.respond(Response::builder().status(500).body(Vec::new()).expect("Failed to build error response"));
|
responder.respond(
|
||||||
|
Response::builder()
|
||||||
|
.status(500)
|
||||||
|
.body(Vec::new())
|
||||||
|
.expect("Failed to build error response"),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn fetch_object(request: http::Request<Vec<u8>>) -> Result<Response<Vec<u8>>, CacheError>
|
pub async fn fetch_object(
|
||||||
{
|
request: http::Request<Vec<u8>>,
|
||||||
|
) -> Result<Response<Vec<u8>>, CacheError> {
|
||||||
// Drop leading /
|
// Drop leading /
|
||||||
let object_id = &request.uri().path()[1..];
|
let object_id = &request.uri().path()[1..];
|
||||||
|
|
||||||
@ -47,13 +54,13 @@ pub async fn fetch_object(request: http::Request<Vec<u8>>) -> Result<Response<Ve
|
|||||||
let data = match r.bytes().await {
|
let data = match r.bytes().await {
|
||||||
Ok(data) => Vec::from(data),
|
Ok(data) => Vec::from(data),
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
warn!(
|
warn!("Could not get data from cache object {object_id} with error {e}",);
|
||||||
"Could not get data from cache object {object_id} with error {e}",
|
|
||||||
);
|
|
||||||
Vec::new()
|
Vec::new()
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
let resp = resp_builder.body(data).expect("Failed to build object cache response body");
|
let resp = resp_builder
|
||||||
|
.body(data)
|
||||||
|
.expect("Failed to build object cache response body");
|
||||||
if cache_result.map_or(true, |x| x.has_expired()) {
|
if cache_result.map_or(true, |x| x.has_expired()) {
|
||||||
cache_object::<ObjectCache>(object_id, &resp.clone().try_into()?)
|
cache_object::<ObjectCache>(object_id, &resp.clone().try_into()?)
|
||||||
.expect("Failed to create cached object");
|
.expect("Failed to create cached object");
|
||||||
@ -1,8 +1,10 @@
|
|||||||
pub mod auth;
|
pub mod auth;
|
||||||
#[macro_use]
|
#[macro_use]
|
||||||
pub mod cache;
|
pub mod cache;
|
||||||
pub mod commands;
|
pub mod error;
|
||||||
pub mod fetch_object;
|
pub mod fetch_object;
|
||||||
pub mod requests;
|
pub mod requests;
|
||||||
pub mod server_proto;
|
pub mod server_proto;
|
||||||
pub mod utils;
|
pub mod utils;
|
||||||
|
|
||||||
|
pub use auth::setup;
|
||||||
@ -1,10 +1,8 @@
|
|||||||
|
use database::{DB, interface::DatabaseImpls};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
DB,
|
auth::generate_authorization_header, error::RemoteAccessError, utils::DROP_CLIENT_ASYNC,
|
||||||
database::db::DatabaseImpls,
|
|
||||||
error::remote_access_error::RemoteAccessError,
|
|
||||||
remote::{auth::generate_authorization_header, utils::DROP_CLIENT_ASYNC},
|
|
||||||
};
|
};
|
||||||
|
|
||||||
pub fn generate_url<T: AsRef<str>>(
|
pub fn generate_url<T: AsRef<str>>(
|
||||||
@ -1,24 +1,30 @@
|
|||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use http::{uri::PathAndQuery, Request, Response, StatusCode, Uri};
|
use database::borrow_db_checked;
|
||||||
|
use http::{Request, Response, StatusCode, Uri, uri::PathAndQuery};
|
||||||
use log::{error, warn};
|
use log::{error, warn};
|
||||||
use tauri::UriSchemeResponder;
|
use tauri::UriSchemeResponder;
|
||||||
|
use utils::webbrowser_open::webbrowser_open;
|
||||||
|
|
||||||
use crate::{database::db::borrow_db_checked, remote::utils::DROP_CLIENT_SYNC, utils::webbrowser_open::webbrowser_open};
|
use crate::utils::DROP_CLIENT_SYNC;
|
||||||
|
|
||||||
pub async fn handle_server_proto_offline_wrapper(request: Request<Vec<u8>>, responder: UriSchemeResponder) {
|
pub async fn handle_server_proto_offline_wrapper(
|
||||||
|
request: Request<Vec<u8>>,
|
||||||
|
responder: UriSchemeResponder,
|
||||||
|
) {
|
||||||
responder.respond(match handle_server_proto_offline(request).await {
|
responder.respond(match handle_server_proto_offline(request).await {
|
||||||
Ok(res) => res,
|
Ok(res) => res,
|
||||||
Err(_) => unreachable!()
|
Err(_) => unreachable!(),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_server_proto_offline(_request: Request<Vec<u8>>) -> Result<Response<Vec<u8>>, StatusCode>{
|
pub async fn handle_server_proto_offline(
|
||||||
|
_request: Request<Vec<u8>>,
|
||||||
|
) -> Result<Response<Vec<u8>>, StatusCode> {
|
||||||
Ok(Response::builder()
|
Ok(Response::builder()
|
||||||
.status(StatusCode::NOT_FOUND)
|
.status(StatusCode::NOT_FOUND)
|
||||||
.body(Vec::new())
|
.body(Vec::new())
|
||||||
.expect("Failed to build error response for proto offline"))
|
.expect("Failed to build error response for proto offline"))
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_server_proto_wrapper(request: Request<Vec<u8>>, responder: UriSchemeResponder) {
|
pub async fn handle_server_proto_wrapper(request: Request<Vec<u8>>, responder: UriSchemeResponder) {
|
||||||
@ -26,7 +32,12 @@ pub async fn handle_server_proto_wrapper(request: Request<Vec<u8>>, responder: U
|
|||||||
Ok(r) => responder.respond(r),
|
Ok(r) => responder.respond(r),
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
warn!("Cache error: {e}");
|
warn!("Cache error: {e}");
|
||||||
responder.respond(Response::builder().status(e).body(Vec::new()).expect("Failed to build error response"));
|
responder.respond(
|
||||||
|
Response::builder()
|
||||||
|
.status(e)
|
||||||
|
.body(Vec::new())
|
||||||
|
.expect("Failed to build error response"),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -37,20 +48,25 @@ async fn handle_server_proto(request: Request<Vec<u8>>) -> Result<Response<Vec<u
|
|||||||
Some(auth) => auth,
|
Some(auth) => auth,
|
||||||
None => {
|
None => {
|
||||||
error!("Could not find auth in database");
|
error!("Could not find auth in database");
|
||||||
return Err(StatusCode::UNAUTHORIZED)
|
return Err(StatusCode::UNAUTHORIZED);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
let web_token = match &auth.web_token {
|
let web_token = match &auth.web_token {
|
||||||
Some(token) => token,
|
Some(token) => token,
|
||||||
None => return Err(StatusCode::UNAUTHORIZED),
|
None => return Err(StatusCode::UNAUTHORIZED),
|
||||||
};
|
};
|
||||||
let remote_uri = db_handle.base_url.parse::<Uri>().expect("Failed to parse base url");
|
let remote_uri = db_handle
|
||||||
|
.base_url
|
||||||
|
.parse::<Uri>()
|
||||||
|
.expect("Failed to parse base url");
|
||||||
|
|
||||||
let path = request.uri().path();
|
let path = request.uri().path();
|
||||||
|
|
||||||
let mut new_uri = request.uri().clone().into_parts();
|
let mut new_uri = request.uri().clone().into_parts();
|
||||||
new_uri.path_and_query =
|
new_uri.path_and_query = Some(
|
||||||
Some(PathAndQuery::from_str(&format!("{path}?noWrapper=true")).expect("Failed to parse request path in proto"));
|
PathAndQuery::from_str(&format!("{path}?noWrapper=true"))
|
||||||
|
.expect("Failed to parse request path in proto"),
|
||||||
|
);
|
||||||
new_uri.authority = remote_uri.authority().cloned();
|
new_uri.authority = remote_uri.authority().cloned();
|
||||||
new_uri.scheme = remote_uri.scheme().cloned();
|
new_uri.scheme = remote_uri.scheme().cloned();
|
||||||
let err_msg = &format!("Failed to build new uri from parts {new_uri:?}");
|
let err_msg = &format!("Failed to build new uri from parts {new_uri:?}");
|
||||||
@ -60,7 +76,7 @@ async fn handle_server_proto(request: Request<Vec<u8>>) -> Result<Response<Vec<u
|
|||||||
|
|
||||||
if whitelist_prefix.iter().all(|f| !path.starts_with(f)) {
|
if whitelist_prefix.iter().all(|f| !path.starts_with(f)) {
|
||||||
webbrowser_open(new_uri.to_string());
|
webbrowser_open(new_uri.to_string());
|
||||||
return Ok(Response::new(Vec::new()))
|
return Ok(Response::new(Vec::new()));
|
||||||
}
|
}
|
||||||
|
|
||||||
let client = DROP_CLIENT_SYNC.clone();
|
let client = DROP_CLIENT_SYNC.clone();
|
||||||
@ -68,12 +84,13 @@ async fn handle_server_proto(request: Request<Vec<u8>>) -> Result<Response<Vec<u
|
|||||||
.request(request.method().clone(), new_uri.to_string())
|
.request(request.method().clone(), new_uri.to_string())
|
||||||
.header("Authorization", format!("Bearer {web_token}"))
|
.header("Authorization", format!("Bearer {web_token}"))
|
||||||
.headers(request.headers().clone())
|
.headers(request.headers().clone())
|
||||||
.send() {
|
.send()
|
||||||
|
{
|
||||||
Ok(response) => response,
|
Ok(response) => response,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
warn!("Could not send response. Got {e} when sending");
|
warn!("Could not send response. Got {e} when sending");
|
||||||
return Err(e.status().unwrap_or(StatusCode::BAD_REQUEST))
|
return Err(e.status().unwrap_or(StatusCode::BAD_REQUEST));
|
||||||
},
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let response_status = response.status();
|
let response_status = response.status();
|
||||||
@ -1,25 +1,24 @@
|
|||||||
use std::{
|
use std::{
|
||||||
fs::{self, File},
|
fs::{self, File},
|
||||||
io::Read,
|
io::Read,
|
||||||
sync::{LazyLock, Mutex},
|
sync::LazyLock,
|
||||||
time::Duration,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use database::db::DATA_ROOT_DIR;
|
||||||
use log::{debug, info, warn};
|
use log::{debug, info, warn};
|
||||||
use reqwest::Certificate;
|
use reqwest::Certificate;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use url::Url;
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
database::db::{borrow_db_mut_checked, DATA_ROOT_DIR}, error::remote_access_error::RemoteAccessError, lock, AppState, AppStatus
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
struct DropHealthcheck {
|
pub struct DropHealthcheck {
|
||||||
app_name: String,
|
app_name: String,
|
||||||
}
|
}
|
||||||
|
impl DropHealthcheck {
|
||||||
|
pub fn app_name(&self) -> &String {
|
||||||
|
&self.app_name
|
||||||
|
}
|
||||||
|
}
|
||||||
static DROP_CERT_BUNDLE: LazyLock<Vec<Certificate>> = LazyLock::new(fetch_certificates);
|
static DROP_CERT_BUNDLE: LazyLock<Vec<Certificate>> = LazyLock::new(fetch_certificates);
|
||||||
pub static DROP_CLIENT_SYNC: LazyLock<reqwest::blocking::Client> = LazyLock::new(get_client_sync);
|
pub static DROP_CLIENT_SYNC: LazyLock<reqwest::blocking::Client> = LazyLock::new(get_client_sync);
|
||||||
pub static DROP_CLIENT_ASYNC: LazyLock<reqwest::Client> = LazyLock::new(get_client_async);
|
pub static DROP_CLIENT_ASYNC: LazyLock<reqwest::Client> = LazyLock::new(get_client_async);
|
||||||
@ -47,11 +46,13 @@ fn fetch_certificates() -> Vec<Certificate> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
.read_to_end(&mut buf)
|
.read_to_end(&mut buf)
|
||||||
.unwrap_or_else(|e| panic!(
|
.unwrap_or_else(|e| {
|
||||||
|
panic!(
|
||||||
"Failed to read to end of certificate file {} with error {}",
|
"Failed to read to end of certificate file {} with error {}",
|
||||||
c.path().display(),
|
c.path().display(),
|
||||||
e
|
e
|
||||||
));
|
)
|
||||||
|
});
|
||||||
|
|
||||||
match Certificate::from_pem_bundle(&buf) {
|
match Certificate::from_pem_bundle(&buf) {
|
||||||
Ok(certificates) => {
|
Ok(certificates) => {
|
||||||
@ -88,7 +89,10 @@ pub fn get_client_sync() -> reqwest::blocking::Client {
|
|||||||
for cert in DROP_CERT_BUNDLE.iter() {
|
for cert in DROP_CERT_BUNDLE.iter() {
|
||||||
client = client.add_root_certificate(cert.clone());
|
client = client.add_root_certificate(cert.clone());
|
||||||
}
|
}
|
||||||
client.use_rustls_tls().build().expect("Failed to build synchronous client")
|
client
|
||||||
|
.use_rustls_tls()
|
||||||
|
.build()
|
||||||
|
.expect("Failed to build synchronous client")
|
||||||
}
|
}
|
||||||
pub fn get_client_async() -> reqwest::Client {
|
pub fn get_client_async() -> reqwest::Client {
|
||||||
let mut client = reqwest::ClientBuilder::new();
|
let mut client = reqwest::ClientBuilder::new();
|
||||||
@ -96,7 +100,10 @@ pub fn get_client_async() -> reqwest::Client {
|
|||||||
for cert in DROP_CERT_BUNDLE.iter() {
|
for cert in DROP_CERT_BUNDLE.iter() {
|
||||||
client = client.add_root_certificate(cert.clone());
|
client = client.add_root_certificate(cert.clone());
|
||||||
}
|
}
|
||||||
client.use_rustls_tls().build().expect("Failed to build asynchronous client")
|
client
|
||||||
|
.use_rustls_tls()
|
||||||
|
.build()
|
||||||
|
.expect("Failed to build asynchronous client")
|
||||||
}
|
}
|
||||||
pub fn get_client_ws() -> reqwest::Client {
|
pub fn get_client_ws() -> reqwest::Client {
|
||||||
let mut client = reqwest::ClientBuilder::new();
|
let mut client = reqwest::ClientBuilder::new();
|
||||||
@ -110,36 +117,3 @@ pub fn get_client_ws() -> reqwest::Client {
|
|||||||
.build()
|
.build()
|
||||||
.expect("Failed to build websocket client")
|
.expect("Failed to build websocket client")
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn use_remote_logic(
|
|
||||||
url: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<(), RemoteAccessError> {
|
|
||||||
debug!("connecting to url {url}");
|
|
||||||
let base_url = Url::parse(&url)?;
|
|
||||||
|
|
||||||
// Test Drop url
|
|
||||||
let test_endpoint = base_url.join("/api/v1")?;
|
|
||||||
let client = DROP_CLIENT_ASYNC.clone();
|
|
||||||
let response = client
|
|
||||||
.get(test_endpoint.to_string())
|
|
||||||
.timeout(Duration::from_secs(3))
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let result: DropHealthcheck = response.json().await?;
|
|
||||||
|
|
||||||
if result.app_name != "Drop" {
|
|
||||||
warn!("user entered drop endpoint that connected, but wasn't identified as Drop");
|
|
||||||
return Err(RemoteAccessError::InvalidEndpoint);
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut app_state = lock!(state);
|
|
||||||
app_state.status = AppStatus::SignedOut;
|
|
||||||
drop(app_state);
|
|
||||||
|
|
||||||
let mut db_state = borrow_db_mut_checked();
|
|
||||||
db_state.base_url = base_url.to_string();
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
268
src-tauri/Cargo.lock
generated
268
src-tauri/Cargo.lock
generated
@ -740,7 +740,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "02260d489095346e5cafd04dea8e8cb54d1d74fcd759022a9b72986ebe9a1257"
|
checksum = "02260d489095346e5cafd04dea8e8cb54d1d74fcd759022a9b72986ebe9a1257"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"toml",
|
"toml 0.8.22",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -808,6 +808,16 @@ dependencies = [
|
|||||||
"windows-link",
|
"windows-link",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "client"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"database 0.5.0",
|
||||||
|
"log",
|
||||||
|
"tauri",
|
||||||
|
"tauri-plugin-autostart",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "combine"
|
name = "combine"
|
||||||
version = "4.6.7"
|
version = "4.6.7"
|
||||||
@ -1059,6 +1069,16 @@ version = "2.9.0"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2a2330da5de22e8a3cb63252ce2abb30116bf5265e89c0e01bc17015ce30a476"
|
checksum = "2a2330da5de22e8a3cb63252ce2abb30116bf5265e89c0e01bc17015ce30a476"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "database"
|
||||||
|
version = "0.1.0"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "database"
|
||||||
|
version = "0.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "f4851b30af1e6d2db292e4873b486fa2ad00a6fb6466da44f827e18bdac62f50"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "der-parser"
|
name = "der-parser"
|
||||||
version = "9.0.0"
|
version = "9.0.0"
|
||||||
@ -1237,9 +1257,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "dlopen2"
|
name = "dlopen2"
|
||||||
version = "0.7.0"
|
version = "0.8.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "9e1297103d2bbaea85724fcee6294c2d50b1081f9ad47d0f6f6f61eda65315a6"
|
checksum = "b54f373ccf864bf587a89e880fb7610f8d73f3045f13580948ccbcaff26febff"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"dlopen2_derive",
|
"dlopen2_derive",
|
||||||
"libc",
|
"libc",
|
||||||
@ -1292,6 +1312,8 @@ dependencies = [
|
|||||||
"bytes",
|
"bytes",
|
||||||
"cacache 13.1.0",
|
"cacache 13.1.0",
|
||||||
"chrono",
|
"chrono",
|
||||||
|
"client",
|
||||||
|
"database 0.1.0",
|
||||||
"deranged",
|
"deranged",
|
||||||
"dirs 6.0.0",
|
"dirs 6.0.0",
|
||||||
"droplet-rs",
|
"droplet-rs",
|
||||||
@ -1311,9 +1333,11 @@ dependencies = [
|
|||||||
"native_model",
|
"native_model",
|
||||||
"page_size",
|
"page_size",
|
||||||
"parking_lot 0.12.3",
|
"parking_lot 0.12.3",
|
||||||
|
"process",
|
||||||
"rand 0.9.1",
|
"rand 0.9.1",
|
||||||
"rayon",
|
"rayon",
|
||||||
"regex",
|
"regex",
|
||||||
|
"remote",
|
||||||
"reqwest 0.12.22",
|
"reqwest 0.12.22",
|
||||||
"reqwest-middleware 0.4.2",
|
"reqwest-middleware 0.4.2",
|
||||||
"reqwest-middleware-cache",
|
"reqwest-middleware-cache",
|
||||||
@ -1345,6 +1369,7 @@ dependencies = [
|
|||||||
"umu-wrapper-lib",
|
"umu-wrapper-lib",
|
||||||
"url",
|
"url",
|
||||||
"urlencoding",
|
"urlencoding",
|
||||||
|
"utils",
|
||||||
"uuid",
|
"uuid",
|
||||||
"walkdir",
|
"walkdir",
|
||||||
"webbrowser",
|
"webbrowser",
|
||||||
@ -1423,7 +1448,7 @@ dependencies = [
|
|||||||
"cc",
|
"cc",
|
||||||
"memchr",
|
"memchr",
|
||||||
"rustc_version",
|
"rustc_version",
|
||||||
"toml",
|
"toml 0.8.22",
|
||||||
"vswhom",
|
"vswhom",
|
||||||
"winreg 0.52.0",
|
"winreg 0.52.0",
|
||||||
]
|
]
|
||||||
@ -2113,7 +2138,7 @@ dependencies = [
|
|||||||
"futures-sink",
|
"futures-sink",
|
||||||
"futures-util",
|
"futures-util",
|
||||||
"http 0.2.12",
|
"http 0.2.12",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"slab",
|
"slab",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tokio-util",
|
"tokio-util",
|
||||||
@ -2132,7 +2157,7 @@ dependencies = [
|
|||||||
"futures-core",
|
"futures-core",
|
||||||
"futures-sink",
|
"futures-sink",
|
||||||
"http 1.3.1",
|
"http 1.3.1",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"slab",
|
"slab",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tokio-util",
|
"tokio-util",
|
||||||
@ -2592,13 +2617,14 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "indexmap"
|
name = "indexmap"
|
||||||
version = "2.9.0"
|
version = "2.11.4"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "cea70ddb795996207ad57735b50c5982d8844f38ba9ee5f1aedcfb708a2aa11e"
|
checksum = "4b0f83760fb341a774ed326568e19f5a863af4a952def8c39f9ab92fd95b88e5"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"equivalent",
|
"equivalent",
|
||||||
"hashbrown 0.15.3",
|
"hashbrown 0.15.3",
|
||||||
"serde",
|
"serde",
|
||||||
|
"serde_core",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -2775,7 +2801,7 @@ checksum = "02cb977175687f33fa4afa0c95c112b987ea1443e5a51c8f8ff27dc618270cc2"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"cssparser",
|
"cssparser",
|
||||||
"html5ever",
|
"html5ever",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"selectors",
|
"selectors",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -2881,9 +2907,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "log"
|
name = "log"
|
||||||
version = "0.4.27"
|
version = "0.4.28"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "13dc2df351e3202783a1fe0d44375f7295ffb4049267b0f3018346dc122a1d94"
|
checksum = "34080505efa8e45a4b816c349525ebe327ceaa8559756f0356cba97ef3bf7432"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"value-bag",
|
"value-bag",
|
||||||
@ -3437,6 +3463,16 @@ dependencies = [
|
|||||||
"objc2-core-foundation",
|
"objc2-core-foundation",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "objc2-javascript-core"
|
||||||
|
version = "0.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "9052cb1bb50a4c161d934befcf879526fb87ae9a68858f241e693ca46225cf5a"
|
||||||
|
dependencies = [
|
||||||
|
"objc2 0.6.1",
|
||||||
|
"objc2-core-foundation",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "objc2-metal"
|
name = "objc2-metal"
|
||||||
version = "0.2.2"
|
version = "0.2.2"
|
||||||
@ -3473,6 +3509,17 @@ dependencies = [
|
|||||||
"objc2-foundation 0.3.1",
|
"objc2-foundation 0.3.1",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "objc2-security"
|
||||||
|
version = "0.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e1f8e0ef3ab66b08c42644dcb34dba6ec0a574bbd8adbb8bdbdc7a2779731a44"
|
||||||
|
dependencies = [
|
||||||
|
"bitflags 2.9.1",
|
||||||
|
"objc2 0.6.1",
|
||||||
|
"objc2-core-foundation",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "objc2-ui-kit"
|
name = "objc2-ui-kit"
|
||||||
version = "0.3.1"
|
version = "0.3.1"
|
||||||
@ -3497,6 +3544,8 @@ dependencies = [
|
|||||||
"objc2-app-kit",
|
"objc2-app-kit",
|
||||||
"objc2-core-foundation",
|
"objc2-core-foundation",
|
||||||
"objc2-foundation 0.3.1",
|
"objc2-foundation 0.3.1",
|
||||||
|
"objc2-javascript-core",
|
||||||
|
"objc2-security",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -3943,7 +3992,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "eac26e981c03a6e53e0aee43c113e3202f5581d5360dae7bd2c70e800dd0451d"
|
checksum = "eac26e981c03a6e53e0aee43c113e3202f5581d5360dae7bd2c70e800dd0451d"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"base64 0.22.1",
|
"base64 0.22.1",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"quick-xml",
|
"quick-xml",
|
||||||
"serde",
|
"serde",
|
||||||
"time",
|
"time",
|
||||||
@ -4074,6 +4123,10 @@ dependencies = [
|
|||||||
"unicode-ident",
|
"unicode-ident",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "process"
|
||||||
|
version = "0.1.0"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "quick-xml"
|
name = "quick-xml"
|
||||||
version = "0.32.0"
|
version = "0.32.0"
|
||||||
@ -4384,6 +4437,10 @@ version = "0.8.5"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2b15c43186be67a4fd63bee50d0303afffcef381492ebe2c5d87f324e1b8815c"
|
checksum = "2b15c43186be67a4fd63bee50d0303afffcef381492ebe2c5d87f324e1b8815c"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "remote"
|
||||||
|
version = "0.1.0"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "reqwest"
|
name = "reqwest"
|
||||||
version = "0.11.27"
|
version = "0.11.27"
|
||||||
@ -4852,10 +4909,11 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serde"
|
name = "serde"
|
||||||
version = "1.0.219"
|
version = "1.0.228"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "5f0e2c6ed6606019b4e29e69dbaba95b11854410e5347d525002456dbbb786b6"
|
checksum = "9a8e94ea7f378bd32cbbd37198a4a91436180c5bb472411e48b5ec2e2124ae9e"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
|
"serde_core",
|
||||||
"serde_derive",
|
"serde_derive",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -4881,10 +4939,19 @@ dependencies = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serde_derive"
|
name = "serde_core"
|
||||||
version = "1.0.219"
|
version = "1.0.228"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "5b0276cf7f2c73365f7157c8123c21cd9a50fbbd844757af28ca1f5925fc2a00"
|
checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
|
||||||
|
dependencies = [
|
||||||
|
"serde_derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_derive"
|
||||||
|
version = "1.0.228"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
@ -4934,6 +5001,15 @@ dependencies = [
|
|||||||
"serde",
|
"serde",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "serde_spanned"
|
||||||
|
version = "1.0.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "5417783452c2be558477e104686f7de5dae53dba813c28435e0e70f82d9b04ee"
|
||||||
|
dependencies = [
|
||||||
|
"serde_core",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serde_urlencoded"
|
name = "serde_urlencoded"
|
||||||
version = "0.7.1"
|
version = "0.7.1"
|
||||||
@ -4956,7 +5032,7 @@ dependencies = [
|
|||||||
"chrono",
|
"chrono",
|
||||||
"hex 0.4.3",
|
"hex 0.4.3",
|
||||||
"indexmap 1.9.3",
|
"indexmap 1.9.3",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_derive",
|
"serde_derive",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
@ -4982,7 +5058,7 @@ version = "0.9.34+deprecated"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "6a8b1a1a2ebf674015cc02edccce75287f1a0130d394307b36743c2f5d504b47"
|
checksum = "6a8b1a1a2ebf674015cc02edccce75287f1a0130d394307b36743c2f5d504b47"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"itoa",
|
"itoa",
|
||||||
"ryu",
|
"ryu",
|
||||||
"serde",
|
"serde",
|
||||||
@ -4991,9 +5067,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serialize-to-javascript"
|
name = "serialize-to-javascript"
|
||||||
version = "0.1.1"
|
version = "0.1.2"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c9823f2d3b6a81d98228151fdeaf848206a7855a7a042bbf9bf870449a66cafb"
|
checksum = "04f3666a07a197cdb77cdf306c32be9b7f598d7060d50cfd4d5aa04bfd92f6c5"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
@ -5002,13 +5078,13 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "serialize-to-javascript-impl"
|
name = "serialize-to-javascript-impl"
|
||||||
version = "0.1.1"
|
version = "0.1.2"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "74064874e9f6a15f04c1f3cb627902d0e6b410abbf36668afa873c61889f1763"
|
checksum = "772ee033c0916d670af7860b6e1ef7d658a4629a6d0b4c8c3e67f09b3765b75d"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
"quote",
|
"quote",
|
||||||
"syn 1.0.109",
|
"syn 2.0.101",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -5424,17 +5500,18 @@ dependencies = [
|
|||||||
"cfg-expr",
|
"cfg-expr",
|
||||||
"heck 0.5.0",
|
"heck 0.5.0",
|
||||||
"pkg-config",
|
"pkg-config",
|
||||||
"toml",
|
"toml 0.8.22",
|
||||||
"version-compare",
|
"version-compare",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tao"
|
name = "tao"
|
||||||
version = "0.34.0"
|
version = "0.34.3"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "49c380ca75a231b87b6c9dd86948f035012e7171d1a7c40a9c2890489a7ffd8a"
|
checksum = "959469667dbcea91e5485fc48ba7dd6023face91bb0f1a14681a70f99847c3f7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"bitflags 2.9.1",
|
"bitflags 2.9.1",
|
||||||
|
"block2 0.6.1",
|
||||||
"core-foundation 0.10.1",
|
"core-foundation 0.10.1",
|
||||||
"core-graphics",
|
"core-graphics",
|
||||||
"crossbeam-channel",
|
"crossbeam-channel",
|
||||||
@ -5506,12 +5583,13 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri"
|
name = "tauri"
|
||||||
version = "2.7.0"
|
version = "2.8.5"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "352a4bc7bf6c25f5624227e3641adf475a6535707451b09bb83271df8b7a6ac7"
|
checksum = "d4d1d3b3dc4c101ac989fd7db77e045cc6d91a25349cd410455cb5c57d510c1c"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"bytes",
|
"bytes",
|
||||||
|
"cookie",
|
||||||
"dirs 6.0.0",
|
"dirs 6.0.0",
|
||||||
"dunce",
|
"dunce",
|
||||||
"embed_plist",
|
"embed_plist",
|
||||||
@ -5530,6 +5608,7 @@ dependencies = [
|
|||||||
"objc2-app-kit",
|
"objc2-app-kit",
|
||||||
"objc2-foundation 0.3.1",
|
"objc2-foundation 0.3.1",
|
||||||
"objc2-ui-kit",
|
"objc2-ui-kit",
|
||||||
|
"objc2-web-kit",
|
||||||
"percent-encoding",
|
"percent-encoding",
|
||||||
"plist",
|
"plist",
|
||||||
"raw-window-handle",
|
"raw-window-handle",
|
||||||
@ -5557,9 +5636,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-build"
|
name = "tauri-build"
|
||||||
version = "2.3.1"
|
version = "2.4.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "182d688496c06bf08ea896459bf483eb29cdff35c1c4c115fb14053514303064"
|
checksum = "9c432ccc9ff661803dab74c6cd78de11026a578a9307610bbc39d3c55be7943f"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"cargo_toml",
|
"cargo_toml",
|
||||||
@ -5573,15 +5652,15 @@ dependencies = [
|
|||||||
"serde_json",
|
"serde_json",
|
||||||
"tauri-utils",
|
"tauri-utils",
|
||||||
"tauri-winres",
|
"tauri-winres",
|
||||||
"toml",
|
"toml 0.9.7",
|
||||||
"walkdir",
|
"walkdir",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-codegen"
|
name = "tauri-codegen"
|
||||||
version = "2.3.1"
|
version = "2.4.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b54a99a6cd8e01abcfa61508177e6096a4fe2681efecee9214e962f2f073ae4a"
|
checksum = "1ab3a62cf2e6253936a8b267c2e95839674e7439f104fa96ad0025e149d54d8a"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"base64 0.22.1",
|
"base64 0.22.1",
|
||||||
"brotli",
|
"brotli",
|
||||||
@ -5606,9 +5685,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-macros"
|
name = "tauri-macros"
|
||||||
version = "2.3.2"
|
version = "2.4.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "7945b14dc45e23532f2ded6e120170bbdd4af5ceaa45784a6b33d250fbce3f9e"
|
checksum = "4368ea8094e7045217edb690f493b55b30caf9f3e61f79b4c24b6db91f07995e"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"heck 0.5.0",
|
"heck 0.5.0",
|
||||||
"proc-macro2",
|
"proc-macro2",
|
||||||
@ -5620,9 +5699,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-plugin"
|
name = "tauri-plugin"
|
||||||
version = "2.3.0"
|
version = "2.4.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "1d9a0bd00bf1930ad1a604d08b0eb6b2a9c1822686d65d7f4731a7723b8901d3"
|
checksum = "9946a3cede302eac0c6eb6c6070ac47b1768e326092d32efbb91f21ed58d978f"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"glob",
|
"glob",
|
||||||
@ -5631,15 +5710,15 @@ dependencies = [
|
|||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"tauri-utils",
|
"tauri-utils",
|
||||||
"toml",
|
"toml 0.9.7",
|
||||||
"walkdir",
|
"walkdir",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-plugin-autostart"
|
name = "tauri-plugin-autostart"
|
||||||
version = "2.3.0"
|
version = "2.5.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c58593aafcb03892dbf9998b35a96ead3b8e597435c7af46aff1654d076d5d03"
|
checksum = "062cdcd483d5e3148c9a64dabf8c574e239e2aa1193cf208d95cf89a676f87a5"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"auto-launch",
|
"auto-launch",
|
||||||
"serde",
|
"serde",
|
||||||
@ -5705,7 +5784,7 @@ dependencies = [
|
|||||||
"tauri-plugin",
|
"tauri-plugin",
|
||||||
"tauri-utils",
|
"tauri-utils",
|
||||||
"thiserror 2.0.12",
|
"thiserror 2.0.12",
|
||||||
"toml",
|
"toml 0.8.22",
|
||||||
"url",
|
"url",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -5733,9 +5812,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-plugin-os"
|
name = "tauri-plugin-os"
|
||||||
version = "2.2.1"
|
version = "2.3.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "424f19432397850c2ddd42aa58078630c15287bbce3866eb1d90e7dbee680637"
|
checksum = "77a1c77ebf6f20417ab2a74e8c310820ba52151406d0c80fbcea7df232e3f6ba"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"gethostname",
|
"gethostname",
|
||||||
"log",
|
"log",
|
||||||
@ -5788,9 +5867,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-runtime"
|
name = "tauri-runtime"
|
||||||
version = "2.7.1"
|
version = "2.8.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2b1cc885be806ea15ff7b0eb47098a7b16323d9228876afda329e34e2d6c4676"
|
checksum = "d4cfc9ad45b487d3fded5a4731a567872a4812e9552e3964161b08edabf93846"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"cookie",
|
"cookie",
|
||||||
"dpi",
|
"dpi",
|
||||||
@ -5799,20 +5878,23 @@ dependencies = [
|
|||||||
"jni",
|
"jni",
|
||||||
"objc2 0.6.1",
|
"objc2 0.6.1",
|
||||||
"objc2-ui-kit",
|
"objc2-ui-kit",
|
||||||
|
"objc2-web-kit",
|
||||||
"raw-window-handle",
|
"raw-window-handle",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"tauri-utils",
|
"tauri-utils",
|
||||||
"thiserror 2.0.12",
|
"thiserror 2.0.12",
|
||||||
"url",
|
"url",
|
||||||
|
"webkit2gtk",
|
||||||
|
"webview2-com",
|
||||||
"windows",
|
"windows",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-runtime-wry"
|
name = "tauri-runtime-wry"
|
||||||
version = "2.7.2"
|
version = "2.8.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "fe653a2fbbef19fe898efc774bc52c8742576342a33d3d028c189b57eb1d2439"
|
checksum = "c1fe9d48bd122ff002064e88cfcd7027090d789c4302714e68fcccba0f4b7807"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"gtk",
|
"gtk",
|
||||||
"http 1.3.1",
|
"http 1.3.1",
|
||||||
@ -5837,9 +5919,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tauri-utils"
|
name = "tauri-utils"
|
||||||
version = "2.6.0"
|
version = "2.7.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "9330c15cabfe1d9f213478c9e8ec2b0c76dab26bb6f314b8ad1c8a568c1d186e"
|
checksum = "41a3852fdf9a4f8fbeaa63dc3e9a85284dd6ef7200751f0bd66ceee30c93f212"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"brotli",
|
"brotli",
|
||||||
@ -5866,7 +5948,7 @@ dependencies = [
|
|||||||
"serde_with",
|
"serde_with",
|
||||||
"swift-rs",
|
"swift-rs",
|
||||||
"thiserror 2.0.12",
|
"thiserror 2.0.12",
|
||||||
"toml",
|
"toml 0.9.7",
|
||||||
"url",
|
"url",
|
||||||
"urlpattern",
|
"urlpattern",
|
||||||
"uuid",
|
"uuid",
|
||||||
@ -5880,8 +5962,8 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "e8d321dbc6f998d825ab3f0d62673e810c861aac2d0de2cc2c395328f1d113b4"
|
checksum = "e8d321dbc6f998d825ab3f0d62673e810c861aac2d0de2cc2c395328f1d113b4"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"embed-resource",
|
"embed-resource",
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"toml",
|
"toml 0.8.22",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -6105,11 +6187,26 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "05ae329d1f08c4d17a59bed7ff5b5a769d062e64a62d34a3261b219e62cd5aae"
|
checksum = "05ae329d1f08c4d17a59bed7ff5b5a769d062e64a62d34a3261b219e62cd5aae"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"serde_spanned",
|
"serde_spanned 0.6.8",
|
||||||
"toml_datetime",
|
"toml_datetime 0.6.9",
|
||||||
"toml_edit 0.22.26",
|
"toml_edit 0.22.26",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "toml"
|
||||||
|
version = "0.9.7"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "00e5e5d9bf2475ac9d4f0d9edab68cc573dc2fd644b0dba36b0c30a92dd9eaa0"
|
||||||
|
dependencies = [
|
||||||
|
"indexmap 2.11.4",
|
||||||
|
"serde_core",
|
||||||
|
"serde_spanned 1.0.2",
|
||||||
|
"toml_datetime 0.7.2",
|
||||||
|
"toml_parser",
|
||||||
|
"toml_writer",
|
||||||
|
"winnow 0.7.13",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "toml_datetime"
|
name = "toml_datetime"
|
||||||
version = "0.6.9"
|
version = "0.6.9"
|
||||||
@ -6119,14 +6216,23 @@ dependencies = [
|
|||||||
"serde",
|
"serde",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "toml_datetime"
|
||||||
|
version = "0.7.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "32f1085dec27c2b6632b04c80b3bb1b4300d6495d1e129693bdda7d91e72eec1"
|
||||||
|
dependencies = [
|
||||||
|
"serde_core",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "toml_edit"
|
name = "toml_edit"
|
||||||
version = "0.19.15"
|
version = "0.19.15"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "1b5bb770da30e5cbfde35a2d7b9b8a2c4b8ef89548a7a6aeab5c9a576e3e7421"
|
checksum = "1b5bb770da30e5cbfde35a2d7b9b8a2c4b8ef89548a7a6aeab5c9a576e3e7421"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"toml_datetime",
|
"toml_datetime 0.6.9",
|
||||||
"winnow 0.5.40",
|
"winnow 0.5.40",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -6136,8 +6242,8 @@ version = "0.20.7"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "70f427fce4d84c72b5b732388bf4a9f4531b53f74e2887e3ecb2481f68f66d81"
|
checksum = "70f427fce4d84c72b5b732388bf4a9f4531b53f74e2887e3ecb2481f68f66d81"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"toml_datetime",
|
"toml_datetime 0.6.9",
|
||||||
"winnow 0.5.40",
|
"winnow 0.5.40",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -6147,12 +6253,21 @@ version = "0.22.26"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "310068873db2c5b3e7659d2cc35d21855dbafa50d1ce336397c666e3cb08137e"
|
checksum = "310068873db2c5b3e7659d2cc35d21855dbafa50d1ce336397c666e3cb08137e"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"indexmap 2.9.0",
|
"indexmap 2.11.4",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_spanned",
|
"serde_spanned 0.6.8",
|
||||||
"toml_datetime",
|
"toml_datetime 0.6.9",
|
||||||
"toml_write",
|
"toml_write",
|
||||||
"winnow 0.7.10",
|
"winnow 0.7.13",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "toml_parser"
|
||||||
|
version = "1.0.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "4cf893c33be71572e0e9aa6dd15e6677937abd686b066eac3f8cd3531688a627"
|
||||||
|
dependencies = [
|
||||||
|
"winnow 0.7.13",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@ -6161,6 +6276,12 @@ version = "0.1.1"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "bfb942dfe1d8e29a7ee7fcbde5bd2b9a25fb89aa70caea2eba3bee836ff41076"
|
checksum = "bfb942dfe1d8e29a7ee7fcbde5bd2b9a25fb89aa70caea2eba3bee836ff41076"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "toml_writer"
|
||||||
|
version = "1.0.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d163a63c116ce562a22cda521fcc4d79152e7aba014456fb5eb442f6d6a10109"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tower"
|
name = "tower"
|
||||||
version = "0.5.2"
|
version = "0.5.2"
|
||||||
@ -6454,6 +6575,10 @@ version = "1.0.4"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b6c140620e7ffbb22c2dee59cafe6084a59b5ffc27a8859a5f0d494b5d52b6be"
|
checksum = "b6c140620e7ffbb22c2dee59cafe6084a59b5ffc27a8859a5f0d494b5d52b6be"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "utils"
|
||||||
|
version = "0.1.0"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "uuid"
|
name = "uuid"
|
||||||
version = "1.17.0"
|
version = "1.17.0"
|
||||||
@ -7202,9 +7327,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "winnow"
|
name = "winnow"
|
||||||
version = "0.7.10"
|
version = "0.7.13"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c06928c8748d81b05c9be96aad92e1b6ff01833332f281e8cfca3be4b35fc9ec"
|
checksum = "21a0236b59786fed61e2a80582dd500fe61f18b5dca67a4a067d0bc9039339cf"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"memchr",
|
"memchr",
|
||||||
]
|
]
|
||||||
@ -7255,14 +7380,15 @@ checksum = "ea2f10b9bb0928dfb1b42b65e1f9e36f7f54dbdf08457afefb38afcdec4fa2bb"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wry"
|
name = "wry"
|
||||||
version = "0.52.1"
|
version = "0.53.4"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "12a714d9ba7075aae04a6e50229d6109e3d584774b99a6a8c60de1698ca111b9"
|
checksum = "6d78ec082b80fa088569a970d043bb3050abaabf4454101d44514ee8d9a8c9f6"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"base64 0.22.1",
|
"base64 0.22.1",
|
||||||
"block2 0.6.1",
|
"block2 0.6.1",
|
||||||
"cookie",
|
"cookie",
|
||||||
"crossbeam-channel",
|
"crossbeam-channel",
|
||||||
|
"dirs 6.0.0",
|
||||||
"dpi",
|
"dpi",
|
||||||
"dunce",
|
"dunce",
|
||||||
"gdkx11",
|
"gdkx11",
|
||||||
@ -7431,7 +7557,7 @@ dependencies = [
|
|||||||
"tracing",
|
"tracing",
|
||||||
"uds_windows",
|
"uds_windows",
|
||||||
"windows-sys 0.59.0",
|
"windows-sys 0.59.0",
|
||||||
"winnow 0.7.10",
|
"winnow 0.7.13",
|
||||||
"zbus_macros",
|
"zbus_macros",
|
||||||
"zbus_names",
|
"zbus_names",
|
||||||
"zvariant",
|
"zvariant",
|
||||||
@ -7460,7 +7586,7 @@ checksum = "7be68e64bf6ce8db94f63e72f0c7eb9a60d733f7e0499e628dfab0f84d6bcb97"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"static_assertions",
|
"static_assertions",
|
||||||
"winnow 0.7.10",
|
"winnow 0.7.13",
|
||||||
"zvariant",
|
"zvariant",
|
||||||
]
|
]
|
||||||
|
|
||||||
@ -7582,7 +7708,7 @@ dependencies = [
|
|||||||
"enumflags2",
|
"enumflags2",
|
||||||
"serde",
|
"serde",
|
||||||
"url",
|
"url",
|
||||||
"winnow 0.7.10",
|
"winnow 0.7.13",
|
||||||
"zvariant_derive",
|
"zvariant_derive",
|
||||||
"zvariant_utils",
|
"zvariant_utils",
|
||||||
]
|
]
|
||||||
@ -7611,5 +7737,5 @@ dependencies = [
|
|||||||
"serde",
|
"serde",
|
||||||
"static_assertions",
|
"static_assertions",
|
||||||
"syn 2.0.101",
|
"syn 2.0.101",
|
||||||
"winnow 0.7.10",
|
"winnow 0.7.13",
|
||||||
]
|
]
|
||||||
|
|||||||
@ -78,6 +78,16 @@ futures-core = "0.3.31"
|
|||||||
bytes = "1.10.1"
|
bytes = "1.10.1"
|
||||||
# tailscale = { path = "./tailscale" }
|
# tailscale = { path = "./tailscale" }
|
||||||
|
|
||||||
|
|
||||||
|
# Workspaces
|
||||||
|
client = { version = "0.1.0", path = "../client" }
|
||||||
|
database = { path = "../database" }
|
||||||
|
process = { path = "../process" }
|
||||||
|
remote = { version = "0.1.0", path = "../remote" }
|
||||||
|
utils = { path = "../utils" }
|
||||||
|
games = { version = "0.1.0", path = "../games" }
|
||||||
|
download_manager = { version = "0.1.0", path = "../download_manager" }
|
||||||
|
|
||||||
[dependencies.dynfmt]
|
[dependencies.dynfmt]
|
||||||
version = "0.1.5"
|
version = "0.1.5"
|
||||||
features = ["curly"]
|
features = ["curly"]
|
||||||
|
|||||||
@ -1,9 +1,41 @@
|
|||||||
use crate::database::db::{borrow_db_checked, borrow_db_mut_checked};
|
use std::sync::nonpoison::Mutex;
|
||||||
use log::debug;
|
|
||||||
|
use database::{borrow_db_checked, borrow_db_mut_checked};
|
||||||
|
use download_manager::DOWNLOAD_MANAGER;
|
||||||
|
use log::{debug, error};
|
||||||
use tauri::AppHandle;
|
use tauri::AppHandle;
|
||||||
use tauri_plugin_autostart::ManagerExt;
|
use tauri_plugin_autostart::ManagerExt;
|
||||||
|
|
||||||
pub fn toggle_autostart_logic(app: AppHandle, enabled: bool) -> Result<(), String> {
|
use crate::AppState;
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn fetch_state(state: tauri::State<'_, Mutex<AppState>>) -> Result<String, String> {
|
||||||
|
let guard = state.lock();
|
||||||
|
let cloned_state = serde_json::to_string(&guard.clone()).map_err(|e| e.to_string())?;
|
||||||
|
drop(guard);
|
||||||
|
Ok(cloned_state)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn quit(app: tauri::AppHandle) {
|
||||||
|
cleanup_and_exit(&app);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn cleanup_and_exit(app: &AppHandle) {
|
||||||
|
debug!("cleaning up and exiting application");
|
||||||
|
match DOWNLOAD_MANAGER.ensure_terminated() {
|
||||||
|
Ok(res) => match res {
|
||||||
|
Ok(()) => debug!("download manager terminated correctly"),
|
||||||
|
Err(()) => error!("download manager failed to terminate correctly"),
|
||||||
|
},
|
||||||
|
Err(e) => panic!("{e:?}"),
|
||||||
|
}
|
||||||
|
|
||||||
|
app.exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn toggle_autostart(app: AppHandle, enabled: bool) -> Result<(), String> {
|
||||||
let manager = app.autolaunch();
|
let manager = app.autolaunch();
|
||||||
if enabled {
|
if enabled {
|
||||||
manager.enable().map_err(|e| e.to_string())?;
|
manager.enable().map_err(|e| e.to_string())?;
|
||||||
@ -16,13 +48,11 @@ pub fn toggle_autostart_logic(app: AppHandle, enabled: bool) -> Result<(), Strin
|
|||||||
// Store the state in DB
|
// Store the state in DB
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
db_handle.settings.autostart = enabled;
|
db_handle.settings.autostart = enabled;
|
||||||
drop(db_handle);
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_autostart_enabled_logic(app: AppHandle) -> Result<bool, tauri_plugin_autostart::Error> {
|
#[tauri::command]
|
||||||
// First check DB state
|
pub fn get_autostart_enabled(app: AppHandle) -> Result<bool, tauri_plugin_autostart::Error> {
|
||||||
let db_handle = borrow_db_checked();
|
let db_handle = borrow_db_checked();
|
||||||
let db_state = db_handle.settings.autostart;
|
let db_state = db_handle.settings.autostart;
|
||||||
drop(db_handle);
|
drop(db_handle);
|
||||||
@ -42,34 +72,3 @@ pub fn get_autostart_enabled_logic(app: AppHandle) -> Result<bool, tauri_plugin_
|
|||||||
|
|
||||||
Ok(db_state)
|
Ok(db_state)
|
||||||
}
|
}
|
||||||
|
|
||||||
// New function to sync state on startup
|
|
||||||
pub fn sync_autostart_on_startup(app: &AppHandle) -> Result<(), String> {
|
|
||||||
let db_handle = borrow_db_checked();
|
|
||||||
let should_be_enabled = db_handle.settings.autostart;
|
|
||||||
drop(db_handle);
|
|
||||||
|
|
||||||
let manager = app.autolaunch();
|
|
||||||
let current_state = manager.is_enabled().map_err(|e| e.to_string())?;
|
|
||||||
|
|
||||||
if current_state != should_be_enabled {
|
|
||||||
if should_be_enabled {
|
|
||||||
manager.enable().map_err(|e| e.to_string())?;
|
|
||||||
debug!("synced autostart: enabled");
|
|
||||||
} else {
|
|
||||||
manager.disable().map_err(|e| e.to_string())?;
|
|
||||||
debug!("synced autostart: disabled");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn toggle_autostart(app: AppHandle, enabled: bool) -> Result<(), String> {
|
|
||||||
toggle_autostart_logic(app, enabled)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn get_autostart_enabled(app: AppHandle) -> Result<bool, tauri_plugin_autostart::Error> {
|
|
||||||
get_autostart_enabled_logic(app)
|
|
||||||
}
|
|
||||||
@ -1,23 +0,0 @@
|
|||||||
use log::{debug, error};
|
|
||||||
use tauri::AppHandle;
|
|
||||||
|
|
||||||
use crate::{lock, AppState};
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn quit(app: tauri::AppHandle, state: tauri::State<'_, std::sync::Mutex<AppState<'_>>>) {
|
|
||||||
cleanup_and_exit(&app, &state);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn cleanup_and_exit(app: &AppHandle, state: &tauri::State<'_, std::sync::Mutex<AppState<'_>>>) {
|
|
||||||
debug!("cleaning up and exiting application");
|
|
||||||
let download_manager = lock!(state).download_manager.clone();
|
|
||||||
match download_manager.ensure_terminated() {
|
|
||||||
Ok(res) => match res {
|
|
||||||
Ok(()) => debug!("download manager terminated correctly"),
|
|
||||||
Err(()) => error!("download manager failed to terminate correctly"),
|
|
||||||
},
|
|
||||||
Err(e) => panic!("{e:?}"),
|
|
||||||
}
|
|
||||||
|
|
||||||
app.exit(0);
|
|
||||||
}
|
|
||||||
@ -1,11 +0,0 @@
|
|||||||
use crate::{lock, AppState};
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn fetch_state(
|
|
||||||
state: tauri::State<'_, std::sync::Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<String, String> {
|
|
||||||
let guard = lock!(state);
|
|
||||||
let cloned_state = serde_json::to_string(&guard.clone()).map_err(|e| e.to_string())?;
|
|
||||||
drop(guard);
|
|
||||||
Ok(cloned_state)
|
|
||||||
}
|
|
||||||
@ -1,3 +0,0 @@
|
|||||||
pub mod autostart;
|
|
||||||
pub mod cleanup;
|
|
||||||
pub mod commands;
|
|
||||||
@ -1,102 +0,0 @@
|
|||||||
use std::{collections::HashMap, path::PathBuf, str::FromStr};
|
|
||||||
|
|
||||||
use log::warn;
|
|
||||||
|
|
||||||
use crate::{database::db::{GameVersion, DATA_ROOT_DIR}, error::backup_error::BackupError, process::process_manager::Platform};
|
|
||||||
|
|
||||||
use super::path::CommonPath;
|
|
||||||
|
|
||||||
pub struct BackupManager<'a> {
|
|
||||||
pub current_platform: Platform,
|
|
||||||
pub sources: HashMap<(Platform, Platform), &'a (dyn BackupHandler + Sync + Send)>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl BackupManager<'_> {
|
|
||||||
pub fn new() -> Self {
|
|
||||||
BackupManager {
|
|
||||||
#[cfg(target_os = "windows")]
|
|
||||||
current_platform: Platform::Windows,
|
|
||||||
|
|
||||||
#[cfg(target_os = "macos")]
|
|
||||||
current_platform: Platform::MacOs,
|
|
||||||
|
|
||||||
#[cfg(target_os = "linux")]
|
|
||||||
current_platform: Platform::Linux,
|
|
||||||
|
|
||||||
sources: HashMap::from([
|
|
||||||
// Current platform to target platform
|
|
||||||
(
|
|
||||||
(Platform::Windows, Platform::Windows),
|
|
||||||
&WindowsBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
|
||||||
),
|
|
||||||
(
|
|
||||||
(Platform::Linux, Platform::Linux),
|
|
||||||
&LinuxBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
|
||||||
),
|
|
||||||
(
|
|
||||||
(Platform::MacOs, Platform::MacOs),
|
|
||||||
&MacBackupManager {} as &(dyn BackupHandler + Sync + Send),
|
|
||||||
),
|
|
||||||
|
|
||||||
]),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait BackupHandler: Send + Sync {
|
|
||||||
fn root_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { Ok(DATA_ROOT_DIR.lock().unwrap().join("games")) }
|
|
||||||
fn game_translate(&self, _path: &PathBuf, game: &GameVersion) -> Result<PathBuf, BackupError> { Ok(PathBuf::from_str(&game.game_id).unwrap()) }
|
|
||||||
fn base_translate(&self, path: &PathBuf, game: &GameVersion) -> Result<PathBuf, BackupError> { Ok(self.root_translate(path, game)?.join(self.game_translate(path, game)?)) }
|
|
||||||
fn home_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { let c = CommonPath::Home.get().ok_or(BackupError::NotFound); println!("{:?}", c); c }
|
|
||||||
fn store_user_id_translate(&self, _path: &PathBuf, game: &GameVersion) -> Result<PathBuf, BackupError> { PathBuf::from_str(&game.game_id).map_err(|_| BackupError::ParseError) }
|
|
||||||
fn os_user_name_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { Ok(PathBuf::from_str(&whoami::username()).unwrap()) }
|
|
||||||
fn win_app_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winAppData>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_local_app_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winLocalAppData>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_local_app_data_low_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winLocalAppDataLow>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_documents_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winDocuments>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_public_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winPublic>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_program_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winProgramData>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn win_dir_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected Windows Reference in Backup <winDir>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn xdg_data_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected XDG Reference in Backup <xdgData>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn xdg_config_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> { warn!("Unexpected XDG Reference in Backup <xdgConfig>"); Err(BackupError::InvalidSystem) }
|
|
||||||
fn skip_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> { Ok(PathBuf::new()) }
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct LinuxBackupManager {}
|
|
||||||
impl BackupHandler for LinuxBackupManager {
|
|
||||||
fn xdg_config_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::Data.get().ok_or(BackupError::NotFound)?)
|
|
||||||
}
|
|
||||||
fn xdg_data_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::Config.get().ok_or(BackupError::NotFound)?)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
pub struct WindowsBackupManager {}
|
|
||||||
impl BackupHandler for WindowsBackupManager {
|
|
||||||
fn win_app_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::Config.get().ok_or(BackupError::NotFound)?)
|
|
||||||
}
|
|
||||||
fn win_local_app_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::DataLocal.get().ok_or(BackupError::NotFound)?)
|
|
||||||
}
|
|
||||||
fn win_local_app_data_low_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::DataLocalLow.get().ok_or(BackupError::NotFound)?)
|
|
||||||
}
|
|
||||||
fn win_dir_translate(&self, _path: &PathBuf,_game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(PathBuf::from_str("C:/Windows").unwrap())
|
|
||||||
}
|
|
||||||
fn win_documents_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::Document.get().ok_or(BackupError::NotFound)?)
|
|
||||||
|
|
||||||
}
|
|
||||||
fn win_program_data_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(PathBuf::from_str("C:/ProgramData").unwrap())
|
|
||||||
}
|
|
||||||
fn win_public_translate(&self, _path: &PathBuf, _game: &GameVersion) -> Result<PathBuf, BackupError> {
|
|
||||||
Ok(CommonPath::Public.get().ok_or(BackupError::NotFound)?)
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
pub struct MacBackupManager {}
|
|
||||||
impl BackupHandler for MacBackupManager {}
|
|
||||||
@ -1,16 +1,12 @@
|
|||||||
use serde_json::json;
|
use games::collections::collection::{Collection, Collections};
|
||||||
|
use remote::{
|
||||||
use crate::{
|
|
||||||
error::remote_access_error::RemoteAccessError,
|
|
||||||
remote::{
|
|
||||||
auth::generate_authorization_header,
|
auth::generate_authorization_header,
|
||||||
cache::{cache_object, get_cached_object},
|
cache::{cache_object, get_cached_object},
|
||||||
|
error::RemoteAccessError,
|
||||||
requests::{generate_url, make_authenticated_get},
|
requests::{generate_url, make_authenticated_get},
|
||||||
utils::DROP_CLIENT_ASYNC,
|
utils::DROP_CLIENT_ASYNC,
|
||||||
},
|
|
||||||
};
|
};
|
||||||
|
use serde_json::json;
|
||||||
use super::collection::{Collection, Collections};
|
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub async fn fetch_collections(
|
pub async fn fetch_collections(
|
||||||
@ -1,5 +0,0 @@
|
|||||||
pub mod commands;
|
|
||||||
pub mod db;
|
|
||||||
pub mod debug;
|
|
||||||
pub mod models;
|
|
||||||
pub mod scan;
|
|
||||||
22
src-tauri/src/download_manager.rs
Normal file
22
src-tauri/src/download_manager.rs
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
use database::DownloadableMetadata;
|
||||||
|
use download_manager::DOWNLOAD_MANAGER;
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn pause_downloads() {
|
||||||
|
DOWNLOAD_MANAGER.pause_downloads();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn resume_downloads() {
|
||||||
|
DOWNLOAD_MANAGER.resume_downloads();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn move_download_in_queue(old_index: usize, new_index: usize) {
|
||||||
|
DOWNLOAD_MANAGER.rearrange(old_index, new_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn cancel_game(meta: DownloadableMetadata) {
|
||||||
|
DOWNLOAD_MANAGER.cancel(meta);
|
||||||
|
}
|
||||||
@ -1,29 +0,0 @@
|
|||||||
use std::sync::Mutex;
|
|
||||||
|
|
||||||
use crate::{AppState, database::models::data::DownloadableMetadata, lock};
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn pause_downloads(state: tauri::State<'_, Mutex<AppState>>) {
|
|
||||||
lock!(state).download_manager.pause_downloads();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn resume_downloads(state: tauri::State<'_, Mutex<AppState>>) {
|
|
||||||
lock!(state).download_manager.resume_downloads();
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn move_download_in_queue(
|
|
||||||
state: tauri::State<'_, Mutex<AppState>>,
|
|
||||||
old_index: usize,
|
|
||||||
new_index: usize,
|
|
||||||
) {
|
|
||||||
lock!(state)
|
|
||||||
.download_manager
|
|
||||||
.rearrange(old_index, new_index);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn cancel_game(state: tauri::State<'_, Mutex<AppState>>, meta: DownloadableMetadata) {
|
|
||||||
lock!(state).download_manager.cancel(meta);
|
|
||||||
}
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
pub mod commands;
|
|
||||||
pub mod download_manager_builder;
|
|
||||||
pub mod download_manager_frontend;
|
|
||||||
pub mod downloadable;
|
|
||||||
pub mod util;
|
|
||||||
@ -1,34 +1,31 @@
|
|||||||
use std::{
|
use std::{path::PathBuf, sync::Arc};
|
||||||
path::PathBuf,
|
|
||||||
sync::{Arc, Mutex},
|
use database::{GameDownloadStatus, borrow_db_checked};
|
||||||
|
use download_manager::{
|
||||||
|
DOWNLOAD_MANAGER, downloadable::Downloadable, error::ApplicationDownloadError,
|
||||||
};
|
};
|
||||||
|
use games::downloads::download_agent::GameDownloadAgent;
|
||||||
|
|
||||||
use crate::{
|
|
||||||
database::{
|
|
||||||
db::borrow_db_checked,
|
|
||||||
models::data::GameDownloadStatus,
|
|
||||||
}, download_manager::downloadable::Downloadable, error::application_download_error::ApplicationDownloadError, lock, AppState
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::download_agent::GameDownloadAgent;
|
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub async fn download_game(
|
pub async fn download_game(
|
||||||
game_id: String,
|
game_id: String,
|
||||||
game_version: String,
|
game_version: String,
|
||||||
install_dir: usize,
|
install_dir: usize,
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<(), ApplicationDownloadError> {
|
) -> Result<(), ApplicationDownloadError> {
|
||||||
let sender = { lock!(state).download_manager.get_sender().clone() };
|
let sender = { DOWNLOAD_MANAGER.get_sender().clone() };
|
||||||
|
|
||||||
let game_download_agent =
|
let game_download_agent = GameDownloadAgent::new_from_index(
|
||||||
GameDownloadAgent::new_from_index(game_id.clone(), game_version.clone(), install_dir, sender).await?;
|
game_id.clone(),
|
||||||
|
game_version.clone(),
|
||||||
|
install_dir,
|
||||||
|
sender,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
let game_download_agent =
|
let game_download_agent =
|
||||||
Arc::new(Box::new(game_download_agent) as Box<dyn Downloadable + Send + Sync>);
|
Arc::new(Box::new(game_download_agent) as Box<dyn Downloadable + Send + Sync>);
|
||||||
lock!(state)
|
|
||||||
.download_manager
|
DOWNLOAD_MANAGER
|
||||||
.queue_download(game_download_agent.clone())
|
.queue_download(game_download_agent.clone())
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
@ -36,10 +33,7 @@ pub async fn download_game(
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub async fn resume_download(
|
pub async fn resume_download(game_id: String) -> Result<(), ApplicationDownloadError> {
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<(), ApplicationDownloadError> {
|
|
||||||
let s = borrow_db_checked()
|
let s = borrow_db_checked()
|
||||||
.applications
|
.applications
|
||||||
.game_statuses
|
.game_statuses
|
||||||
@ -57,21 +51,25 @@ pub async fn resume_download(
|
|||||||
} => (version_name, install_dir),
|
} => (version_name, install_dir),
|
||||||
};
|
};
|
||||||
|
|
||||||
let sender = lock!(state).download_manager.get_sender();
|
let sender = DOWNLOAD_MANAGER.get_sender();
|
||||||
let parent_dir: PathBuf = install_dir.into();
|
let parent_dir: PathBuf = install_dir.into();
|
||||||
|
|
||||||
let game_download_agent = Arc::new(Box::new(
|
let game_download_agent = Arc::new(Box::new(
|
||||||
GameDownloadAgent::new(
|
GameDownloadAgent::new(
|
||||||
game_id,
|
game_id,
|
||||||
version_name.clone(),
|
version_name.clone(),
|
||||||
parent_dir.parent().unwrap_or_else(|| panic!("Failed to get parent directry of {}", parent_dir.display())).to_path_buf(),
|
parent_dir
|
||||||
|
.parent()
|
||||||
|
.unwrap_or_else(|| {
|
||||||
|
panic!("Failed to get parent directry of {}", parent_dir.display())
|
||||||
|
})
|
||||||
|
.to_path_buf(),
|
||||||
sender,
|
sender,
|
||||||
)
|
)
|
||||||
.await?,
|
.await?,
|
||||||
) as Box<dyn Downloadable + Send + Sync>);
|
) as Box<dyn Downloadable + Send + Sync>);
|
||||||
|
|
||||||
lock!(state)
|
DOWNLOAD_MANAGER
|
||||||
.download_manager
|
|
||||||
.queue_download(game_download_agent)
|
.queue_download(game_download_agent)
|
||||||
.unwrap();
|
.unwrap();
|
||||||
Ok(())
|
Ok(())
|
||||||
@ -1,26 +0,0 @@
|
|||||||
use std::fmt::Display;
|
|
||||||
|
|
||||||
use http::{header::ToStrError, HeaderName};
|
|
||||||
use serde_with::SerializeDisplay;
|
|
||||||
|
|
||||||
use crate::error::remote_access_error::RemoteAccessError;
|
|
||||||
|
|
||||||
#[derive(Debug, SerializeDisplay)]
|
|
||||||
pub enum CacheError {
|
|
||||||
HeaderNotFound(HeaderName),
|
|
||||||
ParseError(ToStrError),
|
|
||||||
Remote(RemoteAccessError),
|
|
||||||
ConstructionError(http::Error)
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Display for CacheError {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
let s = match self {
|
|
||||||
CacheError::HeaderNotFound(header_name) => format!("Could not find header {header_name} in cache"),
|
|
||||||
CacheError::ParseError(to_str_error) => format!("Could not parse cache with error {to_str_error}"),
|
|
||||||
CacheError::Remote(remote_access_error) => format!("Cache got remote access error: {remote_access_error}"),
|
|
||||||
CacheError::ConstructionError(error) => format!("Could not construct cache body with error {error}"),
|
|
||||||
};
|
|
||||||
write!(f, "{s}")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,27 +0,0 @@
|
|||||||
use std::{fmt::Display, io, sync::mpsc::SendError};
|
|
||||||
|
|
||||||
use serde_with::SerializeDisplay;
|
|
||||||
|
|
||||||
#[derive(SerializeDisplay)]
|
|
||||||
pub enum DownloadManagerError<T> {
|
|
||||||
IOError(io::Error),
|
|
||||||
SignalError(SendError<T>),
|
|
||||||
}
|
|
||||||
impl<T> Display for DownloadManagerError<T> {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
match self {
|
|
||||||
DownloadManagerError::IOError(error) => write!(f, "{error}"),
|
|
||||||
DownloadManagerError::SignalError(send_error) => write!(f, "{send_error}"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
impl<T> From<SendError<T>> for DownloadManagerError<T> {
|
|
||||||
fn from(value: SendError<T>) -> Self {
|
|
||||||
DownloadManagerError::SignalError(value)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
impl<T> From<io::Error> for DownloadManagerError<T> {
|
|
||||||
fn from(value: io::Error) -> Self {
|
|
||||||
DownloadManagerError::IOError(value)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,10 +0,0 @@
|
|||||||
use serde::Deserialize;
|
|
||||||
|
|
||||||
#[derive(Deserialize, Debug, Clone)]
|
|
||||||
#[serde(rename_all = "camelCase")]
|
|
||||||
pub struct DropServerError {
|
|
||||||
pub status_code: usize,
|
|
||||||
pub status_message: String,
|
|
||||||
// pub message: String,
|
|
||||||
// pub url: String,
|
|
||||||
}
|
|
||||||
@ -1,21 +0,0 @@
|
|||||||
use std::fmt::{Display};
|
|
||||||
|
|
||||||
use serde_with::SerializeDisplay;
|
|
||||||
|
|
||||||
#[derive(SerializeDisplay)]
|
|
||||||
pub enum LibraryError {
|
|
||||||
MetaNotFound(String),
|
|
||||||
VersionNotFound(String),
|
|
||||||
}
|
|
||||||
impl Display for LibraryError {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
write!(f, "{}", match self {
|
|
||||||
LibraryError::MetaNotFound(id) => {
|
|
||||||
format!("Could not locate any installed version of game ID {id} in the database")
|
|
||||||
}
|
|
||||||
LibraryError::VersionNotFound(game_id) => {
|
|
||||||
format!("Could not locate any installed version for game id {game_id} in the database")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,7 +0,0 @@
|
|||||||
pub mod application_download_error;
|
|
||||||
pub mod download_manager_error;
|
|
||||||
pub mod drop_server_error;
|
|
||||||
pub mod library_error;
|
|
||||||
pub mod process_error;
|
|
||||||
pub mod remote_access_error;
|
|
||||||
pub mod cache_error;
|
|
||||||
344
src-tauri/src/games.rs
Normal file
344
src-tauri/src/games.rs
Normal file
@ -0,0 +1,344 @@
|
|||||||
|
use std::sync::nonpoison::Mutex;
|
||||||
|
|
||||||
|
use database::{GameDownloadStatus, GameVersion, borrow_db_checked, borrow_db_mut_checked};
|
||||||
|
use games::{
|
||||||
|
downloads::error::LibraryError,
|
||||||
|
library::{FetchGameStruct, FrontendGameOptions, Game, get_current_meta, uninstall_game_logic},
|
||||||
|
state::{GameStatusManager, GameStatusWithTransient},
|
||||||
|
};
|
||||||
|
use log::warn;
|
||||||
|
use process::PROCESS_MANAGER;
|
||||||
|
use remote::{
|
||||||
|
auth::generate_authorization_header,
|
||||||
|
cache::{cache_object, cache_object_db, get_cached_object, get_cached_object_db},
|
||||||
|
error::{DropServerError, RemoteAccessError},
|
||||||
|
offline,
|
||||||
|
requests::generate_url,
|
||||||
|
utils::DROP_CLIENT_ASYNC,
|
||||||
|
};
|
||||||
|
use tauri::AppHandle;
|
||||||
|
|
||||||
|
use crate::AppState;
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub async fn fetch_library(
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
hard_refresh: Option<bool>,
|
||||||
|
) -> Result<Vec<Game>, RemoteAccessError> {
|
||||||
|
offline!(
|
||||||
|
state,
|
||||||
|
fetch_library_logic,
|
||||||
|
fetch_library_logic_offline,
|
||||||
|
state,
|
||||||
|
hard_refresh
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn fetch_library_logic(
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
hard_fresh: Option<bool>,
|
||||||
|
) -> Result<Vec<Game>, RemoteAccessError> {
|
||||||
|
let do_hard_refresh = hard_fresh.unwrap_or(false);
|
||||||
|
if !do_hard_refresh && let Ok(library) = get_cached_object("library") {
|
||||||
|
return Ok(library);
|
||||||
|
}
|
||||||
|
|
||||||
|
let client = DROP_CLIENT_ASYNC.clone();
|
||||||
|
let response = generate_url(&["/api/v1/client/user/library"], &[])?;
|
||||||
|
let response = client
|
||||||
|
.get(response)
|
||||||
|
.header("Authorization", generate_authorization_header())
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if response.status() != 200 {
|
||||||
|
let err = response.json().await.unwrap_or(DropServerError {
|
||||||
|
status_code: 500,
|
||||||
|
status_message: "Invalid response from server.".to_owned(),
|
||||||
|
});
|
||||||
|
warn!("{err:?}");
|
||||||
|
return Err(RemoteAccessError::InvalidResponse(err));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut games: Vec<Game> = response.json().await?;
|
||||||
|
|
||||||
|
let mut handle = state.lock();
|
||||||
|
|
||||||
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
|
|
||||||
|
for game in &games {
|
||||||
|
handle.games.insert(game.id().clone(), game.clone());
|
||||||
|
if !db_handle.applications.game_statuses.contains_key(game.id()) {
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.game_statuses
|
||||||
|
.insert(game.id().clone(), GameDownloadStatus::Remote {});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add games that are installed but no longer in library
|
||||||
|
for meta in db_handle.applications.installed_game_version.values() {
|
||||||
|
if games.iter().any(|e| *e.id() == meta.id) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
// We should always have a cache of the object
|
||||||
|
// Pass db_handle because otherwise we get a gridlock
|
||||||
|
let game = match get_cached_object_db::<Game>(&meta.id.clone(), &db_handle) {
|
||||||
|
Ok(game) => game,
|
||||||
|
Err(err) => {
|
||||||
|
warn!(
|
||||||
|
"{} is installed, but encountered error fetching its error: {}.",
|
||||||
|
meta.id, err
|
||||||
|
);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
games.push(game);
|
||||||
|
}
|
||||||
|
|
||||||
|
drop(handle);
|
||||||
|
drop(db_handle);
|
||||||
|
cache_object("library", &games)?;
|
||||||
|
|
||||||
|
Ok(games)
|
||||||
|
}
|
||||||
|
pub async fn fetch_library_logic_offline(
|
||||||
|
_state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
_hard_refresh: Option<bool>,
|
||||||
|
) -> Result<Vec<Game>, RemoteAccessError> {
|
||||||
|
let mut games: Vec<Game> = get_cached_object("library")?;
|
||||||
|
|
||||||
|
let db_handle = borrow_db_checked();
|
||||||
|
|
||||||
|
games.retain(|game| {
|
||||||
|
matches!(
|
||||||
|
&db_handle
|
||||||
|
.applications
|
||||||
|
.game_statuses
|
||||||
|
.get(game.id())
|
||||||
|
.unwrap_or(&GameDownloadStatus::Remote {}),
|
||||||
|
GameDownloadStatus::Installed { .. } | GameDownloadStatus::SetupRequired { .. }
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
Ok(games)
|
||||||
|
}
|
||||||
|
pub async fn fetch_game_logic(
|
||||||
|
id: String,
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<FetchGameStruct, RemoteAccessError> {
|
||||||
|
let version = {
|
||||||
|
let state_handle = state.lock();
|
||||||
|
|
||||||
|
let db_lock = borrow_db_checked();
|
||||||
|
|
||||||
|
let metadata_option = db_lock.applications.installed_game_version.get(&id);
|
||||||
|
let version = match metadata_option {
|
||||||
|
None => None,
|
||||||
|
Some(metadata) => db_lock
|
||||||
|
.applications
|
||||||
|
.game_versions
|
||||||
|
.get(&metadata.id)
|
||||||
|
.map(|v| v.get(metadata.version.as_ref().unwrap()).unwrap())
|
||||||
|
.cloned(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let game = state_handle.games.get(&id);
|
||||||
|
if let Some(game) = game {
|
||||||
|
let status = GameStatusManager::fetch_state(&id, &db_lock);
|
||||||
|
|
||||||
|
let data = FetchGameStruct::new(game.clone(), status, version);
|
||||||
|
|
||||||
|
cache_object_db(&id, game, &db_lock)?;
|
||||||
|
|
||||||
|
return Ok(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
version
|
||||||
|
};
|
||||||
|
|
||||||
|
let client = DROP_CLIENT_ASYNC.clone();
|
||||||
|
let response = generate_url(&["/api/v1/client/game/", &id], &[])?;
|
||||||
|
let response = client
|
||||||
|
.get(response)
|
||||||
|
.header("Authorization", generate_authorization_header())
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if response.status() == 404 {
|
||||||
|
let offline_fetch = fetch_game_logic_offline(id.clone(), state).await;
|
||||||
|
if let Ok(fetch_data) = offline_fetch {
|
||||||
|
return Ok(fetch_data);
|
||||||
|
}
|
||||||
|
|
||||||
|
return Err(RemoteAccessError::GameNotFound(id));
|
||||||
|
}
|
||||||
|
if response.status() != 200 {
|
||||||
|
let err = response.json().await?;
|
||||||
|
warn!("{err:?}");
|
||||||
|
return Err(RemoteAccessError::InvalidResponse(err));
|
||||||
|
}
|
||||||
|
|
||||||
|
let game: Game = response.json().await?;
|
||||||
|
|
||||||
|
let mut state_handle = state.lock();
|
||||||
|
state_handle.games.insert(id.clone(), game.clone());
|
||||||
|
|
||||||
|
let mut db_handle = borrow_db_mut_checked();
|
||||||
|
|
||||||
|
db_handle
|
||||||
|
.applications
|
||||||
|
.game_statuses
|
||||||
|
.entry(id.clone())
|
||||||
|
.or_insert(GameDownloadStatus::Remote {});
|
||||||
|
|
||||||
|
let status = GameStatusManager::fetch_state(&id, &db_handle);
|
||||||
|
|
||||||
|
drop(db_handle);
|
||||||
|
|
||||||
|
let data = FetchGameStruct::new(game.clone(), status, version);
|
||||||
|
|
||||||
|
cache_object(&id, &game)?;
|
||||||
|
|
||||||
|
Ok(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn fetch_game_version_options_logic(
|
||||||
|
game_id: String,
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<Vec<GameVersion>, RemoteAccessError> {
|
||||||
|
let client = DROP_CLIENT_ASYNC.clone();
|
||||||
|
|
||||||
|
let response = generate_url(&["/api/v1/client/game/versions"], &[("id", &game_id)])?;
|
||||||
|
let response = client
|
||||||
|
.get(response)
|
||||||
|
.header("Authorization", generate_authorization_header())
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if response.status() != 200 {
|
||||||
|
let err = response.json().await?;
|
||||||
|
warn!("{err:?}");
|
||||||
|
return Err(RemoteAccessError::InvalidResponse(err));
|
||||||
|
}
|
||||||
|
|
||||||
|
let data: Vec<GameVersion> = response.json().await?;
|
||||||
|
|
||||||
|
let state_lock = state.lock();
|
||||||
|
let process_manager_lock = PROCESS_MANAGER.lock();
|
||||||
|
let data: Vec<GameVersion> = data
|
||||||
|
.into_iter()
|
||||||
|
.filter(|v| process_manager_lock.valid_platform(&v.platform))
|
||||||
|
.collect();
|
||||||
|
drop(process_manager_lock);
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn fetch_game_logic_offline(
|
||||||
|
id: String,
|
||||||
|
_state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<FetchGameStruct, RemoteAccessError> {
|
||||||
|
let db_handle = borrow_db_checked();
|
||||||
|
let metadata_option = db_handle.applications.installed_game_version.get(&id);
|
||||||
|
let version = match metadata_option {
|
||||||
|
None => None,
|
||||||
|
Some(metadata) => db_handle
|
||||||
|
.applications
|
||||||
|
.game_versions
|
||||||
|
.get(&metadata.id)
|
||||||
|
.map(|v| v.get(metadata.version.as_ref().unwrap()).unwrap())
|
||||||
|
.cloned(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let status = GameStatusManager::fetch_state(&id, &db_handle);
|
||||||
|
let game = get_cached_object::<Game>(&id)?;
|
||||||
|
|
||||||
|
drop(db_handle);
|
||||||
|
|
||||||
|
Ok(FetchGameStruct::new(game, status, version))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub async fn fetch_game(
|
||||||
|
game_id: String,
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<FetchGameStruct, RemoteAccessError> {
|
||||||
|
offline!(
|
||||||
|
state,
|
||||||
|
fetch_game_logic,
|
||||||
|
fetch_game_logic_offline,
|
||||||
|
game_id,
|
||||||
|
state
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn fetch_game_status(id: String) -> GameStatusWithTransient {
|
||||||
|
let db_handle = borrow_db_checked();
|
||||||
|
GameStatusManager::fetch_state(&id, &db_handle)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn uninstall_game(game_id: String, app_handle: AppHandle) -> Result<(), LibraryError> {
|
||||||
|
let meta = match get_current_meta(&game_id) {
|
||||||
|
Some(data) => data,
|
||||||
|
None => return Err(LibraryError::MetaNotFound(game_id)),
|
||||||
|
};
|
||||||
|
uninstall_game_logic(meta, &app_handle);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub async fn fetch_game_version_options(
|
||||||
|
game_id: String,
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<Vec<GameVersion>, RemoteAccessError> {
|
||||||
|
fetch_game_version_options_logic(game_id, state).await
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn update_game_configuration(
|
||||||
|
game_id: String,
|
||||||
|
options: FrontendGameOptions,
|
||||||
|
) -> Result<(), LibraryError> {
|
||||||
|
let mut handle = borrow_db_mut_checked();
|
||||||
|
let installed_version = handle
|
||||||
|
.applications
|
||||||
|
.installed_game_version
|
||||||
|
.get(&game_id)
|
||||||
|
.ok_or(LibraryError::MetaNotFound(game_id))?;
|
||||||
|
|
||||||
|
let id = installed_version.id.clone();
|
||||||
|
let version = installed_version
|
||||||
|
.version
|
||||||
|
.clone()
|
||||||
|
.ok_or(LibraryError::VersionNotFound(id.clone()))?;
|
||||||
|
|
||||||
|
let mut existing_configuration = handle
|
||||||
|
.applications
|
||||||
|
.game_versions
|
||||||
|
.get(&id)
|
||||||
|
.unwrap()
|
||||||
|
.get(&version)
|
||||||
|
.unwrap()
|
||||||
|
.clone();
|
||||||
|
|
||||||
|
// Add more options in here
|
||||||
|
existing_configuration.launch_command_template = options.launch_string().clone();
|
||||||
|
|
||||||
|
// Add no more options past here
|
||||||
|
|
||||||
|
handle
|
||||||
|
.applications
|
||||||
|
.game_versions
|
||||||
|
.get_mut(&id)
|
||||||
|
.unwrap()
|
||||||
|
.insert(version.to_string(), existing_configuration);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@ -1,78 +0,0 @@
|
|||||||
use std::sync::Mutex;
|
|
||||||
|
|
||||||
use tauri::AppHandle;
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
AppState,
|
|
||||||
database::{
|
|
||||||
db::borrow_db_checked,
|
|
||||||
models::data::GameVersion,
|
|
||||||
},
|
|
||||||
error::{library_error::LibraryError, remote_access_error::RemoteAccessError},
|
|
||||||
games::library::{
|
|
||||||
fetch_game_logic_offline, fetch_library_logic_offline, get_current_meta,
|
|
||||||
uninstall_game_logic,
|
|
||||||
},
|
|
||||||
offline,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
|
||||||
library::{
|
|
||||||
FetchGameStruct, Game, fetch_game_logic, fetch_game_version_options_logic,
|
|
||||||
fetch_library_logic,
|
|
||||||
},
|
|
||||||
state::{GameStatusManager, GameStatusWithTransient},
|
|
||||||
};
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub async fn fetch_library(
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
hard_refresh: Option<bool>,
|
|
||||||
) -> Result<Vec<Game>, RemoteAccessError> {
|
|
||||||
offline!(
|
|
||||||
state,
|
|
||||||
fetch_library_logic,
|
|
||||||
fetch_library_logic_offline,
|
|
||||||
state,
|
|
||||||
hard_refresh
|
|
||||||
).await
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub async fn fetch_game(
|
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<FetchGameStruct, RemoteAccessError> {
|
|
||||||
offline!(
|
|
||||||
state,
|
|
||||||
fetch_game_logic,
|
|
||||||
fetch_game_logic_offline,
|
|
||||||
game_id,
|
|
||||||
state
|
|
||||||
).await
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn fetch_game_status(id: String) -> GameStatusWithTransient {
|
|
||||||
let db_handle = borrow_db_checked();
|
|
||||||
GameStatusManager::fetch_state(&id, &db_handle)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn uninstall_game(game_id: String, app_handle: AppHandle) -> Result<(), LibraryError> {
|
|
||||||
let meta = match get_current_meta(&game_id) {
|
|
||||||
Some(data) => data,
|
|
||||||
None => return Err(LibraryError::MetaNotFound(game_id)),
|
|
||||||
};
|
|
||||||
uninstall_game_logic(meta, &app_handle);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub async fn fetch_game_version_options(
|
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<Vec<GameVersion>, RemoteAccessError> {
|
|
||||||
fetch_game_version_options_logic(game_id, state).await
|
|
||||||
}
|
|
||||||
@ -1,588 +0,0 @@
|
|||||||
use std::fs::remove_dir_all;
|
|
||||||
use std::sync::Mutex;
|
|
||||||
use std::thread::spawn;
|
|
||||||
|
|
||||||
use log::{debug, error, warn};
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use tauri::AppHandle;
|
|
||||||
use tauri::Emitter;
|
|
||||||
|
|
||||||
use crate::AppState;
|
|
||||||
use crate::app_emit;
|
|
||||||
use crate::database::db::{borrow_db_checked, borrow_db_mut_checked};
|
|
||||||
use crate::database::models::data::Database;
|
|
||||||
use crate::database::models::data::{
|
|
||||||
ApplicationTransientStatus, DownloadableMetadata, GameDownloadStatus, GameVersion,
|
|
||||||
};
|
|
||||||
use crate::download_manager::download_manager_frontend::DownloadStatus;
|
|
||||||
use crate::error::drop_server_error::DropServerError;
|
|
||||||
use crate::error::library_error::LibraryError;
|
|
||||||
use crate::error::remote_access_error::RemoteAccessError;
|
|
||||||
use crate::games::state::{GameStatusManager, GameStatusWithTransient};
|
|
||||||
use crate::lock;
|
|
||||||
use crate::remote::auth::generate_authorization_header;
|
|
||||||
use crate::remote::cache::cache_object_db;
|
|
||||||
use crate::remote::cache::{cache_object, get_cached_object, get_cached_object_db};
|
|
||||||
use crate::remote::requests::generate_url;
|
|
||||||
use crate::remote::utils::DROP_CLIENT_ASYNC;
|
|
||||||
use crate::remote::utils::DROP_CLIENT_SYNC;
|
|
||||||
use bitcode::{Decode, Encode};
|
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug)]
|
|
||||||
pub struct FetchGameStruct {
|
|
||||||
game: Game,
|
|
||||||
status: GameStatusWithTransient,
|
|
||||||
version: Option<GameVersion>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Clone, Debug, Default, Encode, Decode)]
|
|
||||||
#[serde(rename_all = "camelCase")]
|
|
||||||
pub struct Game {
|
|
||||||
id: String,
|
|
||||||
m_name: String,
|
|
||||||
m_short_description: String,
|
|
||||||
m_description: String,
|
|
||||||
// mDevelopers
|
|
||||||
// mPublishers
|
|
||||||
m_icon_object_id: String,
|
|
||||||
m_banner_object_id: String,
|
|
||||||
m_cover_object_id: String,
|
|
||||||
m_image_library_object_ids: Vec<String>,
|
|
||||||
m_image_carousel_object_ids: Vec<String>,
|
|
||||||
}
|
|
||||||
#[derive(serde::Serialize, Clone)]
|
|
||||||
pub struct GameUpdateEvent {
|
|
||||||
pub game_id: String,
|
|
||||||
pub status: (
|
|
||||||
Option<GameDownloadStatus>,
|
|
||||||
Option<ApplicationTransientStatus>,
|
|
||||||
),
|
|
||||||
pub version: Option<GameVersion>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Serialize, Clone)]
|
|
||||||
pub struct QueueUpdateEventQueueData {
|
|
||||||
pub meta: DownloadableMetadata,
|
|
||||||
pub status: DownloadStatus,
|
|
||||||
pub progress: f64,
|
|
||||||
pub current: usize,
|
|
||||||
pub max: usize,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, Clone)]
|
|
||||||
pub struct QueueUpdateEvent {
|
|
||||||
pub queue: Vec<QueueUpdateEventQueueData>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, Clone)]
|
|
||||||
pub struct StatsUpdateEvent {
|
|
||||||
pub speed: usize,
|
|
||||||
pub time: usize,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn fetch_library_logic(
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
hard_fresh: Option<bool>,
|
|
||||||
) -> Result<Vec<Game>, RemoteAccessError> {
|
|
||||||
let do_hard_refresh = hard_fresh.unwrap_or(false);
|
|
||||||
if !do_hard_refresh && let Ok(library) = get_cached_object("library") {
|
|
||||||
return Ok(library);
|
|
||||||
}
|
|
||||||
|
|
||||||
let client = DROP_CLIENT_ASYNC.clone();
|
|
||||||
let response = generate_url(&["/api/v1/client/user/library"], &[])?;
|
|
||||||
let response = client
|
|
||||||
.get(response)
|
|
||||||
.header("Authorization", generate_authorization_header())
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if response.status() != 200 {
|
|
||||||
let err = response.json().await.unwrap_or(DropServerError {
|
|
||||||
status_code: 500,
|
|
||||||
status_message: "Invalid response from server.".to_owned(),
|
|
||||||
});
|
|
||||||
warn!("{err:?}");
|
|
||||||
return Err(RemoteAccessError::InvalidResponse(err));
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut games: Vec<Game> = response.json().await?;
|
|
||||||
|
|
||||||
let mut handle = lock!(state);
|
|
||||||
|
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
|
||||||
|
|
||||||
for game in &games {
|
|
||||||
handle.games.insert(game.id.clone(), game.clone());
|
|
||||||
if !db_handle.applications.game_statuses.contains_key(&game.id) {
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.game_statuses
|
|
||||||
.insert(game.id.clone(), GameDownloadStatus::Remote {});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add games that are installed but no longer in library
|
|
||||||
for meta in db_handle.applications.installed_game_version.values() {
|
|
||||||
if games.iter().any(|e| e.id == meta.id) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
// We should always have a cache of the object
|
|
||||||
// Pass db_handle because otherwise we get a gridlock
|
|
||||||
let game = match get_cached_object_db::<Game>(&meta.id.clone(), &db_handle) {
|
|
||||||
Ok(game) => game,
|
|
||||||
Err(err) => {
|
|
||||||
warn!(
|
|
||||||
"{} is installed, but encountered error fetching its error: {}.",
|
|
||||||
meta.id, err
|
|
||||||
);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
games.push(game);
|
|
||||||
}
|
|
||||||
|
|
||||||
drop(handle);
|
|
||||||
drop(db_handle);
|
|
||||||
cache_object("library", &games)?;
|
|
||||||
|
|
||||||
Ok(games)
|
|
||||||
}
|
|
||||||
pub async fn fetch_library_logic_offline(
|
|
||||||
_state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
_hard_refresh: Option<bool>,
|
|
||||||
) -> Result<Vec<Game>, RemoteAccessError> {
|
|
||||||
let mut games: Vec<Game> = get_cached_object("library")?;
|
|
||||||
|
|
||||||
let db_handle = borrow_db_checked();
|
|
||||||
|
|
||||||
games.retain(|game| {
|
|
||||||
matches!(
|
|
||||||
&db_handle
|
|
||||||
.applications
|
|
||||||
.game_statuses
|
|
||||||
.get(&game.id)
|
|
||||||
.unwrap_or(&GameDownloadStatus::Remote {}),
|
|
||||||
GameDownloadStatus::Installed { .. } | GameDownloadStatus::SetupRequired { .. }
|
|
||||||
)
|
|
||||||
});
|
|
||||||
|
|
||||||
Ok(games)
|
|
||||||
}
|
|
||||||
pub async fn fetch_game_logic(
|
|
||||||
id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<FetchGameStruct, RemoteAccessError> {
|
|
||||||
let version = {
|
|
||||||
let state_handle = lock!(state);
|
|
||||||
|
|
||||||
let db_lock = borrow_db_checked();
|
|
||||||
|
|
||||||
let metadata_option = db_lock.applications.installed_game_version.get(&id);
|
|
||||||
let version = match metadata_option {
|
|
||||||
None => None,
|
|
||||||
Some(metadata) => db_lock
|
|
||||||
.applications
|
|
||||||
.game_versions
|
|
||||||
.get(&metadata.id)
|
|
||||||
.map(|v| v.get(metadata.version.as_ref().unwrap()).unwrap())
|
|
||||||
.cloned(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let game = state_handle.games.get(&id);
|
|
||||||
if let Some(game) = game {
|
|
||||||
let status = GameStatusManager::fetch_state(&id, &db_lock);
|
|
||||||
|
|
||||||
let data = FetchGameStruct {
|
|
||||||
game: game.clone(),
|
|
||||||
status,
|
|
||||||
version,
|
|
||||||
};
|
|
||||||
|
|
||||||
cache_object_db(&id, game, &db_lock)?;
|
|
||||||
|
|
||||||
return Ok(data);
|
|
||||||
}
|
|
||||||
|
|
||||||
version
|
|
||||||
};
|
|
||||||
|
|
||||||
let client = DROP_CLIENT_ASYNC.clone();
|
|
||||||
let response = generate_url(&["/api/v1/client/game/", &id], &[])?;
|
|
||||||
let response = client
|
|
||||||
.get(response)
|
|
||||||
.header("Authorization", generate_authorization_header())
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if response.status() == 404 {
|
|
||||||
let offline_fetch = fetch_game_logic_offline(id.clone(), state).await;
|
|
||||||
if let Ok(fetch_data) = offline_fetch {
|
|
||||||
return Ok(fetch_data);
|
|
||||||
}
|
|
||||||
|
|
||||||
return Err(RemoteAccessError::GameNotFound(id));
|
|
||||||
}
|
|
||||||
if response.status() != 200 {
|
|
||||||
let err = response.json().await?;
|
|
||||||
warn!("{err:?}");
|
|
||||||
return Err(RemoteAccessError::InvalidResponse(err));
|
|
||||||
}
|
|
||||||
|
|
||||||
let game: Game = response.json().await?;
|
|
||||||
|
|
||||||
let mut state_handle = lock!(state);
|
|
||||||
state_handle.games.insert(id.clone(), game.clone());
|
|
||||||
|
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
|
||||||
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.game_statuses
|
|
||||||
.entry(id.clone())
|
|
||||||
.or_insert(GameDownloadStatus::Remote {});
|
|
||||||
|
|
||||||
let status = GameStatusManager::fetch_state(&id, &db_handle);
|
|
||||||
|
|
||||||
drop(db_handle);
|
|
||||||
|
|
||||||
let data = FetchGameStruct {
|
|
||||||
game: game.clone(),
|
|
||||||
status,
|
|
||||||
version,
|
|
||||||
};
|
|
||||||
|
|
||||||
cache_object(&id, &game)?;
|
|
||||||
|
|
||||||
Ok(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn fetch_game_logic_offline(
|
|
||||||
id: String,
|
|
||||||
_state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<FetchGameStruct, RemoteAccessError> {
|
|
||||||
let db_handle = borrow_db_checked();
|
|
||||||
let metadata_option = db_handle.applications.installed_game_version.get(&id);
|
|
||||||
let version = match metadata_option {
|
|
||||||
None => None,
|
|
||||||
Some(metadata) => db_handle
|
|
||||||
.applications
|
|
||||||
.game_versions
|
|
||||||
.get(&metadata.id)
|
|
||||||
.map(|v| v.get(metadata.version.as_ref().unwrap()).unwrap())
|
|
||||||
.cloned(),
|
|
||||||
};
|
|
||||||
|
|
||||||
let status = GameStatusManager::fetch_state(&id, &db_handle);
|
|
||||||
let game = get_cached_object::<Game>(&id)?;
|
|
||||||
|
|
||||||
drop(db_handle);
|
|
||||||
|
|
||||||
Ok(FetchGameStruct {
|
|
||||||
game,
|
|
||||||
status,
|
|
||||||
version,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn fetch_game_version_options_logic(
|
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
|
||||||
) -> Result<Vec<GameVersion>, RemoteAccessError> {
|
|
||||||
let client = DROP_CLIENT_ASYNC.clone();
|
|
||||||
|
|
||||||
let response = generate_url(&["/api/v1/client/game/versions"], &[("id", &game_id)])?;
|
|
||||||
let response = client
|
|
||||||
.get(response)
|
|
||||||
.header("Authorization", generate_authorization_header())
|
|
||||||
.send()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if response.status() != 200 {
|
|
||||||
let err = response.json().await?;
|
|
||||||
warn!("{err:?}");
|
|
||||||
return Err(RemoteAccessError::InvalidResponse(err));
|
|
||||||
}
|
|
||||||
|
|
||||||
let data: Vec<GameVersion> = response.json().await?;
|
|
||||||
|
|
||||||
let state_lock = lock!(state);
|
|
||||||
let process_manager_lock = lock!(state_lock.process_manager);
|
|
||||||
let data: Vec<GameVersion> = data
|
|
||||||
.into_iter()
|
|
||||||
.filter(|v| process_manager_lock.valid_platform(&v.platform, &state_lock))
|
|
||||||
.collect();
|
|
||||||
drop(process_manager_lock);
|
|
||||||
drop(state_lock);
|
|
||||||
|
|
||||||
Ok(data)
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Called by:
|
|
||||||
* - on_cancel, when cancelled, for obvious reasons
|
|
||||||
* - when downloading, so if drop unexpectedly quits, we can resume the download. hidden by the "Downloading..." transient state, though
|
|
||||||
* - when scanning, to import the game
|
|
||||||
*/
|
|
||||||
pub fn set_partially_installed(
|
|
||||||
meta: &DownloadableMetadata,
|
|
||||||
install_dir: String,
|
|
||||||
app_handle: Option<&AppHandle>,
|
|
||||||
) {
|
|
||||||
set_partially_installed_db(&mut borrow_db_mut_checked(), meta, install_dir, app_handle);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn set_partially_installed_db(
|
|
||||||
db_lock: &mut Database,
|
|
||||||
meta: &DownloadableMetadata,
|
|
||||||
install_dir: String,
|
|
||||||
app_handle: Option<&AppHandle>,
|
|
||||||
) {
|
|
||||||
db_lock.applications.transient_statuses.remove(meta);
|
|
||||||
db_lock.applications.game_statuses.insert(
|
|
||||||
meta.id.clone(),
|
|
||||||
GameDownloadStatus::PartiallyInstalled {
|
|
||||||
version_name: meta.version.as_ref().unwrap().clone(),
|
|
||||||
install_dir,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
db_lock
|
|
||||||
.applications
|
|
||||||
.installed_game_version
|
|
||||||
.insert(meta.id.clone(), meta.clone());
|
|
||||||
|
|
||||||
if let Some(app_handle) = app_handle {
|
|
||||||
push_game_update(
|
|
||||||
app_handle,
|
|
||||||
&meta.id,
|
|
||||||
None,
|
|
||||||
GameStatusManager::fetch_state(&meta.id, db_lock),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn uninstall_game_logic(meta: DownloadableMetadata, app_handle: &AppHandle) {
|
|
||||||
debug!("triggered uninstall for agent");
|
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.transient_statuses
|
|
||||||
.insert(meta.clone(), ApplicationTransientStatus::Uninstalling {});
|
|
||||||
|
|
||||||
push_game_update(
|
|
||||||
app_handle,
|
|
||||||
&meta.id,
|
|
||||||
None,
|
|
||||||
GameStatusManager::fetch_state(&meta.id, &db_handle),
|
|
||||||
);
|
|
||||||
|
|
||||||
let previous_state = db_handle.applications.game_statuses.get(&meta.id).cloned();
|
|
||||||
|
|
||||||
let previous_state = if let Some(state) = previous_state {
|
|
||||||
state
|
|
||||||
} else {
|
|
||||||
warn!("uninstall job doesn't have previous state, failing silently");
|
|
||||||
return;
|
|
||||||
};
|
|
||||||
|
|
||||||
if let Some((_, install_dir)) = match previous_state {
|
|
||||||
GameDownloadStatus::Installed {
|
|
||||||
version_name,
|
|
||||||
install_dir,
|
|
||||||
} => Some((version_name, install_dir)),
|
|
||||||
GameDownloadStatus::SetupRequired {
|
|
||||||
version_name,
|
|
||||||
install_dir,
|
|
||||||
} => Some((version_name, install_dir)),
|
|
||||||
GameDownloadStatus::PartiallyInstalled {
|
|
||||||
version_name,
|
|
||||||
install_dir,
|
|
||||||
} => Some((version_name, install_dir)),
|
|
||||||
_ => None,
|
|
||||||
} {
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.transient_statuses
|
|
||||||
.insert(meta.clone(), ApplicationTransientStatus::Uninstalling {});
|
|
||||||
|
|
||||||
drop(db_handle);
|
|
||||||
|
|
||||||
let app_handle = app_handle.clone();
|
|
||||||
spawn(move || {
|
|
||||||
if let Err(e) = remove_dir_all(install_dir) {
|
|
||||||
error!("{e}");
|
|
||||||
} else {
|
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
|
||||||
db_handle.applications.transient_statuses.remove(&meta);
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.installed_game_version
|
|
||||||
.remove(&meta.id);
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.game_statuses
|
|
||||||
.insert(meta.id.clone(), GameDownloadStatus::Remote {});
|
|
||||||
let _ = db_handle.applications.transient_statuses.remove(&meta);
|
|
||||||
|
|
||||||
push_game_update(
|
|
||||||
&app_handle,
|
|
||||||
&meta.id,
|
|
||||||
None,
|
|
||||||
GameStatusManager::fetch_state(&meta.id, &db_handle),
|
|
||||||
);
|
|
||||||
|
|
||||||
debug!("uninstalled game id {}", &meta.id);
|
|
||||||
app_emit!(app_handle, "update_library", ());
|
|
||||||
}
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
warn!("invalid previous state for uninstall, failing silently.");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_current_meta(game_id: &String) -> Option<DownloadableMetadata> {
|
|
||||||
borrow_db_checked()
|
|
||||||
.applications
|
|
||||||
.installed_game_version
|
|
||||||
.get(game_id)
|
|
||||||
.cloned()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn on_game_complete(
|
|
||||||
meta: &DownloadableMetadata,
|
|
||||||
install_dir: String,
|
|
||||||
app_handle: &AppHandle,
|
|
||||||
) -> Result<(), RemoteAccessError> {
|
|
||||||
// Fetch game version information from remote
|
|
||||||
if meta.version.is_none() {
|
|
||||||
return Err(RemoteAccessError::GameNotFound(meta.id.clone()));
|
|
||||||
}
|
|
||||||
|
|
||||||
let client = DROP_CLIENT_SYNC.clone();
|
|
||||||
let response = generate_url(
|
|
||||||
&["/api/v1/client/game/version"],
|
|
||||||
&[
|
|
||||||
("id", &meta.id),
|
|
||||||
("version", meta.version.as_ref().unwrap()),
|
|
||||||
],
|
|
||||||
)?;
|
|
||||||
let response = client
|
|
||||||
.get(response)
|
|
||||||
.header("Authorization", generate_authorization_header())
|
|
||||||
.send()?;
|
|
||||||
|
|
||||||
let game_version: GameVersion = response.json()?;
|
|
||||||
|
|
||||||
let mut handle = borrow_db_mut_checked();
|
|
||||||
handle
|
|
||||||
.applications
|
|
||||||
.game_versions
|
|
||||||
.entry(meta.id.clone())
|
|
||||||
.or_default()
|
|
||||||
.insert(meta.version.clone().unwrap(), game_version.clone());
|
|
||||||
handle
|
|
||||||
.applications
|
|
||||||
.installed_game_version
|
|
||||||
.insert(meta.id.clone(), meta.clone());
|
|
||||||
|
|
||||||
drop(handle);
|
|
||||||
|
|
||||||
let status = if game_version.setup_command.is_empty() {
|
|
||||||
GameDownloadStatus::Installed {
|
|
||||||
version_name: meta.version.clone().unwrap(),
|
|
||||||
install_dir,
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
GameDownloadStatus::SetupRequired {
|
|
||||||
version_name: meta.version.clone().unwrap(),
|
|
||||||
install_dir,
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut db_handle = borrow_db_mut_checked();
|
|
||||||
db_handle
|
|
||||||
.applications
|
|
||||||
.game_statuses
|
|
||||||
.insert(meta.id.clone(), status.clone());
|
|
||||||
drop(db_handle);
|
|
||||||
app_emit!(
|
|
||||||
app_handle,
|
|
||||||
&format!("update_game/{}", meta.id),
|
|
||||||
GameUpdateEvent {
|
|
||||||
game_id: meta.id.clone(),
|
|
||||||
status: (Some(status), None),
|
|
||||||
version: Some(game_version),
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn push_game_update(
|
|
||||||
app_handle: &AppHandle,
|
|
||||||
game_id: &String,
|
|
||||||
version: Option<GameVersion>,
|
|
||||||
status: GameStatusWithTransient,
|
|
||||||
) {
|
|
||||||
if let Some(GameDownloadStatus::Installed { .. } | GameDownloadStatus::SetupRequired { .. }) =
|
|
||||||
&status.0
|
|
||||||
&& version.is_none()
|
|
||||||
{
|
|
||||||
panic!("pushed game for installed game that doesn't have version information");
|
|
||||||
}
|
|
||||||
|
|
||||||
app_emit!(
|
|
||||||
app_handle,
|
|
||||||
&format!("update_game/{game_id}"),
|
|
||||||
GameUpdateEvent {
|
|
||||||
game_id: game_id.clone(),
|
|
||||||
status,
|
|
||||||
version,
|
|
||||||
}
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
|
||||||
#[serde(rename_all = "camelCase")]
|
|
||||||
pub struct FrontendGameOptions {
|
|
||||||
launch_string: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn update_game_configuration(
|
|
||||||
game_id: String,
|
|
||||||
options: FrontendGameOptions,
|
|
||||||
) -> Result<(), LibraryError> {
|
|
||||||
let mut handle = borrow_db_mut_checked();
|
|
||||||
let installed_version = handle
|
|
||||||
.applications
|
|
||||||
.installed_game_version
|
|
||||||
.get(&game_id)
|
|
||||||
.ok_or(LibraryError::MetaNotFound(game_id))?;
|
|
||||||
|
|
||||||
let id = installed_version.id.clone();
|
|
||||||
let version = installed_version.version.clone().ok_or(LibraryError::VersionNotFound(id.clone()))?;
|
|
||||||
|
|
||||||
let mut existing_configuration = handle
|
|
||||||
.applications
|
|
||||||
.game_versions
|
|
||||||
.get(&id)
|
|
||||||
.unwrap()
|
|
||||||
.get(&version)
|
|
||||||
.unwrap()
|
|
||||||
.clone();
|
|
||||||
|
|
||||||
// Add more options in here
|
|
||||||
existing_configuration.launch_command_template = options.launch_string;
|
|
||||||
|
|
||||||
// Add no more options past here
|
|
||||||
|
|
||||||
handle
|
|
||||||
.applications
|
|
||||||
.game_versions
|
|
||||||
.get_mut(&id)
|
|
||||||
.unwrap()
|
|
||||||
.insert(version.to_string(), existing_configuration);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
@ -3,136 +3,79 @@
|
|||||||
#![feature(duration_constructors)]
|
#![feature(duration_constructors)]
|
||||||
#![feature(duration_millis_float)]
|
#![feature(duration_millis_float)]
|
||||||
#![feature(iterator_try_collect)]
|
#![feature(iterator_try_collect)]
|
||||||
|
#![feature(nonpoison_mutex)]
|
||||||
|
#![feature(sync_nonpoison)]
|
||||||
#![deny(clippy::all)]
|
#![deny(clippy::all)]
|
||||||
|
|
||||||
mod database;
|
|
||||||
mod games;
|
|
||||||
|
|
||||||
mod client;
|
|
||||||
mod download_manager;
|
|
||||||
mod error;
|
|
||||||
mod process;
|
|
||||||
mod remote;
|
|
||||||
mod utils;
|
|
||||||
|
|
||||||
use crate::database::scan::scan_install_dirs;
|
|
||||||
use crate::process::commands::open_process_logs;
|
|
||||||
use crate::process::process_handlers::UMU_LAUNCHER_EXECUTABLE;
|
|
||||||
use crate::remote::commands::auth_initiate_code;
|
|
||||||
use crate::remote::fetch_object::fetch_object_wrapper;
|
|
||||||
use crate::remote::server_proto::handle_server_proto_wrapper;
|
|
||||||
use crate::{database::db::DatabaseImpls, games::downloads::commands::resume_download};
|
|
||||||
use bitcode::{Decode, Encode};
|
|
||||||
use client::commands::fetch_state;
|
|
||||||
use client::{
|
|
||||||
autostart::{get_autostart_enabled, sync_autostart_on_startup, toggle_autostart},
|
|
||||||
cleanup::{cleanup_and_exit, quit},
|
|
||||||
};
|
|
||||||
use database::commands::{
|
|
||||||
add_download_dir, delete_download_dir, fetch_download_dir_stats, fetch_settings,
|
|
||||||
fetch_system_data, update_settings,
|
|
||||||
};
|
|
||||||
use database::db::{DATA_ROOT_DIR, DatabaseInterface, borrow_db_checked, borrow_db_mut_checked};
|
|
||||||
use database::models::data::GameDownloadStatus;
|
|
||||||
use download_manager::commands::{
|
|
||||||
cancel_game, move_download_in_queue, pause_downloads, resume_downloads,
|
|
||||||
};
|
|
||||||
use download_manager::download_manager_builder::DownloadManagerBuilder;
|
|
||||||
use download_manager::download_manager_frontend::DownloadManager;
|
|
||||||
use games::collections::commands::{
|
|
||||||
add_game_to_collection, create_collection, delete_collection, delete_game_in_collection,
|
|
||||||
fetch_collection, fetch_collections,
|
|
||||||
};
|
|
||||||
use games::commands::{
|
|
||||||
fetch_game, fetch_game_status, fetch_game_version_options, fetch_library, uninstall_game,
|
|
||||||
};
|
|
||||||
use games::downloads::commands::download_game;
|
|
||||||
use games::library::{Game, update_game_configuration};
|
|
||||||
use log::{LevelFilter, debug, info, warn};
|
|
||||||
use log4rs::Config;
|
|
||||||
use log4rs::append::console::ConsoleAppender;
|
|
||||||
use log4rs::append::file::FileAppender;
|
|
||||||
use log4rs::config::{Appender, Root};
|
|
||||||
use log4rs::encode::pattern::PatternEncoder;
|
|
||||||
use process::commands::{kill_game, launch_game};
|
|
||||||
use process::process_manager::ProcessManager;
|
|
||||||
use remote::auth::{self, recieve_handshake};
|
|
||||||
use remote::commands::{
|
|
||||||
auth_initiate, fetch_drop_object, gen_drop_url, manual_recieve_handshake, retry_connect,
|
|
||||||
sign_out, use_remote,
|
|
||||||
};
|
|
||||||
use remote::server_proto::handle_server_proto_offline_wrapper;
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
use std::fs::File;
|
|
||||||
use std::io::Write;
|
|
||||||
use std::panic::PanicHookInfo;
|
|
||||||
use std::path::Path;
|
|
||||||
use std::str::FromStr;
|
|
||||||
use std::sync::Arc;
|
|
||||||
use std::time::SystemTime;
|
|
||||||
use std::{
|
use std::{
|
||||||
collections::HashMap,
|
collections::HashMap, env, fs::File, io::Write, panic::PanicHookInfo, path::Path, str::FromStr,
|
||||||
sync::{LazyLock, Mutex},
|
sync::nonpoison::Mutex, time::SystemTime,
|
||||||
|
};
|
||||||
|
|
||||||
|
use ::client::{app_status::AppStatus, autostart::sync_autostart_on_startup, user::User};
|
||||||
|
use ::download_manager::DownloadManagerWrapper;
|
||||||
|
use ::games::{library::Game, scan::scan_install_dirs};
|
||||||
|
use ::process::ProcessManagerWrapper;
|
||||||
|
use ::remote::{
|
||||||
|
auth::{self, HandshakeRequestBody, HandshakeResponse, generate_authorization_header},
|
||||||
|
cache::clear_cached_object,
|
||||||
|
error::RemoteAccessError,
|
||||||
|
fetch_object::fetch_object_wrapper,
|
||||||
|
offline,
|
||||||
|
server_proto::{handle_server_proto_offline_wrapper, handle_server_proto_wrapper},
|
||||||
|
utils::DROP_CLIENT_ASYNC,
|
||||||
|
};
|
||||||
|
use database::{
|
||||||
|
DB, GameDownloadStatus, borrow_db_checked, borrow_db_mut_checked, db::DATA_ROOT_DIR,
|
||||||
|
interface::DatabaseImpls,
|
||||||
|
};
|
||||||
|
use log::{LevelFilter, debug, info, warn};
|
||||||
|
use log4rs::{
|
||||||
|
Config,
|
||||||
|
append::{console::ConsoleAppender, file::FileAppender},
|
||||||
|
config::{Appender, Root},
|
||||||
|
encode::pattern::PatternEncoder,
|
||||||
|
};
|
||||||
|
use serde::Serialize;
|
||||||
|
use tauri::{
|
||||||
|
AppHandle, Manager, RunEvent, WindowEvent,
|
||||||
|
menu::{Menu, MenuItem, PredefinedMenuItem},
|
||||||
|
tray::TrayIconBuilder,
|
||||||
};
|
};
|
||||||
use std::{env, panic};
|
|
||||||
use tauri::menu::{Menu, MenuItem, PredefinedMenuItem};
|
|
||||||
use tauri::tray::TrayIconBuilder;
|
|
||||||
use tauri::{AppHandle, Manager, RunEvent, WindowEvent};
|
|
||||||
use tauri_plugin_deep_link::DeepLinkExt;
|
use tauri_plugin_deep_link::DeepLinkExt;
|
||||||
use tauri_plugin_dialog::DialogExt;
|
use tauri_plugin_dialog::DialogExt;
|
||||||
|
use url::Url;
|
||||||
|
use utils::app_emit;
|
||||||
|
|
||||||
#[derive(Clone, Copy, Serialize, Eq, PartialEq)]
|
use crate::client::cleanup_and_exit;
|
||||||
pub enum AppStatus {
|
|
||||||
NotConfigured,
|
|
||||||
Offline,
|
|
||||||
ServerError,
|
|
||||||
SignedOut,
|
|
||||||
SignedIn,
|
|
||||||
SignedInNeedsReauth,
|
|
||||||
ServerUnavailable,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Serialize, Deserialize, Encode, Decode)]
|
mod client;
|
||||||
#[serde(rename_all = "camelCase")]
|
mod collections;
|
||||||
pub struct User {
|
mod download_manager;
|
||||||
id: String,
|
mod downloads;
|
||||||
username: String,
|
mod games;
|
||||||
admin: bool,
|
mod process;
|
||||||
display_name: String,
|
mod remote;
|
||||||
profile_picture_object_id: String,
|
mod settings;
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
use client::*;
|
||||||
pub struct CompatInfo {
|
use collections::*;
|
||||||
umu_installed: bool,
|
use download_manager::*;
|
||||||
}
|
use downloads::*;
|
||||||
|
use games::*;
|
||||||
fn create_new_compat_info() -> Option<CompatInfo> {
|
use process::*;
|
||||||
#[cfg(target_os = "windows")]
|
use remote::*;
|
||||||
return None;
|
use settings::*;
|
||||||
|
|
||||||
let has_umu_installed = UMU_LAUNCHER_EXECUTABLE.is_some();
|
|
||||||
Some(CompatInfo {
|
|
||||||
umu_installed: has_umu_installed,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Serialize)]
|
#[derive(Clone, Serialize)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub struct AppState<'a> {
|
pub struct AppState {
|
||||||
status: AppStatus,
|
status: AppStatus,
|
||||||
user: Option<User>,
|
user: Option<User>,
|
||||||
games: HashMap<String, Game>,
|
games: HashMap<String, Game>,
|
||||||
|
|
||||||
#[serde(skip_serializing)]
|
|
||||||
download_manager: Arc<DownloadManager>,
|
|
||||||
#[serde(skip_serializing)]
|
|
||||||
process_manager: Arc<Mutex<ProcessManager<'a>>>,
|
|
||||||
#[serde(skip_serializing)]
|
|
||||||
compat_info: Option<CompatInfo>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn setup(handle: AppHandle) -> AppState<'static> {
|
async fn setup(handle: AppHandle) -> AppState {
|
||||||
let logfile = FileAppender::builder()
|
let logfile = FileAppender::builder()
|
||||||
.encoder(Box::new(PatternEncoder::new(
|
.encoder(Box::new(PatternEncoder::new(
|
||||||
"{d} | {l} | {f}:{L} - {m}{n}",
|
"{d} | {l} | {f}:{L} - {m}{n}",
|
||||||
@ -164,9 +107,9 @@ async fn setup(handle: AppHandle) -> AppState<'static> {
|
|||||||
log4rs::init_config(config).expect("Failed to initialise log4rs");
|
log4rs::init_config(config).expect("Failed to initialise log4rs");
|
||||||
|
|
||||||
let games = HashMap::new();
|
let games = HashMap::new();
|
||||||
let download_manager = Arc::new(DownloadManagerBuilder::build(handle.clone()));
|
|
||||||
let process_manager = Arc::new(Mutex::new(ProcessManager::new(handle.clone())));
|
ProcessManagerWrapper::init(handle.clone());
|
||||||
let compat_info = create_new_compat_info();
|
DownloadManagerWrapper::init(handle.clone());
|
||||||
|
|
||||||
debug!("checking if database is set up");
|
debug!("checking if database is set up");
|
||||||
let is_set_up = DB.database_is_set_up();
|
let is_set_up = DB.database_is_set_up();
|
||||||
@ -178,9 +121,6 @@ async fn setup(handle: AppHandle) -> AppState<'static> {
|
|||||||
status: AppStatus::NotConfigured,
|
status: AppStatus::NotConfigured,
|
||||||
user: None,
|
user: None,
|
||||||
games,
|
games,
|
||||||
download_manager,
|
|
||||||
process_manager,
|
|
||||||
compat_info,
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -243,14 +183,9 @@ async fn setup(handle: AppHandle) -> AppState<'static> {
|
|||||||
status: app_status,
|
status: app_status,
|
||||||
user,
|
user,
|
||||||
games,
|
games,
|
||||||
download_manager,
|
|
||||||
process_manager,
|
|
||||||
compat_info,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub static DB: LazyLock<DatabaseInterface> = LazyLock::new(DatabaseInterface::set_up_database);
|
|
||||||
|
|
||||||
pub fn custom_panic_handler(e: &PanicHookInfo) -> Option<()> {
|
pub fn custom_panic_handler(e: &PanicHookInfo) -> Option<()> {
|
||||||
let crash_file = DATA_ROOT_DIR.join(format!(
|
let crash_file = DATA_ROOT_DIR.join(format!(
|
||||||
"crash-{}.log",
|
"crash-{}.log",
|
||||||
@ -269,7 +204,7 @@ pub fn custom_panic_handler(e: &PanicHookInfo) -> Option<()> {
|
|||||||
|
|
||||||
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
#[cfg_attr(mobile, tauri::mobile_entry_point)]
|
||||||
pub fn run() {
|
pub fn run() {
|
||||||
panic::set_hook(Box::new(|e| {
|
std::panic::set_hook(Box::new(|e| {
|
||||||
let _ = custom_panic_handler(e);
|
let _ = custom_panic_handler(e);
|
||||||
println!("{e}");
|
println!("{e}");
|
||||||
}));
|
}));
|
||||||
@ -391,11 +326,14 @@ pub fn run() {
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
let open_menu_item = MenuItem::with_id(app, "open", "Open", true, None::<&str>).expect("Failed to generate open menu item");
|
let open_menu_item = MenuItem::with_id(app, "open", "Open", true, None::<&str>)
|
||||||
|
.expect("Failed to generate open menu item");
|
||||||
|
|
||||||
let sep = PredefinedMenuItem::separator(app).expect("Failed to generate menu separator item");
|
let sep = PredefinedMenuItem::separator(app)
|
||||||
|
.expect("Failed to generate menu separator item");
|
||||||
|
|
||||||
let quit_menu_item = MenuItem::with_id(app, "quit", "Quit", true, None::<&str>).expect("Failed to generate quit menu item");
|
let quit_menu_item = MenuItem::with_id(app, "quit", "Quit", true, None::<&str>)
|
||||||
|
.expect("Failed to generate quit menu item");
|
||||||
|
|
||||||
let menu = Menu::with_items(
|
let menu = Menu::with_items(
|
||||||
app,
|
app,
|
||||||
@ -414,7 +352,11 @@ pub fn run() {
|
|||||||
|
|
||||||
run_on_tray(|| {
|
run_on_tray(|| {
|
||||||
TrayIconBuilder::new()
|
TrayIconBuilder::new()
|
||||||
.icon(app.default_window_icon().expect("Failed to get default window icon").clone())
|
.icon(
|
||||||
|
app.default_window_icon()
|
||||||
|
.expect("Failed to get default window icon")
|
||||||
|
.clone(),
|
||||||
|
)
|
||||||
.menu(&menu)
|
.menu(&menu)
|
||||||
.on_menu_event(|app, event| match event.id.as_ref() {
|
.on_menu_event(|app, event| match event.id.as_ref() {
|
||||||
"open" => {
|
"open" => {
|
||||||
@ -425,7 +367,7 @@ pub fn run() {
|
|||||||
.expect("Failed to show window");
|
.expect("Failed to show window");
|
||||||
}
|
}
|
||||||
"quit" => {
|
"quit" => {
|
||||||
cleanup_and_exit(app, &app.state());
|
cleanup_and_exit(app);
|
||||||
}
|
}
|
||||||
|
|
||||||
_ => {
|
_ => {
|
||||||
@ -511,3 +453,85 @@ fn run_on_tray<T: FnOnce()>(f: T) {
|
|||||||
(f)();
|
(f)();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: Refactor
|
||||||
|
pub async fn recieve_handshake(app: AppHandle, path: String) {
|
||||||
|
// Tell the app we're processing
|
||||||
|
app_emit!(&app, "auth/processing", ());
|
||||||
|
|
||||||
|
let handshake_result = recieve_handshake_logic(&app, path).await;
|
||||||
|
if let Err(e) = handshake_result {
|
||||||
|
warn!("error with authentication: {e}");
|
||||||
|
app_emit!(&app, "auth/failed", e.to_string());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let app_state = app.state::<Mutex<AppState>>();
|
||||||
|
|
||||||
|
let (app_status, user) = auth::setup().await;
|
||||||
|
|
||||||
|
let mut state_lock = app_state.lock();
|
||||||
|
|
||||||
|
state_lock.status = app_status;
|
||||||
|
state_lock.user = user;
|
||||||
|
|
||||||
|
let _ = clear_cached_object("collections");
|
||||||
|
let _ = clear_cached_object("library");
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
app_emit!(&app, "auth/finished", ());
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: Refactor
|
||||||
|
async fn recieve_handshake_logic(app: &AppHandle, path: String) -> Result<(), RemoteAccessError> {
|
||||||
|
let path_chunks: Vec<&str> = path.split('/').collect();
|
||||||
|
if path_chunks.len() != 3 {
|
||||||
|
app_emit!(app, "auth/failed", ());
|
||||||
|
return Err(RemoteAccessError::HandshakeFailed(
|
||||||
|
"failed to parse token".to_string(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let base_url = {
|
||||||
|
let handle = borrow_db_checked();
|
||||||
|
Url::parse(handle.base_url.as_str())?
|
||||||
|
};
|
||||||
|
|
||||||
|
let client_id = path_chunks
|
||||||
|
.get(1)
|
||||||
|
.expect("Failed to get client id from path chunks");
|
||||||
|
let token = path_chunks
|
||||||
|
.get(2)
|
||||||
|
.expect("Failed to get token from path chunks");
|
||||||
|
let body = HandshakeRequestBody::new((client_id).to_string(), (token).to_string());
|
||||||
|
|
||||||
|
let endpoint = base_url.join("/api/v1/client/auth/handshake")?;
|
||||||
|
let client = DROP_CLIENT_ASYNC.clone();
|
||||||
|
let response = client.post(endpoint).json(&body).send().await?;
|
||||||
|
debug!("handshake responsded with {}", response.status().as_u16());
|
||||||
|
if !response.status().is_success() {
|
||||||
|
return Err(RemoteAccessError::InvalidResponse(response.json().await?));
|
||||||
|
}
|
||||||
|
let response_struct: HandshakeResponse = response.json().await?;
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut handle = borrow_db_mut_checked();
|
||||||
|
handle.auth = Some(response_struct.into());
|
||||||
|
}
|
||||||
|
|
||||||
|
let web_token = {
|
||||||
|
let header = generate_authorization_header();
|
||||||
|
let token = client
|
||||||
|
.post(base_url.join("/api/v1/client/user/webtoken")?)
|
||||||
|
.header("Authorization", header)
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
token.text().await?
|
||||||
|
};
|
||||||
|
let mut handle = borrow_db_mut_checked();
|
||||||
|
handle.auth.as_mut().unwrap().web_token = Some(web_token);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|||||||
50
src-tauri/src/process.rs
Normal file
50
src-tauri/src/process.rs
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
use std::sync::nonpoison::Mutex;
|
||||||
|
|
||||||
|
use process::{PROCESS_MANAGER, error::ProcessError};
|
||||||
|
use tauri::AppHandle;
|
||||||
|
use tauri_plugin_opener::OpenerExt;
|
||||||
|
|
||||||
|
use crate::AppState;
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn launch_game(
|
||||||
|
id: String,
|
||||||
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
|
) -> Result<(), ProcessError> {
|
||||||
|
let state_lock = state.lock();
|
||||||
|
let mut process_manager_lock = PROCESS_MANAGER.lock();
|
||||||
|
//let meta = DownloadableMetadata {
|
||||||
|
// id,
|
||||||
|
// version: Some(version),
|
||||||
|
// download_type: DownloadType::Game,
|
||||||
|
//};
|
||||||
|
|
||||||
|
match process_manager_lock.launch_process(id) {
|
||||||
|
Ok(()) => {}
|
||||||
|
Err(e) => return Err(e),
|
||||||
|
}
|
||||||
|
|
||||||
|
drop(process_manager_lock);
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn kill_game(game_id: String) -> Result<(), ProcessError> {
|
||||||
|
PROCESS_MANAGER
|
||||||
|
.lock()
|
||||||
|
.kill_game(game_id)
|
||||||
|
.map_err(ProcessError::IOError)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
pub fn open_process_logs(game_id: String, app_handle: AppHandle) -> Result<(), ProcessError> {
|
||||||
|
let process_manager_lock = PROCESS_MANAGER.lock();
|
||||||
|
|
||||||
|
let dir = process_manager_lock.get_log_dir(game_id);
|
||||||
|
app_handle
|
||||||
|
.opener()
|
||||||
|
.open_path(dir.display().to_string(), None::<&str>)
|
||||||
|
.map_err(ProcessError::OpenerError)
|
||||||
|
}
|
||||||
@ -1,50 +0,0 @@
|
|||||||
use std::sync::Mutex;
|
|
||||||
|
|
||||||
use crate::{error::process_error::ProcessError, lock, AppState};
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn launch_game(
|
|
||||||
id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState>>,
|
|
||||||
) -> Result<(), ProcessError> {
|
|
||||||
let state_lock = lock!(state);
|
|
||||||
let mut process_manager_lock = lock!(state_lock.process_manager);
|
|
||||||
|
|
||||||
//let meta = DownloadableMetadata {
|
|
||||||
// id,
|
|
||||||
// version: Some(version),
|
|
||||||
// download_type: DownloadType::Game,
|
|
||||||
//};
|
|
||||||
|
|
||||||
match process_manager_lock.launch_process(id, &state_lock) {
|
|
||||||
Ok(()) => {}
|
|
||||||
Err(e) => return Err(e),
|
|
||||||
}
|
|
||||||
|
|
||||||
drop(process_manager_lock);
|
|
||||||
drop(state_lock);
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn kill_game(
|
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState>>,
|
|
||||||
) -> Result<(), ProcessError> {
|
|
||||||
let state_lock = lock!(state);
|
|
||||||
let mut process_manager_lock = lock!(state_lock.process_manager);
|
|
||||||
process_manager_lock
|
|
||||||
.kill_game(game_id)
|
|
||||||
.map_err(ProcessError::IOError)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tauri::command]
|
|
||||||
pub fn open_process_logs(
|
|
||||||
game_id: String,
|
|
||||||
state: tauri::State<'_, Mutex<AppState>>,
|
|
||||||
) -> Result<(), ProcessError> {
|
|
||||||
let state_lock = lock!(state);
|
|
||||||
let mut process_manager_lock = lock!(state_lock.process_manager);
|
|
||||||
process_manager_lock.open_process_logs(game_id)
|
|
||||||
}
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
pub mod commands;
|
|
||||||
pub mod process_manager;
|
|
||||||
pub mod process_handlers;
|
|
||||||
pub mod format;
|
|
||||||
pub mod utils;
|
|
||||||
@ -1,37 +1,57 @@
|
|||||||
use std::sync::Mutex;
|
use std::{sync::nonpoison::Mutex, time::Duration};
|
||||||
|
|
||||||
|
use client::app_status::AppStatus;
|
||||||
|
use database::{borrow_db_checked, borrow_db_mut_checked};
|
||||||
use futures_lite::StreamExt;
|
use futures_lite::StreamExt;
|
||||||
use log::{debug, warn};
|
use log::{debug, warn};
|
||||||
|
use remote::{
|
||||||
|
auth::{auth_initiate_logic, generate_authorization_header},
|
||||||
|
cache::{cache_object, get_cached_object},
|
||||||
|
error::RemoteAccessError,
|
||||||
|
requests::generate_url,
|
||||||
|
setup,
|
||||||
|
utils::{DROP_CLIENT_ASYNC, DROP_CLIENT_WS_CLIENT, DropHealthcheck},
|
||||||
|
};
|
||||||
use reqwest_websocket::{Message, RequestBuilderExt};
|
use reqwest_websocket::{Message, RequestBuilderExt};
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use tauri::{AppHandle, Emitter, Manager};
|
use tauri::{AppHandle, Manager};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
use utils::{app_emit, webbrowser_open::webbrowser_open};
|
||||||
|
|
||||||
use crate::{
|
use crate::{AppState, recieve_handshake};
|
||||||
AppState, AppStatus, app_emit,
|
|
||||||
database::db::{borrow_db_checked, borrow_db_mut_checked},
|
|
||||||
error::remote_access_error::RemoteAccessError,
|
|
||||||
lock,
|
|
||||||
remote::{
|
|
||||||
auth::generate_authorization_header,
|
|
||||||
requests::generate_url,
|
|
||||||
utils::{DROP_CLIENT_SYNC, DROP_CLIENT_WS_CLIENT},
|
|
||||||
},
|
|
||||||
utils::webbrowser_open::webbrowser_open,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
|
||||||
auth::{auth_initiate_logic, recieve_handshake, setup},
|
|
||||||
cache::{cache_object, get_cached_object},
|
|
||||||
utils::use_remote_logic,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub async fn use_remote(
|
pub async fn use_remote(
|
||||||
url: String,
|
url: String,
|
||||||
state: tauri::State<'_, Mutex<AppState<'_>>>,
|
state: tauri::State<'_, Mutex<AppState>>,
|
||||||
) -> Result<(), RemoteAccessError> {
|
) -> Result<(), RemoteAccessError> {
|
||||||
use_remote_logic(url, state).await
|
debug!("connecting to url {url}");
|
||||||
|
let base_url = Url::parse(&url)?;
|
||||||
|
|
||||||
|
// Test Drop url
|
||||||
|
let test_endpoint = base_url.join("/api/v1")?;
|
||||||
|
let client = DROP_CLIENT_ASYNC.clone();
|
||||||
|
let response = client
|
||||||
|
.get(test_endpoint.to_string())
|
||||||
|
.timeout(Duration::from_secs(3))
|
||||||
|
.send()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let result: DropHealthcheck = response.json().await?;
|
||||||
|
|
||||||
|
if result.app_name() != "Drop" {
|
||||||
|
warn!("user entered drop endpoint that connected, but wasn't identified as Drop");
|
||||||
|
return Err(RemoteAccessError::InvalidEndpoint);
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut app_state = state.lock();
|
||||||
|
app_state.status = AppStatus::SignedOut;
|
||||||
|
drop(app_state);
|
||||||
|
|
||||||
|
let mut db_state = borrow_db_mut_checked();
|
||||||
|
db_state.base_url = base_url.to_string();
|
||||||
|
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
@ -51,7 +71,7 @@ pub fn gen_drop_url(path: String) -> Result<String, RemoteAccessError> {
|
|||||||
pub fn fetch_drop_object(path: String) -> Result<Vec<u8>, RemoteAccessError> {
|
pub fn fetch_drop_object(path: String) -> Result<Vec<u8>, RemoteAccessError> {
|
||||||
let _drop_url = gen_drop_url(path.clone())?;
|
let _drop_url = gen_drop_url(path.clone())?;
|
||||||
let req = generate_url(&[&path], &[])?;
|
let req = generate_url(&[&path], &[])?;
|
||||||
let req = DROP_CLIENT_SYNC
|
let req = remote::utils::DROP_CLIENT_SYNC
|
||||||
.get(req)
|
.get(req)
|
||||||
.header("Authorization", generate_authorization_header())
|
.header("Authorization", generate_authorization_header())
|
||||||
.send();
|
.send();
|
||||||
@ -78,21 +98,21 @@ pub fn sign_out(app: AppHandle) {
|
|||||||
|
|
||||||
// Update app state
|
// Update app state
|
||||||
{
|
{
|
||||||
let app_state = app.state::<Mutex<AppState>>();
|
let state = app.state::<Mutex<AppState>>();
|
||||||
let mut app_state_handle = lock!(app_state);
|
let mut app_state_handle = state.lock();
|
||||||
app_state_handle.status = AppStatus::SignedOut;
|
app_state_handle.status = AppStatus::SignedOut;
|
||||||
app_state_handle.user = None;
|
app_state_handle.user = None;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Emit event for frontend
|
// Emit event for frontend
|
||||||
app_emit!(app, "auth/signedout", ());
|
app_emit!(&app, "auth/signedout", ());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
pub async fn retry_connect(state: tauri::State<'_, Mutex<AppState<'_>>>) -> Result<(), ()> {
|
pub async fn retry_connect(state: tauri::State<'_, Mutex<AppState>>) -> Result<(), ()> {
|
||||||
let (app_status, user) = setup().await;
|
let (app_status, user) = setup().await;
|
||||||
|
|
||||||
let mut guard = lock!(state);
|
let mut guard = state.lock();
|
||||||
guard.status = app_status;
|
guard.status = app_status;
|
||||||
guard.user = user;
|
guard.user = user;
|
||||||
drop(guard);
|
drop(guard);
|
||||||
@ -168,7 +188,7 @@ pub fn auth_initiate_code(app: AppHandle) -> Result<String, RemoteAccessError> {
|
|||||||
let result = load().await;
|
let result = load().await;
|
||||||
if let Err(err) = result {
|
if let Err(err) = result {
|
||||||
warn!("{err}");
|
warn!("{err}");
|
||||||
app_emit!(app, "auth/failed", err.to_string());
|
app_emit!(&app, "auth/failed", err.to_string());
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
@ -4,20 +4,14 @@ use std::{
|
|||||||
path::{Path, PathBuf},
|
path::{Path, PathBuf},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use database::{
|
||||||
|
Settings, borrow_db_checked, borrow_db_mut_checked, db::DATA_ROOT_DIR, debug::SystemData,
|
||||||
|
};
|
||||||
|
use download_manager::error::DownloadManagerError;
|
||||||
|
use games::scan::scan_install_dirs;
|
||||||
use log::error;
|
use log::error;
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
|
|
||||||
use crate::{
|
|
||||||
database::{db::borrow_db_mut_checked, scan::scan_install_dirs},
|
|
||||||
error::download_manager_error::DownloadManagerError,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
|
||||||
db::{DATA_ROOT_DIR, borrow_db_checked},
|
|
||||||
debug::SystemData,
|
|
||||||
models::data::Settings,
|
|
||||||
};
|
|
||||||
|
|
||||||
// Will, in future, return disk/remaining size
|
// Will, in future, return disk/remaining size
|
||||||
// Just returns the directories that have been set up
|
// Just returns the directories that have been set up
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
@ -1,6 +0,0 @@
|
|||||||
#[macro_export]
|
|
||||||
macro_rules! app_emit {
|
|
||||||
($app:expr, $event:expr, $p:expr) => {
|
|
||||||
$app.emit($event, $p).expect(&format!("Failed to emit event {}", $event));
|
|
||||||
};
|
|
||||||
}
|
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user