const fetch = require('cross-fetch'),
      // relays = require('../relays.yaml'),
      fs = require('fs'),
      YAML = require('yaml')

let object,
    yaml,
    result,
    file = fs.readFileSync('./relays.yaml', 'utf8')

const getIp = async function(relay){
  let ip
  await fetch(`https://1.1.1.1/dns-query?name=${relay.replace('wss://', '')}`, { headers: { 'accept': 'application/dns-json' } })
    .then(response => response.json())
    .then((data) => { ip = data.Answer ? data.Answer[data.Answer.length-1].data : false })
    .catch(err => console.log('./scripts/geo.js', err))
  return ip
}

const getGeo = async function(ip) {
  let geo
  await fetch(`http://ip-api.com/json/${ip}`, { headers: { 'accept': 'application/dns-json' } })
    .then(response => response.json())
    .then((data) => { geo = data })
    .catch(err => console.log('./scripts/geo.js', err))
  return geo
}

const query = async function(){
  const relays = YAML.parse(file).relays,
        result = {}

  for (const relay of relays) {
    ip = await getIp(relay)
    geo = await getGeo(ip)
    if(geo.status == 'success')
      result[relay] = geo
  }

  return result
}

const run = async function(){
  result = await query()
  object = { geo: result }
  yaml = new YAML.Document()
  yaml.contents = object
  // console.log(object)
  fs.writeFile('./geo.yaml', yaml.toString(), (err) => {
    if (err) return console.log('./scripts/geo.js', err);
  });
}

run()