grafbase-sdk 0.23.1

An SDK to implement extensions for the Grafbase Gateway
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

use super::{Component, state};
use crate::{
    types::{Headers, HttpRequestParts},
    wit,
};

impl wit::HooksGuest for Component {
    fn on_request(
        event_queue: wit::EventQueue,
        parts: wit::HttpRequestParts,
    ) -> Result<wit::OnRequestOutput, wit::ErrorResponse> {
        state::with_event_queue(event_queue, || {
            let mut parts: HttpRequestParts = parts.into();

            state::extension()?
                .on_request(&parts.url, parts.method, &mut parts.headers)
                .map(|output| wit::OnRequestOutput {
                    contract_key: output.contract_key,
                    context: output.context,
                    headers: parts.headers.into(),
                })
                .map_err(Into::into)
        })
    }

    fn on_response(
        event_queue: wit::EventQueue,
        ctx: wit::RequestContext,
        status: u16,
        headers: wit::Headers,
    ) -> Result<wit::OnResponseOutput, String> {
        let mut status = http::StatusCode::from_u16(status)
            .expect("we converted this from http::StatusCode in the host, this cannot be invalid");

        let mut headers: Headers = headers.into();

        state::extension()
            .map_err(|err| err.message)?
            .on_response(&(ctx.into()), &mut status, &mut headers, event_queue.into())
            .map(|_| wit::OnResponseOutput {
                headers: headers.into(),
                status: status.as_u16(),
            })
            .map_err(|err| err.0.message)
    }

    fn on_virtual_subgraph_request(
        event_queue: wit::EventQueue,
        ctx: wit::AuthorizedOperationContext,
        subgraph_name: String,
        headers: wit::Headers,
    ) -> Result<wit::Headers, wit::Error> {
        state::with_event_queue(event_queue, || {
            let mut headers: Headers = headers.into();

            state::extension()?
                .on_virtual_subgraph_request(&(ctx.into()), &subgraph_name, &mut headers)
                .map(|_| headers.into())
                .map_err(Into::into)
        })
    }

    fn on_graphql_subgraph_request(
        event_queue: wit::EventQueue,
        ctx: wit::AuthorizedOperationContext,
        subgraph_name: String,
        parts: wit::HttpRequestParts,
    ) -> Result<wit::HttpRequestParts, wit::Error> {
        state::with_event_queue(event_queue, || {
            let mut parts: HttpRequestParts = parts.into();

            state::extension()?
                .on_graphql_subgraph_request(&(ctx.into()), &subgraph_name, &mut parts)
                .map(|_| parts.into())
                .map_err(Into::into)
        })
    }
}