umari 0.0.1

SDK for building event-sourced WASM components for the Umari runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use std::{cell::RefCell, fmt, marker::PhantomData};

pub use self::exports::umari::effect::effect::{Guest, GuestEffect};
use crate::{
    effect::Effect,
    runtime::common::{self, EventQuery, StoredEvent},
};

wit_bindgen::generate!({
    path: "wit/effect",
    pub_export_macro: true,
    with: {
        "umari:command/executor@0.1.0": crate::runtime::command,
        "umari:command/types@0.1.0": crate::runtime::command,
        "umari:common/types@0.1.0": crate::runtime::common,
        "umari:sqlite/types@0.1.0": crate::runtime::sqlite,
        "umari:sqlite/connection@0.1.0": crate::runtime::sqlite,
        "umari:sqlite/statement@0.1.0": crate::runtime::sqlite,
        "wasi:clocks/monotonic-clock@0.2.8": wasip2::clocks::monotonic_clock,
        "wasi:io/error@0.2.8": wasip2::io,
        "wasi:io/poll@0.2.8": wasip2::io::poll,
        "wasi:io/streams@0.2.8": wasip2::io::streams,
        "wasi:http/types@0.2.8": wasip2::http::types,
        "wasi:http/outgoing-handler@0.2.8": wasip2::http::outgoing_handler,
    },
});

#[macro_export]
macro_rules! export_effect {
    ($ty:path) => {
        type ExportedEffect = $crate::runtime::effect::EffectExport<$ty>;
        $crate::runtime::effect::export!(ExportedEffect with_types_in $crate::runtime::effect);

        // $crate::runtime::effect::export!({
        //     ty: $crate::runtime::effect::EffectExport<$ty>,
        //     with_types_in: $crate::runtime::effect,
        // });
    };
}

pub struct EffectExport<T>(PhantomData<T>);

pub struct EffectState<T> {
    inner: RefCell<T>,
}

impl<T> Guest for EffectExport<T>
where
    T: Effect + 'static,
    T::Error: fmt::Display,
{
    type Effect = EffectState<T>;
}

impl<T> GuestEffect for EffectState<T>
where
    T: Effect + 'static,
    T::Error: fmt::Display,
{
    fn new() -> Self
    where
        Self: Sized,
    {
        let state = T::init().expect("effect init failed");
        EffectState {
            inner: RefCell::new(state),
        }
    }

    fn query(&self) -> EventQuery {
        self.inner.borrow().query().into()
    }

    fn partition_key(&self, stored_event: StoredEvent) -> Option<String> {
        let event = common::transform_stored_event::<T::Query>(stored_event)?;
        self.inner.borrow().partition_key(event)
    }

    fn handle(&self, stored_event: StoredEvent) {
        let Some(event) = common::transform_stored_event::<T::Query>(stored_event) else {
            return;
        };

        self.inner
            .borrow_mut()
            .handle(event)
            .unwrap_or_else(|err| panic!("effect handle error: {err}"))
    }
}