120 lines
3.4 KiB
Rust
120 lines
3.4 KiB
Rust
// OmegaUpload Web Frontend
|
|
// Copyright (C) 2021 Edward Shen
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
use std::{hint::unreachable_unchecked, marker::PhantomData};
|
|
|
|
use gloo_console::log;
|
|
use js_sys::{Array, JsString, Object};
|
|
use wasm_bindgen::JsValue;
|
|
|
|
pub struct IdbObject<State>(Array, PhantomData<State>);
|
|
|
|
impl<State: IdbObjectState> IdbObject<State> {
|
|
fn add_tuple<NextState>(self, key: &str, value: &JsValue) -> IdbObject<NextState> {
|
|
let array = Array::new();
|
|
array.push(&JsString::from(key));
|
|
array.push(value);
|
|
self.0.push(&array);
|
|
IdbObject(self.0, PhantomData)
|
|
}
|
|
}
|
|
|
|
impl From<IdbObject<Ready>> for Object {
|
|
fn from(db_object: IdbObject<Ready>) -> Self {
|
|
match Self::from_entries(db_object.as_ref()) {
|
|
Ok(o) => o,
|
|
// SAFETY: IdbObject maintains the invariant that it can eventually
|
|
// be constructed into a JS object.
|
|
_ => {
|
|
log!("IdbObject invariant violated?!");
|
|
unsafe { unreachable_unchecked() }
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl IdbObject<NeedsType> {
|
|
pub fn new() -> Self {
|
|
Self(Array::new(), PhantomData)
|
|
}
|
|
|
|
pub fn archive(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("archive"))
|
|
}
|
|
|
|
pub fn video(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("video"))
|
|
}
|
|
|
|
pub fn audio(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("audio"))
|
|
}
|
|
|
|
pub fn image(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("image"))
|
|
}
|
|
|
|
pub fn blob(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("blob"))
|
|
}
|
|
|
|
pub fn string(self) -> IdbObject<NeedsExpiration> {
|
|
self.add_tuple("type", &JsString::from("string"))
|
|
}
|
|
}
|
|
|
|
impl Default for IdbObject<NeedsType> {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl IdbObject<NeedsExpiration> {
|
|
pub fn expiration_text(self, expires: &str) -> IdbObject<NeedsData> {
|
|
self.add_tuple("expiration", &JsString::from(expires))
|
|
}
|
|
}
|
|
|
|
impl IdbObject<NeedsData> {
|
|
pub fn data(self, value: &JsValue) -> IdbObject<Ready> {
|
|
self.add_tuple("data", value)
|
|
}
|
|
}
|
|
|
|
impl IdbObject<Ready> {
|
|
pub fn extra(self, key: &str, value: impl Into<JsValue>) -> Self {
|
|
self.add_tuple(key, &value.into())
|
|
}
|
|
}
|
|
|
|
impl AsRef<JsValue> for IdbObject<Ready> {
|
|
fn as_ref(&self) -> &JsValue {
|
|
self.0.as_ref()
|
|
}
|
|
}
|
|
|
|
macro_rules! impl_idb_object_state {
|
|
($($ident:ident),*) => {
|
|
pub trait IdbObjectState {}
|
|
$(
|
|
pub enum $ident {}
|
|
impl IdbObjectState for $ident {}
|
|
)*
|
|
};
|
|
}
|
|
|
|
impl_idb_object_state!(NeedsType, NeedsExpiration, NeedsData, Ready);
|