pull_request_formatter.rb 2.23 KB
Newer Older
1
2
3
4
5
module Gitlab
  module GithubImport
    class PullRequestFormatter < BaseFormatter
      def attributes
        {
6
          iid: number,
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
          title: raw_data.title,
          description: description,
          source_project: source_project,
          source_branch: source_branch.name,
          target_project: target_project,
          target_branch: target_branch.name,
          state: state,
          author_id: author_id,
          assignee_id: assignee_id,
          created_at: raw_data.created_at,
          updated_at: updated_at
        }
      end

      def number
        raw_data.number
      end

      def valid?
26
        !cross_project? && source_branch.present? && target_branch.present?
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
      end

      private

      def assigned?
        raw_data.assignee.present?
      end

      def assignee_id
        if assigned?
          gl_user_id(raw_data.assignee.id)
        end
      end

      def author
        raw_data.user.login
      end

      def author_id
        gl_user_id(raw_data.user.id) || project.creator_id
      end

      def body
        raw_data.body || ""
      end

53
54
55
56
      def cross_project?
        source_repo.present? && target_repo.present? && source_repo.id != target_repo.id
      end

57
58
59
60
61
62
63
64
      def description
        formatter.author_line(author) + body
      end

      def source_project
        project
      end

65
66
67
68
      def source_repo
        raw_data.head.repo
      end

69
70
71
72
73
74
75
76
      def source_branch
        source_project.repository.find_branch(raw_data.head.ref)
      end

      def target_project
        project
      end

77
78
79
80
      def target_repo
        raw_data.base.repo
      end

81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
      def target_branch
        target_project.repository.find_branch(raw_data.base.ref)
      end

      def state
        @state ||= case true
                   when raw_data.state == 'closed' && raw_data.merged_at.present?
                     'merged'
                   when raw_data.state == 'closed'
                     'closed'
                   else
                     'opened'
                   end
      end

      def updated_at
        case state
        when 'merged' then raw_data.merged_at
        when 'closed' then raw_data.closed_at
        else
          raw_data.updated_at
        end
      end
    end
  end
end