lib/bankscrap/cli.rb in bankscrap-1.0.2 vs lib/bankscrap/cli.rb in bankscrap-1.0.3

@@ -1,8 +1,12 @@
 require 'thor'
 require 'active_support/core_ext/string'
+Dir[File.expand_path("../../..", __FILE__) + "/generators/*.rb"].each do |generator|
+  require generator
+end
 
+
 module Bankscrap
   class CLI < Thor
     def self.shared_options
       option :user,     default: ENV['BANKSCRAP_USER']
       option :password, default: ENV['BANKSCRAP_PASSWORD']
@@ -60,9 +64,13 @@
       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
+
+
+    register(Bankscrap::AdapterGenerator, "generate_adapter", "generate_adapter MyBankName", 
+      "generates a template for a new Bankscrap bank adapter")
 
     private
 
     def assign_shared_options
       @user       = options[:user]