Add current module tag to Sentry crash reports

Author: hisenb3rg

plain2code_state.py

@@ -24,6 +24,7 @@ def __init__(self, spec_filename: str, replay_with: Optional[str] = None):
         self.render_time_accumulated: int = 0
         self.last_render_start_timestamp: float = time.monotonic()
         # Mirrored from the render state machine for crash reporting; not authoritative.
+        self.current_module: Optional[str] = None
         self.current_frid: Optional[str] = None
         self.current_render_state: Optional[str] = None
 

plain2code_telemetry.py

@@ -95,6 +95,7 @@ def capture_crash(exc_info, run_state: Optional[RunState], args) -> bool:
         with sentry_sdk.new_scope() as scope:
             if run_state is not None:
                 scope.set_tag("render_id", run_state.render_id)
+                scope.set_tag("current_module", run_state.current_module)
                 scope.set_tag("current_frid", run_state.current_frid)
                 scope.set_tag("render_state", run_state.current_render_state)
             scope.set_tag("headless", bool(getattr(args, "headless", False)))

render_machine/code_renderer.py

@@ -42,6 +42,7 @@ def __init__(self, render_context: RenderContext):
     def run(self):
         """Execute the main rendering workflow."""
         self.render_context.event_bus.publish(RenderModuleStarted(module_name=self.render_context.module_name))
+        self.render_context.run_state.current_module = self.render_context.module_name
         previous_action_payload = None
         previous_state = None
 

tests/test_telemetry.py

@@ -77,6 +77,7 @@ def test_capture_crash_sends_event_with_tags(transport):
     init_with_transport(transport)
 
     run_state = RunState(spec_filename="test.plain")
+    run_state.current_module = "my_module"
     run_state.current_frid = "2.1"
     run_state.current_render_state = "IMPLEMENTING_FRID"
 
@@ -86,6 +87,7 @@ def test_capture_crash_sends_event_with_tags(transport):
     assert len(transport.events) == 1
     tags = transport.events[0]["tags"]
     assert tags["render_id"] == run_state.render_id
+    assert tags["current_module"] == "my_module"
     assert tags["current_frid"] == "2.1"
     assert tags["render_state"] == "IMPLEMENTING_FRID"
     assert tags["headless"] is True