debian-mirror-gitlab/lib/gitlab/email/handler/reply_processing.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

129 lines
3.8 KiB
Ruby
Raw Normal View History

2018-11-20 20:47:30 +05:30
# frozen_string_literal: true
2017-08-17 22:00:37 +05:30
module Gitlab
module Email
module Handler
module ReplyProcessing
2019-02-15 15:39:39 +05:30
# rubocop:disable Gitlab/ModuleWithInstanceVariables
2017-08-17 22:00:37 +05:30
def project
2019-02-15 15:39:39 +05:30
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
2017-08-17 22:00:37 +05:30
end
2019-02-15 15:39:39 +05:30
# rubocop:enable Gitlab/ModuleWithInstanceVariables
2017-08-17 22:00:37 +05:30
2021-09-30 23:02:18 +05:30
private
attr_reader :project_id, :project_slug, :project_path, :incoming_email_token
def author
raise NotImplementedError
end
2017-08-17 22:00:37 +05:30
def message
@message ||= process_message
end
2018-11-08 19:23:39 +05:30
def message_including_reply
@message_with_reply ||= process_message(trim_reply: false)
end
2022-01-26 12:08:38 +05:30
def message_including_reply_or_only_quotes
@message_including_reply_or_only_quotes ||= process_message(trim_reply: false, allow_only_quotes: true)
end
2021-10-27 15:23:28 +05:30
def message_with_appended_reply
@message_with_appended_reply ||= process_message(append_reply: true)
end
2018-11-08 19:23:39 +05:30
def process_message(**kwargs)
2021-10-27 15:23:28 +05:30
message, stripped_text = ReplyParser.new(mail, **kwargs).execute
message = message.strip
2020-07-28 23:09:34 +05:30
message_with_attachments = add_attachments(message)
2021-10-27 15:23:28 +05:30
# Support bot is specifically forbidden from using slash commands.
message = strip_quick_actions(message_with_attachments)
return message unless kwargs[:append_reply]
2020-07-28 23:09:34 +05:30
2021-10-27 15:23:28 +05:30
append_reply(message, stripped_text)
2017-08-17 22:00:37 +05:30
end
def add_attachments(reply)
2021-02-22 17:27:13 +05:30
attachments = Email::AttachmentUploader.new(mail).execute(**upload_params)
2017-08-17 22:00:37 +05:30
reply + attachments.map do |link|
"\n\n#{link[:markdown]}"
end.join
end
2020-03-13 15:44:24 +05:30
def upload_params
{
upload_parent: project,
uploader_class: FileUploader
}
end
2017-08-17 22:00:37 +05:30
def validate_permission!(permission)
raise UserNotFoundError unless author
raise UserBlockedError if author.blocked?
2018-10-15 14:42:47 +05:30
if project
raise ProjectNotFound unless author.can?(:read_project, project)
end
2019-02-02 18:00:53 +05:30
raise UserNotAuthorizedError unless author.can?(permission, try(:noteable) || project)
2017-08-17 22:00:37 +05:30
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
2019-02-15 15:39:39 +05:30
def valid_project_slug?(found_project)
2021-09-04 01:27:46 +05:30
return false unless found_project
2019-02-15 15:39:39 +05:30
project_slug == found_project.full_path_slug
end
2020-07-28 23:09:34 +05:30
def strip_quick_actions(content)
return content unless author.support_bot?
2021-10-27 15:23:28 +05:30
quick_actions_extractor.redact_commands(content)
end
def quick_actions_extractor
2020-07-28 23:09:34 +05:30
command_definitions = ::QuickActions::InterpretService.command_definitions
2021-10-27 15:23:28 +05:30
::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?
2020-07-28 23:09:34 +05:30
2021-10-27 15:23:28 +05:30
message + "\n\n<details><summary>...</summary>\n\n#{reply}\n\n</details>"
2020-07-28 23:09:34 +05:30
end
2017-08-17 22:00:37 +05:30
end
end
end
end
2020-03-13 15:44:24 +05:30
2021-06-08 01:23:25 +05:30
Gitlab::Email::Handler::ReplyProcessing.prepend_mod_with('Gitlab::Email::Handler::ReplyProcessing')