use js_sys::Object;
use wasm_bindgen::{JsCast, JsValue};
use web_sys::IdbCursor;
use crate::{CursorDirection, Error, StoreRequest};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct KeyCursor {
inner: IdbCursor,
}
impl KeyCursor {
pub fn source(&self) -> Object {
self.inner.source()
}
pub fn direction(&self) -> Result<CursorDirection, Error> {
self.inner.direction().try_into()
}
pub fn key(&self) -> Result<JsValue, Error> {
self.inner.key().map_err(Error::CursorKeyNotFound)
}
pub fn primary_key(&self) -> Result<JsValue, Error> {
self.inner
.primary_key()
.map_err(Error::CursorPrimaryKeyNotFound)
}
pub fn request(&self) -> StoreRequest {
self.inner.request().into()
}
pub fn advance(&self, count: u32) -> Result<(), Error> {
self.inner
.advance(count)
.map_err(Error::CursorAdvanceFailed)
}
pub fn next(&self, key: Option<&JsValue>) -> Result<(), Error> {
match key {
None => self.inner.continue_().map_err(Error::CursorContinueFailed),
Some(key) => self
.inner
.continue_with_key(key)
.map_err(Error::CursorContinueFailed),
}
}
pub fn next_primary_key(&self, key: &JsValue, primary_key: &JsValue) -> Result<(), Error> {
self.inner
.continue_primary_key(key, primary_key)
.map_err(Error::CursorContinueFailed)
}
pub fn update(&self, value: &JsValue) -> Result<StoreRequest, Error> {
self.inner
.update(value)
.map(Into::into)
.map_err(Error::UpdateFailed)
}
pub fn delete(&self) -> Result<StoreRequest, Error> {
self.inner
.delete()
.map(Into::into)
.map_err(Error::DeleteFailed)
}
}
impl From<IdbCursor> for KeyCursor {
fn from(inner: IdbCursor) -> Self {
Self { inner }
}
}
impl From<KeyCursor> for IdbCursor {
fn from(cursor: KeyCursor) -> Self {
cursor.inner
}
}
impl TryFrom<JsValue> for KeyCursor {
type Error = Error;
fn try_from(value: JsValue) -> Result<Self, Self::Error> {
value
.dyn_into::<IdbCursor>()
.map(Into::into)
.map_err(|value| Error::UnexpectedJsType("IdbCursor", value))
}
}
impl From<KeyCursor> for JsValue {
fn from(cursor: KeyCursor) -> Self {
cursor.inner.into()
}
}