lib/bankscrap/cli.rb in bankscrap-1.0.0 vs lib/bankscrap/cli.rb in bankscrap-1.0.1

@@ -6,11 +6,15 @@
     def self.shared_options
       option :user,     default: ENV['BANKSCRAP_USER']
       option :password, default: ENV['BANKSCRAP_PASSWORD']
       option :log,      default: false
       option :debug,    default: false
+      option :iban,     default: nil
 
+      option :format
+      option :output
+
       # Some bank needs more input, like birthday, this would go here
       # Usage:
       # bankscrap balance BANK_NAME --extra=birthday:01/12/1980
       option :extra, type: :hash, default: {}
     end
@@ -27,47 +31,46 @@
       end
     end
 
     desc 'transactions BANK', "get account's transactions"
     shared_options
-    def transactions(bank, iban = nil)
+    options from: :string, to: :string
+    def transactions(bank)
       assign_shared_options
 
-      begin
-        start_date = @extra_args.key?('from') ? Date.strptime(@extra_args['from'], '%d-%m-%Y') : nil
-        end_date = @extra_args.key?('to') ? Date.strptime(@extra_args['to'], '%d-%m-%Y') : nil
-      rescue ArgumentError
-        say 'Invalid date format. Correct format d-m-Y', :red
-      end
+      start_date = parse_date(options[:from]) if options[:from]
+      end_date = parse_date(options[:to]) if options[:to]
 
       initialize_client_for(bank)
 
-      account = iban ? @client.account_with_iban(iban) : @client.accounts.first
+      account = @iban ? @client.account_with_iban(@iban) : @client.accounts.first
 
-      if !start_date.nil? && !end_date.nil?
+      if start_date && end_date
         if start_date > end_date
           say 'From date must be lower than to date', :red
           exit
         end
 
         transactions = account.fetch_transactions(start_date: start_date, end_date: end_date)
       else
         transactions = account.transactions
       end
 
-      say "Transactions for: #{account.description} (#{account.iban})", :cyan
+      export_to_file(transactions, options[:format], options[:output]) if options[:format]
 
+      say "Transactions for: #{account.description} (#{account.iban})", :cyan
       transactions.each do |transaction|
         say transaction.to_s, (transaction.amount > Money.new(0) ? :green : :red)
       end
     end
 
     private
 
     def assign_shared_options
       @user       = options[:user]
       @password   = options[:password]
+      @iban       = options[:iban]
       @log        = options[:log]
       @debug      = options[:debug]
       @extra_args = options[:extra]
     end
 
@@ -81,8 +84,28 @@
       Object.const_get("Bankscrap::#{bank_name}::Bank")
     rescue LoadError
       raise ArgumentError.new('Invalid bank name.')
     rescue NameError
       raise ArgumentError.new("Invalid bank name. Did you mean \"#{bank_name.upcase}\"?")
+    end
+
+    def parse_date(string)
+      Date.strptime(string, '%d-%m-%Y')
+    rescue ArgumentError
+      say 'Invalid date format. Correct format d-m-Y (eg: 31-12-2016)', :red
+      exit
+    end
+
+    def export_to_file(data, format, path)
+      exporter(format, path).write_to_file(data)
+    end
+
+    def exporter(format, path)
+      case format.downcase
+      when 'csv' then BankScrap::Exporter::Csv.new(path)
+      else
+        say 'Sorry, file format not supported.', :red
+        exit
+      end
     end
   end
 end