lib/openxml/docx/package.rb
require "openxml/package"
require "open-uri"
module OpenXml
module Docx
class Package < OpenXml::Package
attr_reader :document,
:settings,
:headers,
:footers,
:styles,
:fonts,
:numbering,
:image_names
content_types do
default "xml", TYPE_XML
default "odttf", TYPE_OBSCURED_FONT
default "jpeg", TYPE_IMAGE[:jpeg]
default "png", TYPE_IMAGE[:png]
default "gif", TYPE_IMAGE[:gif]
default "bmp", TYPE_IMAGE[:bmp]
default "tiff", TYPE_IMAGE[:tiff]
override "/word/styles.xml", TYPE_STYLES
override "/word/settings.xml", TYPE_SETTINGS
override "/word/fontTable.xml", TYPE_FONT_TABLE
override "/word/numbering.xml", TYPE_NUMBERING
end
def initialize
super
rels.add_relationship REL_DOCUMENT, "/word/document.xml"
@settings = OpenXml::Docx::Parts::Settings.new
@styles = OpenXml::Docx::Parts::Styles.new
@fonts = OpenXml::Docx::Parts::Fonts.new
@numbering = OpenXml::Docx::Parts::Numbering.new
@document = OpenXml::Docx::Parts::Document.new
@headers = []
@footers = []
@image_names = []
document.relationships.add_relationship REL_STYLES, "styles.xml"
document.relationships.add_relationship REL_SETTINGS, "settings.xml"
document.relationships.add_relationship REL_FONT_TABLE, "fontTable.xml"
document.relationships.add_relationship REL_NUMBERING, "numbering.xml"
add_part "word/_rels/document.xml.rels", document.relationships
add_part "word/_rels/fontTable.xml.rels", fonts.relationships
add_part "word/document.xml", document
add_part "word/settings.xml", settings
add_part "word/styles.xml", styles
add_part "word/fontTable.xml", fonts
add_part "word/numbering.xml", numbering
end
def embed_truetype_font(path: nil, name: nil)
URI.open(path, "rb") do |source_font|
obfuscation_data = obfuscate_font source_font
data = obfuscation_data[:bytes] << source_font.read
destination_font_name = "font#{fonts.fonts.count + 1}.odttf"
add_part "word/fonts/#{destination_font_name}", OpenXml::Parts::UnparsedPart.new(data)
font_relationship = fonts.relationships.add_relationship REL_FONT, "fonts/#{destination_font_name}"
font_description = OpenXml::Docx::Elements::Font.new
font_description.font_name = name
embed_tag = OpenXml::Docx::Elements::EmbedRegular.new
embed_tag.font_key = "{#{obfuscation_data[:key]}}"
embed_tag.relationship_id = font_relationship.id
font_description << embed_tag
fonts << font_description
end
end
def embed_image(path: nil, content_type: nil, into_part: nil)
return if path.nil?
extension_match = path.match(/\.(?<extension>[^\.]+?)(?:\?.+)?$/)
content_type ||= extension_match[:extension] if extension_match
return if content_type.nil?
URI.open(path, "rb") do |source_image|
embed_image_data(data: source_image.read, content_type: content_type, into_part: into_part)
end
end
def embed_image_data(data: nil, content_type: nil, into_part: nil)
return if data.nil? || content_type.nil?
into_part = document unless into_part.respond_to?(:relationships)
content_type = "jpeg" if content_type == "jpg"
content_type = content_type.to_sym
destination_image_name = "image#{image_names.count + 1}.#{content_type}"
add_part "word/media/#{destination_image_name}", OpenXml::Parts::UnparsedPart.new(data)
image_names << destination_image_name
image_relationship = into_part.relationships.add_relationship REL_IMAGE, "media/#{destination_image_name}"
image_relationship.id
end
def add_header(header)
headers << header
header_name = "header#{headers.count}.xml"
content_types.add_override "/word/#{header_name}", TYPE_HEADER
add_part "word/#{header_name}", header
add_part "word/_rels/#{header_name}.rels", header.relationships
relationship = document.relationships.add_relationship REL_HEADER, header_name
relationship.id
end
def add_footer(footer)
footers << footer
footer_name = "footer#{footers.count}.xml"
content_types.add_override "/word/#{footer_name}", TYPE_FOOTER
add_part "word/#{footer_name}", footer
add_part "word/_rels/#{footer_name}.rels", footer.relationships
relationship = document.relationships.add_relationship REL_FOOTER, footer_name
relationship.id
end
private
def obfuscate_font(font)
# From the OpenXml spec, section 17.8.1, the algorithm for obfuscating a font:
# - Generate a GUID, which is used and stored as the obfuscation key
# - Reverse the order of the bytes in the GUID (i.e. Big Endian ordering)
# - XOR the value with the first 32 bytes of the binary: once against 0-15, once against 16-31
# - Store the resulting file in the document, and store the obfuscation key in the fontKey attribute
key = SecureRandom::uuid.upcase # Spec requires hex characters be uppercase
raw_key = key.gsub("-", "")
big_endian_key = [raw_key].pack("H*").bytes.reverse
obfuscated_bytes = []
2.times do
bytes = font.read(16).bytes
(0...16).each do |index|
obfuscated_bytes << (bytes[index] ^ big_endian_key[index])
end
end
{ key: key, bytes: obfuscated_bytes.pack("C*") }
end
end
end
end