homebrew/xcfit in xcfit-4.0.5 vs homebrew/xcfit in xcfit-4.0.6

@@ -1,11 +1,11 @@
 #!/usr/bin/env ruby
 
 require 'fileutils'
 require 'tempfile'
 
-VERSION = "4.0.5"
+VERSION = "4.0.6"
 
 @root_dir = File.expand_path('~')
 @root_lib_dir = File.join(@root_dir, 'Library')
 @root_developer_dir = File.join(@root_lib_dir, 'Developer')
 @root_xcode_dir = File.join(@root_developer_dir, 'Xcode')
@@ -25,10 +25,11 @@
   system("cd /tmp  && curl -sL https://github.com/Shashikant86/XCFit/archive/#{VERSION}.tar.gz | tar xz")
   puts "*******======= Downloading XCFit Templates for the Xcode =============***"
   @src_template_dir = "/tmp/XCFit-#{VERSION}/XCFit_Templates"
   @xcfit_template_dir = "/tmp/XCFit-#{VERSION}/XCFit_Templates/XCFit"
   @gherkin_template_dir = "/tmp/XCFit-#{VERSION}/XCFit_Templates/Gherkin"
+  @fastlane_template_dir = "/tmp/XCFit-#{VERSION}/Fastlane"
 end
 
 def clean_xcfit
   system("rm -rf /tmp/XCFit-#{VERSION}/")
 end
@@ -59,20 +60,32 @@
    puts '=======You need to replace targets with your project targets and comment unwanted targets ========'
    system("curl -s -O https://raw.githubusercontent.com/Shashikant86/XCFit/master/Cocoapods/Podfile")
    puts "Podfile successfully created in the current working directory here at #{Dir.getwd}/Podfile"
 end
 
+def setup_xcfit_fastfile
+   puts '=======Creating Template Podfile for the XCFit project'
+   puts '=======You need to replace schemes and other varibales as required ========'
+   clone_xcfit
+   puts "==================XXXXXXXX==========================="
+   puts "==================XXXXXXXX==========================="
+   FileUtils.cp_r(@fastlane_template_dir, FileUtils.pwd())
+   clean_xcfit
+end
+
 def print_usage
   puts <<EOF
 
   Usage: xcfit <command-name>
 
   <command-name> can be one of
     setup_xcode_templates
-      generate a Xcode 8 Templates for the XCUI, Fitnesse and Cucumberish
+      generate a Xcode Templates for the XCUI, Fitnesse and Cucumberish
     setup_xcfit_podfile
       Creates template Podfile for the XCFit project covering all targets.
+    setup_xcfit_fastfile
+      Creates template Fastfile to run tests with fastlane.
     version
       prints the XCFit version
     help
       prints more detailed help information.
 
@@ -88,19 +101,24 @@
   Usage: xcfit <command-name>
 
   <command-name> can be one of
     help
     setup_xcode_templates
+    setup_xcfit_podfile
+    setup_xcfit_fastfile
+
     version
 
     Commands:
       help : prints more detailed help information.
 
       setup_xcode_templates  : Generate a Xcode Target and File Templates for the XCUI, Fitnesse and Cucumberish
 
       setup_xcfit_podfile : Creates template Podfile for the XCFit project covering all targets.
 
+      setup_xcfit_fastfile : Creates template Fastfile to run tests with fastlane.
+
       version : prints the XCFit version
 
   <Options>
       -v, --verbose  Turns on verbose logging
 EOF
@@ -114,9 +132,11 @@
     print_help
   elsif cmd == 'setup_xcode_templates'
     setup_xcode_templates
   elsif cmd == 'setup_xcfit_podfile'
     setup_xcfit_podfile
+  elsif cmd == 'setup_xcfit_fastfile'
+    setup_xcfit_fastfile
   elsif cmd == 'version'
     puts "#{VERSION}"
   else
     print_usage
   end