command_kit 0.1.0 → 0.2.0

data/lib/command_kit/os.rb

@@ -4,19 +4,102 @@ module CommandKit
   #
   # ## Examples
   #
-  #     include CommandKit::OS
+  #     require 'command_kit/command'
+  #     require 'command_kit/os'
   #     
-  #     def main(*argv)
-  #       if linux?
-  #         # ...
-  #       elsif macos?
-  #         # ...
-  #       elsif windows?
-  #         # ...
+  #     class Command < CommandKit::Command
+  #     
+  #       include CommandKit::OS
+  #     
+  #       def main(*argv)
+  #         if linux?
+  #           # ...
+  #         elsif macos?
+  #           # ...
+  #         elsif freebsd?
+  #           # ...
+  #         elsif windows?
+  #           # ...
+  #         end
   #       end
+  #     
   #     end
   #
   module OS
+    #
+    # @api private
+    #
+    module ModuleMethods
+      #
+      # Extends {ClassMethods} or {ModuleMethods}, depending on whether
+      # {OS} is being included into a class or a module..
+      #
+      # @param [Class, Module] context
+      #   The class or module which is including {OS}.
+      #
+      def included(context)
+        super(context)
+
+        if context.class == Module
+          context.extend ModuleMethods
+        else
+          context.extend ClassMethods
+        end
+      end
+    end
+
+    extend ModuleMethods
+
+    module ClassMethods
+      #
+      # Determines the current OS.
+      #
+      # @return [:linux, :macos, :freebsd, :openbsd, :netbsd, :windows, nil]
+      #   The OS type or `nil` if the OS could not be determined.
+      #
+      # @api semipublic
+      #
+      # @since 0.2.0
+      #
+      def os
+        if    RUBY_PLATFORM.include?('linux')   then :linux
+        elsif RUBY_PLATFORM.include?('darwin')  then :macos
+        elsif RUBY_PLATFORM.include?('freebsd') then :freebsd
+        elsif RUBY_PLATFORM.include?('openbsd') then :openbsd
+        elsif RUBY_PLATFORM.include?('netbsd')  then :netbsd
+        elsif Gem.win_platform?                 then :windows
+        end
+      end
+    end
+
+    # The current OS.
+    #
+    # @return [:linux, :macos, :freebsd, :openbsd, :netbsd, :windows, nil]
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    attr_reader :os
+
+    #
+    # Initializes the command.
+    #
+    # @param [:linux, :macos, :freebsd, :openbsd, :netbsd, :windows, nil] os
+    #   Overrides the default OS.
+    #   
+    # @param [Hash{Symbol => Object}] kwargs
+    #   Additional keyword arguments.
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    #
+    def initialize(os: self.class.os, **kwargs)
+      super(**kwargs)
+
+      @os = os
+    end
+
     #
     # Determines if the current OS is Linux.
     #
@@ -25,7 +108,7 @@ module CommandKit
     # @api public
     #
     def linux?
-      RUBY_PLATFORM.include?('linux')
+      @os == :linux
     end
 
     #
@@ -36,7 +119,72 @@ module CommandKit
     # @api public
     #
     def macos?
-      RUBY_PLATFORM.include?('darwin')
+      @os == :macos
+    end
+
+    #
+    # Determines if the current OS is FreeBSD.
+    #
+    # @return [Boolean]
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    #
+    def freebsd?
+      @os == :freebsd
+    end
+
+    #
+    # Determines if the current OS is OpenBSD.
+    #
+    # @return [Boolean]
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    #
+    def openbsd?
+      @os == :openbsd
+    end
+
+    #
+    # Determines if the current OS is NetBSD.
+    #
+    # @return [Boolean]
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    #
+    def netbsd?
+      @os == :netbsd
+    end
+
+    #
+    # Determines if the current OS is BSD based.
+    #
+    # @return [Boolean]
+    #
+    # @since 0.2.0
+    #
+    # @api public
+    #
+    def bsd?
+      freebsd? || openbsd? || netbsd?
+    end
+
+    #
+    # Determines if the current OS is UNIX based.
+    #
+    # @return [Boolean]
+    #
+    # @since 0.2.0
+    #
+    # @api public
+    #
+    def unix?
+      linux? || macos? || bsd?
     end
 
     #
@@ -47,7 +195,7 @@ module CommandKit
     # @api public
     #
     def windows?
-      Gem.win_platform?
+      @os == :windows
     end
   end
 end

data/lib/command_kit/package_manager.rb

@@ -0,0 +1,200 @@
+require 'command_kit/os'
+require 'command_kit/os/linux'
+require 'command_kit/env/path'
+require 'command_kit/sudo'
+
+module CommandKit
+  #
+  # Allows installing packages using the system's package manager.
+  #
+  # Supports the following package managers:
+  #
+  # * Linux
+  #   * Debian / Ubuntu
+  #     * `apt`
+  #   * RedHat / Fedora
+  #     * `dnf`
+  #     * `yum`
+  #   * Arch
+  #     * `pacman`
+  #   * SUSE / OpenSUSE
+  #     * `zypper`
+  # * macOS
+  #   * `brew`
+  #   * `port`
+  # * FreeBSD
+  #   * `pkg`
+  # * OpenBSD
+  #   * `pkg_add`
+  #
+  # ## Examples
+  #
+  #     unless install_packages("nmap")
+  #       print_error "failed to install nmap"
+  #       exit -1
+  #     end
+  #
+  # ### Installing multiple packages
+  #
+  #     install_packages apt: ["libxml2-dev", ...],
+  #                      dnf: ["libxml2-devel", ...],
+  #                      brew: ["libxml2", ...],
+  #                      ...
+  #
+  # @since 0.2.0
+  #
+  module PackageManager
+    include OS
+    include OS::Linux
+    include Env::Path
+    include Sudo
+
+    # The detected package manager.
+    #
+    # @return [:apt, :dnf, :yum, :zypper, :pacman, :brew, :pkg, :pkg_add, nil]
+    attr_reader :package_manager
+
+    #
+    # Initializes the command and determines which open command to use.
+    #
+    # @param [:apt, :dnf, :yum, :zypper, :pacman, :brew, :pkg, :pkg_add, nil] package_manager
+    #   The explicit package manager to use. If `nil`, the package manager will
+    #   be detected.
+    #
+    def initialize(package_manager: nil, **kwargs)
+      super(**kwargs)
+
+      @package_manager = package_manager || begin
+        if macos?
+          if    command_installed?('brew') then :brew
+          elsif command_installed?('port') then :port
+          end
+        elsif linux?
+          if redhat_linux?
+            if    command_installed?('dnf') then :dnf
+            elsif command_installed?('yum') then :yum
+            end
+          elsif debian_linux?
+            :apt if command_installed?('apt')
+          elsif suse_linux?
+            :zypper if command_installed?('zypper')
+          elsif arch_linux?
+            :pacman if command_installed?('pacman')
+          end
+        elsif freebsd?
+          :pkg if command_installed?('pkg')
+        elsif openbsd?
+          :pkg_add if command_installed?('pkg_add')
+        end
+      end
+    end
+
+    #
+    # Installs the packages using the system's package manager.
+    #
+    # @param [Array<String>, String] packages
+    #   A list of package name(s) to install.
+    #
+    # @param [Boolean] yes
+    #   Assume yes for all user prompts.
+    #
+    # @param [Array<String>, String] apt
+    #   List of `apt` specific package names.
+    #
+    # @param [Array<String>, String] brew
+    #   List of `brew` specific package names.
+    #
+    # @param [Array<String>, String] dnf
+    #   List of `dnf` specific package names.
+    #
+    # @param [Array<String>, String] pacman
+    #   List of `pacman` specific package names.
+    #
+    # @param [Array<String>, String] pkg
+    #   List of `pkg` specific package names.
+    #
+    # @param [Array<String>, String] pkg_add
+    #   List of `pkg_add` specific package names.
+    #
+    # @param [Array<String>, String] port
+    #   List of `port` specific package names.
+    #
+    # @param [Array<String>, String] yum
+    #   List of `yum` specific package names.
+    #
+    # @param [Array<String>, String] zypper
+    #   List of `zypper` specific package names.
+    #
+    # @return [Boolean, nil]
+    #   Specifies whether the packages were successfully installed or not.
+    #   If the package manager command could not be determined, `nil` is
+    #   returned.
+    #
+    # @example Install a package
+    #   install_packages "nmap", ...
+    #
+    # @example Install a list of packages per package-manager
+    #   install_packages apt: ["libxml2-dev", ...],
+    #                    dnf: ["libxml2-devel", ...],
+    #                    brew: ["libxml2", ...],
+    #                    ...
+    #
+    def install_packages(*packages, yes: false,
+                                    apt: nil,
+                                    brew: nil,
+                                    dnf: nil,
+                                    pacman: nil,
+                                    pkg: nil,
+                                    pkg_add: nil,
+                                    port: nil,
+                                    yum: nil,
+                                    zypper: nil)
+      specific_package_names = case @package_manager
+                               when :apt     then apt
+                               when :brew    then brew
+                               when :dnf     then dnf
+                               when :pacman  then pacman
+                               when :pkg     then pkg
+                               when :pkg_add then pkg_add
+                               when :port    then port
+                               when :yum     then yum
+                               when :zypper  then zypper
+                               end
+      packages += Array(specific_package_names)
+
+      case @package_manager
+      when :apt
+        args = []
+        args << '-y' if yes
+
+        sudo('apt','install',*args,*packages)
+      when :brew
+        system('brew','install',*packages)
+      when :dnf, :yum
+        args = []
+        args << '-y' if yes
+
+        sudo(@package_manager.to_s,'install',*args,*packages)
+      when :pacman
+        missing_packages = `pacman -T #{Shellwords.shelljoin(packages)}`.split
+
+        if missing_packages.empty?
+          return true
+        end
+
+        sudo('pacman','-S',*missing_packages)
+      when :pkg
+        args = []
+        args << '-y' if yes
+
+        sudo('pkg','install',*args,*packages)
+      when :pkg_add
+        sudo('pkg_add',*packages)
+      when :port
+        sudo('port','install',*packages)
+      when :zypper
+        sudo('zypper','-n','in','-l',*packages)
+      end
+    end
+  end
+end

data/lib/command_kit/pager.rb

@@ -5,6 +5,8 @@ require 'command_kit/env/path'
 require 'command_kit/stdio'
 require 'command_kit/terminal'
 
+require 'shellwords'
+
 module CommandKit
   #
   # Allows opening a pager, such as `less` or `more`.
@@ -50,7 +52,7 @@ module CommandKit
     #   Keyword arguments.
     #
     # @note
-    #   Respects the `PAGER` env variable, or attemps to find `less` or
+    #   Respects the `PAGER` env variable, or attempts to find `less` or
     #   `more` by searching the `PATH` env variable.
     #
     # @api public
@@ -58,7 +60,7 @@ module CommandKit
     def initialize(**kwargs)
       super(**kwargs)
 
-      @pager = env.fetch('PAGER') do
+      @pager_command = env.fetch('PAGER') do
         PAGERS.find do |command|
           bin = command.split(' ',2).first
 
@@ -94,14 +96,14 @@ module CommandKit
     # @api public
     #
     def pager
-      if !stdout.tty? || @pager.nil?
+      if !stdout.tty? || @pager_command.nil?
         # fallback to stdout if the process does not have a terminal or we could
         # not find a suitable pager command.
         yield stdout
         return
       end
 
-      io  = IO.popen(@pager,'w')
+      io  = IO.popen(@pager_command,'w')
       pid = io.pid
 
       begin
@@ -144,5 +146,45 @@ module CommandKit
         stdout.puts(data)
       end
     end
+
+    #
+    # Pipes a command into the pager.
+    #
+    # @param [#to_s] command
+    #   The program or command to run.
+    #
+    # @param [Array<#to_s>] arguments
+    #   Additional arguments for the program.
+    #
+    # @return [Boolean]
+    #   Indicates whether the command exited successfully or not.
+    #
+    # @note
+    #   If multiple arguments are given, they will be shell-escaped and executed
+    #   as a single command.
+    #   If a single command is given, it will not be shell-escaped to allow
+    #   executing compound shell commands.
+    #
+    # @example Pipe a single command into the pager:
+    #   run_in_pager 'find', '.', '-name', '*.md'
+    # 
+    # @example Pipe a compound command into the pager:
+    #   run_in_pager "wc -l /path/to/wordlists/*.txt | sort -n"
+    #
+    # @api public
+    #
+    # @since 0.2.0
+    #
+    def pipe_to_pager(command,*arguments)
+      if @pager_command
+        unless arguments.empty?
+          command = Shellwords.shelljoin([command, *arguments])
+        end
+
+        system("#{command} | #{@pager_command}")
+      else
+        system(command.to_s,*arguments.map(&:to_s))
+      end
+    end
   end
 end