# main.py - Dynamic Stock Input + Test Alert Dashboard
import yfinance as yf
import pandas as pd
import ta
import requests
import time
import re
from datetime import datetime
from flask import Flask, request, render_template_string, flash, get_flashed_messages
from threading import Thread
import logging

# === CONFIGURE LOGGING ===
logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s [%(levelname)s] %(message)s',
                    datefmt='%Y-%m-%d %H:%M:%S')

# === CONFIGURATION ===
DISCORD_WEBHOOK_URL = "https://discord.com/api/webhooks/1402083180785176637/hm5Efv4z3-OwCJpTJWQB78bvUixMIItPKiPdsUCjmmakr4MXOk_thRk5aBdn3uswIGVV"  # ← REPLACE
SECRET_KEY = "bigly#Redis#NodJSforWin@GOLangSucks"  # ← CHANGE THIS! Use a strong key
ATR_PERIOD = 7
RSI_PERIOD = 7
SUPER_TREND_FACTOR = 3
INTERVAL = "5m"

# 📝 Runtime Data
monitored_stocks = ["SPY", "QQQ"]  # Default list
alert_history = []

# 🔐 Thread Locks
import threading

stocks_lock = threading.Lock()
history_lock = threading.Lock()


# === Validate Stock Symbol ===
def is_valid_symbol(symbol):
    """Allow only 1–5 uppercase letters (e.g., AAPL, MSFT)"""
    return bool(re.fullmatch(r"^[A-Z]{1,5}$", symbol.strip()))


# === Is Market Open? (US: 9:30 AM - 4:00 PM EST) ===
def is_market_open():
    from pytz import timezone
    from datetime import time as dt_time
    est = timezone('US/Eastern')
    now = datetime.now(est)

    if now.weekday() >= 5:
        return False

    market_open_time = dt_time(9, 30)
    market_close_time = dt_time(16, 0)
    return market_open_time <= now.time() <= market_close_time


# === SuperTrend Calculation ===
def calculate_supertrend(df, factor=SUPER_TREND_FACTOR, period=ATR_PERIOD):
    data = df.copy()
    hl2 = (data['high'] + data['low']) / 2
    atr = ta.volatility.average_true_range(data['high'], data['low'], data['close'], window=period)
    data['upper_band'] = hl2 + (factor * atr)
    data['lower_band'] = hl2 - (factor * atr)
    data['supertrend_trend'] = 1
    for i in range(1, len(data)):
        prev_trend = data['supertrend_trend'].iloc[i - 1]
        curr_close = data['close'].iloc[i]
        if prev_trend == 1:
            if curr_close < data['lower_band'].iloc[i - 1]:
                data.loc[data.index[i], 'supertrend_trend'] = -1
            else:
                data.loc[data.index[i], 'lower_band'] = max(data['lower_band'].iloc[i], data['lower_band'].iloc[i - 1])
        else:
            if curr_close > data['upper_band'].iloc[i - 1]:
                data.loc[data.index[i], 'supertrend_trend'] = 1
            else:
                data.loc[data.index[i], 'upper_band'] = min(data['upper_band'].iloc[i], data['upper_band'].iloc[i - 1])

    data.ffill(inplace=True)
    return data


# === Send to Discord ===
def send_discord_alert(symbol, signal, price, rsi):
    if not DISCORD_WEBHOOK_URL:
        logging.warning("Discord webhook URL not set. Skipping alert.")
        return
    color = 0x00ff00 if signal == "BUY" else 0xff0000
    embed = {
        "title": f"{signal} Signal",
        "description": f"**{symbol}**",
        "color": color,
        "fields": [
            {"name": "Price", "value": f"${price:.2f}", "inline": True},
            {"name": f"RSI ({RSI_PERIOD})", "value": f"{rsi:.2f}", "inline": True},
            {"name": "Interval", "value": INTERVAL, "inline": True},
        ],
        "timestamp": datetime.utcnow().isoformat(),
        "footer": {"text": "SuperTrend ATR+RSI Bot"}
    }
    try:
        response = requests.post(DISCORD_WEBHOOK_URL, json={"embeds": [embed]})
        if response.status_code == 204:
            logging.info(f"✅ {signal} alert sent for {symbol} at ${price:.2f}")
        else:
            logging.error(f"❌ Discord API error: {response.status_code} - {response.text}")
    except Exception as e:
        logging.error(f"❌ Failed to send alert: {e}")


# === Send Test Message ===
def send_test_alert():
    if not DISCORD_WEBHOOK_URL:
        logging.error("❌ Test alert failed: Discord Webhook URL is not set.")
        return False
    embed = {
        "title": "✅ Test Alert",
        "description": "**Your bot is working perfectly!**",
        "color": 0x3498db,
        "fields": [
            {"name": "Status", "value": "All systems operational", "inline": False},
            {"name": "Time", "value": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "inline": False},
        ],
        "footer": {"text": "SuperTrend Signal Bot"}
    }
    try:
        response = requests.post(DISCORD_WEBHOOK_URL, json={"embeds": [embed]})
        return response.status_code == 204
    except Exception as e:
        logging.error(f"❌ Test alert failed: {e}")
        return False


# === Check One Stock ===
def check_stock(symbol):
    try:
        if not is_market_open():
            return

        ticker = yf.Ticker(symbol)
        data = ticker.history(period="2d", interval=INTERVAL)
        data.columns = data.columns.str.lower()

        if data.empty or len(data) < RSI_PERIOD + 10:
            logging.warning(f"Insufficient data for {symbol} on {INTERVAL} interval. Skipping.")
            return

        rsi_indicator = ta.momentum.RSIIndicator(data['close'], window=RSI_PERIOD)
        data['rsi'] = rsi_indicator.rsi()
        data = calculate_supertrend(data, factor=SUPER_TREND_FACTOR, period=ATR_PERIOD)

        data.dropna(inplace=True)
        if len(data) < 2:
            logging.warning(f"Not enough data for {symbol} after indicator calculation.")
            return

        latest = data.iloc[-1]
        previous = data.iloc[-2]
        current_trend = latest['supertrend_trend']
        prev_trend = previous['supertrend_trend']
        price = latest['close']
        rsi_val = latest['rsi']

        # ADDED THIS LINE TO PROVIDE A LIVE HEARTBEAT FOR EACH STOCK
        logging.info(f"{symbol}: Price=${price:.2f}, RSI={rsi_val:.2f}, Trend={int(current_trend)}")

        signal = None
        if current_trend == 1 and prev_trend == -1:
            signal = "BUY"
        elif current_trend == -1 and prev_trend == 1:
            signal = "SELL"

        if signal:
            # --- TREND FILTER LOGIC ---
            logging.info(f"Potential {signal} for {symbol}. Checking 4-hour trend filter...")
            long_term_data = ticker.history(period="1mo", interval="4h")
            long_term_data.columns = long_term_data.columns.str.lower()

            if len(long_term_data) < 50:
                logging.warning(f"Not enough 4h data for {symbol} to apply trend filter. Allowing signal.")
            else:
                long_term_data['ma50'] = long_term_data['close'].rolling(window=50).mean()
                last_long_term_price = long_term_data['close'].iloc[-1]
                last_ma_value = long_term_data['ma50'].iloc[-1]

                if signal == "BUY" and last_long_term_price < last_ma_value:
                    logging.info(f"BUY signal for {symbol} blocked by 4-hour trend filter (price is below MA).")
                    return

                if signal == "SELL" and last_long_term_price > last_ma_value:
                    logging.info(f"SELL signal for {symbol} blocked by 4-hour trend filter (price is above MA).")
                    return

            logging.info(f"✅ Signal for {symbol} passed trend filter. Sending alert.")
            # --- END OF TREND FILTER LOGIC ---

            alert = {
                "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                "symbol": symbol,
                "signal": signal,
                "price": f"${price:.2f}",
                "rsi": f"{rsi_val:.2f}",
                "interval": INTERVAL
            }
            with history_lock:
                alert_history.insert(0, alert)
                if len(alert_history) > 50:
                    alert_history.pop()
            send_discord_alert(symbol, signal, price, rsi_val)

    except Exception as e:
        logging.error(f"💥 Error processing {symbol}: {e}")


# === Scanner Loop ===
def run_scanner():
    logging.info("🚀 Scanner started...")
    while True:
        with stocks_lock:
            current_stocks = monitored_stocks.copy()

        if not current_stocks:
            logging.info("No stocks to monitor. Waiting...")
        elif not is_market_open():
            logging.info("Market is currently closed. Scanner is sleeping.")
        else:
            logging.info(f"Market is open. Checking {len(current_stocks)} stocks...")
            for symbol in current_stocks:
                check_stock(symbol)
            logging.info("Finished checking all stocks for this interval.")

        time.sleep(300)


# === FLASK WEB APP (HTML is unchanged) ===
app = Flask(__name__)
app.secret_key = SECRET_KEY

HTML_TEMPLATE = """
<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8" />
  <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
  <title>Stock Signal Dashboard</title>
  <style>
    body { font-family: Arial, sans-serif; margin: 20px; background: #f4f6f9; color: #333; }
    h1, h2 { color: #2c3e50; }
    .container { max-width: 1000px; margin: 0 auto; }
    .section { background: #fff; padding: 20px; border-radius: 8px; box-shadow: 0 2px 5px rgba(0,0,0,0.1); margin-bottom: 20px; }
    input[type="text"], input[type="password"] { width: calc(100% - 22px); padding: 10px; margin: 5px 0 10px 0; border: 1px solid #ddd; border-radius: 4px; }
    button { padding: 10px 15px; border: none; border-radius: 4px; cursor: pointer; color: white; font-weight: bold; }
    .btn-add { background: #27ae60; } .btn-add:hover { background: #2ecc71; }
    .btn-remove { background: #e74c3c; padding: 5px 10px; font-size: 0.9em; } .btn-remove:hover { background: #c0392b; }
    .btn-test { background: #9b59b6; } .btn-test:hover { background: #8e44ad; }
    table { width: 100%; border-collapse: collapse; margin-top: 15px; }
    th, td { padding: 12px; text-align: left; border-bottom: 1px solid #ddd; }
    th { background-color: #34495e; color: white; }
    tr:nth-child(even) { background-color: #f9f9f9; }
    tr:hover { background-color: #f1f1f1; }
    .buy { color: #27ae60; font-weight: bold; }
    .sell { color: #c0392b; font-weight: bold; }
    .empty { color: #7f8c8d; font-style: italic; text-align: center; padding: 20px; }
    .flash { padding: 15px; margin-bottom: 20px; border-radius: 4px; border: 1px solid transparent; }
    .flash.success { background: #d4edda; color: #155724; border-color: #c3e6cb; }
    .flash.error { background: #f8d7da; color: #721c24; border-color: #f5c6cb; }
    .flash.info { background: #d1ecf1; color: #0c5460; border-color: #bee5eb; }
    .footer { margin-top: 30px; color: #7f8c8d; font-size: 0.9em; text-align: center; }
    .form-inline { display: flex; align-items: center; gap: 10px; }
    .form-inline input[type="password"] { width: 120px; margin: 0; }
  </style>
</head>
<body>
  <div class="container">
    <h1>📈 SuperTrend ATR+RSI Dashboard</h1>
    {% with messages = get_flashed_messages(with_categories=true) %}
      {% if messages %}
        {% for category, msg in messages %}
          <div class="flash {{ category }}">{{ msg }}</div>
        {% endfor %}
      {% endif %}
    {% endwith %}
    <div class="section">
      <h2>➕ Add Stock to Watchlist</h2>
      <form method="POST" action="/add_stock">
        <input type="text" name="symbol" placeholder="Enter symbol (e.g., AAPL)" required autocomplete="off" />
        <input type="password" name="key" placeholder="Secret Key" required />
        <button type="submit" class="btn-add">Add Stock</button>
      </form>
    </div>
    <div class="section">
      <h2>📋 Monitored Stocks</h2>
      {% if stocks %}
        <table>
          <thead><tr><th>Symbol</th><th>Action</th></tr></thead>
          <tbody>
            {% for symbol in stocks %}
            <tr>
              <td><strong>{{ symbol }}</strong></td>
              <td>
                <form method="POST" action="/remove_stock" class="form-inline">
                  <input type="hidden" name="symbol" value="{{ symbol }}" />
                  <input type="password" name="key" placeholder="Key" required />
                  <button type="submit" class="btn-remove">Remove</button>
                </form>
              </td>
            </tr>
            {% endfor %}
          </tbody>
        </table>
      {% else %}
        <p class="empty">🚫 No stocks are being monitored. Add one above to get started!</p>
      {% endif %}
    </div>
    <div class="section">
      <h2>🧪 Test Discord Webhook</h2>
      <form method="POST" action="/test_discord">
        <input type="password" name="key" placeholder="Secret Key" required />
        <button type="submit" class="btn-test">📤 Send Test Message</button>
      </form>
    </div>
    <div class="section">
      <h2>🔔 Recent Alert History</h2>
      {% if alerts %}
        <table>
          <thead><tr><th>Time</th><th>Symbol</th><th>Signal</th><th>Price</th><th>RSI</th><th>Interval</th></tr></thead>
          <tbody>
            {% for alert in alerts %}
            <tr>
              <td>{{ alert.time }}</td>
              <td><strong>{{ alert.symbol }}</strong></td>
              <td class="{{ alert.signal.lower() }}">{{ alert.signal }}</td>
              <td>{{ alert.price }}</td>
              <td>{{ alert.rsi }}</td>
              <td>{{ alert.interval }}</td>
            </tr>
            {% endfor %}
          </tbody>
        </table>
      {% else %}
        <p class="empty">No alerts generated yet. Waiting for a signal...</p>
      {% endif %}
    </div>
    <div class="footer">
      Bot status as of: {{ now }} | Scan Interval: {{ interval }}
    </div>
  </div>
</body>
</html>
"""


@app.route('/', methods=['GET'])
def home():
    with stocks_lock:
        stocks = sorted(monitored_stocks.copy())
    with history_lock:
        alerts = alert_history.copy()
    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    return render_template_string(HTML_TEMPLATE,
                                  stocks=stocks,
                                  alerts=alerts,
                                  now=now,
                                  interval=INTERVAL)


@app.route('/add_stock', methods=['POST'])
def add_stock():
    symbol = request.form.get('symbol', '').strip().upper()
    key = request.form.get('key', '').strip()
    if key != SECRET_KEY:
        flash("🔒 Access denied: Invalid secret key.", "error")
    elif not symbol:
        flash("⚠️ Please enter a stock symbol.", "error")
    elif not is_valid_symbol(symbol):
        flash(f"❌ Invalid symbol: '{symbol}'. Use 1–5 uppercase letters.", "error")
    else:
        with stocks_lock:
            if symbol in monitored_stocks:
                flash(f"🟡 '{symbol}' is already being monitored.", "info")
            else:
                monitored_stocks.append(symbol)
                flash(f"✅ Successfully added '{symbol}' to the watchlist!", "success")
                logging.info(f"➕ User added stock: {symbol}")
    return home()


@app.route('/remove_stock', methods=['POST'])
def remove_stock():
    symbol = request.form.get('symbol', '').strip().upper()
    key = request.form.get('key', '').strip()
    if key != SECRET_KEY:
        flash("🔒 Access denied: Invalid secret key.", "error")
    else:
        with stocks_lock:
            if symbol not in monitored_stocks:
                flash(f"🔍 '{symbol}' is not on the list.", "info")
            else:
                monitored_stocks.remove(symbol)
                flash(f"🗑️ Removed '{symbol}' from monitoring.", "success")
                logging.info(f"➖ User removed stock: {symbol}")
    return home()


@app.route('/test_discord', methods=['POST'])
def test_discord():
    key = request.form.get('key', '').strip()
    if key != SECRET_KEY:
        flash("🔒 Access denied: Invalid secret key.", "error")
    else:
        success = send_test_alert()
        if success:
            flash("✅ Test message sent to Discord! Check your server.", "success")
        else:
            flash("❌ Failed to send test message. Check your Discord Webhook URL.", "error")
    return home()


def start():
    scanner_thread = Thread(target=run_scanner, daemon=True)
    scanner_thread.start()
    logging.info("🌐 Starting Flask web server...")
    app.run(host='0.0.0.0', port=8080)


if __name__ == '__main__':
    start()