# Axlsx is a gem or generating excel spreadsheets with charts, images and many other features. 
# 
# acts_as_xlsx provides integration into active_record for Axlsx.
# 
require 'axlsx'

# Adding to the Axlsx module 
# @see http://github.com/randym/axlsx
module Axlsx
  # === Overview
  # This module defines the acts_as_xlsx class method and provides to_xlsx support to both AR classes and instances
  module Ar
    
    def self.included(base) # :nodoc:
      base.send :extend, ClassMethods
    end
    
    # Class methods for the mixin
    module ClassMethods

      # defines the class method to inject to_xlsx
      # @option options [Array, Symbol] columns an array of symbols defining the columns and methods to call in generating sheet data for each row.
      # @option options [String] i18n (default nil) The path to search for localization. When this is specified your i18n.t will be used to determine the labels for columns.
      # @example
      #       class MyModel < ActiveRecord::Base
      #          acts_as_xlsx :columns=> [:id, :created_at, :updated_at], :i18n => 'activerecord.attributes'
      def acts_as_xlsx(options={})
        cattr_accessor :xlsx_i18n, :xlsx_columns
        self.xlsx_i18n = options.delete(:i18n) || false
        self.xlsx_columns = options.delete(:columns) ||  self.column_names.map { |c| c = c.to_sym }
        extend Axlsx::Ar::SingletonMethods
      end
    end

    # Singleton methods for the mixin
    module SingletonMethods

      # Maps the AR class to an Axlsx package
      # options are passed into AR find
      # @param [Array, Array] columns as an array of symbols or a symbol that defines the attributes or methods to render in the sheet.
      # @option options [Integer] header_style to apply to the first row of field names
      # @option options [Array, Symbol] an array of Axlsx types for each cell in data rows or a single type that will be applied to all types.
      # @option options [Integer, Array] style The style to pass to Worksheet#add_row
      # @option options [String] i18n The path to i18n attributes. (usually activerecord.attributes)
      # @see Worksheet#add_row
      def to_xlsx(options = {})

        row_style = options.delete(:style)
        header_style = options.delete(:header_style) || row_style
        types = [options.delete(:types) || []].flatten

        i18n = options.delete(:i18n) || self.xlsx_i18n
        columns = options.delete(:columns) || self.xlsx_columns

        p = Package.new
        row_style = p.workbook.styles.add_style(row_style) unless row_style.nil?
        header_style = p.workbook.styles.add_style(header_style) unless header_style.nil?

        data = [*find(:all, options)]
        data.compact!
        data.flatten!

        return p if data.empty?
        p.workbook.add_worksheet(:name=>table_name.humanize) do |sheet|
          
          col_labels = if i18n
                         columns.map { |c| I18n.t("#{i18n}.#{self.name.underscore}.#{c}") }                         
                       else
                         columns.map { |c| c.to_s.humanize }
                       end
          
          sheet.add_row col_labels, :style=>header_style
          
          data.each do |r|
            sheet.add_row columns.map { |c| r.send(c) }, :style=>row_style, :types=>types
          end
        end
        p
      end
    end
  end
end

require 'active_record'
ActiveRecord::Base.send :include, Axlsx::Ar