ronin-db 0.1.3 → 0.2.0

data/lib/ronin/db/cli/database_command.rb

@@ -1,71 +0,0 @@
-# frozen_string_literal: true
-#
-# ronin-db - A common database library for managing and querying security data.
-#
-# Copyright (c) 2006-2023 Hal Brodigan (postmodern.mod3 at gmail.com)
-#
-# ronin-db is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Lesser General Public License as published
-# by the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
-#
-# ronin-db 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 Lesser General Public License for more details.
-#
-# You should have received a copy of the GNU Lesser General Public License
-# along with ronin-db.  If not, see <https://www.gnu.org/licenses/>.
-#
-
-require 'ronin/db/cli/command'
-require 'ronin/db/cli/uri_methods'
-require 'ronin/db'
-
-module Ronin
-  module DB
-    class CLI
-      #
-      # Base class for all commands that access the database.
-      #
-      class DatabaseCommand < Command
-
-        include URIMethods
-
-        option :db, value: {
-                      type:    DB.config.keys,
-                      default: :default,
-                      usage:   'NAME'
-                    },
-                    desc: 'The database to connect to'
-
-        option :db_uri, value: {
-                          type:  String,
-                          usage: 'URI'
-                        },
-                        desc: 'The database URI to connect to'
-
-        #
-        # The database connection configuration.
-        #
-        # @return [Hash{Symbol => String,Integer}]
-        #
-        def config
-          if options[:db_uri]
-            parse_uri(options[:db_uri])
-          else
-            DB.config[options[:db]]
-          end
-        end
-
-        #
-        # Connects to the database.
-        #
-        def connect
-          DB.connect(config)
-        end
-
-      end
-    end
-  end
-end

data/lib/ronin/db/cli/resources_command.rb

@@ -1,118 +0,0 @@
-# frozen_string_literal: true
-#
-# ronin-db - A common database library for managing and querying security data.
-#
-# Copyright (c) 2006-2023 Hal Brodigan (postmodern.mod3 at gmail.com)
-#
-# ronin-db is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Lesser General Public License as published
-# by the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
-#
-# ronin-db 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 Lesser General Public License for more details.
-#
-# You should have received a copy of the GNU Lesser General Public License
-# along with ronin-db.  If not, see <https://www.gnu.org/licenses/>.
-#
-
-require 'ronin/ui/cli/model_command'
-require 'ronin/db/model/importable'
-
-require 'dm-serializer'
-
-module Ronin
-  module DB
-    module CLI
-      #
-      # A base-command class for listing Database Resources.
-      #
-      class ResourcesCommand < ModelCommand
-
-        option :csv, type:        true,
-                     description: 'CSV output'
-
-        option :xml, type:        true,
-                     description: 'XML output'
-
-        option :yaml, type:        true,
-                      description: 'YAML output'
-
-        option :json, type:        true,
-                      description: 'JSON output'
-
-        #
-        # Default method performs the query and prints the found resources.
-        #
-        # @since 1.1.0
-        #
-        # @api semipublic
-        #
-        def execute
-          if @import
-            self.class.model.import(@import) do |resource|
-              print_info "Imported #{resource}"
-            end
-          else
-            print_resources(query)
-          end
-        end
-
-        #
-        # Sets the model used by the command.
-        #
-        # @see ModelCommand.model
-        #
-        # @since 1.3.0
-        #
-        def self.model(model=nil)
-          if (model && model < Model::Importable)
-            option :import, type:        String,
-                            flag:        '-i',
-                            usage:       'FILE',
-                            description: 'The file to import'
-          end
-
-          return super(model)
-        end
-
-        #
-        # Default method which will print every queried resource.
-        #
-        # @param [DataMapper::Resource] resource
-        #   A queried resource from the Database.
-        #
-        # @since 1.1.0
-        #
-        # @api semipublic
-        #
-        def print_resource(resource)
-          puts resource
-        end
-
-        #
-        # Prints multiple resources.
-        #
-        # @param [DataMapper::Collection] resources
-        #   The query to print.
-        #
-        # @since 1.1.0
-        #
-        # @api semipublic
-        #
-        def print_resources(resources)
-          if    csv?  then puts resources.to_csv
-          elsif xml?  then puts resources.to_xml
-          elsif yaml? then puts resources.to_yaml
-          elsif json? then puts resources.to_json
-          else
-            resources.each { |resource| print_resource(resource) }
-          end
-        end
-
-      end
-    end
-  end
-end