module ChinaCity

# File lib/china_city.rb, line 57
def city(code)
  id_match = match(code)
  "#{id_match[1]}#{id_match[2]}".ljust(6, '0')
end

# File lib/china_city.rb, line 66
def data
  unless @list
    #{ '440000' =>
    #  {
    #    :text => '广东',
    #    :children =>
    #      {
    #        '440300' =>
    #          {
    #            :text => '深圳',
    #            :children =>
    #              {
    #                '440305' => { :text => '南山' }
    #              }
    #           }
    #       }
    #   }
    # }
    @list = {}
    #@see: https://github.com/cn/GB2260
    json = JSON.parse(File.read("#{Engine.root}/db/areas.json"))
    streets = json.values.flatten
    streets.each do |street|
      id = street['id']
      text = street['text']
      sensitive_areas = street['sensitive_areas'] || false
      if id.size == 6    # 省市区
        if id.end_with?('0000')                           # 省
          @list[id] =  {:text => text, :sensitive_areas => sensitive_areas, :children => {}}
        elsif id.end_with?('00')                          # 市
          province_id = province(id)
          @list[province_id] = {:text => nil, :children => {}} unless @list.has_key?(province_id)
          @list[province_id][:children][id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}}
        else
          province_id = province(id)
          city_id     = city(id)
          @list[province_id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}} unless @list.has_key?(province_id)
          @list[province_id][:children][city_id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}} unless @list[province_id][:children].has_key?(city_id)
          @list[province_id][:children][city_id][:children][id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}}
        end
      else               # 街道
        province_id = province(id)
        city_id     = city(id)
        district_id = district(id)
        @list[province_id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}} unless @list.has_key?(province_id)
        @list[province_id][:children][city_id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}} unless @list[province_id][:children].has_key?(city_id)
        @list[province_id][:children][city_id][:children][district_id] = {:text => text, :sensitive_areas => sensitive_areas, :children => {}} unless @list[province_id][:children][city_id][:children].has_key?(district_id)
        @list[province_id][:children][city_id][:children][district_id][:children][id] = {:text => text, :sensitive_areas => sensitive_areas}
      end
    end
  end
  @list
end

# File lib/china_city.rb, line 62
def district(code)
  code[0..5].rjust(6,'0')
end

# File lib/china_city.rb, line 34
def get(id, options = {})
  return '' if id.blank?
  prepend_parent = options[:prepend_parent] || false
  children = data
  return children[id][:text] if children.has_key?(id)
  province_id = province(id)
  province_text = children[province_id][:text]
  children = children[province_id][:children]
  return "#{prepend_parent ? province_text : ''}#{children[id][:text]}" if children.has_key?(id)
  city_id = city(id)
  city_text = children[city_id][:text]
  children = children[city_id][:children]
  return "#{prepend_parent ? (province_text + city_text) : ''}#{children[id][:text]}" if children.has_key?(id)
  district_id = district(id)
  district_text = children[district_id][:text]
  children = children[district_id][:children]
  return "#{prepend_parent ? (province_text + city_text + district_text) : ''}#{children[id][:text]}"
end

# File lib/china_city.rb, line 10
def list(parent_id='000000', options = {})
  # parent_id ||= '000000'
  show_all = options[:show_all] || false

  result = []
  return result if parent_id.blank?
  province_id = province(parent_id)
  city_id     = city(parent_id)
  district_id = district(parent_id)
  children = data
  children = children[province_id][:children] if children.has_key?(province_id)
  children = children[city_id][:children] if children.has_key?(city_id)
  children = children[district_id][:children] if children.has_key?(district_id)
  children.each_key do |id|
    next if (!show_all && children[id][:sensitive_areas])
    result.push [children[id][:text], id]
  end

  #sort
  result.sort! {|a, b| a[1] <=> b[1]}
  result
end

# File lib/china_city.rb, line 120
def match(code)
  code.match(PATTERN)
end

# File lib/china_city.rb, line 53
def province(code)
  match(code)[1].ljust(6, '0')
end