mirror of
https://github.com/discourse/discourse.git
synced 2024-12-21 09:43:48 +08:00
43 lines
1.0 KiB
Ruby
43 lines
1.0 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
module Email
|
||
|
class Cleaner
|
||
|
def initialize(mail, remove_attachments: true, truncate: true, rejected: false)
|
||
|
@mail = Mail.new(mail)
|
||
|
@mail.charset = 'UTF-8'
|
||
|
@remove_attachments = remove_attachments
|
||
|
@truncate = truncate
|
||
|
@rejected = rejected
|
||
|
end
|
||
|
|
||
|
def execute
|
||
|
@mail.without_attachments! if @remove_attachments
|
||
|
truncate! if @truncate
|
||
|
remove_null_byte(@mail.to_s)
|
||
|
end
|
||
|
|
||
|
def self.delete_rejected!
|
||
|
IncomingEmail.delete_by('rejection_message IS NOT NULL AND created_at < ?', SiteSetting.delete_rejected_email_after_days.days.ago)
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def truncate!
|
||
|
parts.each { |part| part.body = part.body.decoded.truncate(truncate_limit, omission: '') }
|
||
|
end
|
||
|
|
||
|
def parts
|
||
|
@mail.multipart? ? @mail.parts : [@mail]
|
||
|
end
|
||
|
|
||
|
def truncate_limit
|
||
|
@rejected ? SiteSetting.raw_rejected_email_max_length : SiteSetting.raw_email_max_length
|
||
|
end
|
||
|
|
||
|
def remove_null_byte(message)
|
||
|
message.gsub!("\x00", "")
|
||
|
message
|
||
|
end
|
||
|
end
|
||
|
end
|