Overleaf/services/web/app/src/infrastructure/GeoIpLookup.js

const request = require('request')
const settings = require('@overleaf/settings')
const _ = require('lodash')
const logger = require('@overleaf/logger')
const { URL } = require('url')
const { promisify, promisifyMultiResult } = require('@overleaf/promise-utils')

const DEFAULT_CURRENCY_CODE = 'USD'

const currencyMappings = {
  GB: 'GBP',
  US: 'USD',
  CH: 'CHF',
  NZ: 'NZD',
  AU: 'AUD',
  DK: 'DKK',
  NO: 'NOK',
  CA: 'CAD',
  SE: 'SEK',
  SG: 'SGD',
  IN: 'INR',
  BR: 'BRL',
  MX: 'MXN',
  CO: 'COP',
  CL: 'CLP',
  PE: 'PEN',
}

const validCurrencyParams = Object.values(currencyMappings).concat([
  'EUR',
  'INR',
  'BRL',
  'MXN',
  'COP',
  'CLP',
  'PEN',
])

// Countries which would likely prefer Euro's
const EuroCountries = [
  'AT',
  'BE',
  'BG',
  'HR',
  'CY',
  'CZ',
  'EE',
  'FI',
  'FR',
  'DE',
  'EL',
  'HU',
  'IE',
  'IT',
  'LV',
  'LT',
  'LU',
  'MT',
  'NL',
  'PL',
  'PT',
  'RO',
  'SK',
  'SI',
  'ES',
]

_.forEach(EuroCountries, country => (currencyMappings[country] = 'EUR'))

function isValidCurrencyParam(currency) {
  if (!currency) {
    return false
  }
  return validCurrencyParams.includes(currency)
}

function getDetails(ip, callback) {
  if (!ip) {
    return callback(new Error('no ip passed'))
  }
  ip = ip.trim().split(' ')[0]
  const opts = {
    url: new URL(ip, settings.apis.geoIpLookup.url).href,
    timeout: 1000,
    json: true,
  }
  logger.debug({ ip, opts }, 'getting geo ip details')
  request.get(opts, function (err, res, ipDetails) {
    if (err) {
      logger.warn({ err, ip }, 'error getting ip details')
    }
    callback(err, ipDetails)
  })
}

function getCurrencyCode(ip, callback) {
  getDetails(ip, function (err, ipDetails) {
    if (err || !ipDetails) {
      logger.err(
        { err, ip },
        `problem getting currencyCode for ip, defaulting to ${DEFAULT_CURRENCY_CODE}`
      )
      return callback(null, DEFAULT_CURRENCY_CODE)
    }
    const countryCode =
      ipDetails && ipDetails.country_code
        ? ipDetails.country_code.toUpperCase()
        : undefined
    const currencyCode = currencyMappings[countryCode] || DEFAULT_CURRENCY_CODE
    logger.debug({ ip, currencyCode, ipDetails }, 'got currencyCode for ip')
    callback(err, currencyCode, countryCode)
  })
}

module.exports = {
  getDetails,
  getCurrencyCode,
  isValidCurrencyParam,
  promises: {
    getDetails: promisify(getDetails),
    getCurrencyCode: promisifyMultiResult(getCurrencyCode, [
      'currencyCode',
      'countryCode',
    ]),
  },
  DEFAULT_CURRENCY_CODE,
}