lib/ronin/ui/command_line/commands/install.rb in ronin-0.1.4 vs lib/ronin/ui/command_line/commands/install.rb in ronin-0.2.0

@@ -20,23 +20,34 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 #++
 #
 
 require 'ronin/ui/command_line/command'
-require 'ronin/cache/overlay'
+require 'ronin/platform/overlay'
 
 module Ronin
   module UI
     module CommandLine
       class InstallCommand < Command
 
         command :install
 
+        def initialize
+          @cache = nil
+          @media = nil
+
+          super
+        end
+
         def define_options(opts)
           opts.usage = 'URI [options]'
 
           opts.options do
+            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+              @cache = dir
+            end
+
             opts.on('-m','--media [MEDIA]','Spedify the media-type of the overlay') do |media|
               @media = media
             end
           end
 
@@ -52,13 +63,13 @@
             fail('only one overlay URI maybe specified')
           end
 
           uri = args.first
 
-          Cache::Overlay.save_cache do
-            Cache::Overlay.install(:uri => uri, :media => @media) do |overlay|
-              puts "Overlay #{overlay.name.dump} has been installed."
-            end
+          Platform.load_overlays(@cache) if @cache
+
+          Platform.install(:uri => uri, :media => @media) do |overlay|
+            puts "Overlay #{overlay.name.dump} has been installed."
           end
         end
 
       end
     end