import { cityData, hotCityData } from './city'

// 根据城市首字母获取地区数据
export const getDictDataByFirstLetter = (type: string) => {
  const dictData = {}

  if (typeof type !== 'string') return []

  cityData.forEach(() => {
    const typeStr = type.replace(/\s+/g, '')
    if (typeStr === '所有城市') {
      dictData[type] = cityData
    } else if (typeStr === '热门城市') {
      dictData[type] = hotCityData
    } else {
      const dictTypeData = cityData.filter(item => {
        let cityItem = Object.assign({}, item)
        if (item.name === '广州市') {
          cityItem = {
            ...item,
            group: 'G',
            pinyin: 'guang zhou shi',
          }
        }
        return ~typeStr.indexOf(cityItem.group) && item.name !== '县'
      })
      dictData[type] = dictTypeData
    }
  })
  return dictData
}

export const getDictInfoByCode = (code: number | string) => {
  const dictInfo = { name: '' }

  if (typeof code !== 'number' && typeof code !== 'string') return dictInfo

  if (code === '00') {
    return {
      code: '00',
      name: '全国',
      group: 'Q',
      pinyin: 'quan guo',
    }
  } else {
    return cityData.find(item => {
      const cityCode = item.code
      return ~~code === ~~cityCode
    })
  }
}