#!/usr/bin/env python3
"""BALROG's drops live in Java's createCustomQuestDrops() (gated behind
WANT_CUSTOM_QUESTS), which the parser doesn't read. Add manually:
  TEDDY_HEAD (1363) guaranteed
  ASHES (181) guaranteed   <- via ashesNpcs set
"""
import sys, os, shutil
sys.path.insert(0, '/Users/tomassimkus/VS/openrsc-develop')
os.chdir('/Users/tomassimkus/VS/openrsc-develop')
from update_drops import parse_update_sql, write_update_sql

shutil.copy("pk_npcdrops.sql", "pk_npcdrops_update.sql")

header, rows, footer = parse_update_sql()
BALROG = 809

# Sanity: ensure no existing rows
existing = [r for r in rows if r[0] == BALROG]
print(f"Existing BALROG rows: {len(existing)}")

# Add ASHES + TEDDY_HEAD as guaranteed drops
rows.append((BALROG, '1', 181,  0, 0))   # ASHES (bone)
rows.append((BALROG, '1', 1363, 0, 0))   # TEDDY_HEAD (quest drop)

rows.sort(key=lambda r: (r[0], 0 if r[3] == 0 else 1, r[3], r[2]))
write_update_sql(header, rows, footer)

shutil.copy("pk_npcdrops_update.sql", "pk_npcdrops.sql")
shutil.copy("pk_npcdrops_update.sql", "pk_npcdrops_p2p_update.sql")
print("Added BALROG (id=809) drops:")
print("  ASHES (id 181) - guaranteed")
print("  TEDDY_HEAD (id 1363) - guaranteed")