# -*- coding: utf-8 -*-
# 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] types 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)
      # @option options [Package] package An Axlsx::Package. When this is provided the output will be added to the package as a new sheet.  # @option options [String] name This will be used to name the worksheet added to the package. If it is not provided the name of the table name will be humanized when i18n is not specified or the I18n.t for the table name.
      # @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 = options.delete(:package) || 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?
        i18n = self.xlsx_i18n == true ? 'activerecord.attributes' : i18n
        sheet_name = options.delete(:name) || (i18n ? I18n.t("#{i18n}.#{table_name.underscore}") : table_name.humanize) 
        data = [*find(:all, options)]
        data.compact!
        data.flatten!

        return p if data.empty?
        p.workbook.add_worksheet(:name=>sheet_name) 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|
            row_data = columns.map do |c|
              if c.to_s =~ /\./
                v = r; c.to_s.split('.').each { |method| v = v.send(method) }; v
              else
                r.send(c)                
              end
            end
            sheet.add_row row_data, :style=>row_style, :types=>types
          end
        end
        p
      end
    end
  end
end

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