mirror of
https://github.com/tauri-apps/plugins-workspace.git
synced 2026-06-06 13:53:54 +02:00
feat(http): persist cookies on disk (#1978)
* enhance(http): persist cookies on disk closes tauri-apps/tauri#11518 * clippy * inline reqwest_cookie_store to fix clippy * clippy * Update .changes/persist-cookies.md * Update plugins/http/src/reqwest_cookie_store.rs * update example * fallback to empty store if failed to load * fix example * persist cookies immediately * clone * lint * .cookies filename * prevent race condition --------- Co-authored-by: Lucas Nogueira <lucas@tauri.app>
This commit is contained in:
+49
-2
@@ -14,25 +14,72 @@ pub use error::{Error, Result};
|
||||
|
||||
mod commands;
|
||||
mod error;
|
||||
#[cfg(feature = "cookies")]
|
||||
mod reqwest_cookie_store;
|
||||
mod scope;
|
||||
|
||||
#[cfg(feature = "cookies")]
|
||||
const COOKIES_FILENAME: &str = ".cookies";
|
||||
|
||||
pub(crate) struct Http {
|
||||
#[cfg(feature = "cookies")]
|
||||
cookies_jar: std::sync::Arc<reqwest::cookie::Jar>,
|
||||
cookies_jar: std::sync::Arc<crate::reqwest_cookie_store::CookieStoreMutex>,
|
||||
}
|
||||
|
||||
pub fn init<R: Runtime>() -> TauriPlugin<R> {
|
||||
Builder::<R>::new("http")
|
||||
.setup(|app, _| {
|
||||
#[cfg(feature = "cookies")]
|
||||
let cookies_jar = {
|
||||
use crate::reqwest_cookie_store::*;
|
||||
use std::fs::File;
|
||||
use std::io::BufReader;
|
||||
|
||||
let cache_dir = app.path().app_cache_dir()?;
|
||||
std::fs::create_dir_all(&cache_dir)?;
|
||||
|
||||
let path = cache_dir.join(COOKIES_FILENAME);
|
||||
let file = File::options()
|
||||
.create(true)
|
||||
.append(true)
|
||||
.read(true)
|
||||
.open(&path)?;
|
||||
|
||||
let reader = BufReader::new(file);
|
||||
CookieStoreMutex::load(path.clone(), reader).unwrap_or_else(|_e| {
|
||||
#[cfg(feature = "tracing")]
|
||||
tracing::warn!(
|
||||
"failed to load cookie store: {_e}, falling back to empty store"
|
||||
);
|
||||
CookieStoreMutex::new(path, Default::default())
|
||||
})
|
||||
};
|
||||
|
||||
let state = Http {
|
||||
#[cfg(feature = "cookies")]
|
||||
cookies_jar: std::sync::Arc::new(reqwest::cookie::Jar::default()),
|
||||
cookies_jar: std::sync::Arc::new(cookies_jar),
|
||||
};
|
||||
|
||||
app.manage(state);
|
||||
|
||||
Ok(())
|
||||
})
|
||||
.on_event(|app, event| {
|
||||
#[cfg(feature = "cookies")]
|
||||
if let tauri::RunEvent::Exit = event {
|
||||
let state = app.state::<Http>();
|
||||
|
||||
match state.cookies_jar.request_save() {
|
||||
Ok(rx) => {
|
||||
let _ = rx.recv();
|
||||
}
|
||||
Err(_e) => {
|
||||
#[cfg(feature = "tracing")]
|
||||
tracing::error!("failed to save cookie jar: {_e}");
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.invoke_handler(tauri::generate_handler![
|
||||
commands::fetch,
|
||||
commands::fetch_cancel,
|
||||
|
||||
Reference in New Issue
Block a user