mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-09-20 23:16:15 +08:00
Move TinyMCE parser to separate class
This commit is contained in:
Urban Rotnik
parent
ae8b2cbf8f
commit
1b860afe30
|
|
@ -3,245 +3,6 @@
|
|||
module Reports::Docx::PrivateMethods
|
||||
private
|
||||
|
||||
# RTE fields support
|
||||
def html_to_word_converter(text)
|
||||
html = Nokogiri::HTML(text)
|
||||
raw_elements = recursive_children(html.css('body').children, [])
|
||||
|
||||
# Combined raw text blocks in paragraphs
|
||||
elements = combine_docx_elements(raw_elements)
|
||||
|
||||
# Draw elements
|
||||
elements.each do |elem|
|
||||
if elem[:type] == 'p'
|
||||
Reports::Docx.render_p_element(@docx, elem, scinote_url: @scinote_url, link_style: @link_style)
|
||||
elsif elem[:type] == 'table'
|
||||
Reports::Docx.render_table_element(@docx, elem)
|
||||
elsif elem[:type] == 'newline'
|
||||
style = elem[:style] || {}
|
||||
# print heading if its heading
|
||||
# Mixing heading with other style setting causes problems for Word
|
||||
if %w(h1 h2 h3 h4 h5).include?(style[:style])
|
||||
@docx.public_send(style[:style], elem[:value])
|
||||
else
|
||||
@docx.p elem[:value] do
|
||||
align style[:align]
|
||||
color style[:color]
|
||||
bold style[:bold]
|
||||
italic style[:italic]
|
||||
end
|
||||
end
|
||||
elsif elem[:type] == 'image'
|
||||
Reports::Docx.render_img_element(@docx, elem)
|
||||
elsif %w(ul ol).include?(elem[:type])
|
||||
Reports::Docx.render_list_element(@docx, elem)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def combine_docx_elements(raw_elements)
|
||||
# Word does not support some nested elements, move some elements to root level
|
||||
elements = []
|
||||
temp_p = []
|
||||
raw_elements.each do |elem|
|
||||
if %w(image newline table ol ul).include? elem[:type]
|
||||
unless temp_p.empty?
|
||||
elements.push(type: 'p', children: temp_p)
|
||||
temp_p = []
|
||||
end
|
||||
elements.push(elem)
|
||||
elsif %w(br text a).include? elem[:type]
|
||||
temp_p.push(elem)
|
||||
end
|
||||
end
|
||||
elements.push(type: 'p', children: temp_p)
|
||||
elements
|
||||
end
|
||||
|
||||
# Convert HTML structure to plain text structure
|
||||
def recursive_children(children, elements)
|
||||
children.each do |elem|
|
||||
if elem.class == Nokogiri::XML::Text
|
||||
next if elem.text.strip == ' ' # Invisible symbol
|
||||
|
||||
style = paragraph_styling(elem.parent)
|
||||
type = (style[:align] && style[:align] != :justify) || style[:style] ? 'newline' : 'text'
|
||||
|
||||
text = smart_annotation_check(elem)
|
||||
|
||||
elements.push(
|
||||
type: type,
|
||||
value: text.strip.delete(' '), # Invisible symbol
|
||||
style: style
|
||||
)
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'br'
|
||||
elements.push(type: 'br')
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'img'
|
||||
elements.push(img_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'a'
|
||||
elements.push(link_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'table'
|
||||
elements.push(tiny_mce_table_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if %w(ul ol).include?(elem.name)
|
||||
elements.push(list_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
elements = recursive_children(elem.children, elements) if elem.children
|
||||
end
|
||||
elements
|
||||
end
|
||||
|
||||
def img_element(elem)
|
||||
return unless elem.attributes['data-mce-token']
|
||||
|
||||
image = TinyMceAsset.find_by(id: Base62.decode(elem.attributes['data-mce-token'].value))
|
||||
return unless image
|
||||
|
||||
image_path = image_path(image.image)
|
||||
dimension = FastImage.size(image_path)
|
||||
|
||||
return unless dimension
|
||||
|
||||
style = image_styling(elem, dimension)
|
||||
|
||||
{ type: 'image', data: image_path.split('&')[0], blob: image.blob, style: style }
|
||||
end
|
||||
|
||||
def link_element(elem)
|
||||
text = elem.text
|
||||
link = elem.attributes['href'].value if elem.attributes['href']
|
||||
if elem.attributes['class']&.value == 'record-info-link'
|
||||
link = nil
|
||||
text = "##{text}"
|
||||
end
|
||||
text = "##{text}" if elem.parent.attributes['class']&.value == 'atwho-inserted'
|
||||
text = "@#{text}" if elem.attributes['class']&.value == 'atwho-user-popover'
|
||||
{
|
||||
type: 'a',
|
||||
value: text,
|
||||
link: link
|
||||
}
|
||||
end
|
||||
|
||||
def list_element(list_element)
|
||||
data_array = list_element.children.select { |n| %w(li ul ol a img).include?(n.name) }.map do |li_child|
|
||||
li_child.children.map do |item|
|
||||
if item.is_a? Nokogiri::XML::Text
|
||||
item.text.chomp
|
||||
elsif %w(ul ol).include?(item.name)
|
||||
list_element(item)
|
||||
elsif %w(a).include?(item.name)
|
||||
link_element(item)
|
||||
elsif %w(img).include?(item.name)
|
||||
img_element(item).merge(bookmark_id: SecureRandom.hex)
|
||||
elsif %w(table).include?(item.name)
|
||||
tiny_mce_table_element(item).merge(bookmark_id: SecureRandom.hex)
|
||||
end
|
||||
end.reject(&:blank?)
|
||||
end
|
||||
{ type: list_element.name, data: data_array }
|
||||
end
|
||||
|
||||
def smart_annotation_check(elem)
|
||||
return "[#{elem.text}]" if elem.parent.attributes['class']&.value == 'sa-type'
|
||||
|
||||
elem.text
|
||||
end
|
||||
|
||||
# Prepare style for text
|
||||
def paragraph_styling(elem)
|
||||
style = elem.attributes['style']
|
||||
result = {}
|
||||
result[:style] = elem.name if elem.name.include? 'h'
|
||||
result[:bold] = true if elem.name == 'strong'
|
||||
result[:italic] = true if elem.name == 'em'
|
||||
style_keys = %w(text-align color)
|
||||
|
||||
if style
|
||||
style_keys.each do |key|
|
||||
style_el = style.value.split(';').select { |i| (i.include? key) }[0]
|
||||
next unless style_el
|
||||
|
||||
value = style_el.split(':')[1].strip if style_el
|
||||
if key == 'text-align'
|
||||
result[:align] = value.to_sym
|
||||
elsif key == 'color' && calculate_color_hsp(value) < 190
|
||||
result[:color] = value.delete('#')
|
||||
end
|
||||
end
|
||||
end
|
||||
result
|
||||
end
|
||||
|
||||
# Prepare style for images
|
||||
def image_styling(elem, dimension)
|
||||
dimension[0] = elem.attributes['width'].value.to_i if elem.attributes['width']
|
||||
dimension[1] = elem.attributes['height'].value.to_i if elem.attributes['height']
|
||||
|
||||
if elem.attributes['style']
|
||||
align = if elem.attributes['style'].value.include? 'margin-right'
|
||||
:center
|
||||
elsif elem.attributes['style'].value.include? 'float: right'
|
||||
:right
|
||||
else
|
||||
:left
|
||||
end
|
||||
end
|
||||
|
||||
margins = Constants::REPORT_DOCX_MARGIN_LEFT + Constants::REPORT_DOCX_MARGIN_RIGHT
|
||||
max_width = (Constants::REPORT_DOCX_WIDTH - margins) / 20
|
||||
|
||||
if dimension[0] > max_width
|
||||
x = max_width
|
||||
y = dimension[1] * max_width / dimension[0]
|
||||
else
|
||||
x = dimension[0]
|
||||
y = dimension[1]
|
||||
end
|
||||
|
||||
{
|
||||
width: x,
|
||||
height: y,
|
||||
align: align,
|
||||
max_width: max_width
|
||||
}
|
||||
end
|
||||
|
||||
def asset_image_preparing(asset)
|
||||
return unless asset
|
||||
|
||||
image_path = image_path(asset.file)
|
||||
|
||||
dimension = FastImage.size(image_path)
|
||||
x = dimension[0]
|
||||
y = dimension[1]
|
||||
if x > 300
|
||||
y = y * 300 / x
|
||||
x = 300
|
||||
end
|
||||
@docx.img image_path.split('&')[0] do
|
||||
data asset.blob.download
|
||||
width x
|
||||
height y
|
||||
end
|
||||
end
|
||||
|
||||
def initial_document_load
|
||||
@docx.page_size do
|
||||
width Constants::REPORT_DOCX_WIDTH
|
||||
|
|
@ -292,40 +53,4 @@ module Reports::Docx::PrivateMethods
|
|||
green: '2dbe61'
|
||||
}
|
||||
end
|
||||
|
||||
def tiny_mce_table_element(table_element)
|
||||
# array of elements
|
||||
rows = table_element.css('tbody').first.children.map do |row|
|
||||
next unless row.name == 'tr'
|
||||
|
||||
cells = row.children.map do |cell|
|
||||
next unless cell.name == 'td'
|
||||
|
||||
# Parse cell content
|
||||
formated_cell = recursive_children(cell.children, [])
|
||||
|
||||
# Combine text elements to single paragraph
|
||||
formated_cell = combine_docx_elements(formated_cell)
|
||||
formated_cell
|
||||
end.reject(&:blank?)
|
||||
{ type: 'tr', data: cells }
|
||||
end.reject(&:blank?)
|
||||
{ type: 'table', data: rows }
|
||||
end
|
||||
|
||||
def image_path(attachment)
|
||||
attachment.service_url
|
||||
end
|
||||
|
||||
def calculate_color_hsp(color)
|
||||
return 255 if color.length != 7
|
||||
|
||||
color = color.delete('#').scan(/.{1,2}/)
|
||||
rgb = color.map(&:hex)
|
||||
Math.sqrt(
|
||||
0.299 * (rgb[0]**2) +
|
||||
0.587 * (rgb[1]**2) +
|
||||
0.114 * (rgb[2]**2)
|
||||
)
|
||||
end
|
||||
end
|
||||
|
|
|
|||
281
app/services/reports/docx/tiny_mce_parser.rb
Normal file
281
app/services/reports/docx/tiny_mce_parser.rb
Normal file
|
|
@ -0,0 +1,281 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
module Reports::Docx::TinyMceParser
|
||||
private
|
||||
|
||||
def html_to_word_converter(text)
|
||||
html = Nokogiri::HTML(text)
|
||||
raw_elements = recursive_children(html.css('body').children, [])
|
||||
|
||||
# Combined raw text blocks in paragraphs
|
||||
elements = combine_docx_elements(raw_elements)
|
||||
|
||||
# Draw elements
|
||||
elements.each do |elem|
|
||||
if elem[:type] == 'p'
|
||||
Reports::Docx.render_p_element(@docx, elem, scinote_url: @scinote_url, link_style: @link_style)
|
||||
elsif elem[:type] == 'table'
|
||||
Reports::Docx.render_table_element(@docx, elem)
|
||||
elsif elem[:type] == 'newline'
|
||||
style = elem[:style] || {}
|
||||
# print heading if its heading
|
||||
# Mixing heading with other style setting causes problems for Word
|
||||
if %w(h1 h2 h3 h4 h5).include?(style[:style])
|
||||
@docx.public_send(style[:style], elem[:value])
|
||||
else
|
||||
@docx.p elem[:value] do
|
||||
align style[:align]
|
||||
color style[:color]
|
||||
bold style[:bold]
|
||||
italic style[:italic]
|
||||
end
|
||||
end
|
||||
elsif elem[:type] == 'image'
|
||||
Reports::Docx.render_img_element(@docx, elem)
|
||||
elsif %w(ul ol).include?(elem[:type])
|
||||
Reports::Docx.render_list_element(@docx, elem)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def combine_docx_elements(raw_elements)
|
||||
# Word does not support some nested elements, move some elements to root level
|
||||
elements = []
|
||||
temp_p = []
|
||||
raw_elements.each do |elem|
|
||||
if %w(image newline table ol ul).include? elem[:type]
|
||||
unless temp_p.empty?
|
||||
elements.push(type: 'p', children: temp_p)
|
||||
temp_p = []
|
||||
end
|
||||
elements.push(elem)
|
||||
elsif %w(br text a).include? elem[:type]
|
||||
temp_p.push(elem)
|
||||
end
|
||||
end
|
||||
elements.push(type: 'p', children: temp_p)
|
||||
elements
|
||||
end
|
||||
|
||||
# Convert HTML structure to plain text structure
|
||||
# rubocop:disable Metrics/BlockLength
|
||||
def recursive_children(children, elements)
|
||||
children.each do |elem|
|
||||
if elem.class == Nokogiri::XML::Text
|
||||
next if elem.text.strip == ' ' # Invisible symbol
|
||||
|
||||
style = paragraph_styling(elem.parent)
|
||||
type = (style[:align] && style[:align] != :justify) || style[:style] ? 'newline' : 'text'
|
||||
|
||||
text = smart_annotation_check(elem)
|
||||
|
||||
elements.push(
|
||||
type: type,
|
||||
value: text.strip.delete(' '), # Invisible symbol
|
||||
style: style
|
||||
)
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'br'
|
||||
elements.push(type: 'br')
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'img'
|
||||
elements.push(img_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'a'
|
||||
elements.push(link_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if elem.name == 'table'
|
||||
elements.push(tiny_mce_table_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
if %w(ul ol).include?(elem.name)
|
||||
elements.push(list_element(elem))
|
||||
next
|
||||
end
|
||||
|
||||
elements = recursive_children(elem.children, elements) if elem.children
|
||||
end
|
||||
elements
|
||||
end
|
||||
# rubocop:enable Metrics/BlockLength
|
||||
|
||||
def img_element(elem)
|
||||
return unless elem.attributes['data-mce-token']
|
||||
|
||||
image = TinyMceAsset.find_by(id: Base62.decode(elem.attributes['data-mce-token'].value))
|
||||
return unless image
|
||||
|
||||
image_path = image_path(image.image)
|
||||
dimension = FastImage.size(image_path)
|
||||
|
||||
return unless dimension
|
||||
|
||||
style = image_styling(elem, dimension)
|
||||
|
||||
{ type: 'image', data: image_path.split('&')[0], blob: image.blob, style: style }
|
||||
end
|
||||
|
||||
def link_element(elem)
|
||||
text = elem.text
|
||||
link = elem.attributes['href'].value if elem.attributes['href']
|
||||
if elem.attributes['class']&.value == 'record-info-link'
|
||||
link = nil
|
||||
text = "##{text}"
|
||||
end
|
||||
text = "##{text}" if elem.parent.attributes['class']&.value == 'atwho-inserted'
|
||||
text = "@#{text}" if elem.attributes['class']&.value == 'atwho-user-popover'
|
||||
{
|
||||
type: 'a',
|
||||
value: text,
|
||||
link: link
|
||||
}
|
||||
end
|
||||
|
||||
def list_element(list_element)
|
||||
data_array = list_element.children.select { |n| %w(li ul ol a img).include?(n.name) }.map do |li_child|
|
||||
li_child.children.map do |item|
|
||||
if item.is_a? Nokogiri::XML::Text
|
||||
item.text.chomp
|
||||
elsif %w(ul ol).include?(item.name)
|
||||
list_element(item)
|
||||
elsif %w(a).include?(item.name)
|
||||
link_element(item)
|
||||
elsif %w(img).include?(item.name)
|
||||
img_element(item).merge(bookmark_id: SecureRandom.hex)
|
||||
elsif %w(table).include?(item.name)
|
||||
tiny_mce_table_element(item).merge(bookmark_id: SecureRandom.hex)
|
||||
end
|
||||
end.reject(&:blank?)
|
||||
end
|
||||
{ type: list_element.name, data: data_array }
|
||||
end
|
||||
|
||||
def smart_annotation_check(elem)
|
||||
return "[#{elem.text}]" if elem.parent.attributes['class']&.value == 'sa-type'
|
||||
|
||||
elem.text
|
||||
end
|
||||
|
||||
# Prepare style for text
|
||||
def paragraph_styling(elem)
|
||||
style = elem.attributes['style']
|
||||
result = {}
|
||||
result[:style] = elem.name if elem.name.include? 'h'
|
||||
result[:bold] = true if elem.name == 'strong'
|
||||
result[:italic] = true if elem.name == 'em'
|
||||
style_keys = %w(text-align color)
|
||||
|
||||
if style
|
||||
style_keys.each do |key|
|
||||
style_el = style.value.split(';').select { |i| (i.include? key) }[0]
|
||||
next unless style_el
|
||||
|
||||
value = style_el.split(':')[1].strip if style_el
|
||||
if key == 'text-align'
|
||||
result[:align] = value.to_sym
|
||||
elsif key == 'color' && calculate_color_hsp(value) < 190
|
||||
result[:color] = value.delete('#')
|
||||
end
|
||||
end
|
||||
end
|
||||
result
|
||||
end
|
||||
|
||||
# Prepare style for images
|
||||
def image_styling(elem, dimension)
|
||||
dimension[0] = elem.attributes['width'].value.to_i if elem.attributes['width']
|
||||
dimension[1] = elem.attributes['height'].value.to_i if elem.attributes['height']
|
||||
|
||||
if elem.attributes['style']
|
||||
align = if elem.attributes['style'].value.include? 'margin-right'
|
||||
:center
|
||||
elsif elem.attributes['style'].value.include? 'float: right'
|
||||
:right
|
||||
else
|
||||
:left
|
||||
end
|
||||
end
|
||||
|
||||
margins = Constants::REPORT_DOCX_MARGIN_LEFT + Constants::REPORT_DOCX_MARGIN_RIGHT
|
||||
max_width = (Constants::REPORT_DOCX_WIDTH - margins) / 20
|
||||
|
||||
if dimension[0] > max_width
|
||||
x = max_width
|
||||
y = dimension[1] * max_width / dimension[0]
|
||||
else
|
||||
x = dimension[0]
|
||||
y = dimension[1]
|
||||
end
|
||||
|
||||
{
|
||||
width: x,
|
||||
height: y,
|
||||
align: align,
|
||||
max_width: max_width
|
||||
}
|
||||
end
|
||||
|
||||
def asset_image_preparing(asset)
|
||||
return unless asset
|
||||
|
||||
image_path = image_path(asset.file)
|
||||
|
||||
dimension = FastImage.size(image_path)
|
||||
x = dimension[0]
|
||||
y = dimension[1]
|
||||
if x > 300
|
||||
y = y * 300 / x
|
||||
x = 300
|
||||
end
|
||||
@docx.img image_path.split('&')[0] do
|
||||
data asset.blob.download
|
||||
width x
|
||||
height y
|
||||
end
|
||||
end
|
||||
|
||||
def tiny_mce_table_element(table_element)
|
||||
# array of elements
|
||||
rows = table_element.css('tbody').first.children.map do |row|
|
||||
next unless row.name == 'tr'
|
||||
|
||||
cells = row.children.map do |cell|
|
||||
next unless cell.name == 'td'
|
||||
|
||||
# Parse cell content
|
||||
formated_cell = recursive_children(cell.children, [])
|
||||
|
||||
# Combine text elements to single paragraph
|
||||
formated_cell = combine_docx_elements(formated_cell)
|
||||
formated_cell
|
||||
end.reject(&:blank?)
|
||||
{ type: 'tr', data: cells }
|
||||
end.reject(&:blank?)
|
||||
{ type: 'table', data: rows }
|
||||
end
|
||||
|
||||
def image_path(attachment)
|
||||
attachment.service_url
|
||||
end
|
||||
|
||||
def calculate_color_hsp(color)
|
||||
return 255 if color.length != 7
|
||||
|
||||
color = color.delete('#').scan(/.{1,2}/)
|
||||
rgb = color.map(&:hex)
|
||||
Math.sqrt(
|
||||
0.299 * (rgb[0]**2) +
|
||||
0.587 * (rgb[1]**2) +
|
||||
0.114 * (rgb[2]**2)
|
||||
)
|
||||
end
|
||||
end
|
||||
Loading…
Reference in a new issue