feat: Add support for tracepoints in vscode insiders extension

Author: pxor

src/common/ct_event.nim

@@ -60,6 +60,7 @@ type
     CtUpdatedFlow,
     CtRunToEntry,
     CtRunTracepoints,
+    CtRunTraceSession,
     CtSetupTraceSession,
     CtLoadAsmFunction,
     CtLoadAsmFunctionResponse,
@@ -68,6 +69,7 @@ type
     InternalAddToScratchpadFromExpression,
     InternalStatusUpdate,
     InternalNewOperation,
+    InternalTraceMapUpdate,
     CtNotification,
 
 when defined(js):

src/frontend/dap.nim

@@ -100,6 +100,7 @@ const EVENT_KIND_TO_DAP_MAPPING: array[CtEventKind, cstring] = [
   CtUpdatedFlow: "ct/updated-flow",
   CtRunToEntry: "ct/run-to-entry",
   CtRunTracepoints: "ct/run-tracepoints",
+  CtRunTraceSession: "ct/run-trace-session",
   CtSetupTraceSession: "ct/setup-trace-session",
   CtLoadAsmFunction: "ct/load-asm-function",
   CtLoadAsmFunctionResponse: "",
@@ -108,6 +109,7 @@ const EVENT_KIND_TO_DAP_MAPPING: array[CtEventKind, cstring] = [
   InternalAddToScratchpadFromExpression: "",
   InternalStatusUpdate: "",
   InternalNewOperation: "",
+  InternalTraceMapUpdate: "",
   CtNotification: "ct/notification",
 ]
 

src/frontend/middleware.nim

@@ -5,6 +5,38 @@ import communication, dap
 import event_helpers
 
 # backend(dap) <-> middleware <-> view (self-contained, can be separate: 0, 1 or more components);
+var tracepointMap: JsAssoc[cstring, JsAssoc[int, Tracepoint]] = Jsassoc[cstring, JsAssoc[int, Tracepoint]]{}
+var sessionCounter = 0
+
+proc getTraceSession(name: cstring = "/home/nedy/calls.rb"): TraceSession =
+  var results = JsAssoc[int, seq[Stop]]{}
+  var tracepoints: seq[Tracepoint] = @[]
+
+  for line, trace in tracepointMap[name]:
+    if trace.expression != "" and trace.isChanged:
+      tracepoints.add(trace)
+
+  result = TraceSession(
+    tracepoints: tracepoints,
+    lastCount: 0,
+    results: results,
+    id: sessionCounter
+  )
+  sessionCounter += 1
+
+  return result
+
+proc updateTraceMap(tracepoint: Tracepoint) =
+  if not tracepointMap.hasKey(tracepoint.name):
+    tracepointMap[tracepoint.name] = JsAssoc[int, Tracepoint]{}
+    tracepointMap[tracepoint.name][tracepoint.line] = tracepoint
+    return
+
+  if not tracepointMap[tracepoint.name].hasKey(tracepoint.line):
+    tracepointMap[tracepoint.name][tracepoint.line] = tracepoint
+    return
+
+  tracepointMap[tracepoint.name][tracepoint.line] = tracepoint
 
 when not defined(ctInExtension):
   import utils
@@ -149,6 +181,19 @@ proc setupMiddlewareApis*(dapApi: DapApi, viewsApi: MediatorWithSubscribers) {.e
     if not lastCompleteMove.isNil:
       viewsApi.emit(CtCompleteMove, lastCompleteMove.toJs)
   )
+  viewsApi.subscribe(InternalTraceMapUpdate, proc(kind: CtEventKind, value: Tracepoint, sub: Subscriber) =
+    updateTraceMap(value)
+  )
+  viewsApi.subscribe(CtRunTraceSession, proc(kind: CtEventKind, value: EmptyArg, sub: Subscriber) =
+    let traceSession = getTraceSession("/home/nedy/calls.rb".cstring)
+    dapApi.sendCtRequest(
+      CtRunTracepoints,
+      RunTracepointsArg(
+        session: traceSession,
+        stopAfter: NO_LIMIT
+      ).toJs
+    )
+  )
 
 when defined(ctInExtension):
   when defined(ctInCentralExtensionContext):

src/frontend/styles/components/tracepoint.styl

@@ -53,6 +53,8 @@
   padding-right: 1ch
   // height: 36ch
   position: absolute
+  if IS_EXTENSION
+    max-width: calc(100% - 15px)
   // left: -81px
 
   .editor-textarea
@@ -255,10 +257,6 @@
     .fa
       color: DROPDOWN_TEXT_BACKGROUND_COLOR
 
-  .trace-buttons-container
-    display: flex
-    gap: 6px
-
   .run-trace-button
     border-radius: 6px;
     color: BORDER_COLOR;
@@ -301,6 +299,14 @@
     z-index: TRACE_LOG_EDITOR_Z_INDEX + 7
     top: 0
 
+    .trace-hamburger-svg
+      height: 24px
+      width: 24px
+      background-image: TRACE_HAMBURGER_IMAGE
+      background-size: 12px 8px
+      background-repeat: no-repeat
+      background-position: center
+
   .trace-dropdown-menu
     display: none
 
@@ -364,6 +370,16 @@
     margin: 0
     padding: 0
 
+  .trace-buttons-container
+    display: flex
+    gap: 6px
+
+    .trace-run-button-svg
+      height: 24px
+      width: 24px
+      background-image: RUN_TRACEPOINT_IMAGE
+      background-repeat: no-repeat
+
 .kind-dropdown-menu
   background: #b8c0cf !important
   position: absolute

src/frontend/styles/default_dark_theme.styl

@@ -21,6 +21,9 @@ BREAKPOINT_COLOR=#F87171
 TRACEPOINT_COLOR=#6366F1
 TRACEPOINT_LINE_NUMBER_COLOR=#c6c6c6
 
+RUN_TRACEPOINT_IMAGE=url("../../public/resources/tracepoints/run_tracepoints_dark.svg")
+TRACE_HAMBURGER_IMAGE=url("../../public/resources/tracepoints/trace_hamburger_dark.svg")
+
 NOUI_SLIDER_HANDLER_IMAGE=url("../../public/resources/shared/noUiSlider_handle_icon_dark.svg")
 NOUI_SLIDER_ACTIVE_HANDLER_IMAGE=url("../../public/resources/shared/noUiSlider_handle_active_icon_dark.svg")
 NOUI_SLIDER_HANDLER_BACKGROUND=#818181

src/frontend/ui/scratchpad.nim

@@ -84,3 +84,6 @@ method render*(self: ScratchpadComponent): VNode =
         else:
           tdiv(class = "empty-overlay"):
             text "You can add values from other components by right clicking on them and then click on 'Add value to scratchpad'."
+
+proc registerScratchpadComponent*(component: ScratchpadComponent, api: MediatorWithSubscribers) {.exportc.} =
+  component.register(api)