1 files changed, 121 insertions, 1 deletions

diff --git a/app/models/ci/bridge.rb b/app/models/ci/bridge.rb
index e6d41dd2779..abd59741e89 100644
--- a/app/models/ci/bridge.rb
+++ b/app/models/ci/bridge.rb
@@ -4,19 +4,78 @@ module Ci
   class Bridge < Ci::Processable
     include Ci::Contextable
     include Ci::PipelineDelegator
+    include Ci::Metadatable
     include Importable
     include AfterCommitQueue
     include HasRef
     include Gitlab::Utils::StrongMemoize
 
+    InvalidBridgeTypeError = Class.new(StandardError)
+
     belongs_to :project
     belongs_to :trigger_request
+    has_many :sourced_pipelines, class_name: "::Ci::Sources::Pipeline",
+                                  foreign_key: :source_job_id
+
     validates :ref, presence: true
 
+    # rubocop:disable Cop/ActiveRecordSerialize
+    serialize :options
+    serialize :yaml_variables, ::Gitlab::Serializer::Ci::Variables
+    # rubocop:enable Cop/ActiveRecordSerialize
+
+    state_machine :status do
+      event :manual do
+        transition all => :manual
+      end
+
+      event :scheduled do
+        transition all => :scheduled
+      end
+    end
+
     def self.retry(bridge, current_user)
       raise NotImplementedError
     end
 
+    def inherit_status_from_downstream!(pipeline)
+      case pipeline.status
+      when 'success'
+        self.success!
+      when 'failed', 'canceled', 'skipped'
+        self.drop!
+      else
+        false
+      end
+    end
+
+    def downstream_pipeline_params
+      return child_params if triggers_child_pipeline?
+      return cross_project_params if downstream_project.present?
+
+      {}
+    end
+
+    def downstream_project
+      strong_memoize(:downstream_project) do
+        if downstream_project_path
+          ::Project.find_by_full_path(downstream_project_path)
+        elsif triggers_child_pipeline?
+          project
+        end
+      end
+    end
+
+    def downstream_project_path
+      strong_memoize(:downstream_project_path) do
+        options&.dig(:trigger, :project)
+      end
+    end
+
+    def triggers_child_pipeline?
+      yaml_for_downstream.present?
+    end
+
     def tags
       [:bridge]
     end
@@ -55,7 +114,68 @@ module Ci
     end
 
     def yaml_for_downstream
-      nil
+      strong_memoize(:yaml_for_downstream) do
+        includes = options&.dig(:trigger, :include)
+        YAML.dump('include' => includes) if includes
+      end
+    end
+
+    def target_ref
+      branch = options&.dig(:trigger, :branch)
+      return unless branch
+
+      scoped_variables.to_runner_variables.yield_self do |all_variables|
+        ::ExpandVariables.expand(branch, all_variables)
+      end
+    end
+
+    def dependent?
+      strong_memoize(:dependent) do
+        options&.dig(:trigger, :strategy) == 'depend'
+      end
+    end
+
+    def downstream_variables
+      variables = scoped_variables.concat(pipeline.persisted_variables)
+
+      variables.to_runner_variables.yield_self do |all_variables|
+        yaml_variables.to_a.map do |hash|
+          { key: hash[:key], value: ::ExpandVariables.expand(hash[:value], all_variables) }
+        end
+      end
+    end
+
+    private
+
+    def cross_project_params
+      {
+        project: downstream_project,
+        source: :pipeline,
+        target_revision: {
+          ref: target_ref || downstream_project.default_branch
+        },
+        execute_params: { ignore_skip_ci: true }
+      }
+    end
+
+    def child_params
+      parent_pipeline = pipeline
+
+      {
+        project: project,
+        source: :parent_pipeline,
+        target_revision: {
+          ref: parent_pipeline.ref,
+          checkout_sha: parent_pipeline.sha,
+          before: parent_pipeline.before_sha,
+          source_sha: parent_pipeline.source_sha,
+          target_sha: parent_pipeline.target_sha
+        },
+        execute_params: {
+          ignore_skip_ci: true,
+          bridge: self
+        }
+      }
     end
   end
 end