virustotalx 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b93b0c39cdd1dbf4efdeffbe4f9ed6c30892f75a3a73172b1e5eb2db11678334
-  data.tar.gz: 38be1389296d3bd50edbf0bdb8906b83a2639a681b0b91b7a9fc1df8539a35b4
+  metadata.gz: f014d99e8a184fa94a9f2d696209672867c2d437c83bf094b49346078887b077
+  data.tar.gz: 0ff81e8a4a61b7283a8d54fe329127d602b07847f8a62ca70f7aa4112b6a6187
 SHA512:
-  metadata.gz: 86a227bded77dba22902a4531f09683cfc618c850ae178762f6ba5cd1438dbd546313bcd67f27041af3b562dcb0d1e047b570f160ad5a6db91a773b9e8aa6edc
-  data.tar.gz: 1b73b35965ee06d99b02fc5674d30ea234f5a64909a2a8a4dfb12cac6c38df8b648ada8c0aaf981338ca3e61accd267a971ddb3883a8eee0fb25f1774b387d4e
+  metadata.gz: c957e719c7a2f9be34493128a95369f5a0bd0f0a534570f9466c70af1cec243daba9198b516cc21af576bd3f5d31b01a65dc017cd6a946663c6d9f3009906d04
+  data.tar.gz: a0497c176fc774a95193452590226f9887792aed9a42a1e144859b0db99684195c75d1e6ca8eb5de833a00a72b506b86f88bf347f621ea3f84b865c4960ef4e5

data/README.md

@@ -39,12 +39,12 @@ api.ip_address.get("1.1.1.1")
 
 ## Supported API endpoints
 
-* [VirusTotal API reference](https://developers.virustotal.com/v3.0/reference#overview)
+- [VirusTotal API reference](https://developers.virustotal.com/v3.0/reference#overview)
 
 ### Files
 
 | HTTP Method | URL                                | API method                                                 |
-|-------------|------------------------------------|------------------------------------------------------------|
+| ----------- | ---------------------------------- | ---------------------------------------------------------- |
 | POST        | /files                             | api.file.upload(filepath)                                  |
 | GET         | /files/upload_url                  | api.file.upload_url                                        |
 | GET         | /files/{id}                        | api.file.get(id)                                           |
@@ -62,7 +62,7 @@ api.ip_address.get("1.1.1.1")
 ### URLs
 
 | HTTP Method | URL                         | API method                                                     |
-|-------------|-----------------------------|----------------------------------------------------------------|
+| ----------- | --------------------------- | -------------------------------------------------------------- |
 | POST        | /urls                       | N/A                                                            |
 | GET         | /urls/{id}                  | api.url.get(id)                                                |
 | POST        | /urls/{id}/analyse          | api.url.analyse(id)                                            |
@@ -78,7 +78,7 @@ Note: you can use a URL as an id.
 ### Domains
 
 | HTTP Method | URL                              | API method                                                   |
-|-------------|----------------------------------|--------------------------------------------------------------|
+| ----------- | -------------------------------- | ------------------------------------------------------------ |
 | GET         | /domains/{domain}                | api.domain.get(domain)                                       |
 | GET         | /domains/{domain}/comments       | api.domain.comment(domain)                                   |
 | POST        | /domains/{domain}/comments       | api.domain.add_comment(domain, text)                         |
@@ -87,7 +87,7 @@ Note: you can use a URL as an id.
 ### IP addresses
 
 | HTTP Method | URL                               | API method                                                                      |
-|-------------|-----------------------------------|---------------------------------------------------------------------------------|
+| ----------- | --------------------------------- | ------------------------------------------------------------------------------- |
 | GET         | /ip_addresses/{ip}                | api.ip_address.get(ip)                                                          |
 | GET         | /ip_addresses/{ip}/comments       | api.ip_address.comments(id)                                                     |
 | POST        | /ip_addresses/{ip}/comments       | api.ip_address.add_comment(id, text)                                            |
@@ -96,12 +96,26 @@ Note: you can use a URL as an id.
 ### Analyses
 
 | HTTP Method | URL            | API method           |
-|-------------|----------------|----------------------|
+| ----------- | -------------- | -------------------- |
 | GET         | /analyses/{id} | api.analysis.get(ip) |
 
 ## Graphs
 
-N/A.
+| HTTP Method | URL                                                  | API method                                                                          |
+| ----------- | ---------------------------------------------------- | ----------------------------------------------------------------------------------- |
+| GET         | graphs                                               | api.graph.search(filter: nil, limit: nil, cursor: nil, order: nil, attributes: nil) |
+| POST        | graphs                                               | api.graph.create(\*\*params)                                                        |
+| GET         | graphs/{id}                                          | api.graph.get(id)                                                                   |
+| PATCH       | graphs/{id}                                          | api.graph.update(id, \*\*params)                                                    |
+| GET         | graphs/{id}/relationships/viewers                    | api.graph.viewers(id, limit: nil, cursor: nil)                                      |
+| POST        | graphs/{id}/relationships/viewers                    | api.graph.add_viewer(id, \*\*params)                                                |
+| GET         | graphs/{id}/relationships/viewers/{user_or_group_id} | api.graph.check_viewer(id, user_or_group_id)                                        |
+| DELETE      | graphs/{id}/relationships/viewers/{user_or_group_id} | api.graph.delete_viewer(id, user_or_group_id)                                       |
+| GET         | graphs/{id}/relationships/editors                    | api.graph.editors(id, limit: nil, cursor: nil)                                      |
+| POST        | graphs/{id}/relationships/editors                    | api.graph.add_editor(id, \*\*params)                                                |
+| GET         | graphs/{id}/relationships/editors/{user_or_group_id} | api.graph.check_editor(id, user_or_group_id)                                        |
+| DELETE      | graphs/{id}/relationships/editors/{user_or_group_id} | api.graph.delete_editor(id, user_or_group_id)                                       |
+| GET         | graphs/{id}/relationships/{relationship}             | api.graph.`relationship`(id)                                                        |
 
 ## License
 

data/lib/virustotal.rb

@@ -7,11 +7,16 @@ require "virustotal/errors"
 require "virustotal/api"
 
 require "virustotal/clients/base"
-require "virustotal/clients/object"
+
+require "virustotal/clients/actions/comments"
+require "virustotal/clients/actions/get"
+require "virustotal/clients/actions/relationships"
+require "virustotal/clients/actions/votes"
 
 require "virustotal/clients/analysis"
 require "virustotal/clients/domain"
 require "virustotal/clients/file"
+require "virustotal/clients/graph"
 require "virustotal/clients/ip_address"
 require "virustotal/clients/url"
 

data/lib/virustotal/api.rb

@@ -1,23 +1,73 @@
 # frozen_string_literal: true
 
-require_relative "clients/base"
-
 module VirusTotal
   class API
-    attr_reader :analysis
-    attr_reader :domain
-    attr_reader :file
-    attr_reader :ip_address
-    attr_reader :url
+    # @return [String] VirusTotal API key
+    attr_reader :key
 
+    #
+    # @param [String] key VirusTotal API key
+    #
+    # @raise [ArgumentError] When given an empty key
+    #
     def initialize(key: ENV["VIRUSTOTAL_API_KEY"])
+      @key = key
+
       raise ArgumentError, "No API key has been found or provided! (setup your VIRUSTOTAL_API_KEY environment varialbe)" unless key
+    end
+
+    #
+    # Analyses API endpoint client
+    #
+    # @return [VirusTotal::Client::Analysis]
+    #
+    def analysis
+      @analysis ||= Client::Analysis.new(key: key)
+    end
+
+    #
+    # Domains API endpoint client
+    #
+    # @return [VirusTotal::Client::Domain]
+    #
+    def domain
+      @domain ||= Client::Domain.new(key: key)
+    end
+
+    #
+    # Files API endpoint client
+    #
+    # @return [VirusTotal::Client::File]
+    #
+    def file
+      @file ||= Client::File.new(key: key)
+    end
+
+    #
+    # IP addresses API endpoint client
+    #
+    # @return [VirusTotal::Client::IPAddress]
+    #
+    def ip_address
+      @ip_address ||= Client::IPAddress.new(key: key)
+    end
+
+    #
+    # URLs API endpoint client
+    #
+    # @return [VirusTotal::Client::URL]
+    #
+    def url
+      @url ||= Client::URL.new(key: key)
+    end
 
-      @analysis = Client::Analysis.new(key: key)
-      @domain = Client::Domain.new(key: key)
-      @file = Client::File.new(key: key)
-      @ip_address = Client::IPAddress.new(key: key)
-      @url = Client::URL.new(key: key)
+    #
+    # Graphs API endpoint client
+    #
+    # @return [VirusTotal::Client::Graph]
+    #
+    def graph
+      @graph ||= Client::Graph.new(key: key)
     end
   end
 end

data/lib/virustotal/clients/actions/comments.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module VirusTotal
+  module Client
+    module Action
+      module Comments
+        #
+        # Retrieve comments for an object
+        #
+        # @param [String] id Object identifier
+        #
+        # @return [Hash]
+        #
+        def comments(id)
+          id = to_id(id)
+          _get("/#{name}/#{id}/comments") { |json| json }
+        end
+
+        #
+        # Add a comment to an object
+        #
+        # @param [String] id Object identifier
+        # @param [String] text
+        #
+        # @return [Hash]
+        #
+        def add_comment(id, text)
+          id = to_id(id)
+          params = {
+            data: {
+              type: "comment",
+              attributes: {
+                text: text
+              }
+            }
+          }
+          _post("/#{name}/#{id}/comments", params) { |json| json }
+        end
+      end
+    end
+  end
+end

data/lib/virustotal/clients/actions/get.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module VirusTotal
+  module Client
+    module Action
+      module Get
+        #
+        # Retrieve information about an object
+        #
+        # @param [String] id Object identifier
+        #
+        # @return [Hash]
+        #
+        def get(id)
+          id = to_id(id)
+          _get("/#{name}/#{id}") { |json| json }
+        end
+      end
+    end
+  end
+end

data/lib/virustotal/clients/actions/relationships.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module VirusTotal
+  module Client
+    module Action
+      module Relationships
+        def relationships
+          []
+        end
+
+        def method_missing(method, *args)
+          if relationships.include?(method)
+            id = to_id(args.first)
+            params = args.length == 2 ? args[1] : {}
+
+            _get("/#{name}/#{id}/#{method}", params) { |json| json }
+          else
+            super
+          end
+        end
+
+        def respond_to_missing?(method_name, *)
+          relationships.include? method_name
+        end
+      end
+    end
+  end
+end

data/lib/virustotal/clients/actions/votes.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module VirusTotal
+  module Client
+    module Action
+      module Votes
+        #
+        # Retrieve votes for an object
+        #
+        # @param [String] id Object identifier
+        #
+        # @return [Hash]
+        #
+        def votes(id)
+          id = to_id(id)
+          _get("/#{name}/#{id}/votes") { |json| json }
+        end
+
+        #
+        # Add a vote for an object
+        #
+        # @param [String] id Object identifier
+        # @param [String] verdict harmless or malicious
+        #
+        # @return [Hash]
+        #
+        def add_vote(id, verdict)
+          id = to_id(id)
+          params = {
+            data: {
+              type: "vote",
+              attributes: {
+                verdict: verdict
+              }
+            }
+          }
+          _post("/#{name}/#{id}/votes", params) { |json| json }
+        end
+      end
+    end
+  end
+end

data/lib/virustotal/clients/analysis.rb

@@ -3,9 +3,7 @@
 module VirusTotal
   module Client
     class Analysis < Base
-      def get(id)
-        _get("/analyses/#{id}") { |json| json }
-      end
+      include Action::Get
     end
   end
 end

data/lib/virustotal/clients/base.rb

@@ -11,6 +11,15 @@ module VirusTotal
       VERSION = "v3"
       BASE_URL = "https://#{HOST}/api/#{VERSION}"
 
+      CONVERT_TABLE = {
+        analysis: "analyses",
+        domain: "domains",
+        file: "files",
+        graph: "graphs",
+        ipaddress: "ip_addresses",
+        url: "urls",
+      }.freeze
+
       attr_reader :key
 
       def initialize(key: ENV["VIRUSTOTAL_API_KEY"])
@@ -111,6 +120,32 @@ module VirusTotal
 
         request(post, &block)
       end
+
+      def _patch(path, params = {}, &block)
+        patch = Net::HTTP::Patch.new(url_for(path))
+        patch.body = JSON.generate(params) if params
+
+        request(patch, &block)
+      end
+
+      def _delete(path, params = {}, &block)
+        delete = Net::HTTP::Delete.new(url_for(path))
+        delete.body = JSON.generate(params) if params
+
+        request(delete, &block)
+      end
+
+      def to_id(id)
+        id
+      end
+
+      def klass
+        self.class.to_s.split("::").last.to_s.downcase.to_sym
+      end
+
+      def name
+        CONVERT_TABLE.fetch klass
+      end
     end
   end
 end

data/lib/virustotal/clients/domain.rb

@@ -2,11 +2,15 @@
 
 module VirusTotal
   module Client
-    class Domain < Object
+    class Domain < Base
+      include Action::Comments
+      include Action::Get
+      include Action::Relationships
+
       private
 
       def relationships
-        %w(
+        @relationships ||= %w(
           communicating_files
           downloaded_files
           graphs

data/lib/virustotal/clients/file.rb

@@ -2,45 +2,86 @@
 
 module VirusTotal
   module Client
-    class File < Object
+    class File < Base
+      include Action::Comments
+      include Action::Get
+      include Action::Relationships
+      include Action::Votes
+
+      #
+      # Upload and analyse a file
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#files-scan
+      #
+      # @param [String] path File path to be scanned
+      #
+      # @return [Hash]
+      #
       def upload(path)
         name = ::File.basename(path)
         data = ::File.read(path)
         _post_with_file("/files", file: data, filename: name) { |json| json }
       end
 
+      #
+      # Get a URL for uploading files larger than 32MB
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#files-upload-url
+      #
+      # @return [Hash]
+      #
       def upload_url
         _get("/files/upload_url") { |json| json }
       end
 
+      #
+      # Reanalyse a file already in VirusTotal
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#files-analyse
+      #
+      # @param [String] hash SHA-256, SHA-1 or MD5 identifying the file
+      #
+      # @return [Hash]
+      #
       def analyse(hash)
         _post("/files/#{hash}/analyse") { |json| json }
       end
 
-      def votes(hash)
-        _get("/files/#{hash}/votes") { |json| json }
-      end
-
-      def add_vote(hash, verdict)
-        params = {
-          data: {
-            type: "vote",
-            attributes: {
-              verdict: verdict
-            }
-          }
-        }
-        _post("/files/#{hash}/votes", params) { |json| json }
-      end
-
+      #
+      # Get a download URL for a file
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#files-download-url
+      #
+      # @param [String] hash SHA-256, SHA-1 or MD5 identifying the file
+      #
+      # @return [Hash]
+      #
       def download_url(hash)
         _get("/files/#{hash}/download_url") { |json| json }
       end
 
+      #
+      # Download a file
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#files-download
+      #
+      # @param [String] SHA-256, SHA-1 or MD5 identifying the file
+      #
+      # @return [<Type>] <description>
+      #
       def download(hash)
         _get("/files/#{hash}/download") { |location| location }
       end
 
+      #
+      # Retrieve objects related to a file
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#file_behaviours_pcap
+      #
+      # @param [String] id SHA-256, SHA-1 or MD5 identifying the file
+      #
+      # @return [Hash]
+      #
       def pcap(id)
         _get("/file_behaviours/#{id}/pcap") { |raw| raw }
       end
@@ -48,7 +89,7 @@ module VirusTotal
       private
 
       def relationships
-        %w(
+        @relationships ||= %w(
           analyses
           behaviours
           bundled_files

data/lib/virustotal/clients/graph.rb

@@ -0,0 +1,195 @@
+# frozen_string_literal: true
+
+module VirusTotal
+  module Client
+    class Graph < Base
+      include Action::Get
+
+      #
+      # Return graphs.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs
+      #
+      # @param [String, nil] filter Return the graphs matching the given criteria only
+      # @param [Integer, nil] limit Maximum number graphs to retrieve
+      # @param [String, nil] cursor Continuation cursor
+      # @param [String, nil] order Sort order
+      # @param [String, nil] attributes Specific fields to retrieve
+      #
+      # @return [Hash]
+      #
+      def search(filter: nil, limit: nil, cursor: nil, order: nil, attributes: nil)
+        params = {
+          filter: filter,
+          limit: limit,
+          cursor: cursor,
+          order: order,
+          attributes: attributes
+        }.compact
+        _get("/graphs", params) { |json| json }
+      end
+
+      #
+      # Create a graph with the given nodes and links.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#create-graphs
+      #
+      # @param [Hash] **params
+      #
+      # @return [Hash]
+      #
+      def create(**params)
+        _post("/graphs", params) { |json| json }
+      end
+
+      #
+      # Update a graph.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-update
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [Hash] **params
+      #
+      # @return [Hash]
+      #
+      def update(id, **params)
+        _patch("/graphs/#{id}", params) { |json| json }
+      end
+
+      #
+      # Retrieve graph viewers.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-viewers
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String, nil] limit Maximum number of related objects to retrieve
+      # @param [Integer, nil] cursor Continuation cursor
+      #
+      # @return [Hash]
+      #
+      def viewers(id, limit: nil, cursor: nil)
+        params = {
+          limit: limit,
+          cursor: cursor
+        }.compact
+        _get("/graphs/#{id}/relationships/viewers", params) { |json| json }
+      end
+
+      #
+      # Add a user or group as a graph viewer.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-add-viewer
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [Hash] **params
+      #
+      # @return [Hash]
+      #
+      def add_viewer(id, **params)
+        _post("/graphs/#{id}/relationships/viewers", params) { |json| json }
+      end
+
+      #
+      # Check if a user or group is a graph viewer
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-check-viewer
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String] user_or_group_id user or group ID
+      #
+      # @return [<Type>] <description>
+      #
+      def check_viewer(id, user_or_group_id)
+        _get("/graphs/#{id}/relationships/viewers/#{user_or_group_id}") { |json| json }
+      end
+
+      #
+      # Remove a user or group as viewer of a graph.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-delete-viewer
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String] user_or_group_id user or group ID
+      #
+      # @return [<Type>] <description>
+      #
+      def delete_viewer(id, user_or_group_id)
+        _delete("/graphs/#{id}/relationships/viewers/#{user_or_group_id}") { |json| json }
+      end
+
+      #
+      # Retrieve graph editors.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-editors
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String, nil] limit Maximum number of related objects to retrieve
+      # @param [Integer, nil] cursor Continuation cursor
+      #
+      # @return [Hash]
+      #
+      def editors(id, limit: nil, cursor: nil)
+        params = {
+          limit: limit,
+          cursor: cursor
+        }.compact
+        _get("/graphs/#{id}/relationships/editors", params) { |json| json }
+      end
+
+      #
+      # Add a user or group as a graph editor.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-add-editor
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [Hash] **params
+      #
+      # @return [Hash]
+      #
+      def add_editor(id, **params)
+        _post("/graphs/#{id}/relationships/editors", params) { |json| json }
+      end
+
+      #
+      # Check if a user or group is a graph editor.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-check-editor
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String] user_or_group_id user or group ID
+      #
+      # @return [Hash]
+      #
+      def check_editor(id, user_or_group_id)
+        _get("/graphs/#{id}/relationships/editors/#{user_or_group_id}") { |json| json }
+      end
+
+      #
+      # Remove a user or group as editor of a graph.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#graphs-delete-editor
+      #
+      # @param [String] id A 65 char length id which uniquely identify the graph.
+      # @param [String] user_or_group_id user or group ID
+      #
+      # @return [Hash]
+      #
+      def delete_editor(id, user_or_group_id)
+        _delete("/graphs/#{id}/relationships/editors/#{user_or_group_id}") { |json| json }
+      end
+
+      private
+
+      def relationships
+        @relationships ||= %w(
+          comments
+          items
+          viewers
+          editors
+          owner
+          group
+        ).map(&:to_sym)
+      end
+    end
+  end
+end

data/lib/virustotal/clients/ip_address.rb

@@ -2,11 +2,15 @@
 
 module VirusTotal
   module Client
-    class IPAddress < Object
+    class IPAddress < Base
+      include Action::Comments
+      include Action::Get
+      include Action::Relationships
+
       private
 
       def relationships
-        %w(
+        @relationships ||= %w(
           communicating_files
           downloaded_files
           graphs

data/lib/virustotal/clients/url.rb

@@ -4,35 +4,40 @@ require "base64"
 
 module VirusTotal
   module Client
-    class URL < Object
+    class URL < Base
+      include Action::Comments
+      include Action::Get
+      include Action::Relationships
+      include Action::Votes
+
+      #
+      # Analyse an URL.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#urls-analyse
+      #
+      # @param [String] url URL identifier
+      #
+      # @return [Hash]
+      #
       def analyse(url)
         id = to_id(url)
         _post("/urls/#{id}/analyse") { |json| json }
       end
 
-      def votes(url)
-        id = to_id(url)
-        _get("/urls/#{id}/votes") { |json| json }
-      end
-
+      #
+      # Domain or IP address for a URL.
+      #
+      # @see https://developers.virustotal.com/v3.0/reference#urlsidnetwork_location
+      #
+      # @param [String] url URL identifier
+      #
+      # @return [Hash]
+      #
       def network_location(url)
         id = to_id(url)
         _get("/urls/#{id}/network_location") { |json| json }
       end
 
-      def add_vote(url, verdict)
-        id = to_id(url)
-        params = {
-          data: {
-            type: "vote",
-            attributes: {
-              verdict: verdict
-            }
-          }
-        }
-        _post("/urls/#{id}/votes", params) { |json| json }
-      end
-
       private
 
       def to_id(url)
@@ -40,7 +45,7 @@ module VirusTotal
       end
 
       def relationships
-        %w(
+        @relationships ||= %w(
           analyses
           downloaded_files
           graphs

data/lib/virustotal/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module VirusTotal
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

data/virustotalx.gemspec

@@ -26,8 +26,8 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", "~> 2.0"
   spec.add_development_dependency "coveralls", "~> 0.8"
-  spec.add_development_dependency "rake", "~> 12.3"
-  spec.add_development_dependency "rspec", "~> 3.8"
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_development_dependency "rspec", "~> 3.9"
   spec.add_development_dependency "vcr", "~> 5.0"
   spec.add_development_dependency "webmock", "~> 3.7"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: virustotalx
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Manabu Niseki
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-09-26 00:00:00.000000000 Z
+date: 2019-10-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -44,28 +44,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.3'
+        version: '13.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.3'
+        version: '13.0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.8'
+        version: '3.9'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.8'
+        version: '3.9'
 - !ruby/object:Gem::Dependency
   name: vcr
   requirement: !ruby/object:Gem::Requirement
@@ -112,12 +112,16 @@ files:
 - bin/setup
 - lib/virustotal.rb
 - lib/virustotal/api.rb
+- lib/virustotal/clients/actions/comments.rb
+- lib/virustotal/clients/actions/get.rb
+- lib/virustotal/clients/actions/relationships.rb
+- lib/virustotal/clients/actions/votes.rb
 - lib/virustotal/clients/analysis.rb
 - lib/virustotal/clients/base.rb
 - lib/virustotal/clients/domain.rb
 - lib/virustotal/clients/file.rb
+- lib/virustotal/clients/graph.rb
 - lib/virustotal/clients/ip_address.rb
-- lib/virustotal/clients/object.rb
 - lib/virustotal/clients/url.rb
 - lib/virustotal/errors.rb
 - lib/virustotal/version.rb
@@ -142,7 +146,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.4
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Yet another VirusTotal API wrapper for Ruby

data/lib/virustotal/clients/object.rb

@@ -1,72 +0,0 @@
-# frozen_string_literal: true
-
-module VirusTotal
-  module Client
-    class Object < Base
-      CONVERT_TABLE = {
-        ipaddress: "ip_addresses",
-        domain: "domains",
-        url: "urls",
-        file: "file"
-      }.freeze
-
-      def get(id)
-        id = to_id(id)
-        _get("/#{name}/#{id}") { |json| json }
-      end
-
-      def comments(id)
-        id = to_id(id)
-        _get("/#{name}/#{id}/comments") { |json| json }
-      end
-
-      def add_comment(id, text)
-        id = to_id(id)
-        params = {
-          data: {
-            type: "comment",
-            attributes: {
-              text: text
-            }
-          }
-        }
-        _post("/#{name}/#{id}/comments", params) { |json| json }
-      end
-
-      def relationships
-        []
-      end
-
-      def method_missing(method, *args)
-        if relationships.include?(method)
-          id = to_id(args.first)
-          params = args.length == 2 ? args[1] : {}
-
-          _get("/#{name}/#{id}/#{method}", params) { |json| json }
-        else
-          super
-        end
-      end
-
-      def respond_to?(sym, *)
-        return true if relationships.include? sym
-
-        super
-      end
-
-      private
-
-      def to_id(id)
-        id
-      end
-
-      def klass
-        self.class.to_s.split("::").last.to_s.downcase.to_sym
-      end
-
-      def name
-        CONVERT_TABLE.fetch klass
-      end
-    end
-  end
-end