kanamaru 0.1.0

Build typed Tauri plugins with the power of Protobuf Buffers
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

use std::{fmt::Debug, sync::Arc};

use prost::Message;
use serde::Deserialize;
use tauri::{
    http::HeaderMap,
    ipc::{Channel, InvokeBody, InvokeMessage},
    Runtime, Webview,
};

use crate::{utils::CancellationTokenListener, Status};

use super::{InvokeMessageToRequestError, RawRequest, RawRequestToRequestError, RequestBase};

#[derive(Clone)]
pub struct UnaryRequest<R, M>
where
    M: Message + Clone + Default,
    R: Runtime,
{
    token: Arc<CancellationTokenListener<Webview<R>, R>>,
    message: M,
    channel_status: Channel<Status>,
    headers: HeaderMap,
}

impl<R: Runtime, M: Message + Clone + Default> Debug for UnaryRequest<R, M> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("UnaryRequest")
            .field("token", &self.token)
            .field("message", &self.message)
            .field("channel_status", &())
            .field("headers", &self.headers)
            .finish()
    }
}

impl<R, M> UnaryRequest<R, M>
where
    M: Message + Clone + Default,
    R: Runtime,
{
    pub fn new(
        token: Arc<CancellationTokenListener<Webview<R>, R>>,
        raw_reqwest: RawRequest,
    ) -> Result<Self, RawRequestToRequestError> {
        if let Some(payload) = raw_reqwest.payload {
            Ok(Self {
                channel_status: raw_reqwest
                    .status_channel
                    .channel_on(token.listener().clone()),
                token,
                message: payload.extract_message()?,
                headers: payload.metadata,
            })
        } else {
            Err(RawRequestToRequestError::MissingPayload)
        }
    }

    pub fn into_inner(self) -> M {
        self.message
    }
    pub fn message(&self) -> &M {
        self.as_ref()
    }
    pub fn headers_ref(&self) -> &HeaderMap {
        &self.headers
    }
    pub fn headers(&self) -> HeaderMap {
        self.headers_ref().clone()
    }
    pub fn map_message<F, M1>(self, map_fn: F) -> UnaryRequest<R, M1>
    where
        F: FnOnce(M) -> M1,
        M1: Message + Clone + Default,
    {
        UnaryRequest {
            token: self.token,
            message: map_fn(self.message),
            channel_status: self.channel_status,
            headers: self.headers,
        }
    }
    pub fn change_message<F, M1>(self, message: M1) -> UnaryRequest<R, M1>
    where
        F: FnOnce(M) -> M1,
        M1: Message + Clone + Default,
    {
        UnaryRequest {
            token: self.token,
            message,
            channel_status: self.channel_status,
            headers: self.headers,
        }
    }
}

impl<R, M> AsRef<M> for UnaryRequest<R, M>
where
    M: Message + Clone + Default,
    R: Runtime,
{
    fn as_ref(&self) -> &M {
        &self.message
    }
}

impl<M, R> TryFrom<&InvokeMessage<R>> for UnaryRequest<R, M>
where
    M: Message + Clone + Default,
    R: Runtime,
{
    type Error = InvokeMessageToRequestError;
    fn try_from(value: &InvokeMessage<R>) -> Result<Self, Self::Error> {
        if let InvokeBody::Json(payload) = value.payload() {
            let raw_req: RawRequest = Deserialize::deserialize(payload)?;
            let token = raw_req.cancel_token(value.webview());
            Ok(UnaryRequest::new(Arc::new(token), raw_req)?)
        } else {
            Err(InvokeMessageToRequestError::RawRequestToRequest(
                RawRequestToRequestError::InvalidPayloadFormat,
            ))
        }
    }
}

impl<M, R> RequestBase<R> for UnaryRequest<R, M>
where
    M: Message + Clone + Default,
    R: Runtime,
{
    fn token(&self) -> Arc<CancellationTokenListener<Webview<R>, R>> {
        self.token.clone()
    }
    fn status_channel(&self) -> tauri::ipc::Channel<crate::Status> {
        self.channel_status.clone()
    }
}