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
use super::{Ctx, Handler};

use crate::capability::logging::logging;

use async_trait::async_trait;
use tracing::instrument;

pub mod unversioned_logging_bindings {
    wasmtime::component::bindgen!({
        world: "unversioned-logging",
        async: true,
        with: {
           "wasi:logging/logging": crate::capability::unversioned_logging,
        },
    });
}

pub mod logging_bindings {
    wasmtime::component::bindgen!({
        world: "logging",
        async: true,
        with: {
           "wasi:logging/logging": crate::capability::logging,
        },
    });
}

/// `wasi:logging/logging` implementation
#[async_trait]
pub trait Logging {
    /// Handle `wasi:logging/logging.log`
    async fn log(
        &self,
        level: logging::Level,
        context: String,
        message: String,
    ) -> anyhow::Result<()>;
}

#[async_trait]
impl<H: Handler> logging::Host for Ctx<H> {
    #[instrument(skip_all)]
    async fn log(
        &mut self,
        level: logging::Level,
        context: String,
        message: String,
    ) -> anyhow::Result<()> {
        self.handler.log(level, context, message).await
    }
}

#[async_trait]
impl<H: Handler> crate::capability::unversioned_logging::logging::Host for Ctx<H> {
    #[instrument(skip_all)]
    async fn log(
        &mut self,
        level: crate::capability::unversioned_logging::logging::Level,
        context: String,
        message: String,
    ) -> anyhow::Result<()> {
        // NOTE(thomastaylor312): I couldn't figure out the proper incantation for using `with` to
        // avoid this. If there is a better way, we can fix it
        use crate::capability::unversioned_logging::logging::Level;
        let level = match level {
            Level::Trace => logging::Level::Trace,
            Level::Debug => logging::Level::Debug,
            Level::Info => logging::Level::Info,
            Level::Warn => logging::Level::Warn,
            Level::Error => logging::Level::Error,
            Level::Critical => logging::Level::Critical,
        };
        self.handler.log(level, context, message).await
    }
}