From 82cab5342abb6fb3b9f45738fa6bb73b872bffa4 Mon Sep 17 00:00:00 2001
From: Jeffrey Lau <jeffrey.lau@ribose.com>
Date: Mon, 28 Oct 2013 09:21:16 +0800
Subject: [PATCH] attr_masked: Provide options for specifying :column_name for
 an attribute

---
 lib/indigo/attr_masked.rb | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/lib/indigo/attr_masked.rb b/lib/indigo/attr_masked.rb
index b0e6b97..4533f26 100644
--- a/lib/indigo/attr_masked.rb
+++ b/lib/indigo/attr_masked.rb
@@ -104,6 +104,7 @@ def attr_masked(*attributes)
       :if               => true,
       :unless           => false,
       :encode           => false,
+      :column_name      => nil,
       :marshal          => false,
       :marshaler        => Marshal,
       :dump_method      => 'dump',
@@ -241,16 +242,20 @@ def mask!
       return if self.class.masked_attributes.length < 1
 
       sql_snippet = self.class.masked_attributes.map do |masked_attr|
+        # puts masked_attr
         masked_attr[0]
-      end.inject({}) do |acc, attr_name|
+        attr_name, column_name = masked_attr[0], (masked_attr[1][:column_name] || masked_attr[0])
+      end.inject({}) do |acc, (attr_name, column_name)|
 
         # build a map of { attr_name => masked_value }
         masked_value = self.mask(attr_name)
         acc.merge(
-          attr_name => masked_value
+          [attr_name, column_name] => masked_value
         )
-      end.inject([]) do |acc, (attr_name, masked_value)|
-        acc << "#{attr_name}=#{ActiveRecord::Base.sanitize(masked_value)}"
+      end.inject([]) do |acc, ((attr_name, column_name), masked_value)|
+        self.send("#{attr_name}=", masked_value)
+        final_masked_value = self.send(column_name)
+        acc << "#{column_name}=#{ActiveRecord::Base.sanitize(final_masked_value)}"
       end.join(', ')
 
       sql = <<-EOQ