refactor: reformat code

bitcoinvsalts · Mar 3, 2021 · 310b159 · 310b159
1 parent 1dfb156
commit 310b159
Show file tree

Hide file tree

Showing 5 changed files with 310 additions and 219 deletions.
diff --git a/src/backtest.js b/src/backtest.js
@@ -361,7 +361,8 @@ async function run() {
     console.log("starting backtest... ".green)
     await backtest(lines)
     console.log(
-        "Ending backtest... ".green + parseInt((Date.now() - startBacktest) / 1000)
+        "Ending backtest... ".green +
+            parseInt((Date.now() - startBacktest) / 1000)
     )
 
     process.exit(0)

diff --git a/src/notifiers/gmail.js b/src/notifiers/gmail.js
@@ -4,98 +4,133 @@ const gmail_address = env.GMAIL_ADDRESS
 const gmail_app_password = env.GMAIL_APP_PASSWORD
 
 function createMailMessage(subject, html) {
-  return {
-    from: '"🐬  BVA " <[email protected]>',
-    to: gmail_address,
-    subject,
-    text: html,
-    html,
-  }
+    return {
+        from: '"🐬  BVA " <[email protected]>',
+        to: gmail_address,
+        subject,
+        text: html,
+        html,
+    }
 }
 
 function createSignalMessage(base, signal) {
-  const subject =
-    base + " :: " +
-    signal.stratname +
-    " " +
-    signal.pair +
-    " " +
-    signal.price;
+    const subject =
+        base +
+        " :: " +
+        signal.stratname +
+        " " +
+        signal.pair +
+        " " +
+        signal.price
 
-  const text =
-    (signal.score
-      ? "score: " + signal.score
-      : "score: NA") + "\n";
+    const text = (signal.score ? "score: " + signal.score : "score: NA") + "\n"
 
-  return createMailMessage(subject, text)
+    return createMailMessage(subject, text)
 }
 
 module.exports = function () {
-  if (!env.USE_GMAIL) return {};
+    if (!env.USE_GMAIL) return {}
+
+    const mailTransport = require("nodemailer").createTransport(
+        `smtps://${encodeURIComponent(gmail_address)}:${encodeURIComponent(
+            gmail_app_password
+        )}@smtp.gmail.com`
+    )
 
-  const mailTransport = require("nodemailer").createTransport(
-    `smtps://${encodeURIComponent(gmail_address)}:${encodeURIComponent(gmail_app_password)}@smtp.gmail.com`
-  )
+    function send(message) {
+        if (!env.USE_GMAIL) return
 
-  function send(message) {
-    if (!env.USE_GMAIL) return;
+        if (typeof message === "string") {
+            message = createMailMessage("Tading Bot Message", message)
+        }
 
-    if (typeof message === "string") {
-      message = createMailMessage("Tading Bot Message", message)
+        return mailTransport.sendMail(message).catch((error) => {
+            console.error(
+                "There was an error while sending the email ... trying again...",
+                error
+            )
+            setTimeout(() => {
+                mailTransport.sendMail(message).catch((error) => {
+                    console.error(
+                        "There was an error while sending the email: stop trying",
+                        error
+                    )
+                })
+            }, 2000)
+        })
     }
 
-    return mailTransport
-      .sendMail(message)
-      .catch((error) => {
-        console.error(
-          "There was an error while sending the email ... trying again...", error
+    function notifyBuyToCoverSignal(signal) {
+        return send(
+            createSignalMessage(
+                "<i>BUY_SIGNAL :: BUY TO COVER SHORT TRADE</i>",
+                signal
+            )
         )
-        setTimeout(() => {
-          mailTransport
-            .sendMail(message)
-            .catch((error) => {
-              console.error(
-                "There was an error while sending the email: stop trying", error
-              )
-            })
-        }, 2000)
-      })
-  }
-
-  function notifyBuyToCoverSignal(signal) {
-    return send(createSignalMessage("<i>BUY_SIGNAL :: BUY TO COVER SHORT TRADE</i>", signal));
-  }
-  function notifyBuyToCoverTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! BUY_SIGNAL :: BUY TO COVER SHORT TRADE</b>", signal));
-  }
-  function notifyEnterLongSignal(signal) {
-    return send(createSignalMessage("<i>BUY_SIGNAL :: ENTER LONG TRADE</i>", signal));
-  }
-  function notifyEnterLongTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! BUY_SIGNAL :: ENTER LONG TRADE</b>", signal));
-  }
-  function notifyEnterShortSignal(signal) {
-    return send(createSignalMessage("<i>SEL_SIGNAL :: ENTER SHORT TRADE</i>", signal));
-  }
-  function notifyEnterShortTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! SELL_SIGNAL :: ENTER SHORT TRADE</b>", signal));
-  }
-  function notifyExitLongSignal(signal) {
-    return send(createSignalMessage("<i>SELL_SIGNAL :: SELL TO EXIT LONG TRADE</i>", signal));
-  }
-  function notifyExitLongTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! SELL_SIGNAL :: SELL TO EXIT LONG TRADE</b>", signal));
-  }
+    }
+    function notifyBuyToCoverTraded(signal) {
+        return send(
+            createSignalMessage(
+                "<b>>> SUCCESS! BUY_SIGNAL :: BUY TO COVER SHORT TRADE</b>",
+                signal
+            )
+        )
+    }
+    function notifyEnterLongSignal(signal) {
+        return send(
+            createSignalMessage("<i>BUY_SIGNAL :: ENTER LONG TRADE</i>", signal)
+        )
+    }
+    function notifyEnterLongTraded(signal) {
+        return send(
+            createSignalMessage(
+                "<b>>> SUCCESS! BUY_SIGNAL :: ENTER LONG TRADE</b>",
+                signal
+            )
+        )
+    }
+    function notifyEnterShortSignal(signal) {
+        return send(
+            createSignalMessage(
+                "<i>SEL_SIGNAL :: ENTER SHORT TRADE</i>",
+                signal
+            )
+        )
+    }
+    function notifyEnterShortTraded(signal) {
+        return send(
+            createSignalMessage(
+                "<b>>> SUCCESS! SELL_SIGNAL :: ENTER SHORT TRADE</b>",
+                signal
+            )
+        )
+    }
+    function notifyExitLongSignal(signal) {
+        return send(
+            createSignalMessage(
+                "<i>SELL_SIGNAL :: SELL TO EXIT LONG TRADE</i>",
+                signal
+            )
+        )
+    }
+    function notifyExitLongTraded(signal) {
+        return send(
+            createSignalMessage(
+                "<b>>> SUCCESS! SELL_SIGNAL :: SELL TO EXIT LONG TRADE</b>",
+                signal
+            )
+        )
+    }
 
-  return {
-    notifyBuyToCoverSignal,
-    notifyBuyToCoverTraded,
-    notifyEnterLongSignal,
-    notifyEnterLongTraded,
-    notifyEnterShortSignal,
-    notifyEnterShortTraded,
-    notifyExitLongSignal,
-    notifyExitLongTraded,
-    send,
-  }
+    return {
+        notifyBuyToCoverSignal,
+        notifyBuyToCoverTraded,
+        notifyEnterLongSignal,
+        notifyEnterLongTraded,
+        notifyEnterShortSignal,
+        notifyEnterShortTraded,
+        notifyExitLongSignal,
+        notifyExitLongTraded,
+        send,
+    }
 }
diff --git a/src/notifiers/index.js b/src/notifiers/index.js
@@ -1,23 +1,30 @@
 const env = require("./../env")
 
 module.exports = function (trading_pairs) {
-  const notifiers = [];
-  if (env.USE_TELEGRAM)
-    notifiers.push(require('./telegram')(trading_pairs))
-  if (env.USE_GMAIL)
-    notifiers.push(require('./gmail')())
+    const notifiers = []
+    if (env.USE_TELEGRAM) notifiers.push(require("./telegram")(trading_pairs))
+    if (env.USE_GMAIL) notifiers.push(require("./gmail")())
 
-  const notifyAllFor = (method, arg) => notifiers.forEach(n => n[method] && n[method](arg));
+    const notifyAllFor = (method, arg) =>
+        notifiers.forEach((n) => n[method] && n[method](arg))
 
-  return {
-    notifyBuyToCoverSignal: signal => notifyAllFor("notifyBuyToCoverSignal", signal),
-    notifyBuyToCoverTraded: signal => notifyAllFor("notifyBuyToCoverTraded", signal),
-    notifyEnterLongSignal: signal => notifyAllFor("notifyEnterLongSignal", signal),
-    notifyEnterLongTraded: signal => notifyAllFor("notifyEnterLongTraded", signal),
-    notifyEnterShortSignal: signal => notifyAllFor("notifyEnterShortSignal", signal),
-    notifyEnterShortTraded: signal => notifyAllFor("notifyEnterShortTraded", signal),
-    notifyExitLongSignal: signal => notifyAllFor("notifyExitLongSignal", signal),
-    notifyExitLongTraded: signal => notifyAllFor("notifyExitLongTraded", signal),
-    send: message => notifyAllFor("send", message),
-  }
-}
+    return {
+        notifyBuyToCoverSignal: (signal) =>
+            notifyAllFor("notifyBuyToCoverSignal", signal),
+        notifyBuyToCoverTraded: (signal) =>
+            notifyAllFor("notifyBuyToCoverTraded", signal),
+        notifyEnterLongSignal: (signal) =>
+            notifyAllFor("notifyEnterLongSignal", signal),
+        notifyEnterLongTraded: (signal) =>
+            notifyAllFor("notifyEnterLongTraded", signal),
+        notifyEnterShortSignal: (signal) =>
+            notifyAllFor("notifyEnterShortSignal", signal),
+        notifyEnterShortTraded: (signal) =>
+            notifyAllFor("notifyEnterShortTraded", signal),
+        notifyExitLongSignal: (signal) =>
+            notifyAllFor("notifyExitLongSignal", signal),
+        notifyExitLongTraded: (signal) =>
+            notifyAllFor("notifyExitLongTraded", signal),
+        send: (message) => notifyAllFor("send", message),
+    }
+}
diff --git a/src/notifiers/telegram.js b/src/notifiers/telegram.js
@@ -5,10 +5,10 @@ const TeleBot = require("telebot")
 /**
  * @type {TeleBot}
  */
-let telBot;
+let telBot
 
 module.exports = function (trading_pairs) {
-    if (!env.USE_TELEGRAM) return publicMethods;
+    if (!env.USE_TELEGRAM) return publicMethods
 
     telBot = new TeleBot(env.TELEGRAM_API_KEY)
 
@@ -20,52 +20,93 @@ module.exports = function (trading_pairs) {
         return send(response)
     })
 
-
     telBot.on("start", () => {
         send("Trader Bot started!")
     })
 
     telBot.start()
-    return publicMethods;
+    return publicMethods
 }
 
 function createSignalMessage(base, signal) {
-    let msg = base + " :: " + signal.stratname + ' ' + signal.pair + ' ' + signal.price + "\n"
-    msg += (signal.score ? "score: " + signal.score : 'score: NA') + "\n"
+    let msg =
+        base +
+        " :: " +
+        signal.stratname +
+        " " +
+        signal.pair +
+        " " +
+        signal.price +
+        "\n"
+    msg += (signal.score ? "score: " + signal.score : "score: NA") + "\n"
     return msg
 }
 
 function send(message) {
-    if (!env.USE_TELEGRAM || !telBot) return;
+    if (!env.USE_TELEGRAM || !telBot) return
 
     return telBot.sendMessage(env.TELEGRAM_RECEIVER_ID, message, {
-        parseMode: "html"
+        parseMode: "html",
     })
 }
 
 function notifyBuyToCoverSignal(signal) {
-    return send(createSignalMessage("<i>BUY_SIGNAL :: BUY TO COVER SHORT TRADE</i>", signal));
+    return send(
+        createSignalMessage(
+            "<i>BUY_SIGNAL :: BUY TO COVER SHORT TRADE</i>",
+            signal
+        )
+    )
 }
 function notifyBuyToCoverTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! BUY_SIGNAL :: BUY TO COVER SHORT TRADE</b>", signal));
+    return send(
+        createSignalMessage(
+            "<b>>> SUCCESS! BUY_SIGNAL :: BUY TO COVER SHORT TRADE</b>",
+            signal
+        )
+    )
 }
 function notifyEnterLongSignal(signal) {
-    return send(createSignalMessage("<i>BUY_SIGNAL :: ENTER LONG TRADE</i>", signal));
+    return send(
+        createSignalMessage("<i>BUY_SIGNAL :: ENTER LONG TRADE</i>", signal)
+    )
 }
 function notifyEnterLongTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! BUY_SIGNAL :: ENTER LONG TRADE</b>", signal));
+    return send(
+        createSignalMessage(
+            "<b>>> SUCCESS! BUY_SIGNAL :: ENTER LONG TRADE</b>",
+            signal
+        )
+    )
 }
 function notifyEnterShortSignal(signal) {
-    return send(createSignalMessage("<i>SELL_SIGNAL :: ENTER SHORT TRADE</i>", signal));
+    return send(
+        createSignalMessage("<i>SELL_SIGNAL :: ENTER SHORT TRADE</i>", signal)
+    )
 }
 function notifyEnterShortTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! SELL_SIGNAL :: ENTER SHORT TRADE</b>", signal));
+    return send(
+        createSignalMessage(
+            "<b>>> SUCCESS! SELL_SIGNAL :: ENTER SHORT TRADE</b>",
+            signal
+        )
+    )
 }
 function notifyExitLongSignal(signal) {
-    return send(createSignalMessage("<i>SELL_SIGNAL :: SELL TO EXIT LONG TRADE</i>", signal));
+    return send(
+        createSignalMessage(
+            "<i>SELL_SIGNAL :: SELL TO EXIT LONG TRADE</i>",
+            signal
+        )
+    )
 }
 function notifyExitLongTraded(signal) {
-    return send(createSignalMessage("<b>>> SUCCESS! SELL_SIGNAL :: SELL TO EXIT LONG TRADE</b>", signal));
+    return send(
+        createSignalMessage(
+            "<b>>> SUCCESS! SELL_SIGNAL :: SELL TO EXIT LONG TRADE</b>",
+            signal
+        )
+    )
 }
 
 const publicMethods = {