use std::fmt;
use std::ops::Deref;
use std::rc::Rc;
use super::{use_reducer, use_reducer_eq, Reducible, UseReducerDispatcher, UseReducerHandle};
struct UseStateReducer<T> {
value: Rc<T>,
}
impl<T> Reducible for UseStateReducer<T> {
type Action = T;
fn reduce(self: Rc<Self>, action: Self::Action) -> Rc<Self> {
Rc::new(Self {
value: action.into(),
})
}
}
impl<T> PartialEq for UseStateReducer<T>
where
T: PartialEq,
{
fn eq(&self, rhs: &Self) -> bool {
self.value == rhs.value
}
}
pub fn use_state<T, F>(init_fn: F) -> UseStateHandle<T>
where
T: 'static,
F: FnOnce() -> T,
{
let handle = use_reducer(move || UseStateReducer {
value: Rc::new(init_fn()),
});
UseStateHandle { inner: handle }
}
pub fn use_state_eq<T, F>(init_fn: F) -> UseStateHandle<T>
where
T: PartialEq + 'static,
F: FnOnce() -> T,
{
let handle = use_reducer_eq(move || UseStateReducer {
value: Rc::new(init_fn()),
});
UseStateHandle { inner: handle }
}
pub struct UseStateHandle<T> {
inner: UseReducerHandle<UseStateReducer<T>>,
}
impl<T: fmt::Debug> fmt::Debug for UseStateHandle<T> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("UseStateHandle")
.field("value", &format!("{:?}", self.inner.value))
.finish()
}
}
impl<T> UseStateHandle<T> {
pub fn set(&self, value: T) {
self.inner.dispatch(value)
}
pub fn setter(&self) -> UseStateSetter<T> {
UseStateSetter {
inner: self.inner.dispatcher(),
}
}
}
impl<T> Deref for UseStateHandle<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&(*self.inner).value
}
}
impl<T> Clone for UseStateHandle<T> {
fn clone(&self) -> Self {
Self {
inner: self.inner.clone(),
}
}
}
impl<T> PartialEq for UseStateHandle<T>
where
T: PartialEq,
{
fn eq(&self, rhs: &Self) -> bool {
*self.inner == *rhs.inner
}
}
pub struct UseStateSetter<T> {
inner: UseReducerDispatcher<UseStateReducer<T>>,
}
impl<T> Clone for UseStateSetter<T> {
fn clone(&self) -> Self {
Self {
inner: self.inner.clone(),
}
}
}
impl<T> fmt::Debug for UseStateSetter<T>
where
T: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("UseStateSetter").finish()
}
}
impl<T> PartialEq for UseStateSetter<T> {
fn eq(&self, rhs: &Self) -> bool {
self.inner == rhs.inner
}
}
impl<T> UseStateSetter<T> {
pub fn set(&self, value: T) {
self.inner.dispatch(value)
}
}