codemp 0.8.5

codemp -- code multiplexer
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

use std::{io::Write, sync::Mutex};

use mlua::prelude::*;
use tokio::sync::mpsc;

#[derive(Debug, Clone)]
struct LuaLoggerProducer(mpsc::UnboundedSender<String>);
impl Write for LuaLoggerProducer {
	fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
		let _ = self.0.send(String::from_utf8_lossy(buf).to_string());
		Ok(buf.len())
	}

	fn flush(&mut self) -> std::io::Result<()> {
		Ok(())
	}
}

// TODO can we make this less verbose?
pub(crate) fn setup_tracing(
	lua: &Lua,
	(printer, debug): (LuaValue, Option<bool>),
) -> LuaResult<bool> {
	let level = if debug.unwrap_or_default() {
		tracing::Level::DEBUG
	} else {
		tracing::Level::INFO
	};
	let format = tracing_subscriber::fmt::format()
		.with_level(true)
		.with_target(true)
		.with_thread_ids(false)
		.with_thread_names(false)
		.with_file(false)
		.with_line_number(false)
		.with_source_location(false);

	let success = match printer {
		LuaValue::Nil => tracing_subscriber::fmt()
			.event_format(format)
			.with_max_level(level)
			.with_writer(std::sync::Mutex::new(std::io::stderr()))
			.try_init()
			.is_ok(),
		LuaValue::String(path) => {
			let logfile = std::fs::File::create(path.to_string_lossy())
				.map_err(|e| LuaError::RuntimeError(e.to_string()))?;
			tracing_subscriber::fmt()
				.event_format(format)
				.with_max_level(level)
				.with_writer(Mutex::new(logfile))
				.with_ansi(false)
				.try_init()
				.is_ok()
		}
		LuaValue::Function(cb) => {
			let key = uuid::Uuid::new_v4().to_string();
			lua.set_named_registry_value(&key, cb)?;
			let (tx, mut rx) = mpsc::unbounded_channel();
			let res = tracing_subscriber::fmt()
				.event_format(format)
				.with_max_level(level)
				.with_writer(Mutex::new(LuaLoggerProducer(tx)))
				.with_ansi(false)
				.try_init()
				.is_ok();
			if res {
				super::a_sync::tokio().spawn(async move {
					while let Some(msg) = rx.recv().await {
						super::callback().invoke(key.clone(), msg, false);
					}
				});
			}
			res
		},
		_ => return Err(LuaError::BindError), // TODO full BadArgument type??
	};

	Ok(success)
}