lib/dap/event.rb in ruby-dap-0.1.1 vs lib/dap/event.rb in ruby-dap-0.1.2

@@ -17,26 +17,29 @@
 class DAP::Event < DAP::ProtocolMessage
   def self.type
     'event'
   end
 
-  property :event
+  def self.bodies
+    @bodies ||= one_of(
+      initialized: empty,
+      stopped: DAP::StoppedEventBody,
+      continued: DAP::ContinuedEventBody,
+      exited: DAP::ExitedEventBody,
+      terminated: DAP::TerminatedEventBody,
+      thread: DAP::ThreadEventBody,
+      output: DAP::OutputEventBody,
+      breakpoint: DAP::BreakpointEventBody,
+      module: DAP::ModuleEventBody,
+      loadedSource: DAP::LoadedSourceEventBody,
+      process: DAP::ProcessEventBody,
+      capabilities: DAP::CapabilitiesEventBody,
+      progressStart: DAP::ProgressStartEventBody,
+      progressUpdate: DAP::ProgressUpdateEventBody,
+      progressEnd: DAP::ProgressEndEventBody,
+      invalidated: DAP::InvalidatedEventBody,
+    )
+  end
 
-  property :body, as: one_of(
-    initialized: empty,
-    stopped: DAP::StoppedEventBody,
-    continued: DAP::ContinuedEventBody,
-    exited: DAP::ExitedEventBody,
-    terminated: DAP::TerminatedEventBody,
-    thread: DAP::ThreadEventBody,
-    output: DAP::OutputEventBody,
-    breakpoint: DAP::BreakpointEventBody,
-    module: DAP::ModuleEventBody,
-    loadedSource: DAP::LoadedSourceEventBody,
-    process: DAP::ProcessEventBody,
-    capabilities: DAP::CapabilitiesEventBody,
-    progressStart: DAP::ProgressStartEventBody,
-    progressUpdate: DAP::ProgressUpdateEventBody,
-    progressEnd: DAP::ProgressEndEventBody,
-    invalidated: DAP::InvalidatedEventBody,
-  ).with(:event)
+  property :event
+  property :body, as: bodies.with(:event)
 end