lib/sanctuary/planter/cli.rb in sanctuary-planter-0.1.2 vs lib/sanctuary/planter/cli.rb in sanctuary-planter-0.1.3

@@ -1,21 +1,26 @@
 require "sanctuary"
+require "sanctuary/planter/parser"
+require "sanctuary/planter/generator"
 
 module Sanctuary
   module Planter
     class CLI
       def self.start
         if ARGV.include?("-i")
-          File.open(ARGV[1]) do |f|
-           f.each_line do |line|
-             if line[1..4] == "TYPE"
-               puts "generating #{line.split("TYPE:").last.split(" ")[0]} as #{line.split("NAME:").last}"
-               Sanctuary::Generator.start([line.split("TYPE:")[1].split(" ")[0].lstrip.chomp, line.split("NAME:").last.chomp])
-             end
-           end
-         end
+          classes_array = Sanctuary::Planter::Parser.parse_plant_uml(ARGV[1])
+          plant_array = Sanctuary::Planter::Parser.generate_plant_data_objects(classes_array)
+          plant_array.each do |plant|
+            Sanctuary::Planter::Generator.generate_file_from_plant(plant)
+          end
         else
-          puts "NO SUCH INPUT"
+          system("vim", "tmp.puml")
+          classes_array = Sanctuary::Planter::Parser.parse_plant_uml("./tmp.puml")
+          plant_array = Sanctuary::Planter::Parser.generate_plant_data_objects(classes_array)
+          plant_array.each do |plant|
+            Sanctuary::Planter::Generator.generate_file_from_plant(plant)
+          end
+          File.delete("./tmp.puml") if File.exists? ("./tmp.puml")
         end
       end
     end
   end
 end
\ No newline at end of file