require "charwidth"

module Charwidth
  module ActiveRecord
    module InstanceMethods
      def write_attribute_with_normalize_charwidth(attr_name, value)
        value = value.is_a?(String) ? Charwidth.normalize(value) : value
        write_attribute_without_normalize_charwidth(attr_name, value)
      end
    end

    def self.included(base)
      base.class_eval do
        include InstanceMethods
        if ::ActiveRecord.version<Gem::Version.new("5.2")
          alias_method :write_attribute_without_normalize_charwidth, :write_attribute
          alias_method :write_attribute, :write_attribute_with_normalize_charwidth
        else
          alias_method :write_attribute_without_normalize_charwidth, :_write_attribute
          alias_method :_write_attribute, :write_attribute_with_normalize_charwidth
        end
      end
    end
  end
end

ActiveRecord::Base.send(:include, Charwidth::ActiveRecord)