gitlab-org--gitlab-foss/lib/gitlab/github_import/comment_formatter.rb
Alejandro Rodríguez 9fdde3693b Move line code generation into Gitlab::Git
Having a distinct class just for that was a bit overkill
2017-10-12 22:13:05 -03:00

69 lines
1.3 KiB
Ruby

module Gitlab
module GithubImport
class CommentFormatter < BaseFormatter
attr_writer :author_id
def attributes
{
project: project,
note: note,
commit_id: raw_data.commit_id,
line_code: line_code,
author_id: author_id,
type: type,
created_at: raw_data.created_at,
updated_at: raw_data.updated_at
}
end
private
def author
@author ||= UserFormatter.new(client, raw_data.user)
end
def author_id
author.gitlab_id || project.creator_id
end
def body
raw_data.body || ""
end
def line_code
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::Git.diff_line_code(file_path, line.new_pos, line.old_pos)
end
def on_diff?
diff_hunk.present?
end
def diff_hunk
raw_data.diff_hunk
end
def file_path
raw_data.path
end
def note
if author.gitlab_id
body
else
formatter.author_line(author.login) + body
end
end
def type
'LegacyDiffNote' if on_diff?
end
end
end
end