mirror of
https://github.com/discourse/discourse.git
synced 2024-11-27 02:03:44 +08:00
106 lines
2.5 KiB
Ruby
106 lines
2.5 KiB
Ruby
require_dependency 'pretty_text'
|
|
|
|
# A very simple formatter for imported emails
|
|
class EmailCook
|
|
|
|
def self.url_regexp
|
|
@url_regexp ||= /((?:https?:(?:\/{1,3}|[a-z0-9%])|www\d{0,3}[.])(?:[^\s()<>]+|\([^\s()<>]+\))+(?:\([^\s()<>]+\)|[^`!()\[\]{};:'".,<>?«»“”‘’\s]))/
|
|
end
|
|
|
|
def self.raw_regexp
|
|
@raw_regexp ||= /^\[plaintext\]$\n(.*)\n^\[\/plaintext\]$(?:\s^\[attachments\]$\n(.*)\n^\[\/attachments\]$)?(?:\s^\[elided\]$\n(.*)\n^\[\/elided\]$)?/m
|
|
end
|
|
|
|
def initialize(raw)
|
|
@raw = raw
|
|
@body, @attachment_html, @elided = @raw.scan(EmailCook.raw_regexp).first
|
|
end
|
|
|
|
def add_quote(result, buffer)
|
|
if buffer.present?
|
|
return if buffer =~ /\A(<br>)+\z$/
|
|
result << "<blockquote>#{buffer}</blockquote>"
|
|
end
|
|
end
|
|
|
|
def link_string!(line, unescaped_line)
|
|
unescaped_line = unescaped_line.strip
|
|
unescaped_line.scan(EmailCook.url_regexp).each do |m|
|
|
url = m[0]
|
|
|
|
if unescaped_line == url
|
|
# this could be oneboxed
|
|
val = %|<a href="#{url}" class="onebox" target="_blank">#{url}</a>|
|
|
else
|
|
val = %|<a href="#{url}">#{url}</a>|
|
|
end
|
|
|
|
line.gsub!(url, val)
|
|
end
|
|
end
|
|
|
|
def htmlify(text)
|
|
result = ""
|
|
|
|
in_text = false
|
|
in_quote = false
|
|
|
|
quote_buffer = ""
|
|
text.each_line do |line|
|
|
|
|
if line =~ /^\s*>/
|
|
in_quote = true
|
|
line.sub!(/^[\s>]*/, '')
|
|
|
|
unescaped_line = line
|
|
line = CGI.escapeHTML(line)
|
|
link_string!(line, unescaped_line)
|
|
|
|
quote_buffer << line << "<br>"
|
|
elsif in_quote
|
|
add_quote(result, quote_buffer)
|
|
quote_buffer = ""
|
|
in_quote = false
|
|
else
|
|
|
|
sz = line.size
|
|
|
|
unescaped_line = line
|
|
line = CGI.escapeHTML(line)
|
|
link_string!(line, unescaped_line)
|
|
|
|
if sz < 60
|
|
if in_text && line == "\n"
|
|
result << "<br>"
|
|
end
|
|
result << line
|
|
result << "<br>"
|
|
|
|
in_text = false
|
|
else
|
|
result << line
|
|
in_text = true
|
|
end
|
|
end
|
|
end
|
|
|
|
if in_quote && quote_buffer.present?
|
|
add_quote(result, quote_buffer)
|
|
end
|
|
|
|
result.gsub!(/(<br>\n*){3,10}/, '<br><br>')
|
|
result
|
|
end
|
|
|
|
def cook(opts = {})
|
|
# fallback to PrettyText if we failed to detect a body
|
|
return PrettyText.cook(@raw, opts) if @body.nil?
|
|
|
|
result = htmlify(@body)
|
|
result << "\n<br>" << @attachment_html if @attachment_html.present?
|
|
result << "\n<br><br>" << Email::Receiver.elided_html(htmlify(@elided)) if @elided.present?
|
|
result
|
|
end
|
|
|
|
end
|