lib/ronin/ui/command_line/commands/add.rb in ronin-0.2.1 vs lib/ronin/ui/command_line/commands/add.rb in ronin-0.2.2

@@ -43,18 +43,34 @@
               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
+                @media = media.to_sym
               end
 
-              opts.on('-U','--uri URI','Specify the source URI of the overlay') do |uri|
-                @uri = uri
+              opts.on('-L','--local','Similiar to: -m local') do
+                @media = nil
               end
 
-              opts.on('-L','--local','Similiar to: --media local') do
-                @media = :local
+              opts.on('--rsync','Alias for -m rsync') do
+                @media = :rsync
+              end
+
+              opts.on('--svn','Alias for -m svn') do
+                @media = :svn
+              end
+
+              opts.on('--hg','Alias for -m hg') do
+                @media = :hg
+              end
+
+              opts.on('--git','Alias for -m git') do
+                @media = :git
+              end
+
+              opts.on('-U','--uri URI','Specify the source URI of the overlay') do |uri|
+                @uri = uri
               end
             end
 
             opts.arguments(
               'PATH' => 'Add the overlay located at the specified PATH'