feat: log to steam logs

Author: CertainLach

bin/driver-proxy/src/driver/server_tracked_provider.rs

@@ -3,12 +3,14 @@ use std::{os::raw::c_char, sync::Mutex};
 use crate::{
 	driver_context::{try_init_driver_context, DRIVER_CONTEXT},
 	factory::{get_hmd_driver_factory, TOKIO_RUNTIME},
+	log::try_init_driver_log,
 	setting,
 	settings::Setting,
 	try_vr,
 };
 use cppvtbl::{impl_vtables, HasVtable, VtableRef, WithVtables};
 use once_cell::sync::Lazy;
+use openvr::{IVRDriverLogVtable, IVRDriverLog_Version};
 use tokio::task::LocalSet;
 use tracing::info;
 use valve_pm::{start_manager, StationCommand, StationControl, StationState};
@@ -36,8 +38,13 @@ impl IServerTrackedDeviceProvider for ServerTrackedProvider {
 		&self,
 		pDriverContext: *const cppvtbl::VtableRef<IVRDriverContextVtable>,
 	) -> EVRInitError {
-		let _ = try_init_driver_context(unsafe { &*pDriverContext });
+		try_init_driver_context(unsafe { &*pDriverContext });
 		let context = DRIVER_CONTEXT.get().expect("context just initialized");
+		let logger: *const cppvtbl::VtableRef<IVRDriverLogVtable> = context
+			.get_generic_interface(IVRDriverLog_Version)
+			.expect("always able to initialize driver log")
+			.cast();
+		try_init_driver_log(unsafe { &*logger });
 
 		let power_management = POWER_MANAGEMENT.get();
 		*self.standby_state.lock().expect("lock") = match power_management {
@@ -56,8 +63,8 @@ impl IServerTrackedDeviceProvider for ServerTrackedProvider {
 					break 'stations;
 				}
 				let Ok(manager) = TOKIO_RUNTIME.block_on(start_manager()) else {
-				break 'stations;
-			};
+					break 'stations;
+				};
 				let stations: Vec<_> = stations
 					.iter()
 					.filter_map(|line| {

bin/driver-proxy/src/factory.rs

@@ -4,7 +4,9 @@ use std::{
 	ptr::null,
 };
 
-use crate::{server_tracked_provider::SERVER_TRACKED_DEVICE_PROVIDER, Error, Result};
+use crate::{
+	log::LogWriter, server_tracked_provider::SERVER_TRACKED_DEVICE_PROVIDER, Error, Result,
+};
 use cppvtbl::{HasVtable, VtableRef};
 use libloading::{Library, Symbol};
 use once_cell::sync::{Lazy, OnceCell};
@@ -36,7 +38,12 @@ pub static TOKIO_RUNTIME: Lazy<Runtime> =
 
 fn HmdDriverFactory_impl(iface: *const c_char) -> Result<*const c_void> {
 	// May be already installed
-	if tracing_subscriber::fmt().without_time().try_init().is_ok() {
+	if tracing_subscriber::fmt()
+		.without_time()
+		.with_writer(LogWriter::default)
+		.try_init()
+		.is_ok()
+	{
 		// This magic string is also used for installation detection!
 		info!("https://patreon.com/0lach");
 	}

bin/driver-proxy/src/log.rs

@@ -7,7 +7,8 @@ use crate::openvr::{IVRDriverLog, IVRDriverLogVtable};
 
 static DRIVER_LOG: OnceCell<&'static VtableRef<IVRDriverLogVtable>> = OnceCell::new();
 
-struct LogWriter(Vec<u8>);
+#[derive(Default)]
+pub struct LogWriter(Vec<u8>);
 impl LogWriter {
 	fn flush_line(&mut self) {
 		if let Some(driver) = DRIVER_LOG.get() {
@@ -27,11 +28,16 @@ impl Write for LogWriter {
 			self.flush_line();
 			buf = &buf[pos + 1..];
 		}
-		self.0.extend_from_slice(&buf);
+		self.0.extend_from_slice(buf);
 		Ok(buf.len())
 	}
 
 	fn flush(&mut self) -> std::io::Result<()> {
+		self.flush_line();
 		Ok(())
 	}
 }
+
+pub fn try_init_driver_log(log: &'static VtableRef<IVRDriverLogVtable>) {
+	let _ = DRIVER_LOG.set(log);
+}