import streamlit as st
from hummingbot.connector.connector_base import OrderType

from frontend.components.st_inputs import get_distribution, normalize, distribution_inputs


def get_dca_distribution_inputs():
    with st.expander("DCA Distribution", expanded=True):
        st.header("DCA Settings")
        buy_order_levels = st.number_input("Number of Order Levels", min_value=1, value=3,
                                           help="Enter the number of order levels (e.g., 2).")

        col_spreads, col_amounts = st.columns(2)
        with col_spreads:
            buy_spread_dist_type, buy_spread_start, buy_spread_base, buy_spread_scaling, buy_spread_step, buy_spread_ratio, buy_manual_spreads = distribution_inputs(
                col_spreads, "Spread", buy_order_levels)
        with col_amounts:
            buy_amount_dist_type, buy_amount_start, buy_amount_base, buy_amount_scaling, buy_amount_step, buy_amount_ratio, buy_manual_amounts = distribution_inputs(
                col_amounts, "Amount", buy_order_levels)

        # Generate distributions
        spread_distributions = get_distribution(buy_spread_dist_type, buy_order_levels, buy_spread_start,
                                                buy_spread_base, buy_spread_scaling, buy_spread_step, buy_spread_ratio,
                                                buy_manual_spreads)

        amount_distributions = get_distribution(buy_amount_dist_type, buy_order_levels, buy_amount_start,
                                                buy_amount_base, buy_amount_scaling,
                                                buy_amount_step, buy_amount_ratio, buy_manual_amounts)

        # Normalize and calculate order amounts
        orders_amount_normalized = normalize(amount_distributions)
        spreads_normalized = [spread - spread_distributions[0] for spread in spread_distributions]
        st.write("---")
        c1, c2, c3, c4, c5 = st.columns(5)
        with c1:
            sl = st.number_input("Stop Loss (%)", min_value=0.0, max_value=100.0, value=2.0, step=0.1,
                                 help="Enter the stop loss as a percentage (e.g., 2.0 for 2%).") / 100
        with c2:
            tp = st.number_input("Take Profit (%)", min_value=0.0, max_value=100.0, value=1.0, step=0.1,
                                 help="Enter the take profit as a percentage (e.g., 3.0 for 3%).") / 100
        with c3:
            time_limit = st.number_input("Time Limit (minutes)", min_value=0, value=60 * 6,
                                         help="Enter the time limit in minutes (e.g., 360 for 6 hours).") * 60
        with c4:
            ts_ap = st.number_input("Trailing Stop Act. Price (%)", min_value=0.0, max_value=100.0, value=1.8,
                                    step=0.1,
                                    help="Enter the tr  ailing stop activation price as a percentage (e.g., 1.0 for 1%).") / 100
        with c5:
            ts_delta = st.number_input("Trailing Stop Delta (%)", min_value=0.0, max_value=100.0, value=0.2, step=0.1,
                                       help="Enter the trailing stop delta as a percentage (e.g., 0.3 for 0.3%).") / 100

    return {
        "dca_spreads": [spread /100 for spread in spreads_normalized],
        "dca_amounts": orders_amount_normalized,
        "stop_loss": sl,
        "take_profit": tp,
        "time_limit": time_limit,
        "trailing_stop": {
            "activation_price": ts_ap,
            "trailing_delta": ts_delta
        },
    }