lib/gorails/commands/railsbytes.rb in gorails-0.1.2 vs lib/gorails/commands/railsbytes.rb in gorails-0.1.3

@@ -1,8 +1,9 @@
 require "gorails"
-require "net/http"
+require "bundler"
 require "json"
+require "net/http"
 
 module Gorails
   module Commands
     class Railsbytes < Gorails::Command
       def call(args, _name)
@@ -42,24 +43,23 @@
           puts CLI::UI.fmt "{{green:#{byte["name"]}}} by #{byte["created_by"]}"
           puts byte["short_description"]
           puts
 
           CLI::UI::Prompt.ask("What would you like to do?") do |handler|
-            handler.option("View source") do |selection|
-              puts
-              puts byte["script"]
-            end
-
             handler.option("Apply Railsbyte") do |selection|
               puts
               puts "Running Railsbyte..."
+              puts
 
-              Bundler.with_original_env do
-                system "rails app:template LOCATION=\"https://railsbytes.com/script/#{id}\""
-              end
+              system "rails app:template LOCATION=\"https://railsbytes.com/script/#{id}\""
             end
 
-            handler.option("exit") { |selection| exit 0 }
+            handler.option("View source") do |selection|
+              puts
+              puts byte["script"]
+            end
+
+            handler.option("Exit") { |selection| exit 0 }
           end
         end
       end
     end
   end