129 lines
3.8 KiB
Ruby
129 lines
3.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Email
|
|
module Handler
|
|
module ReplyProcessing
|
|
# rubocop:disable Gitlab/ModuleWithInstanceVariables
|
|
def project
|
|
return @project if instance_variable_defined?(:@project)
|
|
|
|
if project_id
|
|
@project = Project.find_by_id(project_id)
|
|
@project = nil unless valid_project_slug?(@project)
|
|
else
|
|
@project = Project.find_by_full_path(project_path)
|
|
end
|
|
|
|
@project
|
|
end
|
|
# rubocop:enable Gitlab/ModuleWithInstanceVariables
|
|
|
|
private
|
|
|
|
attr_reader :project_id, :project_slug, :project_path, :incoming_email_token
|
|
|
|
def author
|
|
raise NotImplementedError
|
|
end
|
|
|
|
def message
|
|
@message ||= process_message
|
|
end
|
|
|
|
def message_including_reply
|
|
@message_with_reply ||= process_message(trim_reply: false)
|
|
end
|
|
|
|
def message_including_reply_or_only_quotes
|
|
@message_including_reply_or_only_quotes ||= process_message(trim_reply: false, allow_only_quotes: true)
|
|
end
|
|
|
|
def message_with_appended_reply
|
|
@message_with_appended_reply ||= process_message(append_reply: true)
|
|
end
|
|
|
|
def process_message(**kwargs)
|
|
message, stripped_text = ReplyParser.new(mail, **kwargs).execute
|
|
message = message.strip
|
|
|
|
message_with_attachments = add_attachments(message)
|
|
# Support bot is specifically forbidden from using slash commands.
|
|
message = strip_quick_actions(message_with_attachments)
|
|
return message unless kwargs[:append_reply]
|
|
|
|
append_reply(message, stripped_text)
|
|
end
|
|
|
|
def add_attachments(reply)
|
|
attachments = Email::AttachmentUploader.new(mail).execute(**upload_params)
|
|
|
|
reply + attachments.map do |link|
|
|
"\n\n#{link[:markdown]}"
|
|
end.join
|
|
end
|
|
|
|
def upload_params
|
|
{
|
|
upload_parent: project,
|
|
uploader_class: FileUploader
|
|
}
|
|
end
|
|
|
|
def validate_permission!(permission)
|
|
raise UserNotFoundError unless author
|
|
raise UserBlockedError if author.blocked?
|
|
|
|
if project
|
|
raise ProjectNotFound unless author.can?(:read_project, project)
|
|
end
|
|
|
|
raise UserNotAuthorizedError unless author.can?(permission, try(:noteable) || project)
|
|
end
|
|
|
|
def verify_record!(record:, invalid_exception:, record_name:)
|
|
return if record.persisted?
|
|
return if record.errors.key?(:commands_only)
|
|
|
|
error_title = "The #{record_name} could not be created for the following reasons:"
|
|
|
|
msg = error_title + record.errors.full_messages.map do |error|
|
|
"\n\n- #{error}"
|
|
end.join
|
|
|
|
raise invalid_exception, msg
|
|
end
|
|
|
|
def valid_project_slug?(found_project)
|
|
return false unless found_project
|
|
|
|
project_slug == found_project.full_path_slug
|
|
end
|
|
|
|
def strip_quick_actions(content)
|
|
return content unless author.support_bot?
|
|
|
|
quick_actions_extractor.redact_commands(content)
|
|
end
|
|
|
|
def quick_actions_extractor
|
|
command_definitions = ::QuickActions::InterpretService.command_definitions
|
|
::Gitlab::QuickActions::Extractor.new(command_definitions)
|
|
end
|
|
|
|
def append_reply(message, reply)
|
|
return message if message.blank? || reply.blank?
|
|
|
|
# Do not append if message only contains slash commands
|
|
body, _commands = quick_actions_extractor.extract_commands(message)
|
|
return message if body.empty?
|
|
|
|
message + "\n\n<details><summary>...</summary>\n\n#{reply}\n\n</details>"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
Gitlab::Email::Handler::ReplyProcessing.prepend_mod_with('Gitlab::Email::Handler::ReplyProcessing')
|