Files
Superalgos/Assistant.js
T

560 lines
23 KiB
JavaScript

exports.newAssistant = function newAssistant(BOT, DEBUG_MODULE) {
/*
This module allows trading bots to execute actions on the exchange, and also on its current recorded state.
*/
const MODULE_NAME = "Assistant";
thisObject = {
initialize: initialize,
putPositionAtExchange: putPositionAtExchange,
movePositionAtExchange: movePositionAtExchange
};
let bot = BOT;
const logger = DEBUG_MODULE.newDebugLog();
logger.fileName = MODULE_NAME;
let exchangePositions = []; // These are the open positions at the exchange at the account the bot is authorized to use.
let openPositions = []; // These are the open positions the bot knows it made by itself.
let botContext;
let processDatetime;
let exchangeAPI;
/* The current bots trades only at one market: USDT_BTC. */
const MARKET = {
assetA: "USDT",
assetB: "BTC",
};
return thisObject;
function initialize(pBotContext, pProcessDatetime, pExchangeAPI, callBackFunction) {
try {
/* Store local values. */
botContext = pBotContext;
processDatetime = pProcessDatetime;
exchangeAPI = pExchangeAPI;
getPositionsAtExchange(onDone);
function onDone(err) {
try {
switch (err) {
case null: {
callBackFunction(null);
}
break;
case 'Retry Later': { // Something bad happened, but if we retry in a while it might go through the next time.
logger.write("[ERROR] initialize -> onDone -> Retry Later. Requesting Execution Retry.");
callBackFunction(err);
return;
}
break;
case 'Retry Later': { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] initialize -> onDone -> Operation Failed. Aborting the process.");
callBackFunction(err);
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] initialize -> onDone -> err = " + err);
callBackFunction("Operation Failed");
}
}
} catch (err) {
logger.write("[ERROR] initialize -> err = " + err);
callBackFunction("Operation Failed");
}
}
function getPositionsAtExchange(callBack) {
try {
/*
Here we grab all the positions at the exchange for the account we are using for trading. We will not asume all the positions
were made by this bot, but we need them all to later check if all the ones made by the bot are still there, were executed or
manually cancelled by the exchange account owner.
*/
exchangeAPI.getOpenPositions(MARKET, onResponse);
function onResponse(err, pExchangePositions) {
switch (err) {
case null: { // Everything went well, we have the information requested.
exchangePositions = pExchangePositions;
ordersExecutionCheck(callBack);
}
break;
case 'Retry Later': { // Something bad happened, but if we retry in a while it might go through the next time.
logger.write("[ERROR] getPositionsAtExchange -> onResponse -> Retry Later. Requesting Execution Retry.");
callBack(err);
return;
}
break;
case 'Operation Failed': { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] getPositionsAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack(err);
return;
}
break;
}
}
} catch (err) {
logger.write("[ERROR] getPositionsAtExchange -> err = " + err);
callBack("Operation Failed");
}
}
function ordersExecutionCheck(callBack) {
try {
/*
Here we check that all the positions we know we have are still at the exchange. If they are not, we will try to take appropiate
actions. Reasons why the positions might not be there are:
1. The user / account owner closed the positions manually.
2. The exchange for some eventuality closed the positions. In some exchanges positions have an expiratin time.
3. The orders were executed.
For as 1 and 2 are similar unexpected situations and we will stop the bot execution when we detect we are there. Number 3 is an
expected behaviour and we take appropiate action.
Also, the position might still be there but it might have been partially executed. To detect that we need to compare the
position amounts we have on file to the one we are receiving from the exchange.
Before we begin, we have to remove all the orders that have been already closed at the last execution of the bot.
*/
let openPositions = [];
for (let a = 0; a < botContext.executionContext.positions.length; a++) {
if (botContext.executionContext.positions[a].status === "open") {
openPositions.push(botContext.executionContext.positions[a]);
}
}
botContext.executionContext.positions = openPositions;
/* Now we can start checking what happened at the exchange. */
let i = 0;
controlLoop();
function loopBody() {
try {
for (let j = 0; j < exchangePositions.length; j++) {
if (botContext.executionContext.positions[i].id === exchangePositions[j].id) {
/*
We need to know if the order was partially executed. To know that, we compare the amounts on file with the ones
received from the exchange. If they are the same, then the order is intact. Otherwise, to confirm a partial execution,
we will request the associated trades from the exchange.
*/
if (botContext.executionContext.positions[i].amountB === parseFloat(exchangePositions[j].amountB)) {
/* Position is still there, untouched. Nothing to do here. */
next();
return;
} else {
getPositionTradesAtExchange(botContext.executionContext.positions[i].id, confirmOrderWasPartiallyExecuted);
return;
function confirmOrderWasPartiallyExecuted(pTrades) {
try {
/*
To confirm everything is ok, we will add all the amounts on trades plus the remaining amounts
at the order at the exchange and they must be equal to the one on file. Otherwise something very strange could
have happened, in which case we will halt the bot execution.
*/
let sumAssetA = 0;
let sumAssetB = 0;
for (k = 0; k < pTrades.length; k++) {
sumAssetA = sumAssetA + pTrades[k].amountA;
sumAssetB = sumAssetB + pTrades[k].amountB;
}
/* To this we add the current position amounts. */
sumAssetA = sumAssetA + exchangePositions[j].amountA;
sumAssetB = sumAssetB + exchangePositions[j].amountB;
/* And finally we add the fees */
sumAssetA = sumAssetA + exchangePositions[j].fee;
if (
botContext.executionContext.positions[i].amountA !== sumAssetA ||
botContext.executionContext.positions[i].amountB !== sumAssetB
) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> confirmOrderWasPartiallyExecuted -> Cannot be confirmed that a partially execution was done well.");
callBack("Operation Failed");
return;
}
/*
Confirmed that order was partially executed. Next thing to do is to remember the trades and the new position.
*/
botContext.executionContext.positions[i].amountA = exchangePositions[j].amountA;
botContext.executionContext.positions[i].amountB = exchangePositions[j].amountB;
botContext.executionContext.positions[i].date = (new Date(exchangePositions[j].date)).valueOf();
for (k = 0; k < pTrades.length; k++) {
botContext.executionContext.positions[i].pTrades.push(pTrades[k]);
botContext.newHistoryRecord.newTrades++;
}
let newTransaction = {
type: botContext.executionContext.positions[i].type + " partially executed",
position: botContext.executionContext.positions[i]
};
botContext.executionContext.transactions.push(newTransaction);
/* All done. */
next();
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> confirmOrderWasPartiallyExecuted -> err = " + err);
callBack("Operation Failed");
return;
}
}
}
}
}
/* Position not found: we need to know if the order was executed. */
getPositionTradesAtExchange(botContext.executionContext.positions[i].id, confirmOrderWasExecuted);
function confirmOrderWasExecuted(pTrades) {
try {
/*
To confirm everything is ok, we will add all the amounts on trades asociated to the order and
they must be equal to the one on file. Otherwise something very strange could have happened,
in which case we will halt the bot execution.
*/
let sumAssetA = 0;
let sumAssetB = 0;
for (k = 0; k < pTrades.length; k++) {
sumAssetA = sumAssetA + pTrades[k].amountA;
sumAssetB = sumAssetB + pTrades[k].amountB;
}
/* We add the fees */
sumAssetA = sumAssetA + exchangePositions[j].fee;
if (
botContext.executionContext.positions[i].amountA !== sumAssetA ||
botContext.executionContext.positions[i].amountB !== sumAssetB
) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> confirmOrderWasExecuted -> Cannot be confirmed that the order was executed. It must be manually cancelled by the user or cancelled by the exchange itself.");
callBack("Operation Failed");
return;
}
/*
Confirmed that order was executed. Next thing to do is to remember the trades and change its status.
*/
botContext.executionContext.positions[i].status = "executed";
for (k = 0; k < pTrades.length; k++) {
botContext.executionContext.positions[i].pTrades.push(pTrades[k]);
botContext.newHistoryRecord.newTrades++;
}
let newTransaction = {
type: botContext.executionContext.positions[i].type + " executed",
position: botContext.executionContext.positions[i]
};
botContext.executionContext.transactions.push(newTransaction);
/* All done. */
next();
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> confirmOrderWasExecuted -> err = " + err);
callBack("Operation Failed");
return;
}
}
function getPositionTradesAtExchange(pPositionId, innerCallBack) {
try {
/*
Given one position, we request all the associated trades to it.
*/
exchangeAPI.getExecutedTrades(pPositionId, onResponse);
function onResponse(err, pTrades) {
try {
switch (err) {
case null: { // Everything went well, we have the information requested.
innerCallBack(pTrades);
}
break;
case 'Retry Later': { // Something bad happened, but if we retry in a while it might go through the next time.
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> onResponse -> Retry Later. Requesting Execution Retry.");
callBack('Retry Later');
return;
}
break;
case 'Operation Failed': { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack('Operation Failed');
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> onResponse -> err = " + err);
callBack("Operation Failed");
}
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> err = " + err);
callBack("Operation Failed");
}
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> err = " + err);
callBack("Operation Failed");
return;
}
}
function next() {
i++;
controlLoop();
}
function controlLoop() {
if (i < botContext.executionContext.positions.length) {
loopBody();
} else {
final();
}
}
function final() {
callBack(null);
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> err = " + err);
callBack("Operation Failed");
}
}
function putPositionAtExchange(pType, pRate, pAmountA, pAmountB, callBack) {
try {
exchangeAPI.putPosition(MARKET, pType, pRate, pAmountA, pAmountB, onResponse);
function onResponse(err, pPositionId) {
try {
switch (err) {
case null: { // Everything went well, we have the information requested.
let position = {
id: pPositionId,
type: pType,
rate: pRate,
amountA: pAmountA,
amountB: pAmountB,
date: (processDatetime.valueOf()),
status: "open",
trades: []
};
botContext.executionContext.positions.push(position);
let newTransaction = {
type: "newPosition",
position: position
};
botContext.executionContext.transactions.push(newTransaction);
botContext.newHistoryRecord.newPositions++;
botContext.newHistoryRecord.rate = pRate;
callBack(null);
}
break;
case 'Retry Later': { // Something bad happened, but if we retry in a while it might go through the next time.
logger.write("[ERROR] putPositionAtExchange -> onResponse -> Retry Later. Requesting Execution Retry.");
callBack('Retry Later');
return;
}
break;
case 'Operation Failed': { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] putPositionAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack('Operation Failed');
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] putPositionAtExchange -> onResponse -> err = " + err);
callBack("Operation Failed");
}
}
} catch (err) {
logger.write("[ERROR] putPositionAtExchange -> err = " + err);
callBack("Operation Failed");
}
}
function movePositionAtExchange(pPosition, pNewRate, callBack) {
try {
exchangeAPI.movePosition(pPosition, pNewRate, onResponse);
function onResponse(err, pPositionId) {
try {
switch (err) {
case null: { // Everything went well, we have the information requested.
let newPosition = {
id: pPositionId,
type: pPosition.type,
rate: pNewRate,
amountA: pPosition.amountB * pNewRate,
amountB: pPosition.amountB,
date: (processDatetime.valueOf()),
status: "open",
trades: []
};
let oldPosition = JSON.parse(JSON.stringify(pPosition));
/* We need to update the position we have on file. */
for (let i = 0; i < botContext.executionContext.positions.length; i++) {
if (botContext.executionContext.positions[i].id === pPosition.id) {
botContext.executionContext.positions[i] = newPosition;
break;
}
}
let newTransaction = {
type: "movePosition",
oldPosition: oldPosition,
newPosition: newPosition
};
botContext.executionContext.transactions.push(newTransaction);
botContext.newHistoryRecord.movedPositions++;
botContext.newHistoryRecord.rate = pNewRate;
callBack(null);
}
break;
case 'Retry Later': { // Something bad happened, but if we retry in a while it might go through the next time.
logger.write("[ERROR] movePositionAtExchange -> onResponse -> Retry Later. Requesting Execution Retry.");
callBack('Retry Later');
return;
}
break;
case 'Operation Failed': { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] movePositionAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack('Operation Failed');
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] movePositionAtExchange -> onResponse -> err = " + err);
callBack("Operation Failed");
}
}
} catch (err) {
logger.write("[ERROR] movePositionAtExchange -> err = " + err);
callBack("Operation Failed");
}
}
};