Add Ci::Status::Stage

3 files changed, 65 insertions, 0 deletions

diff --git a/app/models/ci/stage.rb b/app/models/ci/stage.rb
index 8f7727aebaa..2e7f15a16d7 100644
--- a/app/models/ci/stage.rb
+++ b/app/models/ci/stage.rb
@@ -6,6 +6,8 @@ module Ci
 
     attr_reader :pipeline, :name
 
+    delegate :project, to: :pipeline
+
     def initialize(pipeline, name: name, status: nil)
       @pipeline, @name, @status = pipeline, name, status
     end
diff --git a/lib/gitlab/ci/status/stage/common.rb b/lib/gitlab/ci/status/stage/common.rb
new file mode 100644
index 00000000000..a1513024c6c
--- /dev/null
+++ b/lib/gitlab/ci/status/stage/common.rb
@@ -0,0 +1,24 @@
+module Gitlab
+  module Ci
+    module Status
+      module Stage
+        module Common
+          def has_details?
+            true
+          end
+
+          def details_path
+            namespace_project_pipeline_path(@subject.project.namespace,
+                                            @subject.project,
+                                            @subject,
+                                            anchor: subject.name)
+          end
+
+          def has_action?
+            false
+          end
+        end
+      end
+    end
+  end
+end
diff --git a/lib/gitlab/ci/status/stage/factory.rb b/lib/gitlab/ci/status/stage/factory.rb
new file mode 100644
index 00000000000..2e485ee22a9
--- /dev/null
+++ b/lib/gitlab/ci/status/stage/factory.rb
@@ -0,0 +1,39 @@
+module Gitlab
+  module Ci
+    module Status
+      module Stage
+        class Factory
+          EXTENDED_STATUSES = []
+
+          def initialize(stage)
+            @stage = stage
+            @status = stage.status || :created
+          end
+
+          def fabricate!
+            if extended_status
+              extended_status.new(core_status)
+            else
+              core_status
+            end
+          end
+
+          private
+
+          def core_status
+            Gitlab::Ci::Status
+              .const_get(@status.capitalize)
+              .new(@stage)
+              .extend(Status::Pipeline::Common)
+          end
+
+          def extended_status
+            @extended ||= EXTENDED_STATUSES.find do |status|
+              status.matches?(@stage)
+            end
+          end
+        end
+      end
+    end
+  end
+end