add whitelist builder and dsl option #2

Author: randym

lib/active_admin/axlsx/builder.rb

@@ -106,6 +106,12 @@ def clear_columns
         @columns = []
       end
 
+      # Clears the default columns array so you can whitelist only the columns you
+      # want to export
+      def whitelist
+        @columns = []
+      end
+
       # Add a column
       # @param [Symbol] name The name of the column.
       # @param [Proc] block A block of code that is executed on the resource

lib/active_admin/axlsx/dsl.rb

@@ -1,7 +1,7 @@
 module ActiveAdmin
   module Axlsx
     module DSL
-      delegate :ingnore_columns, :column, :after_filer, :i18n_scope, :header_style, :skip_header, to: :xlsx_builder, prefix: :config
+      delegate :ingnore_columns, :column, :after_filer, :i18n_scope, :header_style, :skip_header, :white_list, to: :xlsx_builder, prefix: :config
       # @see ActiveAdmin::Axlsx::Builder
       def xlsx(options={}, &block)
         config.xlsx_builder = ActiveAdmin::Axlsx::Builder.new(config.resource_class, options, &block)

spec/axlsx/unit/builder_spec.rb

@@ -80,6 +80,36 @@ module Axlsx
         end
       end
 
+      context 'whitelisted sheet generation' do
+        let!(:users) {  [User.new(first_name: 'bob', last_name: 'nancy')] }
+
+        let!(:posts) {  [Post.new(title: 'bob', body: 'is a swell guy', author: users.first)] }
+
+        let!(:builder) {
+          Builder.new(Post, header_style: { sz: 10, fg_color: "FF0000" }, i18n_scope: [:axlsx, :post]) do
+            skip_header
+            whitelist
+            column :title
+          end
+        }
+
+        before do
+          User.stub!(:all) { users }
+          Post.stub!(:all) { posts }
+          # disable clean up so we can get the package.
+          builder.stub(:clean_up) { false }
+          builder.serialize(Post.all)
+          @package = builder.send(:package)
+          @collection = builder.collection
+        end
+
+        it 'does not serialize the header' do
+          sheet = @package.workbook.worksheets.first
+          sheet.rows.first.cells.size.should == 1
+          sheet.rows.first.cells.first.value.should == @collection.first.title
+        end
+      end
+
       context 'Sheet generation with a highly customized configuration.' do
 
         let!(:users) {  [User.new(first_name: 'bob', last_name: 'nancy')] }