summaryrefslogtreecommitdiff
path: root/lib/gitlab/bitbucket_import/importer.rb
blob: 3f483847efaa14f6adf98502cf88bc6c92a8799f (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
module Gitlab
  module BitbucketImport
    class Importer
      attr_reader :project, :client

      def initialize(project)
        @project = project
        import_data = project.import_data.try(:data)
        bb_session = import_data["bb_session"] if import_data
        @client = Client.new(bb_session["bitbucket_access_token"],
                             bb_session["bitbucket_access_token_secret"])
        @formatter = Gitlab::ImportFormatter.new
      end

      def execute
        import_issues if has_issues?

        true
      rescue ActiveRecord::RecordInvalid => e
        raise Projects::ImportService::Error.new, e.message
      ensure
        Gitlab::BitbucketImport::KeyDeleter.new(project).execute
      end

      private

      def gl_user_id(project, bitbucket_id)
        if bitbucket_id
          user = User.joins(:identities).find_by("identities.extern_uid = ? AND identities.provider = 'bitbucket'", bitbucket_id.to_s)
          (user && user.id) || project.creator_id
        else
          project.creator_id
        end
      end

      def identifier
        project.import_source
      end

      def has_issues?
        client.project(identifier)["has_issues"]
      end

      def import_issues
        issues = client.issues(identifier)

        issues.each do |issue|
          body = ''
          reporter = nil
          author = 'Anonymous'

          if issue["reported_by"] && issue["reported_by"]["username"]
            reporter = issue["reported_by"]["username"]
            author = reporter
          end

          body = @formatter.author_line(author)
          body += issue["content"]

          comments = client.issue_comments(identifier, issue["local_id"])

          if comments.any?
            body += @formatter.comments_header
          end

          comments.each do |comment|
            author = 'Anonymous'

            if comment["author_info"] && comment["author_info"]["username"]
              author = comment["author_info"]["username"]
            end

            body += @formatter.comment(author, comment["utc_created_on"], comment["content"])
          end

          project.issues.create!(
            description: body,
            title: issue["title"],
            state: %w(resolved invalid duplicate wontfix).include?(issue["status"]) ? 'closed' : 'opened',
            author_id: gl_user_id(project, reporter)
          )
        end
      rescue ActiveRecord::RecordInvalid => e
        raise Projects::ImportService::Error, e.message
      end
    end
  end
end