import { RetrieveResponseResult, RetrieveArgs } from "./types";

const isNumeric = (str: string) => !isNaN(parseFloat(str));

interface CountryMappers {
  [countryCode: string]: (
    result: RetrieveResponseResult
  ) => RetrieveResponseResult;
}

const countryMappers: CountryMappers = {
  ES(result: RetrieveResponseResult) {
    if (isNumeric(result.building_name) && !result.building_number) {
      return {
        ...result,
        building_name: "",
        building_number: result.building_name
      };
    }
    return result;
  },
  US(result: RetrieveResponseResult) {
    if (isNumeric(result.building_name) && !result.building_number) {
      return {
        ...result,
        building_name: "",
        building_number: result.building_name,
        postal_code: result.postal_code.split("-")[0]
      };
    }
    return {
      ...result,
      postal_code: result.postal_code.split("-")[0]
    };
  }
} as const;

export default function mapRetrievedResult(
  args: RetrieveArgs,
  result: RetrieveResponseResult
) {
  const countryCode = args.country as keyof typeof countryMappers;
  const mapper = countryMappers[`${countryCode}`.toUpperCase()];
  return mapper ? mapper(result) : result;
}