Files
Superalgos/Assistant.js
T

563 lines
24 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.result) {
case DEFAULT_OK_RESPONSE.result: {
callBackFunction(DEFAULT_OK_RESPONSE);
return;
}
break;
case DEFAULT_RETRY_RESPONSE.result: { // 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 DEFAULT_FAIL_RESPONSE.result: { // 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.message);
callBackFunction(DEFAULT_FAIL_RESPONSE);
}
}
} catch (err) {
logger.write("[ERROR] initialize -> err = " + err.message);
callBackFunction(DEFAULT_FAIL_RESPONSE);
}
}
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.result) {
case DEFAULT_OK_RESPONSE.result: { // Everything went well, we have the information requested.
exchangePositions = pExchangePositions;
ordersExecutionCheck(callBack);
return;
}
break;
case DEFAULT_RETRY_RESPONSE.result: { // 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 DEFAULT_FAIL_RESPONSE.result: { // 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.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
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(DEFAULT_FAIL_RESPONSE);
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.message);
callBack(DEFAULT_FAIL_RESPONSE);
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(DEFAULT_FAIL_RESPONSE);
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.message);
callBack(DEFAULT_FAIL_RESPONSE);
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.result) {
case DEFAULT_OK_RESPONSE.result: { // Everything went well, we have the information requested.
innerCallBack(pTrades);
}
break;
case DEFAULT_RETRY_RESPONSE.result: { // 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(DEFAULT_RETRY_RESPONSE);
return;
}
break;
case DEFAULT_FAIL_RESPONSE.result: { // 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(DEFAULT_FAIL_RESPONSE);
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> onResponse -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> getPositionTradesAtExchange -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> loopBody -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
return;
}
}
function next() {
i++;
controlLoop();
}
function controlLoop() {
if (i < botContext.executionContext.positions.length) {
loopBody();
} else {
final();
}
}
function final() {
callBack(DEFAULT_OK_RESPONSE);
}
} catch (err) {
logger.write("[ERROR] ordersExecutionCheck -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
function putPositionAtExchange(pType, pRate, pAmountA, pAmountB, callBack) {
try {
exchangeAPI.putPosition(MARKET, pType, pRate, pAmountA, pAmountB, onResponse);
function onResponse(err, pPositionId) {
try {
switch (err.result) {
case DEFAULT_OK_RESPONSE.result: { // 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(DEFAULT_OK_RESPONSE);
return;
}
break;
case DEFAULT_RETRY_RESPONSE.result: { // 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(DEFAULT_RETRY_RESPONSE);
return;
}
break;
case DEFAULT_FAIL_RESPONSE.result: { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] putPositionAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack(DEFAULT_FAIL_RESPONSE);
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] putPositionAtExchange -> onResponse -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
} catch (err) {
logger.write("[ERROR] putPositionAtExchange -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
function movePositionAtExchange(pPosition, pNewRate, callBack) {
try {
exchangeAPI.movePosition(pPosition, pNewRate, onResponse);
function onResponse(err, pPositionId) {
try {
switch (err.result) {
case DEFAULT_OK_RESPONSE.result: { // 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(DEFAULT_OK_RESPONSE);
}
break;
case DEFAULT_RETRY_RESPONSE.result: { // 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(DEFAULT_RETRY_RESPONSE);
return;
}
break;
case DEFAULT_FAIL_RESPONSE.result: { // This is an unexpected exception that we do not know how to handle.
logger.write("[ERROR] movePositionAtExchange -> onResponse -> Operation Failed. Aborting the process.");
callBack(DEFAULT_FAIL_RESPONSE);
return;
}
break;
}
} catch (err) {
logger.write("[ERROR] movePositionAtExchange -> onResponse -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
} catch (err) {
logger.write("[ERROR] movePositionAtExchange -> err = " + err.message);
callBack(DEFAULT_FAIL_RESPONSE);
}
}
};