module QueryReport
  class ReportPdf
    attr_reader :pdf, :options, :report

    def initialize(report)
      @report = report
      @options = QueryReport.config.pdf_options
      @pdf = Prawn::Document.new
    end

    #render the header from the template class
    def render_header
      template.try(:render_header)
    end

    def render_footer
      template.try(:render_footer)
    end

    def to_pdf
      render_charts_with report
      render_table_with report
      pdf
    end

    private
    def render_charts_with(report)
      return if report.charts.empty? #or !report.chart_on_pdf?
      num_of_column = 2
      height = (pdf.bounds.width/num_of_column - 50) * (report.charts.size.to_f/num_of_column).ceil
      pdf.column_box([0, pdf.cursor], :columns => num_of_column, :width => pdf.bounds.width, :height => height) do
        report.charts.each do |chart|
          render_chart(chart)
        end
      end
    end

    def render_chart(chart)
      if chart.respond_to?(:to_blob)
        blob = chart.to_blob
        data = StringIO.new(blob)
        #pdf.pad_top(10) do
        pdf.image(data, :width => pdf.bounds.width)
        #end
      end
    end

    def table_header_for(table_items)
      table_items.first.keys
    end

    def humanized_table_header
      report_columns.collect(&:humanize)
    end

    def table_content_for(report)
      table_items = report.all_records
      items = table_items.map do |item|
        item_values = []

        report_columns.collect(&:humanize).each do |column|
          item_values << item[column].to_s
        end
        item_values
      end

      if report.has_total?
        items = items << report.column_total_with_colspan.collect do |total|
          total[:colspan] ? total : total[:content]
        end
      end
      items
    end

    def render_table_with(report)
      items = [humanized_table_header]
      items += table_content_for(report)
      render_table(items)
    end

    def render_table(items)
      header_bg_color = options[:table][:header][:bg_color]
      alternate_row_bg_color = [options[:table][:row][:odd_bg_color], options[:table][:row][:even_bg_color]]
      font_size = options[:font_size]
      header_font_size = options[:table][:header][:font_size]
      pdf.move_down 10
      pdf.table(items, :row_colors => alternate_row_bg_color, :header => true, :cell_style => {:inline_format => true, :size => font_size}) do
        row(0).style(:font_style => :bold, :background_color => header_bg_color, :size => header_font_size)
      end
    end

    private
    def report_columns
      report.columns.select { |c| !c.only_on_web? }
    end

    def template
      if options[:template_class]
        @template ||= options[:template_class].to_s.constantize.new(report, pdf)
        return @template
      end
      nil
    end
  end
end