import { withProps } from 'recompose'
import countrylist from 'country-list'

const countryMapper = (c = 'RO') => {
  switch (c) {
    case 'GB':
      return 'UK'
    case 'RS':
      return 'SRB'
    case 'ME':
      return 'MNT'
    default:
      return c
  }
}

const codeMapper = (c = 'RO') => {
  switch (c) {
    case 'UK':
      return 'GB'
    case 'SRB':
      return 'RS'
    case 'MNT':
      return 'ME'
    default:
      return c
  }
}

const countries = countrylist()
const countryList = countries
  .getNames()
  .map(c => ({ value: countryMapper(countries.getCode(c)), label: c }))

export default withProps(() => ({
  countries: countryList,
  countryLabel: code => countries.getName(codeMapper(code)),
}))