UPDATED ON DECEMBER 28, 2025

//@version=6
indicator("WildMA(2) vs EMA(2) State + Trade Labels", overlay=true, max_labels_count=500)

// ─────────────────────────────────────────────
// Settings
// ─────────────────────────────────────────────
src      = input.source(close, "Source")
wildLen  = input.int(2, "WildMA Length", minval=1)
emaLen   = input.int(2, "EMA Length", minval=1)

minDiff  = input.float(0.1, "Min MA Difference for Red/Green Dots", step=0.01, minval=0.0)

confirmed = barstate.isconfirmed

// ─────────────────────────────────────────────
// MAs
// ─────────────────────────────────────────────
wild = ta.rma(src, wildLen)
ema  = ta.ema(src, emaLen)

// Slope colors (for line color + new trade conditions)
wildUp = wild > wild[1]
wildDn = wild < wild[1]
emaUp  = ema  > ema[1]
emaDn  = ema  < ema[1]

maColor = wildUp ? color.green : wildDn ? color.red : color.yellow

// ─────────────────────────────────────────────
// Plots
// ─────────────────────────────────────────────
plot(wild, "WildMA", color=maColor, linewidth=2)
plot(ema,  "EMA",    color=maColor, linewidth=2)

// ─────────────────────────────────────────────
// Cross + State Dots (bottom row) with minDiff rule
// ─────────────────────────────────────────────
crossUp = ta.crossover(ema, wild)
crossDn = ta.crossunder(ema, wild)
isCross = crossUp or crossDn

emaAbove = ema > wild
diffOK   = math.abs(ema - wild) >= minDiff


plotshape(not isCross and diffOK and emaAbove,       title="State Dot Up",   style=shape.circle, location=location.bottom, color=color.green, size=size.tiny)
plotshape(not isCross and diffOK and not emaAbove,   title="State Dot Down", style=shape.circle, location=location.bottom, color=color.red,   size=size.tiny)

// If diff is too small, show white X instead of red/green
plotshape(not isCross and not diffOK, title="Neutral X (Diff < min)", style=shape.xcross, location=location.bottom, color=color.white, size=size.tiny)

// ─────────────────────────────────────────────
// Trade State Machine (no overlap; exits first; no same-bar flip)
// ─────────────────────────────────────────────
// state: 0 = flat, 1 = long, -1 = short
var int state = 0
var int lastExitBar = na

// New “both green/red” conditions
bothGreen = wildUp and emaUp
bothRed   = wildDn and emaDn

// Exits (priority):
// - original cross exits
// - PLUS slope exits: long exits if either turns red, short exits if either turns green
exitLong  = confirmed and state == 1  and (crossDn or wildDn or emaDn)
exitShort = confirmed and state == -1 and (crossUp or wildUp or emaUp)

if exitLong
   state := 0
   lastExitBar := bar_index

if exitShort
   state := 0
   lastExitBar := bar_index
// Block same-bar flip after an exit
canEnter = confirmed and state == 0 and (na(lastExitBar) or bar_index > lastExitBar)

// Entries when flat:
// - original cross entries
// - PLUS slope entries: long if both green, short if both red
enterLong  = canEnter and (crossUp or bothGreen)
enterShort = canEnter and (crossDn or bothRed)

if enterLong
   state := 1

if enterShort
   state := -1

// ─────────────────────────────────────────────
// Trade labels
// ─────────────────────────────────────────────
plotshape(enterLong,  title="BUY LONG",  text="BUY",  style=shape.labelup,   location=location.belowbar, color=color.green,  textcolor=color.white, size=size.small)
plotshape(exitLong,   title="EXIT LONG", text="EXIT", style=shape.labeldown, location=location.belowbar, color=color.yellow, textcolor=color.black, size=size.small)

plotshape(enterShort, title="SELL SHORT", text="SELL", style=shape.labeldown, location=location.abovebar, color=color.red,    textcolor=color.white, size=size.small)
plotshape(exitShort,  title="EXIT SHORT", text="EXIT", style=shape.labelup,   location=location.abovebar, color=color.yellow, textcolor=color.black, size=size.small)