|
| 1 | +use rexie::{Direction, Error as RexieError, ObjectStore, Rexie, TransactionMode}; |
| 2 | +use std::rc::Rc; |
| 3 | +use thiserror::Error; |
| 4 | +use wasm_bindgen::JsValue; |
| 5 | +use wasm_bindgen_futures::JsFuture; |
| 6 | + |
| 7 | +#[allow(missing_docs)] |
| 8 | +#[derive(Debug, Error)] |
| 9 | +pub enum IdbError { |
| 10 | + #[error("couldn't build database")] |
| 11 | + BuildError { |
| 12 | + #[source] |
| 13 | + source: RexieError, |
| 14 | + }, |
| 15 | + // The source of this would be a `JsValue`, which we drop for performance |
| 16 | + #[error("persistence check failed")] |
| 17 | + PersistenceCheckFailed { |
| 18 | + /// Whether or not this persistence check could be retried and might result in a success (this is just an estimation). |
| 19 | + retry: bool, |
| 20 | + }, |
| 21 | + #[error("an error occurred while constructing an IndexedDB transaction")] |
| 22 | + TransactionError { |
| 23 | + #[source] |
| 24 | + source: RexieError, |
| 25 | + }, |
| 26 | + #[error("an error occured while trying to set a new value")] |
| 27 | + SetError { |
| 28 | + #[source] |
| 29 | + source: RexieError, |
| 30 | + }, |
| 31 | + #[error("an error occurred while clearing the store of previous values")] |
| 32 | + ClearError { |
| 33 | + #[source] |
| 34 | + source: RexieError, |
| 35 | + }, |
| 36 | + #[error("an error occurred while trying to get the latest value")] |
| 37 | + GetError { |
| 38 | + #[source] |
| 39 | + source: RexieError, |
| 40 | + }, |
| 41 | +} |
| 42 | + |
| 43 | +/// A frozen state store that uses IndexedDB as a backend. This will only store a single frozen state at a time, removing all previously stored states every time a new one is set. |
| 44 | +/// |
| 45 | +/// TODO Browser compatibility information. |
| 46 | +#[derive(Clone)] |
| 47 | +pub struct IdbFrozenStateStore { |
| 48 | + /// A handle to the database. |
| 49 | + db: Rc<Rexie>, |
| 50 | +} |
| 51 | +impl std::fmt::Debug for IdbFrozenStateStore { |
| 52 | + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
| 53 | + f.debug_struct("IdbFrozenStateStore").finish() |
| 54 | + } |
| 55 | +} |
| 56 | +impl IdbFrozenStateStore { |
| 57 | + /// Creates a new store for this origin. If it already exists from a previous visit, the existing one will be interfaced with. |
| 58 | + pub async fn new() -> Result<Self, IdbError> { |
| 59 | + // Build the database |
| 60 | + let rexie = Rexie::builder("perseus") |
| 61 | + // IndexedDB uses versions to track database schema changes |
| 62 | + // If the structure of this DB ever changes, this MUST be changed, and this should be considered a non-API-breaking, but app-breaking change! |
| 63 | + .version(1) |
| 64 | + .add_object_store( |
| 65 | + // We'll store many versions of frozen state so that the user can revert to previous states |
| 66 | + ObjectStore::new("frozen_state") |
| 67 | + .key_path("id") |
| 68 | + .auto_increment(true), // IndexedDB doesn't need us to register value types, only things that should be indexed (gotta love JS type safety haven't you!) |
| 69 | + ) |
| 70 | + .build() |
| 71 | + .await |
| 72 | + .map_err(|err| IdbError::BuildError { source: err })?; |
| 73 | + |
| 74 | + Ok(Self { db: Rc::new(rexie) }) |
| 75 | + } |
| 76 | + /// Gets the stored frozen state. Be warned that the result of this could be arbitrarily old, or it may have been tampered with by the user (in which case Perseus will either return an |
| 77 | + /// error that you can handle or it'll fall back to the active state). If no state has been stored yet, this will return `Ok(None)`. |
| 78 | + pub async fn get(&self) -> Result<Option<String>, IdbError> { |
| 79 | + let transaction = self |
| 80 | + .db |
| 81 | + .transaction(&["frozen_state"], TransactionMode::ReadOnly) |
| 82 | + .map_err(|err| IdbError::TransactionError { source: err })?; |
| 83 | + let store = transaction |
| 84 | + .store("frozen_state") |
| 85 | + .map_err(|err| IdbError::TransactionError { source: err })?; |
| 86 | + |
| 87 | + // Get the last element from the store by working backwards and getting everything, with a limit of 1 |
| 88 | + let frozen_states = store |
| 89 | + .get_all(None, Some(1), None, Some(Direction::Prev)) |
| 90 | + .await |
| 91 | + .map_err(|err| IdbError::GetError { source: err })?; |
| 92 | + let frozen_state = match frozen_states.get(0) { |
| 93 | + Some((_key, value)) => value, |
| 94 | + None => return Ok(None), |
| 95 | + }; |
| 96 | + // TODO Do this without cloning the whole thing into the Wasm table and then moving it into Rust |
| 97 | + let frozen_state = frozen_state.as_string().unwrap(); |
| 98 | + transaction |
| 99 | + .commit() |
| 100 | + .await |
| 101 | + .map_err(|err| IdbError::TransactionError { source: err })?; |
| 102 | + |
| 103 | + Ok(Some(frozen_state)) |
| 104 | + } |
| 105 | + /// Sets the content to a new frozen state. |
| 106 | + pub async fn set(&self, frozen_state: &str) -> Result<(), IdbError> { |
| 107 | + let transaction = self |
| 108 | + .db |
| 109 | + .transaction(&["frozen_state"], TransactionMode::ReadWrite) |
| 110 | + .map_err(|err| IdbError::TransactionError { source: err })?; |
| 111 | + let store = transaction |
| 112 | + .store("frozen_state") |
| 113 | + .map_err(|err| IdbError::TransactionError { source: err })?; |
| 114 | + |
| 115 | + // We only store a single frozen state, and they can be quite large, so we'll remove any that are already in here |
| 116 | + store |
| 117 | + .clear() |
| 118 | + .await |
| 119 | + .map_err(|err| IdbError::ClearError { source: err })?; |
| 120 | + // We can add the frozen state directly because it's already a serialized string |
| 121 | + // This returns the ID, but we don't need to care about that |
| 122 | + store |
| 123 | + .add(&JsValue::from(frozen_state), None) |
| 124 | + .await |
| 125 | + .map_err(|err| IdbError::SetError { source: err })?; |
| 126 | + transaction |
| 127 | + .commit() |
| 128 | + .await |
| 129 | + .map_err(|err| IdbError::SetError { source: err })?; |
| 130 | + |
| 131 | + Ok(()) |
| 132 | + } |
| 133 | + /// Checks if the storage is persistently stored. If it is, the browser isn't allowed to clear it, the user would have to manually. This doesn't provide a guarantee that all users who've |
| 134 | + /// been to your site before will have previous state stored, you should assume that they could well have cleared it manually (or with very stringent privacy settings). |
| 135 | + /// |
| 136 | + /// If this returns an error, a recommendation about whether or not to retry will be attached. You generally shouldn't retry this more than once if there was an error. |
| 137 | + /// |
| 138 | + /// For more information about persistent storage on the web, see [here](https://web.dev/persistent-storage). |
| 139 | + pub async fn is_persistent() -> Result<bool, IdbError> { |
| 140 | + let storage_manager = web_sys::window().unwrap().navigator().storage(); |
| 141 | + // If we can't access this, we're probably in a very old browser, so retrying isn't worth it in all likelihood |
| 142 | + let persisted = storage_manager |
| 143 | + .persisted() |
| 144 | + .map_err(|_| IdbError::PersistenceCheckFailed { retry: false })?; |
| 145 | + let persisted = JsFuture::from(persisted) |
| 146 | + .await |
| 147 | + .map_err(|_| IdbError::PersistenceCheckFailed { retry: true })?; |
| 148 | + let persisted_bool = persisted |
| 149 | + .as_bool() |
| 150 | + .ok_or(IdbError::PersistenceCheckFailed { retry: true })?; |
| 151 | + |
| 152 | + Ok(persisted_bool) |
| 153 | + } |
| 154 | + /// Requests persistent storage from the browser. In Firefox, the user will be prompted, though in Chrome the browser will automatically accept or deny based on your site's level of |
| 155 | + /// engagement, whether ot not it's been installed or bookmarked, and whether or not it's been granted the permission to show notifications. In other words, do NOT assume that this will |
| 156 | + /// be accepted, even if you ask the user very nicely. That said, especially in Firefox, you should display a custom notification before this with `alert()` or similar that explains why |
| 157 | + /// your site needs persistent storage for frozen state. |
| 158 | + /// |
| 159 | + /// If this returns `false`, the request was rejected, but you can retry in future (for user experience though, it's recommended to only do so very sparingly). If this returns an error, |
| 160 | + /// a recommendation about whether or not to retry will be attached. You generally shouldn't retry this more than once if there was an error. |
| 161 | + /// |
| 162 | + /// For more information about persistent storage on the web, see [here](https://web.dev/persistent-storage). |
| 163 | + pub async fn request_persistence() -> Result<bool, IdbError> { |
| 164 | + let storage_manager = web_sys::window().unwrap().navigator().storage(); |
| 165 | + // If we can't access this, we're probably in a very old browser, so retrying isn't worth it in all likelihood |
| 166 | + let res = storage_manager |
| 167 | + .persist() |
| 168 | + .map_err(|_| IdbError::PersistenceCheckFailed { retry: false })?; |
| 169 | + let res = JsFuture::from(res) |
| 170 | + .await |
| 171 | + .map_err(|_| IdbError::PersistenceCheckFailed { retry: true })?; |
| 172 | + let res_bool = res.as_bool().unwrap(); |
| 173 | + |
| 174 | + Ok(res_bool) |
| 175 | + } |
| 176 | +} |
0 commit comments