WIP

lib/software/event/handler/virus.ex

@@ -1,11 +1,12 @@
 defmodule Helix.Software.Event.Handler.Virus do
 
   alias Helix.Event
-
   alias Helix.Software.Action.Virus, as: VirusAction
 
   alias Helix.Software.Event.File.Install.Processed,
     as: FileInstallProcessedEvent
+  alias Helix.Software.Event.Virus.Collect.Processed,
+    as: VirusCollectProcessedEvent
 
   @doc """
   Handles the completion of FileInstallProcess when the target file is a virus.
@@ -29,4 +30,8 @@ defmodule Helix.Software.Event.Handler.Virus do
   end
   def virus_installed(%FileInstallProcessedEvent{backend: _}),
     do: :noop
+
+  def handle_collect(%VirusCollectProcessedEvent{}) do
+    IO.puts "collected"
+  end
 end

lib/software/model/virus.ex

@@ -99,6 +99,11 @@ defmodule Helix.Software.Model.Virus do
       active: nil}
   end
 
+  def calculate_earnings(_file = %File{}, _virus = %Virus{}, _saved_earnings) do
+    # Obviously TODO
+    5000
+  end
+
   query do
 
     alias Helix.Entity.Model.Entity