BigW Consortium Gitlab

comment_formatter.rb 1.21 KB
Newer Older
1 2
module Gitlab
  module GithubImport
3
    class CommentFormatter < BaseFormatter
4 5 6 7 8 9 10
      def attributes
        {
          project: project,
          note: note,
          commit_id: raw_data.commit_id,
          line_code: line_code,
          author_id: author_id,
11
          type: type,
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
          created_at: raw_data.created_at,
          updated_at: raw_data.updated_at
        }
      end

      private

      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

      def line_code
32 33 34 35 36 37 38 39
        return unless on_diff?

        parsed_lines = Gitlab::Diff::Parser.new.parse(diff_hunk.lines)
        generate_line_code(parsed_lines.to_a.last)
      end

      def generate_line_code(line)
        Gitlab::Diff::LineCode.generate(file_path, line.new_pos, line.old_pos)
40 41 42
      end

      def on_diff?
43 44 45 46 47 48 49 50 51
        diff_hunk.present?
      end

      def diff_hunk
        raw_data.diff_hunk
      end

      def file_path
        raw_data.path
52 53 54
      end

      def note
55
        formatter.author_line(author) + body
56
      end
57 58 59 60

      def type
        'LegacyDiffNote' if on_diff?
      end
61 62 63
    end
  end
end