summaryrefslogtreecommitdiff
path: root/lib/github/representation/pull_request.rb
blob: 55461097e8a9ee05b46c29b556f3b4ab57366d3b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
module Github
  module Representation
    class PullRequest < Representation::Issuable
      delegate :user, :repo, :ref, :sha, to: :source_branch, prefix: true
      delegate :user, :exists?, :repo, :ref, :sha, :short_sha, to: :target_branch, prefix: true

      def source_project
        project
      end

      def source_branch_name
        @source_branch_name ||=
          if cross_project? || !source_branch_exists?
            source_branch_name_prefixed
          else
            source_branch_ref
          end
      end

      def source_branch_exists?
        return @source_branch_exists if defined?(@source_branch_exists)

        @source_branch_exists = !cross_project? && source_branch.exists?
      end

      def target_project
        project
      end

      def target_branch_name
        @target_branch_name ||= target_branch_exists? ? target_branch_ref : target_branch_name_prefixed
      end

      def target_branch_exists?
        @target_branch_exists ||= target_branch.exists?
      end

      def state
        return 'merged' if raw['state'] == 'closed' && raw['merged_at'].present?
        return 'closed' if raw['state'] == 'closed'

        'opened'
      end

      def opened?
        state == 'opened'
      end

      def valid?
        source_branch.valid? && target_branch.valid?
      end

      def restore_branches!
        restore_source_branch!
        restore_target_branch!
      end

      def remove_restored_branches!
        return if opened?

        remove_source_branch!
        remove_target_branch!
      end

      private

      def project
        @project ||= options.fetch(:project)
      end

      def source_branch
        @source_branch ||= Representation::Branch.new(raw['head'], repository: project.repository)
      end

      def source_branch_name_prefixed
        "gh-#{target_branch_short_sha}/#{iid}/#{source_branch_user}/#{source_branch_ref}"
      end

      def target_branch
        @target_branch ||= Representation::Branch.new(raw['base'], repository: project.repository)
      end

      def target_branch_name_prefixed
        "gl-#{target_branch_short_sha}/#{iid}/#{target_branch_user}/#{target_branch_ref}"
      end

      def cross_project?
        return true if source_branch_repo.nil?

        source_branch_repo.id != target_branch_repo.id
      end

      def restore_source_branch!
        return if source_branch_exists?

        source_branch.restore!(source_branch_name)
      end

      def restore_target_branch!
        return if target_branch_exists?

        target_branch.restore!(target_branch_name)
      end

      def remove_source_branch!
        # We should remove the source/target branches only if they were
        # restored. Otherwise, we'll remove branches like 'master' that
        # target_branch_exists? returns true. In other words, we need
        # to clean up only the restored branches that (source|target)_branch_exists?
        # returns false for the first time it has been called, because of
        # this that is important to memoize these values.
        source_branch.remove!(source_branch_name) unless source_branch_exists?
      end

      def remove_target_branch!
        target_branch.remove!(target_branch_name) unless target_branch_exists?
      end
    end
  end
end