# --- Do not remove these libs ---
from freqtrade.strategy.interface import IStrategy
from typing import Dict, List
from functools import reduce
from pandas import DataFrame
# --------------------------------

import talib.abstract as ta
import numpy as np
import freqtrade.vendor.qtpylib.indicators as qtpylib
import datetime
from technical.util import resample_to_interval, resampled_merge
from datetime import datetime, timedelta
from freqtrade.persistence import Trade
from freqtrade.strategy import stoploss_from_open, merge_informative_pair, DecimalParameter, IntParameter, CategoricalParameter

# author @tirail

ma_types = {
        'SMA': ta.SMA,
        'EMA': ta.EMA,
        }

class SMAOffsetBZed(IStrategy):
    INTERFACE_VERSION = 2
    # Optimal timeframe for the strategy
    timeframe = '5m'

    use_sell_signal = True
    sell_profit_only = False

    process_only_new_candles = True
    startup_candle_count = 200
    
    

    plot_config = {
            'main_plot': {
                'ma_offset_buy': {'color': 'orange'},
                'ma_offset_sell': {'color': 'orange'},
                },
            }

    # Buy hyperspace params:
    buy_params = {
        "buy_dip_candles_1": 3,
        "buy_dip_candles_2": 29,
        "buy_dip_candles_3": 130,
        "buy_dip_threshold_1": 0.13,
        "buy_dip_threshold_2": 0.2,
        "buy_dip_threshold_3": 0.25,
        "base_nb_candles_buy": 19,  # value loaded from strategy
        "buy_trigger": "SMA",  # value loaded from strategy
        "low_offset": 0.969,  # value loaded from strategy
    }

    # Sell hyperspace params:
    sell_params = {
        "base_nb_candles_sell": 30,  # value loaded from strategy
        "high_offset": 1.012,  # value loaded from strategy
        "sell_trigger": "EMA",  # value loaded from strategy
    }



    # Stoploss:
    stoploss = -0.1

    # ROI table:
    minimal_roi = {
            "0": 1,
            }

    inf_1h = '1h' # informative tf

    optimize_dip = False
    optimize_non_dip = True
    optimize_roi = False
    optimize_rsi = False
    optimize_profits = False
    
    base_nb_candles_buy = IntParameter(5, 80, default=30, space='buy', optimize=optimize_non_dip, load=True)
    low_offset = DecimalParameter(0.8, 0.99, default=0.958, space='buy', optimize=optimize_non_dip, load=True)

    buy_dip_threshold_1 = DecimalParameter(0.08, 0.2, default=0.12, space='buy', decimals=2, optimize=optimize_dip, load=True)
    buy_dip_threshold_2 = DecimalParameter(0.02, 0.5, default=0.28, space='buy', decimals=2, optimize=optimize_dip, load=True)
    buy_dip_threshold_3 = DecimalParameter(0.02, 0.5, default=0.28, space='buy', decimals=2, optimize=optimize_dip, load=True)
    buy_dip_candles_1 = IntParameter(1, 20, default=2,  space='buy', optimize=optimize_dip, load=True)
    buy_dip_candles_2 = IntParameter(1, 40, default=10, space='buy', optimize=optimize_dip, load=True)
    buy_dip_candles_3 = IntParameter(40, 140, default=132, space='buy', optimize=optimize_dip, load=True)

    # sell params
    base_nb_candles_sell = IntParameter(5, 80, default=30, load=True, optimize=optimize_non_dip, space='sell')
    high_offset = DecimalParameter(0.8, 1.1, default=1.012, load=True, optimize=optimize_non_dip, space='sell')

    # they are fine as they are.
    sell_trigger = CategoricalParameter(ma_types.keys(), default='EMA', space='sell', load=optimize_non_dip, optimize=False)
    buy_trigger = CategoricalParameter(ma_types.keys(), default='SMA', space='buy', load=optimize_non_dip, optimize=False)
    

    
    #roi and profit from bhc8
    sell_custom_roi_profit_1 = DecimalParameter(0.01, 0.02, default=0.01, space='sell', decimals=3, optimize=optimize_roi, load=True)
    sell_custom_roi_profit_2 = DecimalParameter(0.01, 0.05, default=0.04, space='sell', decimals=2, optimize=optimize_roi, load=True)
    sell_custom_roi_profit_3 = DecimalParameter(0.05, 0.1, default=0.08, space='sell', decimals=2, optimize=optimize_roi, load=True)
    sell_custom_roi_profit_4 = DecimalParameter(0.1, 0.2, default=0.14, space='sell', decimals=2, optimize=optimize_roi, load=True)
    sell_custom_roi_profit_5 = DecimalParameter(0.01, 0.1, default=0.04, space='sell', decimals=2, optimize=optimize_roi, load=True)
        
    sell_custom_roi_rsi_1 = DecimalParameter(40, 56, default=50, space='sell', decimals=0, optimize=optimize_rsi, load=True)
    sell_custom_roi_rsi_2 = DecimalParameter(42, 56, default=50, space='sell', decimals=0, optimize=optimize_rsi, load=True)
    sell_custom_roi_rsi_3 = DecimalParameter(44, 58, default=56, space='sell', decimals=0, optimize=optimize_rsi, load=True)
    sell_custom_roi_rsi_4 = DecimalParameter(44, 60, default=58, space='sell', decimals=0, optimize=optimize_rsi, load=True)
    
    sell_trail_profit_min_1 = DecimalParameter(0.1, 0.25, default=0.1, space='sell', decimals=3, optimize=optimize_profits, load=True)
    sell_trail_profit_max_1 = DecimalParameter(0.3, 0.5, default=0.4, space='sell', decimals=2, optimize=optimize_profits, load=True)
    sell_trail_down_1 = DecimalParameter(0.04, 0.1, default=0.03, space='sell', decimals=3, optimize=optimize_profits, load=True)

    sell_trail_profit_min_2 = DecimalParameter(0.01, 0.1, default=0.02, space='sell', decimals=3, optimize=optimize_profits, load=True)
    sell_trail_profit_max_2 = DecimalParameter(0.08, 0.25, default=0.1, space='sell', decimals=2, optimize=optimize_profits, load=True)
    sell_trail_down_2 = DecimalParameter(0.04, 0.1, default=0.08, space='sell', decimals=3, optimize=optimize_profits, load=True)

    def informative_pairs(self):
        pairs = self.dp.current_whitelist()
        informative_pairs = [(pair, self.inf_1h) for pair in pairs]
        return informative_pairs

    def informative_1h_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        assert self.dp, "DataProvider is required for multiple timeframes."
        # Get the informative pair
        informative_1h = self.dp.get_pair_dataframe(pair=metadata['pair'], timeframe=self.inf_1h)
        # EMA

        informative_1h['ema_50'] = ta.EMA(informative_1h, timeperiod=50)
        informative_1h['ema_100'] = ta.EMA(informative_1h, timeperiod=100)
        informative_1h['ema_200'] = ta.EMA(informative_1h, timeperiod=200)

        return informative_1h

    def custom_stoploss(self, pair: str, trade: 'Trade', current_time: datetime, current_rate: float, current_profit: float, **kwargs) -> float:
        return 1

    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        if not self.config['runmode'].value == 'hyperopt':
            dataframe['ma_offset_buy'] = ma_types[self.buy_trigger.value](dataframe, int(self.base_nb_candles_buy.value)) * self.low_offset.value
            dataframe['ma_offset_sell'] = ma_types[self.sell_trigger.value](dataframe, int(self.base_nb_candles_sell.value)) * self.high_offset.value
        informative_1h = self.informative_1h_indicators(dataframe, metadata)
        dataframe = merge_informative_pair(dataframe, informative_1h, self.timeframe, self.inf_1h, ffill=True)
        
        dataframe['ema_200'] = ta.EMA(dataframe, timeperiod=200)
        
        # RSI
        dataframe['rsi'] = ta.RSI(dataframe, timeperiod=14)
        
        informative_1h = self.dp.get_pair_dataframe(pair=metadata['pair'], timeframe=self.inf_1h)
        
        # SMA
        informative_1h['sma_200_1h'] = ta.SMA(informative_1h, timeperiod=200)
        informative_1h['sma_200_dec_1h'] = informative_1h['sma_200_1h'] < informative_1h['sma_200_1h'].shift(20)
        
        # SMA
        dataframe['sma_200'] = ta.SMA(dataframe, timeperiod=200)
        dataframe['sma_200_dec'] = dataframe['sma_200'] < dataframe['sma_200'].shift(20)
        return dataframe

    def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        if self.config['runmode'].value == 'hyperopt':
            dataframe['ma_offset_buy'] = ma_types[self.buy_trigger.value](dataframe, int(self.base_nb_candles_buy.value)) * self.low_offset.value

        dataframe.loc[
                (
                    (dataframe['close'] > dataframe['ema_200']) &
                    (dataframe['close'] > dataframe['ema_200_1h']) &
                    (dataframe['ema_50_1h'] > dataframe['ema_100_1h']) &
                    (dataframe['ema_50_1h'] > dataframe['ema_200_1h']) &

                    (((dataframe['open'].rolling(int(self.buy_dip_candles_1.value)).max() - dataframe['close']) / dataframe['close']) < self.buy_dip_threshold_1.value) &
                    (((dataframe['open'].rolling(int(self.buy_dip_candles_1.value + self.buy_dip_candles_2.value)).max() - dataframe['close']) / dataframe['close']) < self.buy_dip_threshold_2.value) &
                    (((dataframe['open'].rolling(int(self.buy_dip_candles_1.value + self.buy_dip_candles_2.value + self.buy_dip_candles_3.value)).max() - dataframe['close']) / dataframe['close']) < self.buy_dip_threshold_3.value) &

                    (dataframe['close'] < dataframe['ma_offset_buy']) &
                    (dataframe['volume'] > 0)
                    ),
                'buy'] = 1
        return dataframe

    def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        if self.config['runmode'].value == 'hyperopt':
            dataframe['ma_offset_sell'] = ma_types[self.sell_trigger.value](dataframe, int(self.base_nb_candles_sell.value)) * self.high_offset.value

        dataframe.loc[
                (
                    (dataframe['close'] > dataframe['ma_offset_sell']) &
                    (dataframe['volume'] > 0)
                    ),
                'sell'] = 1
        return dataframe

#stolen from bhc8
    def custom_sell(self, pair: str, trade: 'Trade', current_time: 'datetime', current_rate: float,
                    current_profit: float, **kwargs):
        dataframe, _ = self.dp.get_analyzed_dataframe(pair, self.timeframe)
        last_candle = dataframe.iloc[-1].squeeze()

        if (last_candle is not None):
            if (current_profit > self.sell_custom_roi_profit_4.value) & (last_candle['rsi'] < self.sell_custom_roi_rsi_4.value):
                return 'roi_target_4'
            elif (current_profit > self.sell_custom_roi_profit_3.value) & (last_candle['rsi'] < self.sell_custom_roi_rsi_3.value):
                return 'roi_target_3'
            elif (current_profit > self.sell_custom_roi_profit_2.value) & (last_candle['rsi'] < self.sell_custom_roi_rsi_2.value):
                return 'roi_target_2'
            elif (current_profit > self.sell_custom_roi_profit_1.value) & (last_candle['rsi'] < self.sell_custom_roi_rsi_1.value):
                return 'roi_target_1'
            elif (current_profit > 0) & (current_profit < self.sell_custom_roi_profit_5.value) & (last_candle['sma_200_dec']):
                return 'roi_target_5'

            elif (current_profit > self.sell_trail_profit_min_1.value) & (current_profit < self.sell_trail_profit_max_1.value) & (((trade.max_rate - trade.open_rate) / trade.open_rate) > (current_profit + self.sell_trail_down_1.value)):
                return 'trail_target_1'
            elif (current_profit > self.sell_trail_profit_min_2.value) & (current_profit < self.sell_trail_profit_max_2.value) & (((trade.max_rate - trade.open_rate) / trade.open_rate) > (current_profit + self.sell_trail_down_2.value)):
                return 'trail_target_2'

        return None
        
        
class SMAOffset_OptimizeNonDip(SMAOffsetBZed):
    optimize_dip = False
    optimize_non_dip = True
    optimize_roi = False
    optimize_rsi = False
    optimize_profits = False

class SMAOffset_OptimizeDip(SMAOffsetBZed):
    optimize_dip = False
    optimize_non_dip = False
    optimize_roi = True
    optimize_rsi = True
    optimize_profits = False

    
class SMAOffset_OptimizeRoiRsi(SMAOffsetBZed):
    optimize_dip = False
    optimize_non_dip = False
    optimize_roi = True
    optimize_rsi = True
    optimize_profits = False
    
    
class SMAOffset_OptimizeProfits(SMAOffsetBZed):
    optimize_dip = False
    optimize_non_dip = False
    optimize_roi = True
    optimize_rsi = True
    optimize_profits = False
    
class SMAOffset_OptimizeNonDip1(SMAOffsetBZed):
    #Best result:
    #415/512:    456 trades. 409/0/47 Wins/Draws/Losses. Avg profit   1.05%. Median profit   1.44%. Total profit  119.92876131 USDT ( 159.91%). Avg duration 1:27:00 min. Objective: -42984.08811
    #docker-compose run freqtrade hyperopt --config user_data/config_max3Trades_USDT_IterativTop90.json --hyperopt-loss WinRatioAndProfitRatioLoss --spaces buy sell --strategy SMAOffsetBZed  --epochs 512 --timerange 20210418- --timeframe 5m --print-all --j 3
    # Buy hyperspace params:
    buy_params = {
        "base_nb_candles_buy": 24,
        "low_offset": 0.965,
        "buy_dip_candles_1": 3,  # value loaded from strategy
        "buy_dip_candles_2": 29,  # value loaded from strategy
        "buy_dip_candles_3": 130,  # value loaded from strategy
        "buy_dip_threshold_1": 0.13,  # value loaded from strategy
        "buy_dip_threshold_2": 0.2,  # value loaded from strategy
        "buy_dip_threshold_3": 0.25,  # value loaded from strategy
        "buy_trigger": "SMA",  # value loaded from strategy
    }

    # Sell hyperspace params:
    sell_params = {
        "base_nb_candles_sell": 36,
        "high_offset": 1.027,
        "sell_custom_roi_profit_1": 0.01,  # value loaded from strategy
        "sell_custom_roi_profit_2": 0.04,  # value loaded from strategy
        "sell_custom_roi_profit_3": 0.08,  # value loaded from strategy
        "sell_custom_roi_profit_4": 0.14,  # value loaded from strategy
        "sell_custom_roi_profit_5": 0.04,  # value loaded from strategy
        "sell_custom_roi_rsi_1": 50,  # value loaded from strategy
        "sell_custom_roi_rsi_2": 50,  # value loaded from strategy
        "sell_custom_roi_rsi_3": 56,  # value loaded from strategy
        "sell_custom_roi_rsi_4": 58,  # value loaded from strategy
        "sell_trail_down_1": 0.03,  # value loaded from strategy
        "sell_trail_down_2": 0.08,  # value loaded from strategy
        "sell_trail_profit_max_1": 0.4,  # value loaded from strategy
        "sell_trail_profit_max_2": 0.1,  # value loaded from strategy
        "sell_trail_profit_min_1": 0.1,  # value loaded from strategy
        "sell_trail_profit_min_2": 0.02,  # value loaded from strategy
        "sell_trigger": "EMA",  # value loaded from strategy
    }
