ronin 1.3.0 → 1.4.0.rc1

data/lib/ronin/ui/cli/commands/install.rb

@@ -0,0 +1,104 @@
+#
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This file is part of Ronin.
+#
+# Ronin is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ronin is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+require 'ronin/ui/cli/command'
+require 'ronin/repository'
+require 'ronin/database'
+
+module Ronin
+  module UI
+    module CLI
+      module Commands
+        #
+        # The `ronin-install` command.
+        #
+        class Install < Command
+
+          summary 'Installs Ronin Repositories'
+
+          option :rsync, :type => true
+          option :svn,   :type => true
+          option :hg,    :type => true
+          option :git,   :type => true
+
+          argument :uri, :type     => String,
+                         :banner   => '[URI|PATH]'
+
+          #
+          # Sets up the install command.
+          #
+          def setup
+            super
+
+            Database.setup
+          end
+
+          #
+          # Executes the command.
+          #
+          def execute
+            unless uri?
+              print_error "Must specify the URI argument"
+              exit -1
+            end
+
+            scm = if rsync?
+                    :rsync
+                  elsif svn?
+                    :sub_version
+                  elsif hg?
+                    :mercurial
+                  elsif git?
+                    :git
+                  end
+
+            repository = begin
+                           if File.directory?(@uri)
+                             # add local repositories
+                             Repository.add(:path => @uri, :scm => scm)
+                           else
+                             # install remote repositories
+                             Repository.install(:uri => @uri, :scm => scm)
+                           end
+                         rescue DuplicateRepository => e
+                           print_error e.message
+                           exit -1
+                         end
+
+            print_info "Repository #{repository} installed."
+
+            # print any caching exceptions/errors
+            repository.script_paths.each do |script_path|
+              if script_path.cache_exception
+                print_exception script_path.cache_exception
+              end
+
+              if script_path.cache_errors
+                script_path.cache_errors.each do |error|
+                  print_error error
+                end
+              end
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/ui/cli/commands/ips.rb

@@ -32,32 +32,32 @@ module Ronin
 
           model IPAddress
 
-          query_option :v4, :type => :boolean, :aliases => '-4'
-          query_option :v6, :type => :boolean, :aliases => '-6'
+          query_option :v4, :type => true, :flag => '-4'
+          query_option :v6, :type => true, :flag => '-6'
 
-          query_option :with_ports, :type => :array,
-                                    :aliases => '-p',
-                                    :banner => 'PORT [...]'
+          query_option :with_ports, :type  => Array[Integer],
+                                    :flag  => '-p',
+                                    :usage => 'PORT [...]'
 
-          query_option :with_macs, :type => :array,
-                                   :aliases => '-M',
-                                   :banner => 'MAC [...]'
+          query_option :with_macs, :type  => Array,
+                                   :flag  => '-M',
+                                   :usage => 'MAC [...]'
 
-          query_option :with_hosts, :type => :array,
-                                    :aliases => '-H',
-                                    :banner => 'HOST [...]'
+          query_option :with_hosts, :type  => Array,
+                                    :flag  => '-H',
+                                    :usage => 'HOST [...]'
 
-          class_option :list, :type => :boolean,
-                              :default => true,
-                              :aliases => '-l'
+          option :list, :type    => true,
+                        :default => true,
+                        :aliases => '-l'
 
-          class_option :lookup, :type => :string,
-                                :aliases => '-L',
-                                :banner => 'HOST'
+          option :lookup, :type  => String,
+                          :flag  => '-L',
+                          :usage => 'HOST'
 
-          class_option :import, :type => :string,
-                                :aliases => '-i',
-                                :banner => 'FILE'
+          option :import, :type  => String,
+                          :flag  => '-i',
+                          :usage => 'FILE'
 
           #
           # Queries the {IPAddress} model.
@@ -65,8 +65,8 @@ module Ronin
           # @since 1.0.0
           #
           def execute
-            if options[:lookup]
-              lookup options[:lookup]
+            if lookup?
+              lookup(@lookup)
             else
               super
             end
@@ -101,7 +101,7 @@ module Ronin
           # @since 1.0.0
           #
           def print_resource(ip)
-            return super(ip) unless options.verbose?
+            return super(ip) unless verbose?
 
             print_title ip.address
 

data/lib/ronin/ui/cli/commands/repos.rb

@@ -29,226 +29,113 @@ module Ronin
         #
         class Repos < ModelCommand
 
-          desc 'Manages Ronin repositories'
-          class_option :add, :type => :string,
-                             :aliases => '-A',
-                             :banner => 'PATH'
-          class_option :install, :type => :string,
-                                 :aliases => '-I',
-                                 :banner => 'URI'
-          class_option :list, :type => :boolean,
-                              :aliases => '-l'
-          class_option :update, :type => :boolean,
-                                :aliases => '-u'
-          class_option :uninstall, :type => :string,
-                                   :aliases => '-U',
-                                   :banner => 'REPO'
-          class_option :scm, :type => :string, :aliases => '-S'
+          summary 'Lists Ronin Repositories'
 
-          class_option :rsync, :type => :boolean
-          class_option :svn, :type => :boolean
-          class_option :hg, :type => :boolean
-          class_option :git, :type => :boolean
+          model Repository
 
-          argument :name, :type => :string, :required => false
+          query_option :domain, :type        => String,
+                                :description => 'Domain to filter by'
+
+          query_option :named, :type        => String,
+                               :flag        => '-n',
+                               :usage       => 'NAME',
+                               :description => 'Name to filter by'
+
+          query_option :titled, :type        => String,
+                                :flag        => '-t',
+                                :usage       => 'TITLE',
+                                :description => 'Title to filter by'
+
+          query_option :describing, :type    => String,
+                                    :flag    => '-d',
+                                    :usage   => 'DESC',
+                                    :desc    => 'Description to filter by'
+
+          query_option :licensed_under, :type        => String,
+                                        :flag        => '-L',
+                                        :usage       => 'LICENSE',
+                                        :description => 'License to filter by'
+
+          argument :repo, :type        => String,
+                          :description => 'Repository to list'
 
           #
           # Executes the command.
           #
           def execute
-            if options[:add]
-              add options[:add]
-            elsif options[:install]
-              install options[:install]
-            elsif options.update?
-              update
-            elsif options[:uninstall]
-              uninstall options[:uninstall]
+            if repo?
+              repository = begin
+                             Repository.find(@repo)
+                           rescue RepositoryNotFound => e
+                             print_error e.message
+                             exit -1
+                           end
+
+              print_repository(repository)
             else
-              list
+              print_array query
             end
           end
 
           protected
 
           #
-          # The selected SCM.
-          #
-          # @return [Symbol]
-          #   The name of the selected SCM.
-          #
-          def scm
-            @scm ||= if options[:scm]
-                       options[:scm].to_sym
-                     elsif options.rsync?
-                       :rsync
-                     elsif options.svn?
-                       :sub_version
-                     elsif options.hg?
-                       :mercurial
-                     elsif options.git?
-                       :git
-                     end
-          end
-
-          #
-          # Print out any exceptions or validation errors encountered
-          # when caching the files of the repository.
+          # Prints information about a Repository.
           #
           # @param [Repository] repo
-          #   The repository that was updated.
+          #   The repository to print.
           #
-          def print_cache_errors(repo)
-            repo.cached_files.each do |cached_file|
-              if cached_file.cache_exception
-                print_exception cached_file.cache_exception
-              end
+          def print_repository(repo)
+            print_title repo.name
 
-              if cached_file.cache_errors
-                cached_file.cache_errors.each do |error|
-                  print_error error
-                end
+            indent do
+              if repo.installed?
+                puts "Domain: #{repo.domain}"
+              else
+                puts "Path: #{repo.path}"
               end
-            end
-          end
 
-          #
-          # Lists installed or added Repositories.
-          #
-          def list
-            unless name
-              print_array Repository.all
-            else
-              repo = begin
-                       [Repository.find(name)]
-                     rescue RepositoryNotFound => e
-                       print_error e.message
-                       exit -1
-                     end
-
-              print_title repo.name
-
-              indent do
-                if repo.installed?
-                  puts "Domain: #{repo.domain}"
-                else
-                  puts "Path: #{repo.path}"
-                end
+              puts "SCM: #{repo.scm}" if repo.scm
 
-                puts "SCM: #{repo.scm}" if repo.scm
+              if repo.verbose?
+                putc "\n"
 
-                if repo.verbose?
-                  putc "\n"
+                puts "Title: #{repo.title}" if repo.title
+                puts "URI: #{repo.uri}" if repo.uri
+                puts "Source URI: #{repo.source}" if repo.source
+                puts "Website: #{repo.website}" if repo.website
 
-                  puts "Title: #{repo.title}" if repo.title
-                  puts "URI: #{repo.uri}" if repo.uri
-                  puts "Source URI: #{repo.source}" if repo.source
-                  puts "Website: #{repo.website}" if repo.website
+                executables = repo.executables
 
-                  executables = repo.executables
+                unless executables.empty?
+                  puts "Executables: #{executables.join(', ')}"
+                end
 
-                  unless executables.empty?
-                    puts "Executables: #{executables.join(', ')}"
-                  end
+                putc "\n"
 
-                  putc "\n"
+                unless repo.script_paths.empty?
+                  print_title 'Cached Files'
 
-                  unless repo.cached_files.empty?
-                    print_title 'Cached Files'
-
-                    indent do
-                      repo.cached_files.each do |cached_file|
-                        puts cached_file.path
-                      end
+                  indent do
+                    repo.script_paths.each do |script_path|
+                      puts script_path.path
                     end
                   end
+                end
 
-                  if repo.description
-                    print_title "Description"
+                if repo.description
+                  print_title "Description"
 
-                    indent { puts "#{repo.description}\n\n" }
-                  else
-                    putc "\n"
-                  end
+                  indent { puts "#{repo.description}\n\n" }
                 else
                   putc "\n"
                 end
+              else
+                putc "\n"
               end
             end
           end
 
-          #
-          # Adds a Repository.
-          #
-          # @param [String] path
-          #   The path of the local repository.
-          #
-          def add(path)
-            repo = begin
-                     Repository.add!(:path => path, :scm => @scm)
-                   rescue DuplicateRepository => e
-                     print_error e.message
-                     exit -1
-                   end
-
-            print_info "Repository #{repo} added."
-          end
-
-          #
-          # Installs a Repository.
-          #
-          # @param [String, URI] uri
-          #   The URI of the remote repository.
-          #
-          def install(uri)
-            repo = begin
-                     Repository.install!(:uri => uri, :scm => scm)
-                   rescue DuplicateRepository => e
-                     print_error e.message
-                     exit -1
-                   end
-
-            print_cache_errors(repo)
-            print_info "Repository #{repo} has been installed."
-          end
-
-          #
-          # Updates installed Repositories.
-          #
-          def update
-            repos = if name
-                      begin
-                        [Repository.find(name)]
-                      rescue RepositoryNotFound => e
-                        print_error e.message
-                        exit -1
-                      end
-                    else
-                      Repository.all
-                    end
-
-            repos.each do |repo|
-              print_info "Updating repository #{repo} ..."
-
-              repo.update!
-
-              print_cache_errors(repo)
-              print_info "Repository updated."
-            end
-          end
-
-          #
-          # Uninstalls a Repository.
-          #
-          # @param [String] name
-          #   The name of the repository.
-          #
-          def uninstall(name)
-            repo = Repository.uninstall!(name)
-
-            print_info "Uninstalling repository #{repo} ..."
-          end
-
         end
       end
     end