renyijiu/id_validator

module IdValidator
  module Concern
    module Func
      extend self

      # 检测出生日期, example: '20181015'
      def check_birthday(birthday)
        return false if birthday.to_s.length != 8
        date = Date.parse(birthday.to_s) rescue nil

        return false if date.nil?
        return false if date.year < 1800 || date > Date.today

        true
      end

      # 格式化出生日期
      def format_birthday_code(birthday_code)
        "#{birthday_code[0..3]}-#{birthday_code[4..5]}-#{birthday_code[6..7]}"
      end

      # 格式化地址栏，地址三元组（省，市，区）
      def format_address_info(address_info)
        [address_info[:province], address_info[:city], address_info[:district]]
      end

      # 随意获取一个符合要求的地址码
      def get_random_address_code(reg)
        keys = IdValidator::Config.latest_address_info.keys.shuffle

        keys.find { |key| key.to_s =~ reg }.to_s
      end

      # 字符串填充
      def get_str_pad(str, length = 2, character = '0', right = true)
        str = str.to_s
        return str if length <= str.length

        frequency = length - str.length
        right ? (str + character.to_s * frequency) : (character.to_s * frequency + str)
      end

      # 随机获取一个符合条件的数字
      def get_random_right_num(num, min, max)
        return num if (min..max).member?(num)

        rand(min..max)
      end

      # 随意生成一个生日
      def get_random_birthday_code
        Time.at(rand * Time.now.to_i).strftime('%Y%m%d')
      end

      # 获取指定区划代码最新的地址名称
      def get_address_info(address_code)
        address_code = address_code.to_i
        address_arr = IdValidator::Config.total_address_info.fetch(address_code, [])

        address = address_arr.sort_by{|a| a[:end_year]}.last || {}
        address.fetch(:address, nil)
      end
    end

    module Helper
      # 获取身份证信息
      def get_id_argument(id_card)
        id_card = id_card.to_s.upcase

        if id_card.length == 18
          {
              body: id_card[0...17],
              address_code: id_card[0...6],
              birthday_code: id_card[6...14],
              order_code: id_card[14...17],
              check_bit: id_card[17],
              type: 18
          }
        else
          {
              body: id_card,
              address_code: id_card[0...6],
              birthday_code: '19' + id_card[6...12],
              order_code: id_card[12...15],
              check_bit: '',
              type: 15
          }
        end
      end

      # 获取地址信息
      def get_address_info(address_code)
        province_address_code = "#{address_code[0...2]}0000"
        city_address_code = "#{address_code[0...4]}00"

        is_mainland = (address_code[0].to_i != 8)

        {
            province: Func.get_address_info(province_address_code),
            city: is_mainland ? Func.get_address_info(city_address_code) : nil,
            district: is_mainland ? Func.get_address_info(address_code) : nil
        }
      end

      # 获取星座信息
      def get_constellation(birthday_code)
        date = Date.parse(birthday_code.to_s)

        month = date.month
        day = date.day
        constellation = IdValidator::Config.constellation

        start_day = constellation[month][:start_date].split('-').last.to_i
        if day < start_day
          tmp_month = month - 1 == 0 ? 12 : month - 1

          constellation[tmp_month][:name]
        else
          constellation[month][:name]
        end
      end

      # 获取生肖信息
      def get_chinese_zodiac(birthday_code)
        # 1900为子鼠
        index = (birthday_code[0...4].to_i - 1900) % 12

        IdValidator::Config.chinese_zodiac[index]
      end

      # 检查地址码
      def check_address_code(address_code)
        address_info = get_address_info(address_code.to_s)
        return false if address_info[:province].nil?

        true
      end

      # 检查出生日期
      def check_birthday_code(birthday_code)
        return false unless Func.check_birthday(birthday_code.to_s)

        true
      end

      # 检查顺序码
      def check_order_code(order_code)
        return false if order_code.length != 3

        true
      end

      # 检查是否已经废弃
      def check_is_abandoned(address_code)
        address = IdValidator::Config.latest_address_info.fetch(address_code.to_i, nil)

        address.nil?
      end

      # 生成校验码
      def generate_check_bit(body)
        body = body.to_s.reverse
        return nil if body.length != 17

        body_sum = 0
        weight = (2..18).map{ |a| 2**(a - 1) % 11 }
        weight.each_with_index{ |w, i| body_sum += body[i].to_i * w }

        check_bit = (12 - (body_sum % 11)) % 11
        check_bit < 10 ? check_bit.to_s : 'X'
      end

      # 生成地址码
      def generate_address_code(address)
        key = IdValidator::Config.latest_address_info.key(address).to_s

        case
        when key.empty?
          reg = %r(^\d{4}(?!00)\d{2}$)
          return Func.get_random_address_code(reg)
        when key[0] == '8'
          return key
        when key =~ /^\d{2}0000$/
          reg = %r(^#{key[0..1]}\d{2}(?!00)\d{2}$)
          return Func.get_random_address_code(reg)
        when key =~ /^\d{4}00$/
          reg = %r(^#{key[0..3]}(?!00)\d{2}$)
          return Func.get_random_address_code(reg)
        else
          return key.to_s
        end
      end

      # 生成生日信息
      def generate_birthday_code(birthday)
        return birthday if Func.check_birthday(birthday)
        birthday = Func.get_str_pad(birthday.to_s, 8)

        year = Func.get_random_right_num(birthday[0..3].to_i, 1900, Time.now.year - 1)
        month = Func.get_random_right_num(birthday[4..5].to_i, 1, 12)
        day = Func.get_random_right_num(birthday[6..7], 1, 28)

        year = Func.get_str_pad(year.to_s, 4)
        month = Func.get_str_pad(month.to_s, 2, '0', false)
        day = Func.get_str_pad(day.to_s, 2, '0', false)
        result = year + month + day

        if Func.check_birthday(result)
          result
        else
          Func.get_random_birthday_code
        end
      end

      # 生成顺序码
      def generate_order_code(sex)
        order_code = rand(1..999)

        order_code -= 1 if sex != (order_code % 2)

        Func.get_str_pad(order_code, 3, '0', false)
      end

    end
  end
end