From 8542c0f3f9bbf415f67733b8f4f312b8ac976402 Mon Sep 17 00:00:00 2001
From: Jonathan Rainville <rainville.jonathan@gmail.com>
Date: Wed, 11 Jul 2018 12:47:15 -0400
Subject: [PATCH] fix ens by using right addres depending on chain and also add
 isAvailable

---
 js/embark.js                |  4 ++
 lib/modules/ens/embarkjs.js | 91 ++++++++++++++++++++++---------------
 2 files changed, 58 insertions(+), 37 deletions(-)

diff --git a/js/embark.js b/js/embark.js
index 110dac1e6..af7f4ff5b 100644
--- a/js/embark.js
+++ b/js/embark.js
@@ -374,6 +374,10 @@ EmbarkJS.Names.lookup = function (identifier, callback) {
   return this.currentNameSystems.lookup(identifier, callback);
 };
 
+EmbarkJS.Names.isAvailable = function () {
+  return this.currentNameSystems.isAvailable();
+};
+
 // To Implement
 
 
diff --git a/lib/modules/ens/embarkjs.js b/lib/modules/ens/embarkjs.js
index 3cf77e730..90fc0b52b 100644
--- a/lib/modules/ens/embarkjs.js
+++ b/lib/modules/ens/embarkjs.js
@@ -1,6 +1,7 @@
+/*global EmbarkJS, web3*/
+
 import namehash from 'eth-ens-namehash';
 
-/*global EmbarkJS*/
 let __embarkENS = {};
 
 // resolver interface
@@ -157,47 +158,62 @@ const providerNotSetError = 'ENS provider not set';
 const NoDecodeAddrError = 'Error: Couldn\'t decode address from ABI: 0x';
 const NoDecodeStringError = 'ERROR: The returned value is not a convertible string: 0x0';
 
+__embarkENS.registryAddresses = {
+  // Mainnet
+  "1": "0x314159265dd8dbb310642f98f50c066173c1259b",
+  // Ropsten
+  "3": "0x112234455c3a32fd11230c42e7bccd4a84e02010",
+  // Rinkeby
+  "4": "0xe7410170f87102DF0055eB195163A03B7F2Bff4A"
+};
+
 __embarkENS.setProvider = function (config) {
   const self = this;
+  const ERROR_MESSAGE = 'ENS is not available in this chain';
   EmbarkJS.onReady(() => {
-    self.ens = new EmbarkJS.Contract({abi: config.abi, address: config.address});
+    web3.eth.net.getId((err, id) => {
+      if (err) {
+        if (err.message.indexOf('Provider not set or invalid') > -1) {
+          console.warn(ERROR_MESSAGE);
+          return;
+        }
+        console.error(err);
+      }
+      if (!self.registryAddresses[id]) {
+        console.warn(ERROR_MESSAGE);
+        return;
+      }
+      self.isAvailable = true;
+      self.ens = new EmbarkJS.Contract({abi: config.abi, address: self.registryAddresses[id]});
+    });
   });
 };
 
 __embarkENS.resolve = function (name, callback) {
+  callback = callback || function () {};
   if (!this.ens) {
     return callback(providerNotSetError);
   }
-
   let node = namehash.hash(name);
 
-  console.log('Name to check', name);
-  console.log('Node', node);
-  console.log(this.ens);
-  return this.ens.methods.resolver(node).call().then((resolverAddress) => {
-    console.log('address', resolverAddress);
-    let resolverContract = new EmbarkJS.Contract({abi: this.resolverInterface, address: resolverAddress});
-    return resolverContract.methods.addr(node).call()
-      .then((addr) => {
-        console.log('ADRESSS', addr);
-        callback(null, addr);
-      })
-      .catch(err => {
-        if (err === NoDecodeAddrError) {
-          return callback(name + " is not registered", "0x");
-        }
-        callback(err);
-      });
-  }).catch((err) => {
+  function cb(err, addr) {
     if (err === NoDecodeAddrError) {
       return callback(name + " is not registered", "0x");
     }
-    console.log('oldekwe^pfke');
-    callback(err);
+    callback(err, addr);
+  }
+
+  return this.ens.methods.resolver(node).call((err, resolverAddress) => {
+    if (err) {
+      return cb(err);
+    }
+    let resolverContract = new EmbarkJS.Contract({abi: this.resolverInterface, address: resolverAddress});
+    resolverContract.methods.addr(node).call(cb);
   });
 };
 
 __embarkENS.lookup = function (address, callback) {
+  callback = callback || function () {};
   if (!this.ens) {
     return callback(providerNotSetError);
   }
@@ -208,21 +224,22 @@ __embarkENS.lookup = function (address, callback) {
   let node = namehash.hash(address.toLowerCase() + ".addr.reverse");
   console.log('Node', node);
 
-  return this.ens.methods.resolver(node).call().then((resolverAddress) => {
-    console.log('Resolver address', resolverAddress);
-    let resolverContract = new EmbarkJS.Contract({abi: this.resolverInterface, address: resolverAddress});
-    resolverContract.methods.name(node).call()
-      .then(name => {
-        console.log('Name');
-        callback(null, name);
-      })
-      .catch(callback);
-  }).catch((err) => {
-    console.log('Got error', err);
+  function cb(err, name) {
     if (err === NoDecodeStringError || err === NoDecodeAddrError) {
-      console.log('Address does not resolve to name. Try syncing chain.');
-      return "";
+      return callback('Address does not resolve to name. Try syncing chain.');
     }
-    return err;
+    return callback(err, name);
+  }
+
+  return this.ens.methods.resolver(node).call((err, resolverAddress) => {
+    if (err) {
+      return cb(err);
+    }
+    let resolverContract = new EmbarkJS.Contract({abi: this.resolverInterface, address: resolverAddress});
+    resolverContract.methods.name(node).call(cb);
   });
 };
+
+__embarkENS.isAvailable = function () {
+  return Boolean(this.isAvailable);
+};