lib/robot-vim/runner.rb in robot-vim-0.1.1 vs lib/robot-vim/runner.rb in robot-vim-0.9.0

@@ -9,11 +9,23 @@
     def vim_binary
       @vim_binary || DEFAULT_VIM_BINARY
     end
 
     def run(args={})
-      ScriptFile.open(args[:commands]) do |script_file_path|
-        Kernel.send(:`, "#{self.vim_binary} -n -s #{script_file_path} #{args[:input_file]}")
+      output_file_name = UUID.new.generate(:compact)
+      commands = args[:commands] + ":w #{output_file_name}"
+      commands = commands + "\n:%bd!\n:q!\n"
+      invoke_vim(commands, args[:input_file])
+      return File.read(output_file_name)
+    ensure
+      File.delete(output_file_name) if File.exists?(output_file_name)
+    end
+
+    private
+    
+    def invoke_vim(commands, input_file)
+      ScriptFile.open(commands) do |script_file_path|
+        Kernel.send(:`, "#{self.vim_binary} -n -s #{script_file_path} #{input_file}")
       end
     end
   end
 end