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

// Copyright (c) Zefchain Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use std::{
    panic::{self, PanicHookInfo},
    sync::Once,
};

use log::{LevelFilter, Log, Metadata, Record};

use crate::{
    contract::wit::base_runtime_api as contract_wit, service::wit::base_runtime_api as service_wit,
};

static CONTRACT_LOGGER: ContractLogger = ContractLogger;
static SERVICE_LOGGER: ServiceLogger = ServiceLogger;

static INSTALL_LOGGER: Once = Once::new();

/// A logger that uses the system API for contracts.
#[derive(Clone, Copy, Debug)]
pub struct ContractLogger;

impl ContractLogger {
    /// Configures [`log`] to use the log system API for contracts.
    pub fn install() {
        INSTALL_LOGGER.call_once(|| {
            log::set_logger(&CONTRACT_LOGGER).expect("Failed to initialize contract logger");
            log::set_max_level(LevelFilter::Trace);
            panic::set_hook(Box::new(log_panic));
        });
    }
}

impl Log for ContractLogger {
    fn enabled(&self, _: &Metadata) -> bool {
        true
    }

    fn log(&self, record: &Record) {
        contract_wit::log(&record.args().to_string(), record.level().into());
    }

    fn flush(&self) {}
}

/// A logger that uses the system API for services.
#[derive(Clone, Copy, Debug)]
pub struct ServiceLogger;

impl ServiceLogger {
    /// Configures [`log`] to use the log system API for services.
    pub fn install() {
        INSTALL_LOGGER.call_once(|| {
            log::set_logger(&SERVICE_LOGGER).expect("Failed to initialize service logger");
            log::set_max_level(LevelFilter::Trace);
            panic::set_hook(Box::new(log_panic));
        });
    }
}

impl Log for ServiceLogger {
    fn enabled(&self, _: &Metadata) -> bool {
        true
    }

    fn log(&self, record: &Record) {
        service_wit::log(&record.args().to_string(), record.level().into());
    }

    fn flush(&self) {}
}

/// Logs a panic using the [`log`] API.
fn log_panic(info: &PanicHookInfo<'_>) {
    log::error!("{info}");
}