mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-12 20:24:43 +08:00
259 lines
8 KiB
Ruby
259 lines
8 KiB
Ruby
# frozen_string_literal: true
|
||
|
||
module Reports
|
||
class HtmlToWordConverter
|
||
def initialize(document, options = {})
|
||
@docx = document
|
||
@scinote_url = options[:scinote_url]
|
||
@link_style = options[:link_style]
|
||
end
|
||
|
||
def html_to_word_converter(text)
|
||
html = Nokogiri::HTML(text)
|
||
raw_elements = recursive_children(html.css('body').children, []).compact
|
||
|
||
# Combined raw text blocks in paragraphs
|
||
elements = combine_docx_elements(raw_elements)
|
||
|
||
# Draw elements
|
||
elements.each do |elem|
|
||
if elem[:type] == 'p'
|
||
Reports::DocxRenderer.render_p_element(@docx, elem, scinote_url: @scinote_url, link_style: @link_style)
|
||
elsif elem[:type] == 'table'
|
||
Reports::DocxRenderer.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 h6).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::DocxRenderer.render_img_element(@docx, elem)
|
||
elsif %w(ul ol).include?(elem[:type])
|
||
Reports::DocxRenderer.render_list_element(@docx, elem)
|
||
end
|
||
end
|
||
end
|
||
|
||
private
|
||
|
||
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 = Reports::Utils.image_prepare(image).service_url
|
||
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)
|
||
allowed_elements = %w(li ul ol a img strong em h1 h2 h2 h3 h4 h5 span p)
|
||
data_array = list_element.children.select { |n| allowed_elements.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)
|
||
elsif %w(strong em h1 h2 h2 h3 h4 h5 span p).include?(item.name)
|
||
# Pass styles and extend renderer for li with style, some limitations on li items
|
||
# { type: 'text', value: item[:value], style: paragraph_styling(item) }
|
||
item.children.text
|
||
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' && Reports::Utils.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 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
|
||
end
|
||
end
|