From dca52d0d4ebb6a7fde958c1db41076f3e0b1cbf3 Mon Sep 17 00:00:00 2001
From: Jonathan Rainville <rainville.jonathan@gmail.com>
Date: Tue, 6 Nov 2018 15:26:53 -0500
Subject: [PATCH] feat(cockpit): add searching ENS names in search bar

---
 embark-ui/src/reducers/selectors.js |  4 ++++
 embark-ui/src/sagas/searchSaga.js   | 30 ++++++++++++++++++++++-------
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/embark-ui/src/reducers/selectors.js b/embark-ui/src/reducers/selectors.js
index b30348517..a92211c97 100644
--- a/embark-ui/src/reducers/selectors.js
+++ b/embark-ui/src/reducers/selectors.js
@@ -157,6 +157,10 @@ export function getEnsRecords(state) {
   return state.entities.ensRecords;
 }
 
+export function getEnsRecordForName(state, name) {
+  return state.entities.ensRecords.find((record) => record.name === name);
+}
+
 export function getEnsErrors(state) {
   return state.errorEntities.ensRecords;
 }
diff --git a/embark-ui/src/sagas/searchSaga.js b/embark-ui/src/sagas/searchSaga.js
index 52819063f..48bbacfeb 100644
--- a/embark-ui/src/sagas/searchSaga.js
+++ b/embark-ui/src/sagas/searchSaga.js
@@ -1,16 +1,28 @@
 import {put, select} from "redux-saga/effects";
-import {getAccounts, getBlocks, getTransactions, getContracts} from "../reducers/selectors";
-import {fetchAccounts, fetchBlocks, fetchTransactions, fetchContracts} from "./index";
+import {getAccounts, getBlocks, getTransactions, getContracts, getEnsRecordForName} from "../reducers/selectors";
+import {fetchAccounts, fetchBlocks, fetchTransactions, fetchContracts, fetchEnsRecord} from "./index";
 import {ELEMENTS_LIMIT} from '../constants';
 
 export function *searchExplorer(entity, payload) {
   let result;
+  let searchValue = payload.searchValue;
+  let isENSName = false;
+  if (searchValue.endsWith('.eth')) {
+    isENSName = true;
+    yield fetchEnsRecord({name: payload.searchValue});
+    const ensRecord = yield select(getEnsRecordForName, searchValue);
+
+    if (!ensRecord) {
+      return yield put(entity.success({error: 'No ENS record for that name'}));
+    }
+    searchValue = ensRecord.address;
+  }
 
   // Accounts
   yield fetchAccounts({});
   const accounts = yield select(getAccounts);
   result = accounts.find(account => {
-    return account.address === payload.searchValue;
+    return account.address === searchValue;
   });
 
   if (result) {
@@ -21,7 +33,7 @@ export function *searchExplorer(entity, payload) {
   yield fetchContracts({});
   const contracts = yield select(getContracts);
   result = contracts.find(contract => {
-    return contract.address === payload.searchValue;
+    return contract.address === searchValue;
   });
 
   if (result) {
@@ -31,9 +43,9 @@ export function *searchExplorer(entity, payload) {
   // Blocks
   yield fetchBlocks({limit: ELEMENTS_LIMIT});
   const blocks = yield select(getBlocks);
-  const intSearchValue = parseInt(payload.searchValue, 10);
+  const intSearchValue = parseInt(searchValue, 10);
   result = blocks.find(block => {
-    return block.hash === payload.searchValue || block.number === intSearchValue;
+    return block.hash === searchValue || block.number === intSearchValue;
   });
 
   if (result) {
@@ -44,12 +56,16 @@ export function *searchExplorer(entity, payload) {
   yield fetchTransactions({blockLimit: ELEMENTS_LIMIT});
   const transactions = yield select(getTransactions);
   result = transactions.find(transaction => {
-    return transaction.hash === payload.searchValue;
+    return transaction.hash === searchValue;
   });
 
   if (result) {
     return yield put(entity.success(result));
   }
 
+  if (isENSName) {
+    return yield put(entity.success({error: `ENS resolved to ${searchValue}, but Embark couldn't find what this address represents`}));
+  }
+
   return yield put(entity.success({error: `No result found in transactions, accounts, contracts, or blocks. Please note: We limit the search to the last ${ELEMENTS_LIMIT} elements for performance`}));
 }