1 files changed, 38 insertions, 0 deletions
diff --git a/app/serializers/analytics/cycle_analytics/stage_entity.rb b/app/serializers/analytics/cycle_analytics/stage_entity.rb
new file mode 100644
index 00000000000..b24148802d0
--- /dev/null
+++ b/app/serializers/analytics/cycle_analytics/stage_entity.rb
@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Analytics
+  module CycleAnalytics
+    class StageEntity < Grape::Entity
+      expose :title
+      expose :hidden
+      expose :legend
+      expose :description
+      expose :id
+      expose :custom
+      expose :start_event_identifier, if: -> (s) { s.custom? }
+      expose :end_event_identifier, if: -> (s) { s.custom? }
+      expose :start_event_label, using: LabelEntity, if: -> (s) { s.start_event_label_based? }
+      expose :end_event_label, using: LabelEntity, if: -> (s) { s.end_event_label_based? }
+      expose :start_event_html_description
+      expose :end_event_html_description
+
+      def id
+        object.id || object.name
+      end
+
+      def start_event_html_description
+        html_description(object.start_event)
+      end
+
+      def end_event_html_description
+        html_description(object.end_event)
+      end
+
+      private
+
+      def html_description(event)
+        Banzai::Renderer.render(event.markdown_description, { group: object.group, project: nil })
+      end
+    end
+  end
+end