From f6e4443437ccef00fa07549c3d8801e0a73d5da8 Mon Sep 17 00:00:00 2001
From: kikito <kikito@gmail.com>
Date: Sun, 1 Jan 2017 12:34:15 +0100
Subject: [PATCH] Removes sanitizable from budget, replacing it by custom impl

---
 app/models/budget.rb | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/app/models/budget.rb b/app/models/budget.rb
index d03515ae7..d25861e2f 100644
--- a/app/models/budget.rb
+++ b/app/models/budget.rb
@@ -1,6 +1,5 @@
 class Budget < ActiveRecord::Base
 
-  include Sanitizable
   include Measurable
 
   VALID_PHASES = %w(accepting reviewing selecting valuating balloting reviewing_ballots finished).freeze
@@ -16,6 +15,7 @@ class Budget < ActiveRecord::Base
   has_many :headings, through: :groups
 
   scope :on_hold,   -> { where(phase: ["reviewing", "valuating", "reviewing_ballots"]) }
+  before_validation :sanitize_descriptions
   scope :accepting, -> { where(phase: "accepting") }
   scope :reviewing, -> { where(phase: "reviewing") }
   scope :selecting, -> { where(phase: "selecting") }
@@ -84,5 +84,15 @@ class Budget < ActiveRecord::Base
   def formatted_heading_amount_spent(heading)
     formatted_amount(amount_spent(heading))
   end
+
+  private
+
+    def sanitize_descriptions
+      s = WYSIWYGSanitizer.new
+      VALID_PHASES.each do |phase|
+        sanitized = s.sanitize(self.send("description_#{phase}"))
+        self.send("description_#{phase}=", sanitized)
+      end
+    end
 end