From 3db4fa68417dc552183135c0a4c8740e11daa513 Mon Sep 17 00:00:00 2001
From: Tim Stallard <code@timstallard.me.uk>
Date: Mon, 3 Apr 2017 22:08:46 +0100
Subject: [PATCH] Fixes

---
 src/blocks/affineDecrypt.js |  2 +-
 src/blocks/affineEncrypt.js |  2 +-
 src/blocks/substitution.js  | 11 +++--------
 3 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/blocks/affineDecrypt.js b/src/blocks/affineDecrypt.js
index 1d6fb89..06da530 100644
--- a/src/blocks/affineDecrypt.js
+++ b/src/blocks/affineDecrypt.js
@@ -45,7 +45,7 @@ module.exports =  {
     return text
     .split("")
     .map(require("./util/toNum.js"))
-    .map((num)=>(reverseLookupTable[num]))
+    .map((num)=>(Number.isInteger(num) ? reverseLookupTable[num] : num))
     .map(require("./util/toChar.js"))
     .join("");
   },
diff --git a/src/blocks/affineEncrypt.js b/src/blocks/affineEncrypt.js
index e9aaa79..e0dd38d 100644
--- a/src/blocks/affineEncrypt.js
+++ b/src/blocks/affineEncrypt.js
@@ -45,7 +45,7 @@ module.exports =  {
     return text
     .split("")
     .map(require("./util/toNum.js"))
-    .map((num)=>(lookupTable[num]))
+    .map((num)=>(Number.isInteger(num) ? lookupTable[num] : num))
     .map(require("./util/toChar.js"))
     .join("");
   },
diff --git a/src/blocks/substitution.js b/src/blocks/substitution.js
index 600e73c..500a2f6 100644
--- a/src/blocks/substitution.js
+++ b/src/blocks/substitution.js
@@ -18,7 +18,7 @@ module.exports =  {
   execute: function({text, mapping}, elem){
     mapping = mapping.toLowerCase();
     var letterMapping = {};
-    for(var i = 0; i < (mapping.length - 1); i++){
+    for(var i = 0; i < (mapping.length - 1); i = i + 2){
       letterMapping[mapping[i]] = mapping[i+1];
     }
 
@@ -26,15 +26,10 @@ module.exports =  {
     .split("")
     .map((char)=>{
       if(char.toLowerCase() in letterMapping){
-        if(char.toLowerCase() == char){
-          return letterMapping[char];
-        }
-        else{
-          return letterMapping[char.toLowerCase()].toUpperCase();
-        }
+        return letterMapping[char.toLowerCase()]
       }
       else{
-        return char;
+        return char.toLowerCase();
       }
     })
     .join("");