bin/aipim in aipim-rails-0.0.41 vs bin/aipim in aipim-rails-0.0.139

@@ -2,44 +2,84 @@
 
 require 'aipim-rails'
 
 path = File.expand_path(File.dirname(__FILE__))+"/../"
 
-if ARGV[0] == 'generate'
-	system('mkdir -p aipim')
-	system('mkdir -p aipim/screenshots')
-	system('cp '+path+'lib/screenshot.rb features/support/')
-
-elsif ARGV[0] == 'html'
-	%x[rm -rf aipim/markdown]
+def get_files
 	files = []
 	ARGV.delete_at(0)
 	ARGV.each do |arg|
 		ls = %x[ls #{arg}].split("\n")
 		ls.each do |f|
-			f = f.split('/')
-			f.delete_at(0)
-			f = f.join('/')
+			f = f.split("/")
+			f.delete_at(0) if f[0] == "features"
+			f = f[0]
 			files << f if !(f =~ /.feature\z/).nil?
 		end
 	end
-	files.each { |f| Parser.init(f) }
-	files.each { |f| ConvertToHtml.init(f) }
-	system('cp '+path+'lib/bootstrap.min.css aipim/html')
-	system('cp '+path+'lib/jquery-1.9.1.js aipim/html')
+	files
+end
 
+if ARGV[0] == 'generate'
+	system('mkdir -p aipim')
+	system('mkdir -p aipim/screenshots')
+
+	width = "1024"
+	height = "768"
+
+	unless ARGV[1].nil?
+		width  = ARGV[1].split("x")[0]
+		height = ARGV[1].split("x")[1]
+	end
+
+	f = File.open(path+'lib/webdriver/screenshot.rb', "r")
+	f_str = ""
+	f.each_line {|line| f_str = f_str + line}
+	f_str = f_str.gsub("SCREENSHOT_HEIGHT", height).gsub("SCREENSHOT_WIDTH", width)
+	f.close
+	f = File.open("features/support/screenshot.rb", "w")
+	f.puts f_str
+	f.close
+
+elsif ARGV[0] == 'html'
+	%x[rm -rf aipim/html]
+	%x[mkdir -p aipim/html]
+	features = []
+	files = get_files
+	files.each do |f|
+		features << Parser.init(f)
+	end
+	ConvertToHtml.init(features)
+	system('cp '+path+'lib/assets/bootstrap.min.css aipim/html')
+	system('cp '+path+'lib/assets/jquery-1.9.1.js aipim/html')
+
 elsif ARGV[0] == 'markdown' || ARGV[0] == 'md'
 	%x[rm -rf aipim/markdown]
 	%x[mkdir -p aipim/markdown]
-	files = []
-	ARGV.delete_at(0)
-	ARGV.each do |arg|
-		ls = %x[ls #{arg}].split("\n")
-		ls.each do |f|
-			f = f.split('/')
-			f.delete_at(0)
-			f = f.join('/')
-			files << f if !(f =~ /.feature\z/).nil?
+	features = []
+	files = get_files
+	files.each do |f|
+		features << Parser.init(f)
+	end
+	Markdown.init(features)
+
+elsif ARGV[0] == 'parser'
+	%x[rm -rf aipim/markdown]
+	%x[mkdir -p aipim/markdown]
+	files = get_files
+	files.each do |f|
+		puts "Tentando abrir arquivo #{f}"
+		feature = Parser.init(f)
+		puts "Feature name: #{feature[:feature_name]}"
+		puts "Feature description: "
+		feature[:feature_description].each {|t| puts t}
+		puts "Context name: #{feature[:context_name]}"
+		puts "Context description"
+		feature[:context_description].each {|t| puts t}
+		feature[:scenarios].each do |scenario|
+			puts "----------------------------------------------"
+			puts scenario[:name]
+			scenario[:steps].each {|step| puts step}
 		end
 	end
-	files.each { |f| Parser.init(f) }
+
 end