fetch report for file from virus total.

1 files changed, 8 insertions, 0 deletions

diff --git a/app/models/disposition.rb b/app/models/disposition.rb
index 1bf43c7..87ce27e 100644
--- a/app/models/disposition.rb
+++ b/app/models/disposition.rb
@@ -1,6 +1,7 @@
 class Disposition < ActiveRecord::Base
   enum state: [ :clean, :malicious, :unknown ]
   attr_readonly :fingerprint
+  has_many :file_reports
 
   validates_uniqueness_of :fingerprint
   validates_presence_of :fingerprint, :state
@@ -8,4 +9,11 @@ class Disposition < ActiveRecord::Base
   def to_param
     fingerprint
   end
+
+  def self.create_for(fingerprint, report)
+    disposition = Disposition.find_by(fingerprint: fingerprint)
+    disposition = Disposition.new(fingerprint: fingerprint) if disposition.nil?
+    disposition.state = :unknown
+    disposition.file_reports.create!(data: report)
+  end
 end