323 lines
8.7 KiB
Ruby
323 lines
8.7 KiB
Ruby
module Gitlab
|
|
module GoogleCodeImport
|
|
class Importer
|
|
attr_reader :project, :repo
|
|
|
|
def initialize(project)
|
|
@project = project
|
|
@repo = GoogleCodeImport::Repository.new(project.import_data)
|
|
|
|
@closed_statuses = []
|
|
@known_labels = Set.new
|
|
end
|
|
|
|
def execute
|
|
return true unless repo.valid?
|
|
|
|
import_status_labels
|
|
|
|
import_labels
|
|
|
|
import_issues
|
|
|
|
true
|
|
end
|
|
|
|
private
|
|
|
|
def import_status_labels
|
|
repo.raw_data["issuesConfig"]["statuses"].each do |status|
|
|
closed = !status["meansOpen"]
|
|
@closed_statuses << status["status"] if closed
|
|
|
|
name = nice_status_name(status["status"])
|
|
create_label(name)
|
|
@known_labels << name
|
|
end
|
|
end
|
|
|
|
def import_labels
|
|
repo.raw_data["issuesConfig"]["labels"].each do |label|
|
|
name = nice_label_name(label["label"])
|
|
create_label(name)
|
|
@known_labels << name
|
|
end
|
|
end
|
|
|
|
def import_issues
|
|
return unless repo.raw_data["issues"]
|
|
|
|
last_id = 0
|
|
|
|
deleted_issues = []
|
|
|
|
issues = repo.raw_data["issues"]["items"]
|
|
issues.each_with_index do |raw_issue, i|
|
|
while raw_issue["id"] > last_id + 1
|
|
last_id += 1
|
|
|
|
issue = project.issues.create!(
|
|
title: "Deleted issue",
|
|
description: "*This issue has been deleted*",
|
|
author_id: project.creator_id,
|
|
state: "closed"
|
|
)
|
|
deleted_issues << issue
|
|
end
|
|
last_id = raw_issue["id"]
|
|
|
|
author = mask_email(raw_issue["author"]["name"])
|
|
author_link = raw_issue["author"]["htmlLink"]
|
|
date = DateTime.parse(raw_issue["published"]).to_formatted_s(:long)
|
|
|
|
body = []
|
|
body << "*By [#{author}](#{author_link}) on #{date}*"
|
|
body << "---"
|
|
|
|
comments = raw_issue["comments"]["items"]
|
|
|
|
issue_comment = comments.shift
|
|
|
|
content = format_content(issue_comment["content"])
|
|
if content.blank?
|
|
content = "*(No description has been entered for this issue)*"
|
|
end
|
|
body << content
|
|
|
|
attachments = format_attachments(raw_issue["id"], 0, issue_comment["attachments"])
|
|
if attachments.any?
|
|
body << "---"
|
|
body += attachments
|
|
end
|
|
|
|
labels = []
|
|
raw_issue["labels"].each do |label|
|
|
name = nice_label_name(label)
|
|
labels << name
|
|
|
|
unless @known_labels.include?(name)
|
|
create_label(name)
|
|
@known_labels << name
|
|
end
|
|
end
|
|
labels << nice_status_name(raw_issue["status"])
|
|
|
|
issue = project.issues.create!(
|
|
title: raw_issue["title"],
|
|
description: body.join("\n\n"),
|
|
author_id: project.creator_id,
|
|
state: raw_issue["state"] == "closed" ? "closed" : "opened"
|
|
)
|
|
issue.add_labels_by_names(labels)
|
|
|
|
import_issue_comments(issue, comments)
|
|
end
|
|
|
|
deleted_issues.each(&:destroy!)
|
|
end
|
|
|
|
def import_issue_comments(issue, comments)
|
|
comments.each_with_index do |raw_comment, i|
|
|
next if raw_comment.has_key?("deletedBy")
|
|
|
|
author = mask_email(raw_comment["author"]["name"])
|
|
author_link = raw_comment["author"]["htmlLink"]
|
|
date = DateTime.parse(raw_comment["published"]).to_formatted_s(:long)
|
|
|
|
body = []
|
|
body << "*By [#{author}](#{author_link}) on #{date}*"
|
|
body << "---"
|
|
|
|
content = format_content(raw_comment["content"])
|
|
if content.blank?
|
|
content = "*(No comment has been entered for this change)*"
|
|
end
|
|
body << content
|
|
|
|
updates = format_updates(raw_comment["updates"])
|
|
if updates.any?
|
|
body << "---"
|
|
body += updates
|
|
end
|
|
|
|
attachments = format_attachments(issue.iid, raw_comment["id"], raw_comment["attachments"])
|
|
if attachments.any?
|
|
body << "---"
|
|
body += attachments
|
|
end
|
|
|
|
comment = issue.notes.create!(
|
|
project_id: project.id,
|
|
author_id: project.creator_id,
|
|
note: body.join("\n\n")
|
|
)
|
|
end
|
|
end
|
|
|
|
def nice_label_color(name)
|
|
case name
|
|
when /\AComponent:/
|
|
"#fff39e"
|
|
when /\AOpSys:/
|
|
"#e2e2e2"
|
|
when /\AMilestone:/
|
|
"#fee3ff"
|
|
|
|
when *@closed_statuses.map { |s| nice_status_name(s) }
|
|
"#cfcfcf"
|
|
when "Status: New"
|
|
"#428bca"
|
|
when "Status: Accepted"
|
|
"#5cb85c"
|
|
when "Status: Started"
|
|
"#8e44ad"
|
|
|
|
when "Priority: Critical"
|
|
"#ffcfcf"
|
|
when "Priority: High"
|
|
"#deffcf"
|
|
when "Priority: Medium"
|
|
"#fff5cc"
|
|
when "Priority: Low"
|
|
"#cfe9ff"
|
|
|
|
when "Type: Defect"
|
|
"#d9534f"
|
|
when "Type: Enhancement"
|
|
"#44ad8e"
|
|
when "Type: Task"
|
|
"#4b6dd0"
|
|
when "Type: Review"
|
|
"#8e44ad"
|
|
when "Type: Other"
|
|
"#7f8c8d"
|
|
else
|
|
"#e2e2e2"
|
|
end
|
|
end
|
|
|
|
def nice_label_name(name)
|
|
name.sub("-", ": ")
|
|
end
|
|
|
|
def nice_status_name(name)
|
|
"Status: #{name}"
|
|
end
|
|
|
|
def mask_email(author)
|
|
parts = author.split("@", 2)
|
|
parts[0] = "#{parts[0][0...-3]}..."
|
|
parts.join("@")
|
|
end
|
|
|
|
def linkify_issues(s)
|
|
s.gsub(/([Ii]ssue) ([0-9]+)/, '\1 #\2')
|
|
end
|
|
|
|
def escape_for_markdown(s)
|
|
s = s.gsub("*", "\\*")
|
|
s = s.gsub("#", "\\#")
|
|
s = s.gsub("`", "\\`")
|
|
s = s.gsub(":", "\\:")
|
|
s = s.gsub("-", "\\-")
|
|
s = s.gsub("+", "\\+")
|
|
s = s.gsub("_", "\\_")
|
|
s = s.gsub("(", "\\(")
|
|
s = s.gsub(")", "\\)")
|
|
s = s.gsub("[", "\\[")
|
|
s = s.gsub("]", "\\]")
|
|
s = s.gsub("<", "\\<")
|
|
s = s.gsub(">", "\\>")
|
|
s = s.gsub("\r", "")
|
|
s = s.gsub("\n", " \n")
|
|
s
|
|
end
|
|
|
|
def create_label(name)
|
|
color = nice_label_color(name)
|
|
project.labels.create!(name: name, color: color)
|
|
end
|
|
|
|
def format_content(raw_content)
|
|
linkify_issues(escape_for_markdown(raw_content))
|
|
end
|
|
|
|
def format_updates(raw_updates)
|
|
updates = []
|
|
|
|
if raw_updates.has_key?("status")
|
|
updates << "*Status: #{raw_updates["status"]}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("owner")
|
|
updates << "*Owner: #{mask_email(raw_updates["owner"])}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("cc")
|
|
cc = raw_updates["cc"].map do |l|
|
|
deleted = l.start_with?("-")
|
|
l = l[1..-1] if deleted
|
|
l = mask_email(l)
|
|
l = "~~#{l}~~" if deleted
|
|
l
|
|
end
|
|
|
|
updates << "*Cc: #{cc.join(", ")}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("labels")
|
|
labels = raw_updates["labels"].map do |l|
|
|
deleted = l.start_with?("-")
|
|
l = l[1..-1] if deleted
|
|
l = nice_label_name(l)
|
|
l = "~~#{l}~~" if deleted
|
|
l
|
|
end
|
|
|
|
updates << "*Labels: #{labels.join(", ")}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("mergedInto")
|
|
updates << "*Merged into: ##{raw_updates["mergedInto"]}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("blockedOn")
|
|
blocked_ons = raw_updates["blockedOn"].map do |raw_blocked_on|
|
|
name, id = raw_blocked_on.split(":", 2)
|
|
if name == project.import_source
|
|
"##{id}"
|
|
else
|
|
"#{project.namespace.path}/#{name}##{id}"
|
|
end
|
|
end
|
|
updates << "*Blocked on: #{blocked_ons.join(", ")}*"
|
|
end
|
|
|
|
if raw_updates.has_key?("blocking")
|
|
blockings = raw_updates["blocking"].map do |raw_blocked_on|
|
|
name, id = raw_blocked_on.split(":", 2)
|
|
if name == project.import_source
|
|
"##{id}"
|
|
else
|
|
"#{project.namespace.path}/#{name}##{id}"
|
|
end
|
|
end
|
|
updates << "*Blocking: #{blockings.join(", ")}*"
|
|
end
|
|
|
|
updates
|
|
end
|
|
|
|
def format_attachments(issue_id, comment_id, raw_attachments)
|
|
return [] unless raw_attachments
|
|
|
|
raw_attachments.map do |attachment|
|
|
next if attachment["isDeleted"]
|
|
|
|
link = "https://storage.googleapis.com/google-code-attachments/#{@repo.name}/issue-#{issue_id}/comment-#{comment_id}/#{attachment["fileName"]}"
|
|
"[#{attachment["fileName"]}](#{link})"
|
|
end.compact
|
|
end
|
|
end
|
|
end
|
|
end
|