From 9f2e63026696f944d30615a873c27c0deea38144 Mon Sep 17 00:00:00 2001
From: Patrick Gansterer <paroga@paroga.com>
Date: Mon, 8 Feb 2021 01:07:10 +0100
Subject: [PATCH] Add CSV downloads for BankTransaction and Invoice

---
 .../finance/bank_transactions_controller.rb   | 13 +++++--
 .../finance/invoices_controller.rb            | 10 ++++-
 .../bank_transactions/_transactions.html.haml |  4 ++
 .../finance/invoices/_invoices.html.haml      |  9 ++++-
 lib/bank_transactions_csv.rb                  | 29 +++++++++++++++
 lib/invoices_csv.rb                           | 37 +++++++++++++++++++
 6 files changed, 96 insertions(+), 6 deletions(-)
 create mode 100644 lib/bank_transactions_csv.rb
 create mode 100644 lib/invoices_csv.rb

diff --git a/app/controllers/finance/bank_transactions_controller.rb b/app/controllers/finance/bank_transactions_controller.rb
index 555e923a..32ad94a5 100644
--- a/app/controllers/finance/bank_transactions_controller.rb
+++ b/app/controllers/finance/bank_transactions_controller.rb
@@ -17,9 +17,16 @@ class Finance::BankTransactionsController < ApplicationController
     end
 
     @bank_account = BankAccount.find(params[:bank_account_id])
-    @bank_transactions = @bank_account.bank_transactions.order(sort).includes(:financial_link)
-    @bank_transactions = @bank_transactions.where('reference LIKE ? OR text LIKE ?', "%#{params[:query]}%", "%#{params[:query]}%") unless params[:query].nil?
-    @bank_transactions = @bank_transactions.page(params[:page]).per(@per_page)
+    @bank_transactions_all = @bank_account.bank_transactions.order(sort).includes(:financial_link)
+    @bank_transactions_all = @bank_transactions_all.where('reference LIKE ? OR text LIKE ?', "%#{params[:query]}%", "%#{params[:query]}%") unless params[:query].nil?
+    @bank_transactions = @bank_transactions_all.page(params[:page]).per(@per_page)
+
+    respond_to do |format|
+      format.js; format.html { render }
+      format.csv do
+        send_data BankTransactionsCsv.new(@bank_transactions_all).to_csv, filename: 'transactions.csv', type: 'text/csv'
+      end
+    end
   end
 
   def show
diff --git a/app/controllers/finance/invoices_controller.rb b/app/controllers/finance/invoices_controller.rb
index 289684a4..33949017 100644
--- a/app/controllers/finance/invoices_controller.rb
+++ b/app/controllers/finance/invoices_controller.rb
@@ -5,7 +5,15 @@ class Finance::InvoicesController < ApplicationController
   before_action :ensure_can_edit, only: [:edit, :update, :destroy]
 
   def index
-    @invoices = Invoice.includes(:supplier, :deliveries, :orders).order('date DESC').page(params[:page]).per(@per_page)
+    @invoices_all = Invoice.includes(:supplier, :deliveries, :orders).order('date DESC')
+    @invoices = @invoices_all.page(params[:page]).per(@per_page)
+
+    respond_to do |format|
+      format.js; format.html { render }
+      format.csv do
+        send_data InvoicesCsv.new(@invoices_all).to_csv, filename: 'invoices.csv', type: 'text/csv'
+      end
+    end
   end
 
   def unpaid
diff --git a/app/views/finance/bank_transactions/_transactions.html.haml b/app/views/finance/bank_transactions/_transactions.html.haml
index 7d19ae67..389978b8 100644
--- a/app/views/finance/bank_transactions/_transactions.html.haml
+++ b/app/views/finance/bank_transactions/_transactions.html.haml
@@ -1,4 +1,8 @@
 .pull-right
+  .btn-group
+    = link_to url_for(query: params[:query], format: :csv), class: 'btn' do
+      = glyph :download
+      CSV
   - if @bank_transactions.total_pages > 1
     .btn-group= items_per_page wrap: false
 = pagination_links_remote @bank_transactions
diff --git a/app/views/finance/invoices/_invoices.html.haml b/app/views/finance/invoices/_invoices.html.haml
index 1cf2a550..a055b1a2 100644
--- a/app/views/finance/invoices/_invoices.html.haml
+++ b/app/views/finance/invoices/_invoices.html.haml
@@ -1,5 +1,10 @@
-- if Invoice.count > 20
-  = items_per_page
+.pull-right
+  .btn-group
+    = link_to url_for(query: params[:query], format: :csv), class: 'btn' do
+      = glyph :download
+      CSV
+  - if @invoices.total_pages > 1
+    = items_per_page
 = pagination_links_remote @invoices
 
 %table.table.table-striped
diff --git a/lib/bank_transactions_csv.rb b/lib/bank_transactions_csv.rb
new file mode 100644
index 00000000..34c39403
--- /dev/null
+++ b/lib/bank_transactions_csv.rb
@@ -0,0 +1,29 @@
+require 'csv'
+
+class BankTransactionsCsv < RenderCSV
+  include ApplicationHelper
+
+  def header
+    [
+      BankTransaction.human_attribute_name(:external_id),
+      BankTransaction.human_attribute_name(:date),
+      BankTransaction.human_attribute_name(:amount),
+      BankTransaction.human_attribute_name(:iban),
+      BankTransaction.human_attribute_name(:reference),
+      BankTransaction.human_attribute_name(:text)
+    ]
+  end
+
+  def data
+    @object.each do |t|
+      yield [
+        t.external_id,
+        t.date,
+        t.amount,
+        t.iban,
+        t.reference,
+        t.text
+      ]
+    end
+  end
+end
diff --git a/lib/invoices_csv.rb b/lib/invoices_csv.rb
new file mode 100644
index 00000000..90353ced
--- /dev/null
+++ b/lib/invoices_csv.rb
@@ -0,0 +1,37 @@
+require 'csv'
+
+class InvoicesCsv < RenderCSV
+  include ApplicationHelper
+
+  def header
+    [
+      Invoice.human_attribute_name(:created_at),
+      Invoice.human_attribute_name(:created_by),
+      Invoice.human_attribute_name(:date),
+      Invoice.human_attribute_name(:supplier),
+      Invoice.human_attribute_name(:number),
+      Invoice.human_attribute_name(:amount),
+      Invoice.human_attribute_name(:deposit),
+      Invoice.human_attribute_name(:deposit_credit),
+      Invoice.human_attribute_name(:paid_on),
+      Invoice.human_attribute_name(:note)
+    ]
+  end
+
+  def data
+    @object.each do |t|
+      yield [
+        t.created_at,
+        show_user(t.created_by),
+        t.date,
+        t.supplier.name,
+        t.number,
+        t.amount,
+        t.deposit,
+        t.deposit_credit,
+        t.paid_on,
+        t.note,
+      ]
+    end
+  end
+end