lib/pacto/rake_task.rb in pacto-0.3.0.pre vs lib/pacto/rake_task.rb in pacto-0.3.0

@@ -1,15 +1,9 @@
 require 'pacto'
 
-unless String.respond_to?(:colors)
-  class String
-    def colorize(*args)
-      self
-    end
-  end
-end
-
+# FIXME: RakeTask is a huge class, refactor this please
+# rubocop:disable ClassLength
 module Pacto
   class RakeTask
     include Rake::DSL
 
     def initialize
@@ -27,72 +21,79 @@
 
     def validate_task
       desc 'Validates all contracts in a given directory against a given host'
       task :validate, :host, :dir do |t, args|
         if args.to_a.size < 2
-          fail 'USAGE: rake pacto:validate[<host>, <contract_dir>]'.colorize(:yellow)
+          fail Pacto::UI.yellow('USAGE: rake pacto:validate[<host>, <contract_dir>]')
         end
 
         validate_contracts(args[:host], args[:dir])
       end
     end
 
     def generate_task
       desc 'Generates contracts from partial contracts'
       task :generate, :input_dir, :output_dir, :host do |t, args|
         if args.to_a.size < 3
-          fail 'USAGE: rake pacto:generate[<request_contract_dir>, <output_dir>, <record_host>]'.colorize(:yellow)
+          fail Pacto::UI.yellow('USAGE: rake pacto:generate[<request_contract_dir>, <output_dir>, <record_host>]')
         end
 
         generate_contracts(args[:input_dir], args[:output_dir], args[:host])
       end
     end
 
+    # FIXME: meta_validate is a big method =(. Needs refactoring
+    # rubocop:disable MethodLength
     def meta_validate
       desc 'Validates a directory of contract definitions'
       task :meta_validate, :dir do |t, args|
         if args.to_a.size < 1
-          fail 'USAGE: rake pacto:meta_validate[<contract_dir>]'.colorize(:yellow)
+          fail Pacto::UI.yellow('USAGE: rake pacto:meta_validate[<contract_dir>]')
         end
 
         each_contract(args[:dir]) do |contract_file|
-          puts "Validating #{contract_file}"
           fail unless Pacto.validate_contract contract_file
         end
+        puts 'All contracts successfully meta-validated'
       end
     end
 
     def validate_contracts(host, dir)
       WebMock.allow_net_connect!
       puts "Validating contracts in directory #{dir} against host #{host}\n\n"
 
       total_failed = 0
-      each_contract(dir) do |contact_file|
+      contracts = []
+      each_contract(dir) do |contract_file|
+        contracts << contract_file
         print "#{contract_file.split('/').last}:"
-        contract = Pacto.build_from_file(contract_file, host)
-        errors = contract.validate
+        contract = Pacto.load_contract(contract_file, host)
+        errors = contract.validate_provider
 
         if errors.empty?
-          puts ' OK!'.colorize(:green)
+          puts Pacto::UI.green(' OK!')
         else
           @exit_with_error = true
           total_failed += 1
-          puts ' FAILED!'.colorize(:red)
+          puts Pacto::UI.red(' FAILED!')
           errors.each do |error|
-            puts "\t* #{error}".colorize(:light_red)
+            puts Pacto::UI.red("\t* #{error}")
           end
           puts ''
         end
       end
 
       if @exit_with_error
-        fail "#{total_failed} of #{contracts.size} failed. Check output for detailed error messages.".colorize(:red)
+        fail Pacto::UI.red("#{total_failed} of #{contracts.size} failed. Check output for detailed error messages.")
       else
-        puts "#{contracts.size} valid contract#{contracts.size > 1 ? 's' : nil}".colorize(:green)
+        puts Pacto::UI.green("#{contracts.size} valid contract#{contracts.size > 1 ? 's' : nil}")
       end
     end
+    # rubocop:enable MethodLength
 
+    # FIXME: generate_contracts is a big method =(. Needs refactoring
+    # rubocop:disable MethodLength
     def generate_contracts(input_dir, output_dir, host)
       WebMock.allow_net_connect!
       generator = Pacto::Generator.new
       puts "Generating contracts from partial contracts in #{input_dir} and recording to #{output_dir}\n\n"
 
@@ -105,34 +106,36 @@
           output_file.write contract
           output_file.flush
           output_file.close
         rescue InvalidContract => e
           failed_contracts << contract_file
-          puts e.message.colorize(:red)
+          puts Pacto::UI.red(e.message)
         end
       end
 
       if failed_contracts.empty?
-        puts 'Successfully generated all contracts'.colorize(:green)
+        puts Pacto::UI.green('Successfully generated all contracts')
       else
-        fail "The following contracts could not be generated: #{failed_contracts.join ','}".colorize(:red)
+        fail Pacto::UI.red("The following contracts could not be generated: #{failed_contracts.join ','}")
       end
     end
+    # rubocop:enable MethodLength
 
     private
 
     def each_contract(dir)
       if File.file? dir
         yield dir
       else
-        contracts = Dir[File.join(dir, '*{.json.erb,.json}')]
-        fail "No contracts found in directory #{dir}".colorize(:yellow) if contracts.empty?
+        contracts = Dir[File.join(dir, '**/*{.json.erb,.json}')]
+        fail Pacto::UI.yellow("No contracts found in directory #{dir}") if contracts.empty?
 
         contracts.sort.each do |contract_file|
           yield contract_file
         end
       end
     end
   end
 end
+# rubocop:enable ClassLength
 
 Pacto::RakeTask.new.install