require 'tempfile'

module Jpmobile
  module Util
    # SJIS   = "Shift_JIS"
    SJIS   = 'Windows-31J'.freeze
    UTF8   = 'UTF-8'.freeze
    JIS    = 'ISO-2022-JP'.freeze
    JIS_WIN = 'CP50220'.freeze
    BINARY = 'ASCII-8BIT'.freeze

    WAVE_DASH = [0x301c].pack('U')
    FULLWIDTH_TILDE = [0xff5e].pack('U')

    OVERLINE = [0x203e].pack('U')
    FULLWIDTH_MACRON = [0xffe3].pack('U')

    EM_DASH = [0x2014].pack('U')
    HORIZONTAL_BAR = [0x2015].pack('U')

    MINUS_SIGN = [0x2212].pack('U')
    FULLWIDTH_HYPHEN_MINUS = [0xFF0D].pack('U')

    module_function

    def deep_convert(obj, &proc)
      case obj
      when Hash
        new_obj = {}
        obj.each_pair do |key, value|
          new_obj[deep_convert(key.dup, &proc)] = deep_convert(value, &proc)
        end
        new_obj
      when Array
        new_obj = obj.map do |value|
          deep_convert(value, &proc)
        end
      when Symbol
        new_obj = yield(obj.to_s).to_sym
      when String
        obj = obj.to_param if obj.respond_to?(:to_param)
        new_obj = yield(obj)
      else
        # NilClass, TrueClass, FalseClass, Tempfile, StringIO, etc...
        new_obj = obj
      end

      new_obj
    end

    def sjis(str)
      unless shift_jis?(str)
        str = str.dup.force_encoding(SJIS)
      end
      str
    end

    def utf8(str)
      unless utf8?(str)
        str = str.dup.force_encoding(UTF8)
      end
      str
    end

    def jis(str)
      unless jis?(str)
        str = str.dup.force_encoding(JIS)
      end
      str
    end

    def jis_win(str)
      unless jis?(str)
        str = str.dup.force_encoding(JIS_WIN)
      end
      str
    end

    def ascii_8bit(str)
      unless ascii_8bit?(str)
        str = str.dup.force_encoding(BINARY)
      end
      str
    end

    def ascii_compatible!(str)
      unless str.encoding.ascii_compatible?
        str = str.dup.force_encoding(BINARY)
      end
      str
    end

    def utf8_to_sjis(utf8_str)
      # 波ダッシュ対策
      utf8_str = wavedash_to_fullwidth_tilde(utf8_str)
      # オーバーライン対策(不可逆的)
      utf8_str = overline_to_fullwidth_macron(utf8_str)
      # ダッシュ対策（不可逆的）
      utf8_str = emdash_to_horizontal_bar(utf8_str)
      # マイナス対策（不可逆的）
      utf8_str = minus_sign_to_fullwidth_hyphen_minus(utf8_str)

      utf8_str.
        gsub(/(\r\n|\r|\n)/, "\r\n").
        encode(SJIS, undef: :replace, replace: '?')
    end

    def sjis_to_utf8(sjis_str)
      utf8_str = sjis_str.encode('UTF-8', universal_newline: true)

      # 波ダッシュ対策
      fullwidth_tilde_to_wavedash(utf8_str)
    end

    def utf8_to_jis(utf8_str)
      # 波ダッシュ対策
      utf8_str = fullwidth_tilde_to_wavedash(utf8_str)

      utf8_str.
        gsub(/(\r\n|\r|\n)/, "\r\n").
        encode(JIS, undef: :replace, replace: '?')
    end

    def jis_to_utf8(jis_str)
      jis_str.encode(UTF8, universal_newline: true)
    end

    def regexp_utf8_to_sjis(utf8_str)
      Regexp.compile(Regexp.escape(utf8_to_sjis(utf8_str)))
    end

    def regexp_to_sjis(sjis_str)
      Regexp.compile(Regexp.escape(sjis(sjis_str)))
    end

    def hash_to_utf8(hash)
      new_hash = {}
      hash.each do |key, value|
        new_hash[utf8(key)] = utf8(value)
      end
    end

    def sjis_regexp(sjis)
      sjis_str = if sjis.is_a?(Numeric)
                   [sjis].pack('n')
                 else
                   sjis
                 end

      Regexp.compile(Regexp.escape(sjis_str.force_encoding(SJIS)))
    end

    def jis_regexp(jis)
      jis_str = jis.is_a?(Numeric) ? [jis].pack('n') : jis

      Regexp.compile(Regexp.escape(jis_str.force_encoding(BINARY)))
    end

    def jis_string_regexp
      Regexp.compile(Regexp.escape(ascii_8bit("\x1b\x24\x42")) + '(.+?)' + Regexp.escape(ascii_8bit("\x1b\x28\x42")))
    end

    def encode(str, charset)
      return str if charset.nil? || (charset == '') || str.nil? || (str == '')
      return str.encode(charset) unless utf8?(str)

      case charset
      when /iso-2022-jp/i
        utf8_to_jis(str)
      when /shift_jis/i
        utf8_to_sjis(str)
      when /utf-8/i
        str
      end
    end

    def wavedash_to_fullwidth_tilde(utf8_str)
      utf8_str.gsub(WAVE_DASH, FULLWIDTH_TILDE)
    end

    def fullwidth_tilde_to_wavedash(utf8_str)
      utf8_str.gsub(FULLWIDTH_TILDE, WAVE_DASH)
    end

    def overline_to_fullwidth_macron(utf8_str)
      utf8_str.gsub(OVERLINE, FULLWIDTH_MACRON)
    end

    def fullwidth_macron_to_overline(utf8_str)
      utf8_str.gsub(FULLWIDTH_MACRON, OVERLINE)
    end

    def emdash_to_horizontal_bar(utf8_str)
      utf8_str.gsub(EM_DASH, HORIZONTAL_BAR)
    end

    def minus_sign_to_fullwidth_hyphen_minus(utf8_str)
      utf8_str.gsub(MINUS_SIGN, FULLWIDTH_HYPHEN_MINUS)
    end

    def fullwidth_hyphen_minus_to_minus_sign(utf8_str)
      utf8_str.gsub(FULLWIDTH_HYPHEN_MINUS, MINUS_SIGN)
    end

    def force_encode(str, from, to)
      s = str.dup
      return str if detect_encoding(str) == to

      to = SJIS if to.match?(/shift_jis/i)

      to_enc = ::Encoding.find(to)
      return str if s.encoding == to_enc

      if from
        from_enc = ::Encoding.find(from)
        s.force_encoding(from) unless s.encoding == from_enc
      end

      begin
        s.encode(to)
      rescue ::Encoding::InvalidByteSequenceError, ::Encoding::UndefinedConversionError => e
        # iPhone MailがISO-2022-JPに半角カナや①などのCP50220文字を含めてくる問題対策
        raise e unless s.encoding == ::Encoding::ISO2022_JP

        s.force_encoding(::Encoding::CP50220)
        retry
      end
    end

    def set_encoding(str, encoding)
      encoding = SJIS if encoding.match?(/shift_jis/i)
      str.force_encoding(encoding)

      str
    end

    def extract_charset(str)
      case str
      when /iso-2022-jp/i
        'ISO-2022-JP'
      when /shift_jis/i
        'Shift_JIS'
      when /utf-8/i
        'UTF-8'
      else
        ''
      end
    end

    def detect_encoding(str)
      case str.encoding
      when ::Encoding::ISO2022_JP
        JIS
      when ::Encoding::Shift_JIS, ::Encoding::Windows_31J, ::Encoding::CP932
        SJIS
      when ::Encoding::UTF_8
        UTF8
      else
        # 上記以外はすべて BINARY 扱い
        BINARY
      end
    end

    def ascii_8bit?(str)
      detect_encoding(str) == BINARY
    end

    def utf8?(str)
      detect_encoding(str) == UTF8
    end

    def shift_jis?(str)
      detect_encoding(str) == SJIS
    end

    def jis?(str)
      detect_encoding(str) == JIS
    end

    def fold_text(str, size = 15)
      folded_texts = []

      while (texts = split_text(str, size)) && (texts.first.size != 0)
        folded_texts << texts.first
        str = texts.last
      end

      folded_texts
    end

    def split_text(str, size = 15)
      return nil if str.nil? || (str == '')

      [str[0..(size - 1)], str[size..-1]]
    end

    def invert_table(hash)
      result = {}
      hash.each_key do |key|
        if result[hash[key]]
          if !key.is_a?(Array) && !result[hash[key]].is_a?(Array) && result[hash[key]] > key
            result[hash[key]] = key
          end
        else
          result[hash[key]] = key
        end
      end
      result
    end

    def decode(str, encoding, charset)
      _str = case encoding
             when /quoted-printable/i
               str.unpack1('M').strip
             when /base64/i
               str.unpack1('m').strip
             else
               str
             end

      _extract_charset = Jpmobile::Util.extract_charset(_str)
      charset = _extract_charset unless _extract_charset.empty? || (_extract_charset == charset)
      Jpmobile::Util.set_encoding(_str, charset)
    end
  end
end