rsibot-utils 1.0.0

package/config.js

@@ -0,0 +1,5 @@
+module.exports.APIKEY = '09R8VYlX2u5k38wx3HyJdAeR9lrwnepzqwogKyXLLexuJhxNjWZPIUPNweiGqla8';
+module.exports.APISECRET = 'Iy4k6ocN6GUaxMuLTHANS7DKJbI4hYgCz6gvYwiPq7TYbfBSycNah43zSAAq3dlY';
+module.exports.DB_USER = 'shalom';
+module.exports.DB_PASSWORD = 'AaSsDdFf1234';
+

package/db.js

@@ -0,0 +1,177 @@
+const mongoose = require('mongoose');
+const { DB_USER, DB_PASSWORD } = require('./config');
+const { write } = require('./utils');
+
+mongoose.Promise = global.Promise;
+const { Schema } = mongoose;
+// Create a schema
+
+const hotCoinsSchema = new Schema(
+  {
+    symbol: { type: String, unique: true },
+  },
+  {
+    versionKey: false,
+  },
+);
+const activeSchema = new Schema(
+  {
+    symbol: { type: String, unique: true },
+    inPosition: { type: Boolean },
+    lastBuy: { type: Number },
+    abandoned: { type: Boolean },
+  },
+  {
+    versionKey: false,
+  },
+);
+const queueSchema = new Schema(
+  {
+    queue: [],
+    created: { type: Date, default: Date.now },
+  },
+  {
+    versionKey: false,
+  },
+);
+
+// Create a model
+mongoose
+  .connect(`mongodb+srv://${DB_USER}:${DB_PASSWORD}@rsi-bot.d6kbq.mongodb.net/rsi?retryWrites=true&w=majority`, {
+    useNewUrlParser: true,
+    useUnifiedTopology: true,
+  })
+  .then((connection) => {
+    console.log('DB is connected');
+  })
+  .catch((err) => {
+    console.log(err);
+  });
+
+const HotCoin = mongoose.model('hot_coins', hotCoinsSchema);
+const ActiveCoin = mongoose.model('active_coins', activeSchema);
+const Queue = mongoose.model('queue_coins', queueSchema);
+
+module.exports = {
+  insertHotCoin: async function ({ symbol }) {
+    try {
+      const newHotCoin = new HotCoin({ symbol });
+      await newHotCoin.save();
+      return symbol;
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+      return false;
+    }
+  },
+  getHotCoins: async function () {
+    try {
+      const results = await HotCoin.find();
+      return results;
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+      return [];
+    }
+  },
+  removeHotCoin: async function (symbol) {
+    try {
+      await HotCoin.deleteOne({ symbol });
+      return symbol;
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+      return false;
+    }
+  },
+
+  getActiveCoins: async function () {
+    try {
+      const results = await ActiveCoin.find();
+      return results;
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+      return [];
+    }
+  },
+  getQueue: async function () {
+    try {
+      const results = await Queue.find();
+      if (results[0]) {
+        return results[0].queue;
+      } else {
+        return [];
+      }
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+      return [];
+    }
+  },
+  setQueue: async function (queue) {
+    if (queue.length > 0) {
+      try {
+        await Queue.deleteOne({ created: { $exists: true } });
+        const newQueue = new Queue({ queue });
+        await newQueue.save();
+      } catch (error) {
+        console.error(error.stack);
+        write(error.stack, true);
+      }
+    }
+  },
+  removeCoin: async function (symbol, queue) {
+    try {
+      await ActiveCoin.deleteOne({ symbol });
+      await Queue.deleteOne({ created: { $exists: true } });
+      const modifiedQueue = queue.filter((q) => q.symbol !== symbol);
+      const newQueue = new Queue({ queue: modifiedQueue });
+      await newQueue.save();
+    } catch (error) {
+      write(error, true);
+    }
+  },
+  initQueueListener: function () {
+    return Queue.watch();
+  },
+  getAbandonedCoins: async function () {
+    try {
+      const abandonedCoins = await ActiveCoin.find({ abandoned: true });
+      return abandonedCoins;
+    } catch (error) {
+      write(error, true);
+      return [];
+    }
+  },
+  createNewHistory: async function (coinObj) {
+    try {
+      const newHistory = new History(coinObj);
+      await newHistory.save();
+    } catch (error) {
+      console.error(error.stack);
+      write(error.stack, true);
+    }
+  },
+  deleteHotCoins: async function () {
+    try {
+      await HotCoin.deleteMany({ symbol: { $exists: true } });
+    } catch (error) {
+      write(error, true);
+    }
+  },
+  deleteActiveCoins: async function () {
+    try {
+      await ActiveCoin.deleteMany({ symbol: { $exists: true } });
+    } catch (error) {
+      write(error, true);
+    }
+  },
+  deleteQueueCoins: async function () {
+    try {
+      await Queue.deleteMany({ created: { $exists: true } });
+    } catch (error) {
+      write(error, true);
+    }
+  },
+};

package/index.js

@@ -0,0 +1,39 @@
+const fs = require('fs');
+const path = require('path');
+const util = require('util');
+const exec = util.promisify(require('child_process').exec);
+const dbUtils = require('./db')
+const cname = process.env.CNAME;
+const LOGS_PATH = cname?`${cname}_logs.txt`:"commander_logs.txt";
+
+module.exports = {
+  ...dbUtils,
+  sortByPercent: function (a, b) {
+    return b.percentChange - a.percentChange;
+  },
+  write: function (msg, isError = false) {
+    let dateStr = new Date().toLocaleString('he-IL');
+    let prefixType = isError ? '[ERROR]' : '[LOG]';
+
+    fs.appendFileSync(path.resolve(LOGS_PATH), `${dateStr}-${prefixType} ${msg}\n`, 'utf8');
+  },
+  execute: async function (command) {
+    console.log(`$ ${command}`);
+    try {
+      const { error, stdout, stderr } = await exec(command);
+      if (stderr) {
+        console.log(`${stderr}`);
+        return stderr;
+      }
+      // delete coin and archive data
+      console.log(`${stdout}`);
+      return stdout;
+    } catch (error) {
+      console.log(error);
+      throw Error(error);
+    }
+  },
+  getServiceName: function (symbol) {
+    return `rsibot_${symbol.slice(0, -4)}_${symbol.slice(-4)}`.toLowerCase();
+  },
+};

package/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "rsibot-utils",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "mongoose": "^6.2.9"
+  }
+}