You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
3.4 KiB
Rust
123 lines
3.4 KiB
Rust
use tauri::plugin::Plugin;
|
|
use tauri::{AppHandle, Builder, Invoke, Manager, Runtime};
|
|
|
|
use state::ApiState;
|
|
|
|
use crate::tauri_plugin::state::{AppState, BufferState};
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
mod background_tasks;
|
|
pub(crate) mod commands;
|
|
pub mod custom_schemes;
|
|
pub mod error;
|
|
mod settings;
|
|
mod state;
|
|
mod utils;
|
|
|
|
use crate::tauri_plugin::background_tasks::{start_background_task_runtime, TaskContext};
|
|
use commands::*;
|
|
|
|
const MAX_BUFFER_SIZE: usize = 2 * 1024 * 1024 * 1024; // 2GiB
|
|
|
|
pub fn register_plugin<R: Runtime>(builder: Builder<R>) -> Builder<R> {
|
|
let repo_plugin = MediarepoPlugin::new();
|
|
|
|
custom_schemes::register_custom_uri_schemes(builder.plugin(repo_plugin))
|
|
}
|
|
|
|
pub struct MediarepoPlugin<R: Runtime> {
|
|
invoke_handler: Box<dyn Fn(Invoke<R>) + Send + Sync>,
|
|
}
|
|
|
|
impl<R: Runtime> MediarepoPlugin<R> {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
invoke_handler: Box::new(tauri::generate_handler![
|
|
get_all_files,
|
|
find_files,
|
|
get_file_thumbnails,
|
|
get_repositories,
|
|
get_all_tags,
|
|
get_tags_for_file,
|
|
get_tags_for_files,
|
|
get_active_repository,
|
|
add_repository,
|
|
select_repository,
|
|
init_repository,
|
|
start_daemon,
|
|
check_daemon_running,
|
|
stop_daemon,
|
|
disconnect_repository,
|
|
close_local_repository,
|
|
check_local_repository_exists,
|
|
remove_repository,
|
|
change_file_tags,
|
|
create_tags,
|
|
update_file_name,
|
|
resolve_paths_to_files,
|
|
add_local_file,
|
|
save_file_locally,
|
|
delete_thumbnails,
|
|
read_file,
|
|
delete_repository,
|
|
has_executable,
|
|
get_frontend_state,
|
|
set_frontend_state,
|
|
get_all_namespaces,
|
|
get_files,
|
|
get_repo_metadata,
|
|
get_size,
|
|
get_file_metadata,
|
|
run_job,
|
|
update_file_status,
|
|
delete_file,
|
|
get_file_tag_map,
|
|
all_sorting_presets,
|
|
add_sorting_preset,
|
|
delete_sorting_preset,
|
|
is_job_running
|
|
]),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<R: Runtime> Plugin<R> for MediarepoPlugin<R> {
|
|
fn name(&self) -> &'static str {
|
|
"mediarepo"
|
|
}
|
|
|
|
#[tracing::instrument(skip(self, app, _config))]
|
|
fn initialize(
|
|
&mut self,
|
|
app: &AppHandle<R>,
|
|
_config: serde_json::value::Value,
|
|
) -> tauri::plugin::Result<()> {
|
|
let api_state = ApiState::new();
|
|
app.manage(api_state);
|
|
|
|
let buffer_state = BufferState::default();
|
|
app.manage(buffer_state.clone());
|
|
|
|
let repo_state = AppState::load()?;
|
|
app.manage(repo_state);
|
|
|
|
let task_context = TaskContext::new();
|
|
start_background_task_runtime(task_context.clone());
|
|
app.manage(task_context);
|
|
|
|
thread::spawn(move || loop {
|
|
thread::sleep(Duration::from_secs(10));
|
|
buffer_state.clear_expired();
|
|
buffer_state.trim_to_size(MAX_BUFFER_SIZE);
|
|
});
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip_all)]
|
|
fn extend_api(&mut self, message: Invoke<R>) {
|
|
(self.invoke_handler)(message)
|
|
}
|
|
}
|