virus_total 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4cd640a88172cb1d3cd160a5be4ef045d36f4125
-  data.tar.gz: 0074bf1f636a55c21c8bdab4300ba496cbadbda1
+  metadata.gz: 308097c5810248bccc4c3e783656b80fc22ba05d
+  data.tar.gz: 95daa10ff9a9263248c8c664a63ffbc766af731f
 SHA512:
-  metadata.gz: 1b9ea12c003845bb2c1922c67cae1c8fa18a1951073360706da6f86167ff9ea01c20e5bf455ef360f8dbfbe917ebf585398439f8a98b75c2d8e2f29d8c1d4f69
-  data.tar.gz: 505752f17dae22f2a85128bd27fb177b282a32f45f700d85acfe9c876dd2f7d2f67439197cd024a2db0bef480b8b687d70f58317fb4ced50bfc9215175461094
+  metadata.gz: 19a0ea7eab3ebc523f4e71a78c97487d6e443b382ef676ab192e10f8a537d54a1049fdf9a6ee82d532bacc7d2cef0603515f4cc44b3fa6241104105a864f3622
+  data.tar.gz: 5e74412cd54ccf0471b422f4d0c376cc2cc9a4f1142f4a864570c1443b93a87b475ad5954ce55dddfc9a0e88c5118fb6cb8e52a59d8019823919205f6d878881

data/lib/core_extentions.rb

@@ -0,0 +1,7 @@
+class Hash
+  def method_missing(name, *args)
+    name = name.to_s
+    name.gsub!("_", " ")
+    self[name]
+  end
+end

data/lib/virus_total/base.rb

@@ -11,6 +11,12 @@ module VirusTotal
       @apikey   = apikey
     end
 
+    def resource(separator = ", ")
+      return @resource.join(separator) if @resource.is_a?(Array)
+
+      @resource
+    end
+
     def api_response(url, params = {})
       full_url = BASE_URL + url
       params = default_params.merge(params)

data/lib/virus_total/domain.rb

@@ -2,7 +2,7 @@ module VirusTotal
   # to use this feature you must have a private apikey!
   class Domain < Base
     def report
-      api_response('domain/report', resource: @resource.to_str, method: :get)
+      api_response('domain/report', resource: @resource.to_s, method: :get)
     end
   end
 end

data/lib/virus_total/file.rb

@@ -5,18 +5,18 @@ module VirusTotal
     end
 
     def rescan
-      api_response('file/rescan', resource: @resource.to_str)
+      api_response('file/rescan', resource: resource)
     end
 
     def report
-      api_response('file/report', resource: @resource.to_str)
+      api_response('file/report', resource: resource)
     end
 
     private
 
     def get_file
       tmp = Tempfile.open('tmp')
-      tmp.write(IO::File.read(@resource.to_str))
+      tmp.write(IO::File.read(resource))
       tmp
     end
   end

data/lib/virus_total/ip.rb

@@ -2,7 +2,7 @@ module VirusTotal
   # to use this feature you must have a private apikey!
   class Ip < Base
     def report
-      api_response('ip-address/report', resource: @resource.to_str, method: :get)
+      api_response('ip-address/report', resource: resource, method: :get)
     end
   end
 end

data/lib/virus_total/parser.rb

@@ -10,16 +10,17 @@ module VirusTotal
     end
 
     def info
-      @response.except_key("scans")
+      return unless @response
+
+      @response.tap { |hash| hash.delete("scans") }
     end
 
     def dangers
-      scans = @response.scans
       return {} unless scans
 
       dangers = {}
       scans.each_pair do |key, hash|
-        next unless hash.detected
+        next unless hash["detected"]
 
         dangers.merge!({ key => hash })
       end

data/lib/virus_total/url.rb

@@ -1,11 +1,11 @@
 module VirusTotal
   class Url < Base
     def scan
-      api_response("url/scan", url: @resource.to_str("\n"))
+      api_response("url/scan", url: resource("\n"))
     end
 
     def report
-      api_response("url/report", resource: @resource.to_str)
+      api_response("url/report", resource: resource)
     end
   end
 end

data/lib/virus_total/version.rb

@@ -1,3 +1,3 @@
 module VirusTotal
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/virus_total.rb

@@ -1,4 +1,4 @@
-require "core_extensions"
+require "core_extentions"
 require "virus_total/base"
 require "virus_total/url"
 require "virus_total/file"

data/spec/virus_total/base_spec.rb

@@ -2,7 +2,6 @@ require "virus_total"
 
 describe VirusTotal::Base do
   let(:resource_param) { "first, second" }
-  let(:resource) { base.instance_variable_get("@resource") }
   let(:base) { VirusTotal::Base.new(resource_param, "invalid_key") }
 
   it "raises AuthError "\
@@ -13,14 +12,14 @@ describe VirusTotal::Base do
 
   context "for resource parameter" do
     it "returns resource correctly" do
-      resource.to_str.should == "first, second"
+      base.resource.should == "first, second"
     end
 
     let(:resource_param) { ["first", "second"] }
 
     it "split resources array correctly" do
-      resource.to_str.should == "first, second"
-      resource.to_str("\n").should == "first\nsecond"
+      base.resource.should == "first, second"
+      base.resource("\n").should == "first\nsecond"
     end
   end
 end

data/spec/virus_total/parser_spec.rb

@@ -42,8 +42,4 @@ describe Hash do
      "if hash key with a space" do
     hash.key2.key_3.should == "val3"
   end
-
-  it "returns hash w/o key correctly" do
-    hash.except_key("key2").should == {"key1"=>"val1"}
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: virus_total
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Rubycop
@@ -78,7 +78,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
-- lib/core_extensions.rb
+- lib/core_extentions.rb
 - lib/virus_total.rb
 - lib/virus_total/base.rb
 - lib/virus_total/domain.rb

data/lib/core_extensions.rb

@@ -1,23 +0,0 @@
-class Array
-  def to_str(spliter = ", ")
-    join(spliter)
-  end
-end
-
-class String
-  def to_str(*args)
-    self
-  end
-end
-
-class Hash
-  def except_key(hash_key)
-    tap { |hash| hash.delete(hash_key) }
-  end
-
-  def method_missing(name, *args)
-    name = name.to_s
-    name.gsub!("_", " ")
-    self[name]
-  end
-end