use js_sys::Object;
use wasm_bindgen::{JsCast, JsValue};
use web_sys::IdbCursorWithValue;
#[cfg(feature = "futures")]
use crate::ManagedCursor;
use crate::{
request::{DeleteStoreRequest, OpenCursorStoreRequest, UpdateStoreRequest},
CursorDirection, Error,
};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Cursor {
inner: IdbCursorWithValue,
}
impl Cursor {
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 value(&self) -> Result<JsValue, Error> {
self.inner.value().map_err(Error::CursorValueNotFound)
}
pub fn request(&self) -> OpenCursorStoreRequest {
#[allow(deprecated)]
self.inner.request().into()
}
pub fn advance(&self, count: u32) -> Result<OpenCursorStoreRequest, Error> {
self.inner
.advance(count)
.map_err(Error::CursorAdvanceFailed)?;
Ok(self.request())
}
pub fn next(&self, key: Option<&JsValue>) -> Result<OpenCursorStoreRequest, Error> {
match key {
None => self.inner.continue_().map_err(Error::CursorContinueFailed),
Some(key) => self
.inner
.continue_with_key(key)
.map_err(Error::CursorContinueFailed),
}?;
Ok(self.request())
}
pub fn next_primary_key(
&self,
key: &JsValue,
primary_key: &JsValue,
) -> Result<OpenCursorStoreRequest, Error> {
self.inner
.continue_primary_key(key, primary_key)
.map_err(Error::CursorContinueFailed)?;
Ok(self.request())
}
pub fn update(&self, value: &JsValue) -> Result<UpdateStoreRequest, Error> {
self.inner
.update(value)
.map(Into::into)
.map_err(Error::UpdateFailed)
}
pub fn delete(&self) -> Result<DeleteStoreRequest, Error> {
self.inner
.delete()
.map(Into::into)
.map_err(Error::DeleteFailed)
}
#[cfg(feature = "futures")]
#[cfg_attr(any(docsrs, feature = "doc"), doc(cfg(feature = "futures")))]
pub fn into_managed(self) -> ManagedCursor {
self.into()
}
}
impl From<IdbCursorWithValue> for Cursor {
fn from(inner: IdbCursorWithValue) -> Self {
Self { inner }
}
}
impl From<Cursor> for IdbCursorWithValue {
fn from(cursor: Cursor) -> Self {
cursor.inner
}
}
impl TryFrom<JsValue> for Cursor {
type Error = Error;
fn try_from(value: JsValue) -> Result<Self, Self::Error> {
value
.dyn_into::<IdbCursorWithValue>()
.map(Into::into)
.map_err(|value| Error::UnexpectedJsType("IdbCursorWithValue", value))
}
}
impl From<Cursor> for JsValue {
fn from(cursor: Cursor) -> Self {
cursor.inner.into()
}
}