1 files changed, 41 insertions, 0 deletions
diff --git a/ui/app/pages/first-time-flow/seed-phrase/confirm-seed-phrase/confirm-seed-phrase.state.js b/ui/app/pages/first-time-flow/seed-phrase/confirm-seed-phrase/confirm-seed-phrase.state.js
new file mode 100644
index 000000000..f2476fc5c
--- /dev/null
+++ b/ui/app/pages/first-time-flow/seed-phrase/confirm-seed-phrase/confirm-seed-phrase.state.js
@@ -0,0 +1,41 @@
+export function selectSeedWord (word, shuffledIndex) {
+  return function update (state) {
+    const { selectedSeedWords, selectedSeedWordsHash } = state
+    const nextSelectedIndex = selectedSeedWords.length
+
+    return {
+      selectedSeedWords: [ ...selectedSeedWords, word ],
+      selectedSeedWordsHash: { ...selectedSeedWordsHash, [shuffledIndex]: nextSelectedIndex },
+    }
+  }
+}
+
+export function deselectSeedWord (shuffledIndex) {
+  return function update (state) {
+    const {
+      selectedSeedWords: prevSelectedSeedWords,
+      selectedSeedWordsHash: prevSelectedSeedWordsHash,
+    } = state
+
+    const selectedSeedWords = [...prevSelectedSeedWords]
+    const indexToRemove = prevSelectedSeedWordsHash[shuffledIndex]
+    selectedSeedWords.splice(indexToRemove, 1)
+    const selectedSeedWordsHash = Object.keys(prevSelectedSeedWordsHash).reduce((acc, index) => {
+      const output = { ...acc }
+      const selectedSeedWordIndex = prevSelectedSeedWordsHash[index]
+
+      if (selectedSeedWordIndex < indexToRemove) {
+        output[index] = selectedSeedWordIndex
+      } else if (selectedSeedWordIndex > indexToRemove) {
+        output[index] = selectedSeedWordIndex - 1
+      }
+
+      return output
+    }, {})
+
+    return {
+      selectedSeedWords,
+      selectedSeedWordsHash,
+    }
+  }
+}