mrtexport 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d20d61ba920b60a63f93e98d6e2fd3d8a89a4414
+  data.tar.gz: 219915b00d28a897167ffaa34d23530049525a2f
+SHA512:
+  metadata.gz: 0bd1c6eca3cceb8e8ff5ae47a6dcb1b79fd5a96229d6c662df4af3d3ff974aae20a6dd52032ee8b83b861386f873926a1fa2b9dccf599ee1920d3584d8416da3
+  data.tar.gz: 8214d9e736f2b4f18e487b117942b8fa32b3e0a6d42f29ecd5b3de8357b7d457b3987031958cfd69d17df5bae7f6029349ae9abe01bc200e26cd8bdce9ed4311

data/lib/data_band.rb

@@ -0,0 +1,74 @@
+require_relative "util.rb"
+
+class DataBandRenderer
+  def initialize(data, pdf, sql_conn, replacements, data_sources, y_off)
+    @data = data
+    @pdf = pdf
+    @sql_conn = sql_conn
+    @data_sources = data_sources
+    @replacements = replacements
+    @y_off = y_off
+
+    @row_data = prepare_dataset
+  end
+
+  def render
+    @row_data.each do |row|
+      draw_row(row)
+    end
+
+    return @y_off
+  end
+
+  def prepare_dataset
+    data_source = @data.xpath("DataSourceName").text
+    sql = @data_sources[data_source]["sql"]
+    
+    @replacements.each do |key, val|
+      sql.sub!("{#{key}}", val)
+    end
+    
+    return @sql_conn.query sql
+  end
+
+  def draw_row(row)
+    last_y_off = @y_off
+
+    @data.xpath("Components/*[contains(.,'Text')]").each do |node|
+      next if node.name == "Name"
+
+      if node.name.start_with? "Text"
+        x, y, w, h = Stylist.get_dimensions(node, @y_off)
+
+        # Text
+        text = node.xpath("Text").text
+
+        # SQL replacements
+        if text.include?("{")
+          text.gsub!(/{.*}/) do |m|
+            index = m.split(".")[1].sub("}","")
+            text = row[index]
+            text = Util.number_format(text) if Util.is_number?(text)
+            text
+          end
+        end
+
+        # Do painting
+        @pdf.fill_color Stylist.get_text_colour(node)
+        @pdf.text_box text,
+                  :at     => [x + 1, y - 1],
+                  :height => h,
+                  :width  => w,
+                  :align  => Stylist.get_horizontal_align(node),
+                  :valign => Stylist.get_vertical_align(node),
+                  :size   => Stylist.get_font_size(node),
+                  :font   => Stylist.get_font_face(node),
+                  :style  => Stylist.get_font_style(node)
+
+        last_y_off = (720 + h - y) if (720 + h - y) > @y_off
+      end
+    end
+
+    @y_off = last_y_off
+  end
+end

data/lib/dbdb_builder.rb

@@ -0,0 +1,45 @@
+class DatabaseDatabaseBuilder
+  def DatabaseDatabaseBuilder.get_database_database(document)
+    connections = {}
+
+    document.xpath("//Dictionary/Databases/*").each do |database|
+      name = database.xpath("Name").text
+      connection_string = database.xpath("ConnectionString").text
+
+      connections[name] = connection_string
+    end
+
+    return connections
+  end
+
+  def DatabaseDatabaseBuilder.get_data_sources(doc)
+    data_sources = {}
+
+    data_source_container = doc.xpath("//DataSources")
+    data_source_count = data_source_container.attribute("count")
+
+    data_source_container.xpath("./*").each do |data_source_node|
+      # Data source name
+      name = data_source_node.xpath("Name").text
+
+      # Column names
+      column_container = data_source_node.xpath("Columns")
+      columns = []
+      column_container.xpath("./value").each do |column|
+        columns << column.text.split(",")[0]
+      end
+
+      # SQL query
+      sql = data_source_node.xpath("SqlCommand").text
+      sql.gsub!("\n", " ")
+
+      data_source = {}
+      data_source["columns"] = columns
+      data_source["sql"] = sql
+
+      data_sources[name] = data_source
+    end
+
+    return data_sources
+  end
+end

data/lib/mrtexport.rb

@@ -0,0 +1,303 @@
+#!/usr/bin/env ruby
+
+# Dependencies
+require "prawn"
+require "nokogiri"
+require "mysql2"
+require "base64"
+
+# Included files
+require_relative "dbdb_builder"
+require_relative "data_band"
+require_relative "stylist"
+
+class MRTExport
+  def initialize(params)
+    @report_file   = params[:report_file]
+    @output_file   = params[:output_file]
+    @export_format = params[:export_format]
+    @replacements  = params[:replacements]
+    
+    puts "[+] Compiling new report: #{@report_file}"
+
+    @xml_doc = Nokogiri::XML(File.open(@report_file))
+
+    @sql_connections = DatabaseDatabaseBuilder.get_database_database(@xml_doc)
+    @data_sources    = DatabaseDatabaseBuilder.get_data_sources(@xml_doc)
+
+    puts @sql_connections
+
+    initialize_database_connection if @sql_connections.length > 0
+
+    @y_off = 0
+
+    generate_pdf
+  end
+
+  def initialize_database_connection
+    sql_connection_string = @sql_connections["Localhost"]
+    sql_connection_array = {}
+
+    sql_connection_string.split(";").each do |x|
+      sql_connection_array[x.split("=")[0]] = x.split("=")[1]
+    end
+
+    @sql_connection = Mysql2::Client.new(
+      :host     => sql_connection_array["Server"],
+      :username => sql_connection_array["User"],
+      :password => sql_connection_array["Password"],
+      :database => sql_connection_array["Database"]
+    )
+  end
+
+  def get_value_from_db(name)
+    name.sub! "{", ""
+    name.sub! "}", ""
+
+    query_parts = name.split(".")
+    query_name  = query_parts[0]
+    query_field = query_parts[1]
+
+    sql = @data_sources[query_name]["sql"]
+
+    @replacements.each do |key, val|
+      sql.sub!("{#{key}}", val)
+    end
+    
+    text = ""
+
+    rs = @sql_connection.query sql
+    rs.each do |r|
+      text = r[query_field]
+      text = Util.number_format(text) if Util.is_number?(text)
+    end 
+
+    return text
+  end
+
+  def generate_pdf
+    @pdf = Prawn::Document.new
+
+    @xml_doc.xpath("//Pages/*").each do |page|
+      # List bands bands
+      puts "[*] listing report bands"
+      page.xpath("./Components/*[contains(., 'Band')]").each do |band|
+        puts "    - #{band.name}"
+      end
+
+      # render out of band stuff
+      page.xpath("./Components/*[not(contains(.,'Band'))]").each do |component|
+        if component.name.include?("Text")
+          render_background(component)
+          render_text(component)
+        elsif component.name.include?("Image")
+          render_image(component)
+        end
+      end
+
+      # but we want to do bands in a specific order!
+      # Start with header band
+      page.xpath("./Components/*[contains(., 'ReportTitleBand')]").each do |band|
+        puts "[*] Doing band: #{band.name}, ref: #{band.attribute("Ref")}"
+        render_band(band)
+      end
+
+      page.xpath("./Components/*[contains(., 'HeaderBand')]").each do |band|
+        puts "[*] Doing band: #{band.name}, ref: #{band.attribute("Ref")}"
+        render_band(band)
+      end
+
+      page.xpath("./Components/*[contains(., 'DataBand')]").each do |band|
+        puts "[*] Doing band: #{band.name}, ref: #{band.attribute("Ref")}"
+        #render_band(band)
+        data_band_renderer = DataBandRenderer.new(
+          band,
+          @pdf,
+          @sql_connection,
+          @replacements,
+          @data_sources,
+          @y_off
+        )
+
+         @y_off = data_band_renderer.render
+      end
+
+      page.xpath("./Components/*[contains(., 'FooterBand')]").each do |band|
+        puts "[*] Doing band: #{band.name}, ref: #{band.attribute("Ref")}"
+        render_band(band)
+      end
+    end
+
+    @pdf.render_file @output_file
+  end
+
+  def render_band(band)
+    render_backgrounds(band)
+    render_images(band)
+    render_texts(band)
+  end
+
+  def render_backgrounds(page)
+    page.xpath("Components/*[contains(.,'Text')]").each do |node|
+      next if node.name == "Name"
+
+      if node.name.start_with? "Text"
+        x, y, w, h = Stylist.get_dimensions(node, @y_off)
+
+        hue = node.xpath("Brush").text
+
+        if hue != "Transparent"
+          hue = hue.sub("[","").sub("]","").split(":")
+          hex_hue = hue[0].to_i.to_s(16) << hue[1].to_i.to_s(16) << hue[2].to_i.to_s(16)
+
+          @pdf.fill_color hex_hue
+          @pdf.fill_rectangle [x,y], w, h
+        end
+
+        if node.xpath("Border").text != ""
+          border = node.xpath("Border").text
+          border_bits = border.split(";")
+
+          sides = border_bits[0]
+
+          if sides != "None"
+            @pdf.stroke_color = Stylist.mrt_colour_to_hex(border_bits[1])
+            @pdf.line_width = 1
+
+            @pdf.line [x,   y], [x+w,   y] if sides.include? "Top"
+            @pdf.line [x, y-h], [x+w, y-h] if sides.include? "Bottom"
+            @pdf.line [x,   y], [x,   y-h] if sides.include? "Left"
+            @pdf.line [x+w, y], [x+w, y-h] if sides.include? "Right"
+            @pdf.stroke
+          end
+        end
+      end
+    end
+  end
+
+  def render_background(node)
+    return if node.name == "Name"
+
+    if node.name.start_with? "Text"
+      x, y, w, h = Stylist.get_dimensions(node, @y_off)
+
+      hue = node.xpath("Brush").text
+
+      if hue != "Transparent"
+        hue = hue.sub("[","").sub("]","").split(":")
+        hex_hue = hue[0].to_i.to_s(16) << hue[1].to_i.to_s(16) << hue[2].to_i.to_s(16)
+
+        @pdf.fill_color hex_hue
+        @pdf.fill_rectangle [x,y], w, h
+      end
+
+      if node.xpath("Border").text != ""
+        border = node.xpath("Border").text
+        border_bits = border.split(";")
+
+        sides = border_bits[0]
+
+        if sides != "None"
+          @pdf.stroke_color = Stylist.mrt_colour_to_hex(border_bits[1])
+          @pdf.line_width = 1
+
+          @pdf.line [x,   y], [x+w,   y] if sides.include? "Top"
+          @pdf.line [x, y-h], [x+w, y-h] if sides.include? "Bottom"
+          @pdf.line [x,   y], [x,   y-h] if sides.include? "Left"
+          @pdf.line [x+w, y], [x+w, y-h] if sides.include? "Right"
+          @pdf.stroke
+        end
+      end
+    end
+  end
+
+  def render_images(band)
+    band.xpath("Components/*[contains(.,'Image')]").each do |node|
+      render_image(node)
+    end
+  end
+
+  def render_image(node)
+    x, y, w, h = Stylist.get_dimensions(node, @y_off)
+
+    data = Base64::decode64(node.xpath("Image").text)
+
+    File.open("/tmp/mrt_to_pdf_img",'w') { |file| file.puts data }
+
+    @pdf.image "/tmp/mrt_to_pdf_img", :at => [x, 0 + h], :width => w 
+
+    File.delete("/tmp/mrt_to_pdf_img")
+  end
+
+  def render_texts(page)
+    last_y_off = @y_off
+
+    page.xpath("Components/*[contains(.,'Text')]").each do |node|
+      last_y_off = render_text(node)
+    end
+
+    @y_off = last_y_off
+  end
+
+  def render_text(node)
+    last_y_off = @y_off
+
+    return if node.name == "Name"
+
+    if node.name.start_with? "Text"
+      x, y, w, h = Stylist.get_dimensions(node, @y_off)
+
+      # Text
+      text = node.xpath("Text").text
+
+      # SQL replacements
+      if text.include?("{")
+        text.gsub!(/{.*}/) { |m| get_value_from_db(m) }
+      end
+
+      # Do painting
+      @pdf.fill_color = Stylist.get_text_colour(node)
+      @pdf.text_box text,
+                :at     => [x + 1, y - 1],
+                :height => h,
+                :width  => w,
+                :align  => Stylist.get_horizontal_align(node),
+                :valign => Stylist.get_vertical_align(node),
+                :size   => Stylist.get_font_size(node),
+                :font   => Stylist.get_font_face(node),
+                :style  => Stylist.get_font_style(node)
+
+      last_y_off = (740 - y) if (740 - y) > @y_off
+    end
+
+    return last_y_off
+  end
+end
+
+# Handler for command line launch
+if __FILE__ == $0
+  if ARGV[1].nil?
+    puts "USAGE:"
+    puts "./mrttopdf.rb <report file> <output file> <replacement 1> .. <replacement n>"
+    exit
+  elsif ARGV[1] == "--version"
+    puts "0.0.1"
+  end
+
+  report_file = ARGV[0]
+  output_file = ARGV[1]
+
+  replacements = {}
+
+  ARGV[2..ARGV.length].each do |arg|
+    arg_bits = arg.split("=")
+    replacements[arg_bits[0]] = arg_bits[1]
+  end
+
+  MRTExport.new({
+    :report_file   => report_file,
+    :output_file   => output_file,
+    :export_format => "pdf",
+    :replacements  => replacements
+  })
+end

data/lib/stylist.rb

@@ -0,0 +1,61 @@
+class Stylist
+  def Stylist.mrt_colour_to_hex(mrt_colour)
+    mrt_colour = mrt_colour.sub("[","").sub("]","").split(":")
+    return mrt_colour[0].to_i.to_s(16) << mrt_colour[1].to_i.to_s(16) << mrt_colour[2].to_i.to_s(16)
+  end
+
+  def Stylist.get_text_colour(node)
+    hue = node.xpath("TextBrush").text
+    case hue
+    when "Black"
+      "000000"
+    else
+      Stylist.mrt_colour_to_hex(hue)
+    end
+  end
+
+  def Stylist.get_horizontal_align(node)
+    if node.xpath("HorAlignment").text != ""
+      return node.xpath("HorAlignment").text.downcase.to_sym
+    else
+      return :left
+    end
+  end
+
+  def Stylist.get_vertical_align(node)
+    if node.xpath("VertAlignment").text != ""
+      return node.xpath("VertAlignment").text.downcase.to_sym
+    else
+      return :top
+    end
+  end
+
+  def Stylist.get_font_face(node)
+    node.xpath("Font").text.split(",")[0]
+  end
+
+  def Stylist.get_font_size(node)
+    node.xpath("Font").text.split(",")[1].to_i
+  end
+
+  def Stylist.get_font_style(node)
+    font_style = node.xpath("Font").text.split(",")[2]
+    if font_style.nil?
+      return :normal
+    else
+      return font_style.downcase.to_sym
+    end
+  end
+
+  def Stylist.get_dimensions(node, y_off)
+    position = node.xpath("ClientRectangle").text.split(",")
+    position = position.map {|x| x.to_f}
+
+    x = Util.cm_to_px(position[0])
+    y = Util.trans_y(y_off, Util.cm_to_px(position[1]))
+    w = Util.cm_to_px(position[2])
+    h = Util.cm_to_px(position[3])
+
+    return x, y, w, h
+  end
+end

data/lib/util.rb

@@ -0,0 +1,17 @@
+class Util
+  def Util.cm_to_px(cm)
+    return cm / 2.5 * 72
+  end
+
+  def Util.trans_y(off, y)
+    720 - off - y
+  end
+
+  def Util.is_number?(object)
+    true if Float(object) rescue false
+  end
+
+  def Util.number_format(number)
+    '%.2f' % number.to_f
+  end
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: mrtexport
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- jsrn
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-07-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: prawn
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+description: An exporter/renderer for .mrt report files.
+email: james.srn@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/data_band.rb
+- lib/dbdb_builder.rb
+- lib/mrtexport.rb
+- lib/stylist.rb
+- lib/util.rb
+homepage: https://github.com/jsrn/MRTExport
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: A renderer for MRT files.
+test_files: []