Add methods to register, unregister, and reset currency symbols

Author: RubenIgnacio

lib/monetize.rb

@@ -68,5 +68,17 @@ def from_numeric(value, currency = Money.default_currency)
       fail ArgumentError, "'value' should be a type of Numeric" unless value.is_a?(Numeric)
       Money.from_amount(value, currency)
     end
+
+    def register_currency_symbol(symbol, iso_code)
+      Monetize::Parser.register_currency_symbol(symbol, iso_code)
+    end
+
+    def unregister_currency_symbol(symbol)
+      Monetize::Parser.unregister_currency_symbol(symbol)
+    end
+
+    def reset_currency_symbols!
+      Monetize::Parser.reset_currency_symbols!
+    end
   end
 end

lib/monetize/parser.rb

@@ -1,6 +1,12 @@
 module Monetize
   class Parser
-    CURRENCY_SYMBOLS = {
+    MULTIPLIER_SUFFIXES = { 'K' => 3, 'M' => 6, 'B' => 9, 'T' => 12 }
+    MULTIPLIER_SUFFIXES.default = 0
+    MULTIPLIER_REGEXP = /^(.*?\d)(#{MULTIPLIER_SUFFIXES.keys.join('|')})\b([^\d]*)$/i
+
+    DEFAULT_DECIMAL_MARK = '.'.freeze
+
+    @@original_currency_symbols = {
       '$'  => 'USD',
       '€'  => 'EUR',
       '£'  => 'GBP',
@@ -26,15 +32,38 @@ class Parser
       'S$' => 'SGD',
       'HK$'=> 'HKD',
       'NT$'=> 'TWD',
-      '₱'  => 'PHP',
-    }
+      '₱'  => 'PHP'
+    }.freeze
 
-    CURRENCY_SYMBOL_REGEX = /(?<![A-Z])(#{CURRENCY_SYMBOLS.keys.map { |key| Regexp.escape(key) }.join('|')})(?![A-Z])/i
-    MULTIPLIER_SUFFIXES = { 'K' => 3, 'M' => 6, 'B' => 9, 'T' => 12 }
-    MULTIPLIER_SUFFIXES.default = 0
-    MULTIPLIER_REGEXP = Regexp.new(format('^(.*?\d)(%s)\b([^\d]*)$', MULTIPLIER_SUFFIXES.keys.join('|')), 'i')
+    class << self
+      def currency_symbols
+        @@currency_symbols ||= @@original_currency_symbols.dup
+      end
 
-    DEFAULT_DECIMAL_MARK = '.'.freeze
+      def register_currency_symbol(symbol, iso_code)
+        @@currency_symbols[symbol] = iso_code
+
+        reset_currency_symbol_regex
+      end
+
+      def unregister_currency_symbol(symbol)
+        @@currency_symbols.delete(symbol)
+        reset_currency_symbol_regex
+      end
+
+      def reset_currency_symbols!
+        @@currency_symbols = @@original_currency_symbols.dup
+        reset_currency_symbol_regex
+      end
+
+      def currency_symbol_regex
+        @@currency_symbol_regex ||= /(?<![A-Z])(#{currency_symbols.keys.map { |key| Regexp.escape(key) }.join('|')})(?![A-Z])/i
+      end
+
+      def reset_currency_symbol_regex
+        @@currency_symbol_regex = nil
+      end
+    end
 
     def initialize(input, fallback_currency = Money.default_currency, options = {})
       @input = input.to_s.strip
@@ -91,9 +120,9 @@ def compute_currency_from_iso_code
     end
 
     def compute_currency_from_symbol
-      match = input.match(CURRENCY_SYMBOL_REGEX)
+      match = input.match(self.class.currency_symbol_regex)
 
-      CURRENCY_SYMBOLS[match.to_s] if match
+      self.class.currency_symbols[match.to_s] if match
     end
 
     def assume_from_symbol?

spec/monetize_spec.rb

@@ -54,7 +54,7 @@
           Monetize.assume_from_symbol = false
         end
 
-        Monetize::Parser::CURRENCY_SYMBOLS.each_pair do |symbol, iso_code|
+        Monetize::Parser.currency_symbols.each_pair do |symbol, iso_code|
           context iso_code do
             let(:currency) { Money::Currency.find(iso_code) }
             let(:amount) { 5_95 }
@@ -132,6 +132,57 @@
             expect(Monetize.parse('kr9.99')).to eq Money.new(999, 'USD')
           end
         end
+
+        context "registered custom symbols and unregistering" do
+          after { Monetize.reset_currency_symbols! }
+
+          it "parses registered custom currency symbol" do
+            Monetize.register_currency_symbol("S/", "PEN")
+
+            expect(Monetize.parse("S/19.90")).to eq Money.new(1990, "PEN")
+          end
+
+          it "overrides existing currency symbol when registering" do
+            Monetize.register_currency_symbol("€", "USD")
+
+            expect(Monetize.parse("€19.90")).to eq Money.new(1990, "USD")
+          end
+
+          it "unregisters custom currency symbol" do
+            Monetize.register_currency_symbol("S/", "PEN")
+
+            expect(Monetize.parse("S/19.90")).to eq Money.new(1990, "PEN")
+
+            Monetize.unregister_currency_symbol("S/")
+
+            expect(Monetize.parse("S/19.90")).to eq Money.new(1990, "USD")
+          end
+
+          it "unregisters existing currency symbol" do
+            Monetize.unregister_currency_symbol("€")
+
+            expect(Monetize.parse("€19.90")).to eq Money.new(1990, "USD")
+          end
+
+          it "unregisters non-registered currency symbol without errors" do
+            expect {
+              Monetize.unregister_currency_symbol("XYZ")
+            }.not_to raise_error
+          end
+
+          it "resets currency symbols to original set" do
+            Monetize.register_currency_symbol("S/", "PEN")
+            Monetize.unregister_currency_symbol("€")
+
+            expect(Monetize.parse("S/19.90")).to eq Money.new(1990, "PEN")
+            expect(Monetize.parse("€19.90")).to eq Money.new(1990, "USD")
+
+            Monetize.reset_currency_symbols!
+
+            expect(Monetize.parse("€19.90")).to eq Money.new(1990, "EUR")
+            expect(Monetize.parse("S/19.90")).to eq Money.new(1990, "USD")
+          end
+        end
       end
 
       context 'opted out' do