gitlab-org--gitlab-foss/lib/gitlab/ci/pipeline_object_hierarchy.rb

# frozen_string_literal: true

module Gitlab
  module Ci
    class PipelineObjectHierarchy < ::Gitlab::ObjectHierarchy
      private

      def middle_table
        ::Ci::Sources::Pipeline.arel_table
      end

      def from_tables(cte)
        [objects_table, cte.table, middle_table]
      end

      def parent_id_column(_cte)
        middle_table[:source_pipeline_id]
      end

      def ancestor_conditions(cte)
        middle_table[:source_pipeline_id].eq(objects_table[:id]).and(
          middle_table[:pipeline_id].eq(cte.table[:id])
        ).and(
          project_condition
        )
      end

      def descendant_conditions(cte)
        middle_table[:pipeline_id].eq(objects_table[:id]).and(
          middle_table[:source_pipeline_id].eq(cte.table[:id])
        ).and(
          project_condition
        )
      end

      def project_condition
        case options[:project_condition]
        when :same then middle_table[:source_project_id].eq(middle_table[:project_id])
        when :different then middle_table[:source_project_id].not_eq(middle_table[:project_id])
        else Arel.sql('TRUE')
        end
      end
    end
  end
end