From 2419f8bb07789bb4eab1aa1bee1832b3f2616652 Mon Sep 17 00:00:00 2001
From: Mohamed Koubaa <koubaa@github.com>
Date: Tue, 14 May 2024 14:37:29 -0500
Subject: [PATCH] add trace options

---
 clr_loader/__init__.py |  2 ++
 clr_loader/ffi/mono.py |  3 +++
 clr_loader/mono.py     | 12 ++++++++++++
 tests/test_common.py   | 18 ++++++++++++++++++
 4 files changed, 35 insertions(+)

diff --git a/clr_loader/__init__.py b/clr_loader/__init__.py
index 71eb09d..52ce603 100644
--- a/clr_loader/__init__.py
+++ b/clr_loader/__init__.py
@@ -33,6 +33,8 @@ def get_mono(
     assembly_dir: Optional[str] = None,
     config_dir: Optional[str] = None,
     set_signal_chaining: bool = False,
+    trace_mask: Optional[str] = None,
+    trace_level: Optional[str] = None,
 ) -> Runtime:
     """Get a Mono runtime instance
 
diff --git a/clr_loader/ffi/mono.py b/clr_loader/ffi/mono.py
index c194393..a3249a1 100644
--- a/clr_loader/ffi/mono.py
+++ b/clr_loader/ffi/mono.py
@@ -44,5 +44,8 @@
 
 void mono_set_signal_chaining(bool chain_signals);
 
+void mono_trace_set_level_string(const char* value);
+void mono_trace_set_mask_string(const char* value);
+
 """
 )
diff --git a/clr_loader/mono.py b/clr_loader/mono.py
index 1899ea3..f7f90e5 100644
--- a/clr_loader/mono.py
+++ b/clr_loader/mono.py
@@ -27,6 +27,8 @@ def __init__(
         assembly_dir: Optional[str] = None,
         config_dir: Optional[str] = None,
         set_signal_chaining: bool = False,
+        trace_mask: Optional[str] = None,
+        trace_level: Optional[str] = None,
     ):
         self._assemblies: Dict[Path, Any] = {}
 
@@ -39,6 +41,8 @@ def __init__(
             assembly_dir=assembly_dir,
             config_dir=config_dir,
             set_signal_chaining=set_signal_chaining,
+            trace_mask=trace_mask,
+            trace_level=trace_level,
         )
 
         if domain is None:
@@ -131,11 +135,19 @@ def initialize(
     assembly_dir: Optional[str] = None,
     config_dir: Optional[str] = None,
     set_signal_chaining: bool = False,
+    trace_mask: Optional[str] = None,
+    trace_level: Optional[str] = None,
 ) -> str:
     global _MONO, _ROOT_DOMAIN
     if _MONO is None:
         _MONO = load_mono(libmono)
 
+        if trace_mask is not None:
+            _MONO.mono_trace_set_mask_string(trace_mask.encode("utf8"))
+
+        if trace_level is not None:
+            _MONO.mono_trace_set_level_string(trace_level.encode("utf8"))
+
         if assembly_dir is not None and config_dir is not None:
             _MONO.mono_set_dirs(assembly_dir.encode("utf8"), config_dir.encode("utf8"))
 
diff --git a/tests/test_common.py b/tests/test_common.py
index 139f192..3eaf215 100644
--- a/tests/test_common.py
+++ b/tests/test_common.py
@@ -56,6 +56,24 @@ def test_mono_signal_chaining(example_netstandard: Path):
     run_tests(asm)
 
 
+def test_mono_trace_mask(example_netstandard: Path):
+    from clr_loader import get_mono
+
+    mono = get_mono(trace_mask="all")
+    asm = mono.get_assembly(example_netstandard / "example.dll")
+
+    run_tests(asm)
+
+
+def test_mono_trace_level(example_netstandard: Path):
+    from clr_loader import get_mono
+
+    mono = get_mono(trace_level="message")
+    asm = mono.get_assembly(example_netstandard / "example.dll")
+
+    run_tests(asm)
+
+
 def test_mono_set_dir(example_netstandard: Path):
     from clr_loader import get_mono