Superalgos/LowFrequencyTrading/TradingStrategy.js

exports.newTradingStrategy = function newTradingStrategy(bot, logger) {
    /*
    This module packages all functions related to Strategies.
    */
    const MODULE_NAME = 'Trading Strategy'
    let thisObject = {
        openStrategy: openStrategy,
        closeStrategy: closeStrategy,
        updateEnds: updateEnds,
        updateStageType: updateStageType,
        updateStatus: updateStatus,
        updateCounters: updateCounters,
        resetStrategy: resetStrategy,
        initialize: initialize,
        finalize: finalize
    }

    let tradingEngine

    return thisObject

    function initialize() {
        tradingEngine = bot.simulationState.tradingEngine
    }

    function finalize() {
        tradingEngine = undefined
    }

    function openStrategy(stageType, index, situationName, strategyName) {
        tradingEngine.current.strategy.status.value = 'Open'
        tradingEngine.current.strategy.begin.value = tradingEngine.current.candle.begin.value
        tradingEngine.current.strategy.end.value = tradingEngine.current.candle.end.value
        tradingEngine.current.strategy.beginRate.value = tradingEngine.current.candle.min.value
        tradingEngine.current.strategy.endRate.value = tradingEngine.current.candle.min.value

        tradingEngine.current.strategy.stageType.value = stageType
        tradingEngine.current.strategy.index.value = index
        tradingEngine.current.strategy.situationName.value = situationName
        tradingEngine.current.strategy.strategyName.value = strategyName
    }

    function closeStrategy() {
        tradingEngine.current.strategy.status.value = 'Closed'
        tradingEngine.current.strategy.end.value = tradingEngine.current.candle.end.value
        tradingEngine.current.strategy.endRate.value = tradingEngine.current.candle.min.value
    }

    function resetStrategy() {
        tradingEngine.current.strategy.status.value = tradingEngine.current.strategy.status.config.initialValue
        tradingEngine.current.strategy.stageType.value = tradingEngine.current.strategy.stageType.config.initialValue
        tradingEngine.current.strategy.index.value = tradingEngine.current.strategy.index.config.initialValue
        tradingEngine.current.strategy.strategyCounters.periods.value = tradingEngine.current.strategy.strategyCounters.periods.config.initialValue
    }

    function updateEnds() {
        if (tradingEngine.current.strategy.status.value === 'Open') {
            tradingEngine.current.strategy.end.value = tradingEngine.current.candle.end.value
            tradingEngine.current.strategy.endRate.value = tradingEngine.current.candle.min.value
        }
    }

    function updateStageType(stageType) {
        tradingEngine.current.strategy.stageType.value = stageType
    }

    function updateStatus() {
        if (tradingEngine.current.strategy.status.value === 'Closed') {
            resetStrategy()
        }
    }

    function updateCounters() {
        if (tradingEngine.current.strategy.status.value === 'Open') {
            tradingEngine.current.strategy.strategyCounters.periods.value++
        }
    }
}