71 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			71 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| module Gitlab
 | |
|   module GithubImport
 | |
|     class PullRequestFormatter < IssuableFormatter
 | |
|       delegate :exists?, :project, :ref, :repo, :sha, to: :source_branch, prefix: true
 | |
|       delegate :exists?, :project, :ref, :repo, :sha, to: :target_branch, prefix: true
 | |
| 
 | |
|       def attributes
 | |
|         {
 | |
|           iid: number,
 | |
|           title: raw_data.title,
 | |
|           description: description,
 | |
|           source_project: source_branch_project,
 | |
|           source_branch: source_branch_name,
 | |
|           source_branch_sha: source_branch_sha,
 | |
|           target_project: target_branch_project,
 | |
|           target_branch: target_branch_name,
 | |
|           target_branch_sha: target_branch_sha,
 | |
|           state: state,
 | |
|           milestone: milestone,
 | |
|           author_id: author_id,
 | |
|           assignee_id: assignee_id,
 | |
|           created_at: raw_data.created_at,
 | |
|           updated_at: raw_data.updated_at
 | |
|         }
 | |
|       end
 | |
| 
 | |
|       def project_association
 | |
|         :merge_requests
 | |
|       end
 | |
| 
 | |
|       def valid?
 | |
|         source_branch.valid? && target_branch.valid?
 | |
|       end
 | |
| 
 | |
|       def source_branch
 | |
|         @source_branch ||= BranchFormatter.new(project, raw_data.head)
 | |
|       end
 | |
| 
 | |
|       def source_branch_name
 | |
|         @source_branch_name ||= begin
 | |
|           if source_branch.repo.id != target_branch.repo.id
 | |
|             "pull/#{number}/#{source_branch.repo.full_name}/#{source_branch_ref}"
 | |
|           else
 | |
|             source_branch_exists? ? source_branch_ref : "pull/#{number}/#{source_branch_ref}"
 | |
|           end
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       def target_branch
 | |
|         @target_branch ||= BranchFormatter.new(project, raw_data.base)
 | |
|       end
 | |
| 
 | |
|       def target_branch_name
 | |
|         @target_branch_name ||= begin
 | |
|           target_branch_exists? ? target_branch_ref : "pull/#{number}/#{target_branch_ref}"
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       private
 | |
| 
 | |
|       def state
 | |
|         if raw_data.state == 'closed' && raw_data.merged_at.present?
 | |
|           'merged'
 | |
|         else
 | |
|           super
 | |
|         end
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| end
 |