From 1c2b05c2bad4e73ee33c32f55e00f4be2fee9906 Mon Sep 17 00:00:00 2001
From: Andrew Tomaka <atomaka@gmail.com>
Date: Tue, 28 Apr 2015 11:51:18 -0400
Subject: [PATCH] Maybe account for defenses

---
 app.rb         | 26 ++++++++++++++++++++------
 models/raid.rb | 14 +++++++++++++-
 2 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/app.rb b/app.rb
index 0402737..3358994 100644
--- a/app.rb
+++ b/app.rb
@@ -33,15 +33,29 @@ end
 
 get '/bossfight' do
   @username = params['username'] ? params['username'] : 'mafiaman'
-  user_raids = Raid.for_user(@username)
-  @first_negative = user_raids.first_negative
-  @soldiers_killed = user_raids.soldiers_lost_to_date(@first_negative.created_at)
-  @contributor_attacks = user_raids.contributor_attacks(@first_negative.created_at)
-  @contributor_soldiers = user_raids.contributor_soldiers(@first_negative.created_at)
+  user_defenses = Raid.defenses_for_user(@username)
+  user_attacks = Raid.attacks_for_user(@username)
+
+  @first_negative = user_defenses.first_negative
+  @soldiers_killed = user_defenses.soldiers_lost_to_date(@first_negative.created_at)
+
+  @contributor_attacks = user_defenses.contributor_attacks(@first_negative.created_at)
+  @contributor_soldiers = user_defenses.contributor_soldiers(@first_negative.created_at)
+
+  @contributor_defenses = user_attacks.contributor_defenses(@first_negative.created_at)
+  @contributor_defenders = user_attacks.contributor_defenders(@first_negative.created_at)
 
   @contributors = []
   @contributor_attacks.each do |k, v|
-    @contributors << { username: k, attacks: v, lost: @contributor_soldiers[k] }
+    attacks = @contributor_attacks[k] || 0
+    attacks += @contributor_defenses[k] if @contributor_defenses[k]
+    lost = @contributor_soldiers[k] || 0
+    lost += @contributor_defenders[k] if @contributor_defenders[k]
+    @contributors << {
+                       username: k,
+                       attacks: attacks,
+                       lost: lost
+                     }
   end
 
   erb :bossfight
diff --git a/models/raid.rb b/models/raid.rb
index d0169c9..d103bf1 100644
--- a/models/raid.rb
+++ b/models/raid.rb
@@ -9,10 +9,14 @@ class Raid < ActiveRecord::Base
     write_attribute(:soldiers, value)
   end
 
-  def self.for_user(username)
+  def self.defenses_for_user(username)
     where('defender = ?', username)
   end
 
+  def self.attacks_for_user(username)
+    where('attacker = ?', username)
+  end
+
   def self.first_negative
     where('money < 0').order(:created_at).first
   end
@@ -25,7 +29,15 @@ class Raid < ActiveRecord::Base
     where('created_at <= ?', date).group(:attacker).count(:id)
   end
 
+  def self.contributor_defenses(date)
+    where('created_at <= ?', date).group(:defender).count(:id)
+  end
+
   def self.contributor_soldiers(date)
     where('created_at <= ?', date).group(:attacker).sum(:soldiers)
   end
+
+  def self.contributor_defenders(date)
+    where('created_at <= ?', date).group(:defender).sum(:soldiers)
+  end
 end