diff --git a/webserver.py b/webserver.py index 99cb358..cade31a 100644 --- a/webserver.py +++ b/webserver.py @@ -22,6 +22,9 @@ from src.rate_limiter import RateLimiter, CommandCategory class PetBotRequestHandler(BaseHTTPRequestHandler): """HTTP request handler for PetBot web server""" + # Class-level admin sessions storage + admin_sessions = {} + @property def database(self): """Get database instance from server""" @@ -259,10 +262,10 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): animation: fadeIn 0.3s ease; } - @keyframes fadeIn { - from { opacity: 0; transform: translateY(-10px); } - to { opacity: 1; transform: translateY(0); } - } + @keyframes fadeIn {{ + from {{ opacity: 0; transform: translateY(-10px); }} + to {{ opacity: 1; transform: translateY(0); }} + }} .dropdown-item { color: var(--text-primary); @@ -542,6 +545,65 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): margin-bottom: 40px; } + /* IV Display Styles */ + .iv-section { + margin-top: 15px; + padding: 12px; + background: var(--bg-primary); + border-radius: 8px; + border: 1px solid var(--border-color); + } + + .iv-title { + font-size: 0.9em; + font-weight: bold; + color: var(--text-accent); + margin-bottom: 8px; + text-align: center; + } + + .iv-grid { + display: grid; + grid-template-columns: repeat(2, 1fr); + gap: 6px; + margin-bottom: 8px; + } + + .iv-stat { + display: flex; + justify-content: space-between; + align-items: center; + font-size: 0.8em; + padding: 2px 0; + } + + .iv-value { + font-weight: bold; + padding: 2px 6px; + border-radius: 4px; + min-width: 28px; + text-align: center; + } + + .iv-perfect { background: #4caf50; color: white; } + .iv-excellent { background: #2196f3; color: white; } + .iv-good { background: #ff9800; color: white; } + .iv-fair { background: #ff5722; color: white; } + .iv-poor { background: #607d8b; color: white; } + + .iv-total { + text-align: center; + font-size: 0.85em; + padding-top: 8px; + border-top: 1px solid var(--border-color); + color: var(--text-secondary); + } + + .iv-total-value { + font-weight: bold; + color: var(--text-accent); + } + /* Responsive design */ @media (max-width: 768px) { .main-container { @@ -559,6 +621,10 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): .grid-2, .grid-3, .grid-4 { grid-template-columns: 1fr; } + + .iv-grid { + grid-template-columns: 1fr; + } } """ @@ -585,7 +651,10 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ("petdex?sort=all", "๐Ÿ“‹ Show All"), ("petdex#search", "๐Ÿ” Search") ]), - ("help", "๐Ÿ“– Help", []) + ("help", "๐Ÿ“– Help", [ + ("help", "๐Ÿ“‹ Commands"), + ("faq", "โ“ FAQ") + ]) ] nav_links = "" @@ -642,6 +711,8 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): def do_GET(self): """Handle GET requests with rate limiting""" + print(f"GET request: {self.path}") + # Check rate limit first allowed, rate_limit_message = self.check_rate_limit() if not allowed: @@ -650,14 +721,22 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): parsed_path = urlparse(self.path) path = parsed_path.path + print(f"Parsed path: {path}") # Route handling if path == '/': self.serve_index() elif path == '/help': self.serve_help() + elif path == '/faq': + self.serve_faq() elif path == '/players': self.serve_players() + elif path.startswith('/player/') and path.endswith('/pets'): + # Handle /player/{nickname}/pets - must come before general /player/ route + nickname = path[8:-5] # Remove '/player/' prefix and '/pets' suffix + print(f"DEBUG: Route matched! Parsed nickname from path '{path}' as '{nickname}'") + self.serve_player_pets(nickname) elif path.startswith('/player/'): nickname = path[8:] # Remove '/player/' prefix self.serve_player_profile(nickname) @@ -670,7 +749,23 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): elif path.startswith('/teambuilder/'): nickname = path[13:] # Remove '/teambuilder/' prefix self.serve_teambuilder(nickname) + elif path.startswith('/testteambuilder/'): + nickname = path[17:] # Remove '/testteambuilder/' prefix + self.serve_test_teambuilder(nickname) + elif path == '/admin': + self.serve_admin_login() + elif path == '/admin/dashboard': + self.serve_admin_dashboard() + elif path == '/admin/auth': + self.handle_admin_auth() + elif path == '/admin/verify': + self.handle_admin_verify() + elif path.startswith('/admin/api/'): + print(f"Admin API path detected in GET: {path}") + print(f"Extracted endpoint: {path[11:]}") + self.handle_admin_api(path[11:]) # Remove '/admin/api/' prefix else: + print(f"No route found for path: {path}") self.send_error(404, "Page not found") def do_POST(self): @@ -678,7 +773,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): # Check rate limit first (POST requests have stricter limits) allowed, rate_limit_message = self.check_rate_limit() if not allowed: - self.send_json_response({"success": False, "error": rate_limit_message}, 429) + self.send_json_response({"success": False, "error": "Rate limit exceeded"}, 429) return parsed_path = urlparse(self.path) @@ -690,6 +785,20 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): elif path.startswith('/teambuilder/') and path.endswith('/verify'): nickname = path[13:-7] # Remove '/teambuilder/' prefix and '/verify' suffix self.handle_team_verify(nickname) + elif path.startswith('/testteambuilder/') and path.endswith('/save'): + nickname = path[17:-5] # Remove '/testteambuilder/' prefix and '/save' suffix + self.handle_test_team_save(nickname) + elif path.startswith('/testteambuilder/') and path.endswith('/verify'): + nickname = path[17:-7] # Remove '/testteambuilder/' prefix and '/verify' suffix + self.handle_test_team_verify(nickname) + elif path.startswith('/player/') and '/pets/rename' in path: + # Handle pet rename request: /player/{nickname}/pets/rename + nickname = path.split('/')[2] + self.handle_pet_rename_request(nickname) + elif path.startswith('/player/') and '/pets/verify' in path: + # Handle pet rename PIN verification: /player/{nickname}/pets/verify + nickname = path.split('/')[2] + self.handle_pet_rename_verify(nickname) elif path.startswith('/teambuilder/') and '/config/save/' in path: # Handle team configuration save: /teambuilder/{nickname}/config/save/{slot} parts = path.split('/') @@ -717,7 +826,25 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): self.handle_team_config_rename(nickname, slot) else: self.send_error(400, "Invalid configuration rename path") + elif path.startswith('/teambuilder/') and '/config/apply/' in path: + # Handle team configuration apply: /teambuilder/{nickname}/config/apply/{slot} + parts = path.split('/') + if len(parts) >= 6: + nickname = parts[2] + slot = parts[5] + self.handle_team_config_apply(nickname, slot) + else: + self.send_error(400, "Invalid configuration apply path") + elif path == '/admin/auth': + self.handle_admin_auth() + elif path == '/admin/verify': + self.handle_admin_verify() + elif path.startswith('/admin/api/'): + print(f"Admin API path detected: {path}") + print(f"Extracted endpoint: {path[11:]}") + self.handle_admin_api(path[11:]) # Remove '/admin/api/' prefix else: + print(f"No route found for path: {path}") self.send_error(404, "Page not found") def serve_index(self): @@ -1313,6 +1440,63 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): except Exception as e: self.serve_error_page("Help", f"Error loading help file: {str(e)}") + def serve_faq(self): + """Serve the FAQ page using unified template""" + try: + with open('faq.html', 'r', encoding='utf-8') as f: + faq_content = f.read() + + import re + + # Extract CSS from faq.html + css_match = re.search(r']*>(.*?)', faq_content, re.DOTALL) + faq_css = css_match.group(1) if css_match else "" + + # Extract body content (everything between tags) + body_match = re.search(r']*>(.*?)', faq_content, re.DOTALL) + if body_match: + body_content = body_match.group(1) + # Remove the back link since we'll have the navigation bar + body_content = re.sub(r'.*?', '', body_content, flags=re.DOTALL) + else: + # Fallback: use original content if we can't parse it + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(faq_content.encode()) + return + + # Create template with merged CSS + html_content = f""" + + + + + PetBot - FAQ + + + + {self.get_navigation_bar("faq")} +
+ {body_content} +
+ +""" + + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(html_content.encode()) + except FileNotFoundError: + self.serve_error_page("FAQ", "FAQ file not found") + except Exception as e: + self.serve_error_page("FAQ", f"Error loading FAQ file: {str(e)}") + def serve_players(self): """Serve the players page with real data""" # Get database instance from the server class @@ -3265,6 +3449,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): # Get player info import aiosqlite async with aiosqlite.connect(database.db_path) as db: + db.row_factory = aiosqlite.Row # Get player basic info cursor = await db.execute(""" SELECT p.*, l.name as location_name, l.description as location_desc @@ -3276,19 +3461,8 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): if not player: return None - # Convert to dict manually - player_dict = { - 'id': player[0], - 'nickname': player[1], - 'created_at': player[2], - 'last_active': player[3], - 'level': player[4], - 'experience': player[5], - 'money': player[6], - 'current_location_id': player[7], - 'location_name': player[8], - 'location_desc': player[9] - } + # Convert Row to dict + player_dict = dict(player) # Get player pets cursor = await db.execute(""" @@ -3299,18 +3473,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ORDER BY p.is_active DESC, p.level DESC, p.id ASC """, (player_dict['id'],)) pets_rows = await cursor.fetchall() - pets = [] - for row in pets_rows: - pet_dict = { - 'id': row[0], 'player_id': row[1], 'species_id': row[2], - 'nickname': row[3], 'level': row[4], 'experience': row[5], - 'hp': row[6], 'max_hp': row[7], 'attack': row[8], - 'defense': row[9], 'speed': row[10], 'happiness': row[11], - 'caught_at': row[12], 'is_active': bool(row[13]), # Convert to proper boolean - 'team_order': row[14], 'species_name': row[15], 'type1': row[16], 'type2': row[17], - 'emoji': row[18] if row[18] else '๐Ÿพ' # Add emoji support - } - pets.append(pet_dict) + pets = [dict(row) for row in pets_rows] # Get player achievements cursor = await db.execute(""" @@ -3321,13 +3484,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ORDER BY pa.completed_at DESC """, (player_dict['id'],)) achievements_rows = await cursor.fetchall() - achievements = [] - for row in achievements_rows: - achievement_dict = { - 'id': row[0], 'player_id': row[1], 'achievement_id': row[2], - 'completed_at': row[3], 'achievement_name': row[4], 'achievement_desc': row[5] - } - achievements.append(achievement_dict) + achievements = [dict(row) for row in achievements_rows] # Get player inventory cursor = await db.execute(""" @@ -3338,13 +3495,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ORDER BY i.rarity DESC, i.name ASC """, (player_dict['id'],)) inventory_rows = await cursor.fetchall() - inventory = [] - for row in inventory_rows: - item_dict = { - 'name': row[0], 'description': row[1], 'category': row[2], - 'rarity': row[3], 'quantity': row[4] - } - inventory.append(item_dict) + inventory = [dict(row) for row in inventory_rows] # Get player gym badges cursor = await db.execute(""" @@ -3357,14 +3508,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ORDER BY pgb.first_victory_date ASC """, (player_dict['id'],)) gym_badges_rows = await cursor.fetchall() - gym_badges = [] - for row in gym_badges_rows: - badge_dict = { - 'gym_name': row[0], 'badge_name': row[1], 'badge_icon': row[2], - 'location_name': row[3], 'victories': row[4], - 'first_victory_date': row[5], 'highest_difficulty': row[6] - } - gym_badges.append(badge_dict) + gym_badges = [dict(row) for row in gym_badges_rows] # Get player encounters using database method encounters = [] @@ -3412,6 +3556,601 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): print(f"Database error fetching player {nickname}: {e}") return None + def serve_player_pets(self, nickname): + """Serve pet management page for a player""" + try: + print(f"DEBUG: serve_player_pets called with nickname: '{nickname}'") + # Get player data using database method directly + player = asyncio.run(self.database.get_player(nickname)) + print(f"DEBUG: Player result: {player}") + if not player: + print(f"DEBUG: Player not found for: '{nickname}'") + self.serve_player_not_found(nickname) + return + + # Get player pets for management + player_pets = asyncio.run(self.database.get_player_pets_for_rename(player['id'])) + + # Render the pets management page + self.serve_pets_management_interface(nickname, player_pets) + + except Exception as e: + print(f"Error serving player pets page: {e}") + self.serve_player_error(nickname, str(e)) + + def get_iv_grade(self, iv_value): + """Get color grade for IV value""" + if iv_value >= 27: # 27-31 (Perfect) + return "perfect" + elif iv_value >= 21: # 21-26 (Excellent) + return "excellent" + elif iv_value >= 15: # 15-20 (Good) + return "good" + elif iv_value >= 10: # 10-14 (Fair) + return "fair" + else: # 0-9 (Poor) + return "poor" + + def serve_pets_management_interface(self, nickname, pets): + """Serve the pet management interface""" + if not pets: + self.serve_no_pets_error(nickname) + return + + # Generate pet cards + pet_cards = [] + for pet in pets: + status_badge = "" + if pet.get('is_active'): + team_order = pet.get('team_order', 0) + if team_order > 0: + status_badge = f'Team #{team_order}' + else: + status_badge = 'Active' + else: + status_badge = 'Storage' + + fainted_badge = "" + if pet.get('fainted_at'): + fainted_badge = '๐Ÿ’€ Fainted' + + current_name = pet.get('nickname') or pet.get('species_name') + pet_id = pet.get('id') + + pet_card = f""" +
+
+
+
{pet.get('emoji', '๐Ÿพ')} {current_name}
+
Level {pet.get('level', 1)} {pet.get('species_name')}
+
+
+ {status_badge} + {fainted_badge} +
+
+ +
+
+ HP: + {pet.get('hp', 0)}/{pet.get('max_hp', 0)} +
+
+ ATK: + {pet.get('attack', 0)} +
+
+ DEF: + {pet.get('defense', 0)} +
+
+ SPD: + {pet.get('speed', 0)} +
+
+ +
+
+ Individual Values (IVs) + โ„น๏ธ +
+
+
+ HP: + {pet.get('iv_hp', 15)} +
+
+ ATK: + {pet.get('iv_attack', 15)} +
+
+ DEF: + {pet.get('iv_defense', 15)} +
+
+ SPD: + {pet.get('iv_speed', 15)} +
+
+
+ Total IV: + {pet.get('iv_hp', 15) + pet.get('iv_attack', 15) + pet.get('iv_defense', 15) + pet.get('iv_speed', 15)}/124 +
+
+ +
+ +
+ + +
+ """ + pet_cards.append(pet_card) + + pets_html = "".join(pet_cards) + + content = f""" +
+

๐Ÿพ My Pets - {nickname}

+

Manage your pet collection and customize their names

+
+ +
+ {pets_html} +
+ +
+ โ† Back to Profile +
+

๐Ÿ’ก Tips:

+
    +
  • Click "Rename" to change a pet's nickname
    • +
  • You'll receive a PIN via IRC for security
    • +
  • PIN expires in 15 seconds
    • +
  • Names must be unique among your pets
    • +
+
+
+ + + + + + + + """ + + page_html = self.get_page_template("My Pets - " + nickname, content, "pets") + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(page_html.encode()) + + def serve_no_pets_error(self, nickname): + """Serve error page when player has no pets""" + content = f""" +
+

๐Ÿพ No Pets Found

+
+ +
+

You don't have any pets yet!

+

Start your journey by using !start in #petz to get your first pet.

+

Then explore locations and catch more pets with !explore and !catch.

+
+ +
+ โ† Back to Profile +
+ """ + + page_html = self.get_page_template("No Pets - " + nickname, content, "pets") + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(page_html.encode()) + def serve_player_not_found(self, nickname): """Serve player not found page using unified template""" content = f""" @@ -3690,6 +4429,12 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): ๐Ÿ”ง Team Builder + + ๐Ÿงช Test Team Builder + + + ๐Ÿพ My Pets + @@ -4254,6 +4999,24 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): active_pets = [pet for pet in pets if pet['is_active']] inactive_pets = [pet for pet in pets if not pet['is_active']] + # Get team configurations for team selection interface + import asyncio + from src.database import Database + database = Database() + + # Get event loop + try: + loop = asyncio.get_running_loop() + except RuntimeError: + loop = asyncio.new_event_loop() + asyncio.set_event_loop(loop) + + # Get player and team configurations + player = loop.run_until_complete(database.get_player(nickname)) + team_configs = [] + if player: + team_configs = loop.run_until_complete(database.get_player_team_configurations(player['id'])) + # Debug logging print(f"Team Builder Debug for {nickname}:") print(f"Total pets: {len(pets)}") @@ -4333,1555 +5096,7 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): storage_cards = ''.join(make_pet_card(pet, False) for pet in inactive_pets) - html = f""" - - - - - Team Builder - {nickname} - - - - โ† Back to {nickname}'s Profile - -
-

๐Ÿพ Team Builder

-

Drag pets between Active and Storage to build your perfect team

-

{nickname} | Active: {len(active_pets)} pets | Storage: {len(inactive_pets)} pets

-
- -
-
๐Ÿ’พ Team Configurations
-

- Save up to 3 different team setups for quick switching between strategies -

- -
- - - - -
- -
- Editing current team (not saved to any configuration) -
- -
- - - -
-
- -
-
-
โญ Active Team
-
-
-
1
-
{team_slots[0]}
-
-
-
2
-
{team_slots[1]}
-
-
-
3
-
{team_slots[2]}
-
-
-
4
-
{team_slots[3]}
-
-
-
5
-
{team_slots[4]}
-
-
-
6
-
{team_slots[5]}
-
-
-
- -
-
๐Ÿ“ฆ Storage
-
- {storage_cards} -
-
- Drop pets here to store them -
-
-
- -
- - โ† Back to Profile -
- Changes are saved securely with PIN verification via IRC -
-
- - -
-

๐Ÿ” PIN Verification Required

-

A 6-digit PIN has been sent to you via IRC private message.

-

Enter the PIN below to confirm your team changes:

- - -
-
- - - -""" + # Old template removed - using new unified template system below # Generate storage pets HTML first storage_pets_html = "" @@ -5915,6 +5130,138 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): margin: 0; } + /* Storage Controls */ + .storage-controls { + display: flex; + gap: 15px; + margin-bottom: 20px; + align-items: center; + flex-wrap: wrap; + } + + .search-container input, .sort-container select { + background: var(--bg-tertiary); + border: 2px solid var(--border-color); + color: var(--text-primary); + padding: 8px 12px; + border-radius: 8px; + font-size: 0.9em; + } + + .search-container input { + min-width: 250px; + } + + .sort-container select { + min-width: 180px; + } + + .search-container input:focus, .sort-container select:focus { + outline: none; + border-color: var(--text-accent); + } + + /* Team Selection Interface */ + .team-selector-section { + background: var(--bg-secondary); + border-radius: 15px; + padding: 25px; + margin: 30px 0; + border: 1px solid var(--border-color); + } + + .team-selector-section h2 { + color: var(--text-accent); + margin-bottom: 20px; + text-align: center; + } + + .team-cards { + display: grid; + grid-template-columns: repeat(3, 1fr); + gap: 20px; + } + + .team-card { + background: var(--bg-tertiary); + border-radius: 12px; + padding: 20px; + border: 2px solid var(--border-color); + transition: all 0.3s ease; + cursor: pointer; + } + + .team-card:hover { + border-color: var(--text-accent); + transform: translateY(-2px); + } + + .team-card.selected { + border-color: var(--text-accent); + background: var(--bg-primary); + } + + .team-card-header { + display: flex; + justify-content: space-between; + align-items: center; + margin-bottom: 15px; + } + + .team-card h3 { + margin: 0; + color: var(--text-accent); + } + + .team-status { + color: var(--text-secondary); + font-size: 0.9em; + } + + .team-preview { + display: grid; + grid-template-columns: repeat(3, 1fr); + gap: 5px; + margin: 15px 0; + min-height: 60px; + } + + .mini-pet { + background: var(--bg-primary); + border-radius: 6px; + padding: 8px; + text-align: center; + font-size: 0.8em; + border: 1px solid var(--border-color); + } + + .mini-pet.empty { + color: var(--text-secondary); + font-style: italic; + } + + .edit-team-btn { + width: 100%; + background: var(--primary-color); + color: white; + border: none; + border-radius: 8px; + padding: 12px; + cursor: pointer; + font-size: 1em; + transition: all 0.3s ease; + } + + .edit-team-btn:hover { + background: var(--secondary-color); + transform: translateY(-1px); + } + + .edit-team-btn.active { + background: var(--text-accent); + color: var(--bg-primary); + } + .team-sections { margin-top: 30px; } @@ -6295,10 +5642,26 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): } .config-quick-actions { - display: flex; - gap: 10px; - margin-top: 15px; - flex-wrap: wrap; + display: grid; + grid-template-columns: 1fr 1fr; + gap: 30px; + margin-top: 20px; + } + + .config-slot-actions { + text-align: center; + } + + .config-slot-actions h4 { + color: var(--text-accent); + margin: 0 0 15px 0; + font-size: 1.1em; + } + + .config-slot-actions button { + display: block; + width: 100%; + margin: 8px 0; } .config-action-btn { @@ -6323,12 +5686,95 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): cursor: not-allowed; opacity: 0.5; } + + .config-apply-btn { + background: var(--secondary-color); + color: white; + border: none; + border-radius: 8px; + padding: 8px 16px; + cursor: pointer; + font-size: 0.9em; + transition: all 0.3s ease; + min-width: 120px; + } + + .config-apply-btn:hover:not(:disabled) { + background: #4CAF50; + transform: translateY(-1px); + } + + .config-apply-btn:disabled { + background: var(--text-secondary); + cursor: not-allowed; + opacity: 0.5; + } + + .save-config-buttons { + display: grid; + grid-template-columns: repeat(3, 1fr); + gap: 10px; + margin: 15px 0; + } + + .config-save-slot-btn { + background: #FF9800; + color: white; + border: none; + border-radius: 8px; + padding: 10px 16px; + cursor: pointer; + font-size: 0.9em; + transition: all 0.3s ease; + } + + .config-save-slot-btn:hover { + background: #F57C00; + transform: translateY(-1px); + } + + .saved-configs-list { + background: var(--bg-tertiary); + border-radius: 8px; + padding: 15px; + margin-top: 10px; + } + + .saved-config-item { + margin: 8px 0; + display: flex; + align-items: center; + } + + .config-slot-label { + font-weight: bold; + color: var(--text-accent); + margin-right: 10px; + min-width: 60px; + } + + .config-name { + color: var(--text-primary); + } + + .config-name.empty { + color: var(--text-secondary); + font-style: italic; + }

๐Ÿพ Team Builder

-

Drag pets between Active Team and Storage. Double-click as backup.

+

Choose a team to edit, then drag pets between Active Team and Storage.

+
+ + +
+

Select Team to Edit

+
+ +
@@ -6376,46 +5822,26 @@ class PetBotRequestHandler(BaseHTTPRequestHandler):

๐Ÿ“ฆ Storage

+
+
+ +
+
+ +
+
""" + storage_pets_html + active_pets_html + """
-
-

๐Ÿ’พ Team Configurations

-

- Save up to 3 different team setups for quick switching between strategies -

- -
- - - - -
- -
- Editing current team (not saved to any configuration) -
- -
- - - -
-
+
@@ -6448,14 +5874,344 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): let draggedElement = null; // Initialize when DOM is ready + // Global variables for team management + let currentEditingTeam = 1; // Default to team 1 + document.addEventListener('DOMContentLoaded', function() { console.log('Team Builder: DOM loaded, initializing...'); initializeTeamBuilder(); }); + function selectTeam(teamSlot) { + console.log('Selecting team slot:', teamSlot); + + // Update UI to show which team is selected + document.querySelectorAll('.team-card').forEach(card => { + card.classList.remove('selected'); + const btn = card.querySelector('.edit-team-btn'); + btn.classList.remove('active'); + btn.textContent = btn.textContent.replace('๐ŸŸข Currently Editing', '๐Ÿ“ Edit ' + card.querySelector('h3').textContent); + }); + + // Mark selected team + const selectedCard = document.querySelector(`[data-slot="${teamSlot}"]`); + if (selectedCard) { + selectedCard.classList.add('selected'); + const btn = selectedCard.querySelector('.edit-team-btn'); + btn.classList.add('active'); + btn.textContent = '๐ŸŸข Currently Editing'; + } + + // Set current editing team + currentEditingTeam = teamSlot; + + // Load team data for this slot (to be implemented) + loadTeamConfiguration(teamSlot); + } + + function loadTeamConfiguration(teamSlot) { + console.log('Loading team configuration for slot:', teamSlot); + + // Update dynamic headers and button text + updateDynamicElements(teamSlot); + + // Clear current team slots + for (let i = 1; i <= 6; i++) { + const slot = document.getElementById(`slot-${i}`); + if (slot) { + const slotContent = slot.querySelector('.slot-content'); + slotContent.innerHTML = '
Drop pet here
'; + } + } + + // Move all pets back to storage + const storageContainer = document.getElementById('storage-container'); + const allPetCards = document.querySelectorAll('.pet-card'); + allPetCards.forEach(card => { + if (storageContainer && !storageContainer.contains(card)) { + storageContainer.appendChild(card); + // Update pet card status + card.classList.remove('active'); + card.classList.add('storage'); + const statusDiv = card.querySelector('.pet-status'); + if (statusDiv) { + statusDiv.textContent = 'Storage'; + statusDiv.className = 'pet-status status-storage'; + } + } + }); + + // Load team data from server for the selected slot + if (teamSlot === 1) { + // For Team 1, load current active pets (default behavior) + loadCurrentActiveTeam(); + } else { + // For Teams 2 and 3, load saved configuration if exists + loadSavedTeamConfiguration(teamSlot); + } + + // Reset team state + currentTeam = {}; + originalTeam = {}; + + // Re-initialize team state + updateTeamState(); + } + + function updateDynamicElements(teamSlot) { + // Update team header + const teamHeader = document.querySelector('h2'); + if (teamHeader && teamHeader.textContent.includes('Active Team')) { + teamHeader.textContent = `โš”๏ธ Team ${teamSlot} Selection (1-6 pets)`; + } + + // Update save button + const saveBtn = document.getElementById('save-btn'); + if (saveBtn) { + saveBtn.textContent = `๐Ÿ”’ Save Changes to Team ${teamSlot}`; + } + } + + function loadCurrentActiveTeam() { + // Load the player's current active pets back into team slots + console.log('Loading current active team (Team 1)'); + + // Find all pet cards that should be active based on their original data attributes + const allCards = document.querySelectorAll('.pet-card'); + console.log(`Found ${allCards.length} total pet cards`); + + allCards.forEach(card => { + const isActive = card.dataset.active === 'true'; + const teamOrder = card.dataset.teamOrder; + const petId = card.dataset.petId; + + console.log(`Pet ${petId}: active=${isActive}, teamOrder=${teamOrder}`); + + if (isActive && teamOrder && teamOrder !== 'None' && teamOrder !== '' && teamOrder !== 'null') { + const slot = document.getElementById(`slot-${teamOrder}`); + if (slot) { + // Move pet from storage back to team slot + const slotContent = slot.querySelector('.slot-content'); + slotContent.innerHTML = ''; + slotContent.appendChild(card); + + // Update pet visual status + card.classList.remove('storage'); + card.classList.add('active'); + const statusDiv = card.querySelector('.pet-status'); + if (statusDiv) { + statusDiv.textContent = 'Active'; + statusDiv.className = 'pet-status status-active'; + } + + // Update team state tracking + currentTeam[petId] = parseInt(teamOrder); + originalTeam[petId] = parseInt(teamOrder); + + console.log(`โœ… Restored pet ${petId} to slot ${teamOrder}`); + } else { + console.log(`โŒ Could not find slot ${teamOrder} for pet ${petId}`); + } + } else { + // This pet should stay in storage + currentTeam[petId] = false; + if (!originalTeam.hasOwnProperty(petId)) { + originalTeam[petId] = false; + } + console.log(`Pet ${petId} staying in storage`); + } + }); + + console.log('Current team state after restoration:', currentTeam); + updateSaveButton(); + } + + async function loadSavedTeamConfiguration(teamSlot) { + console.log(`Loading saved configuration for team ${teamSlot}`); + try { + const response = await fetch(`/teambuilder/{nickname}/config/load/${teamSlot}`, { + method: 'GET', + headers: { 'Content-Type': 'application/json' } + }); + + if (!response.ok) { + console.error(`Failed to load team config: ${response.status} ${response.statusText}`); + return; + } + + const contentType = response.headers.get('content-type'); + if (!contentType || !contentType.includes('application/json')) { + console.error(`Expected JSON response but got: ${contentType}`); + const text = await response.text(); + console.error('Response body:', text); + return; + } + + const result = await response.json(); + + if (result.success && result.team_data) { + // Load pets into team slots based on saved configuration + for (const [position, petData] of Object.entries(result.team_data)) { + if (petData && position >= 1 && position <= 6) { + const petCard = document.querySelector(`[data-pet-id="${petData.pet_id}"]`); + const slot = document.getElementById(`slot-${position}`); + + if (petCard && slot) { + // Move pet to team slot + const slotContent = slot.querySelector('.slot-content'); + slotContent.innerHTML = ''; + slotContent.appendChild(petCard); + + // Update pet status + petCard.classList.remove('storage'); + petCard.classList.add('active'); + const statusDiv = petCard.querySelector('.pet-status'); + if (statusDiv) { + statusDiv.textContent = 'Active'; + statusDiv.className = 'pet-status status-active'; + } + + // Update team state + currentTeam[petData.pet_id] = parseInt(position); + originalTeam[petData.pet_id] = parseInt(position); + } + } + } + } else { + console.log(`No saved configuration found for team ${teamSlot} - starting with empty team`); + // Team is already cleared, just update team state for empty team + currentTeam = {}; + originalTeam = {}; + updateSaveButton(); + } + } catch (error) { + console.error('Error loading team configuration:', error); + } + } + + function updateTeamState() { + // Update team state tracking + const allCards = document.querySelectorAll('.pet-card'); + allCards.forEach(card => { + const petId = card.dataset.petId; + const isActive = card.classList.contains('active'); + const teamOrder = card.dataset.teamOrder; + + if (isActive && teamOrder) { + currentTeam[petId] = parseInt(teamOrder); + if (!originalTeam.hasOwnProperty(petId)) { + originalTeam[petId] = parseInt(teamOrder); + } + } else { + currentTeam[petId] = false; + if (!originalTeam.hasOwnProperty(petId)) { + originalTeam[petId] = false; + } + } + }); + + updateSaveButton(); + } + + function updateTeamCard(teamSlot) { + // Update the team card display to reflect current team composition + const teamCard = document.querySelector(`[data-slot="${teamSlot}"]`); + if (!teamCard) return; + + // Count active pets in current team + let petCount = 0; + let petPreviews = ''; + + // Generate mini pet previews for the team card + for (let i = 1; i <= 6; i++) { + const slot = document.getElementById(`slot-${i}`); + if (slot) { + const petCard = slot.querySelector('.pet-card'); + if (petCard) { + const petName = petCard.querySelector('.pet-name').textContent; + petPreviews += `
${petName}
`; + petCount++; + } else { + petPreviews += '
Empty
'; + } + } + } + + // Update team card content + const statusSpan = teamCard.querySelector('.team-status'); + const previewDiv = teamCard.querySelector('.team-preview'); + + if (statusSpan) { + statusSpan.textContent = petCount > 0 ? `${petCount}/6 pets` : 'Empty team'; + } + + if (previewDiv) { + previewDiv.innerHTML = petPreviews; + } + } + + function filterPets() { + const searchTerm = document.getElementById('pet-search').value.toLowerCase(); + const storageContainer = document.getElementById('storage-container'); + const petCards = storageContainer.querySelectorAll('.pet-card'); + + petCards.forEach(card => { + const petName = card.querySelector('.pet-name').textContent.toLowerCase(); + const petSpecies = card.dataset.species ? card.dataset.species.toLowerCase() : ''; + const petTypes = card.querySelectorAll('.type-badge'); + let typeText = ''; + petTypes.forEach(badge => typeText += badge.textContent.toLowerCase() + ' '); + + const matches = petName.includes(searchTerm) || + petSpecies.includes(searchTerm) || + typeText.includes(searchTerm); + + card.style.display = matches ? 'block' : 'none'; + }); + } + + function sortPets() { + const sortBy = document.getElementById('pet-sort').value; + const storageContainer = document.getElementById('storage-container'); + const petCards = Array.from(storageContainer.querySelectorAll('.pet-card')); + + petCards.sort((a, b) => { + switch (sortBy) { + case 'name': + const nameA = a.querySelector('.pet-name').textContent.toLowerCase(); + const nameB = b.querySelector('.pet-name').textContent.toLowerCase(); + return nameA.localeCompare(nameB); + + case 'level': + const levelA = parseInt(a.querySelector('.pet-level').textContent.replace('Level ', '')); + const levelB = parseInt(b.querySelector('.pet-level').textContent.replace('Level ', '')); + return levelB - levelA; // Descending order + + case 'type': + const typeA = a.querySelector('.type-badge').textContent.toLowerCase(); + const typeB = b.querySelector('.type-badge').textContent.toLowerCase(); + return typeA.localeCompare(typeB); + + case 'species': + const speciesA = a.dataset.species ? a.dataset.species.toLowerCase() : ''; + const speciesB = b.dataset.species ? b.dataset.species.toLowerCase() : ''; + return speciesA.localeCompare(speciesB); + + default: + return 0; + } + }); + + // Re-append sorted cards to container + petCards.forEach(card => storageContainer.appendChild(card)); + } + function initializeTeamBuilder() { console.log('Team Builder: Starting initialization...'); + // Initialize dynamic elements for Team 1 (default) + updateDynamicElements(1); + // Initialize team state const allCards = document.querySelectorAll('.pet-card'); console.log(`Found ${allCards.length} pet cards`); @@ -6647,7 +6403,12 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): const saveBtn = document.getElementById('save-btn'); const hasChanges = JSON.stringify(originalTeam) !== JSON.stringify(currentTeam); saveBtn.disabled = !hasChanges; - saveBtn.textContent = hasChanges ? '๐Ÿ”’ Save Team Changes' : 'โœ… No Changes'; + // Preserve the dynamic team number text + if (hasChanges) { + saveBtn.textContent = `๐Ÿ”’ Save Changes to Team ${currentEditingTeam}`; + } else { + saveBtn.textContent = `โœ… No Changes (Team ${currentEditingTeam})`; + } } async function saveTeam() { @@ -6656,11 +6417,21 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): teamData[petId] = position; }); + // Include the current editing team slot + const saveData = { + teamData: teamData, + teamSlot: currentEditingTeam + }; + + console.log('๐Ÿ” SAVE DEBUG: Saving team data:', saveData); + console.log('๐Ÿ” SAVE DEBUG: Current editing team:', currentEditingTeam); + console.log('๐Ÿ” SAVE DEBUG: Team data entries:', Object.keys(teamData).length); + try { const response = await fetch('/teambuilder/""" + nickname + """/save', { method: 'POST', headers: { 'Content-Type': 'application/json' }, - body: JSON.stringify(teamData) + body: JSON.stringify(saveData) }); const result = await response.json(); @@ -6681,6 +6452,9 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): showMessage('Please enter a 6-digit PIN', 'error'); return; } + + console.log('๐Ÿ” PIN DEBUG: Verifying PIN for team:', currentEditingTeam); + console.log('๐Ÿ” PIN DEBUG: PIN entered:', pin); try { const response = await fetch('/teambuilder/""" + nickname + """/verify', { @@ -6697,6 +6471,9 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): document.getElementById('pin-section').style.display = 'none'; document.getElementById('pin-input').value = ''; + // Update team card display after successful save + updateTeamCard(currentEditingTeam); + // Celebration animation document.querySelectorAll('.pet-card').forEach(card => { card.style.animation = 'bounce 0.6s ease-in-out'; @@ -6734,6 +6511,102 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): """ + # Generate team cards HTML + print(f"Debug: Generating team cards for {len(team_configs)} configs") + team_cards_html = "" + + # If no team configs exist, create default slots with Team 1 showing current active team + if not team_configs: + print("Debug: No team configs found, creating default empty slots") + for slot in range(1, 4): + # For Team 1, show current active pets + if slot == 1: + pet_previews = "" + active_count = 0 + for pos in range(1, 7): + # Find pet in position pos + pet_in_slot = next((pet for pet in active_pets if pet.get('team_order') == pos), None) + if pet_in_slot: + pet_name = pet_in_slot['nickname'] or pet_in_slot['species_name'] + pet_emoji = pet_in_slot.get('emoji', '๐Ÿพ') + pet_previews += f'
{pet_emoji} {pet_name}
' + active_count += 1 + else: + pet_previews += '
Empty
' + + status_text = f"{active_count}/6 pets" if active_count > 0 else "Empty team" + else: + # Teams 2 and 3 are empty by default + pet_previews = '
Empty
' * 6 + status_text = "Empty team" + + team_cards_html += f''' +
+
+

Team {slot}

+ {status_text} +
+
+ {pet_previews} +
+ +
+ ''' + else: + for config in team_configs: + print(f"Debug: Processing config: {config}") + pet_previews = "" + + # Special handling for Team 1 - show current active team instead of saved config + if config['slot'] == 1: + active_count = 0 + for pos in range(1, 7): + # Find pet in position pos from current active team + pet_in_slot = next((pet for pet in active_pets if pet.get('team_order') == pos), None) + if pet_in_slot: + pet_name = pet_in_slot['nickname'] or pet_in_slot['species_name'] + pet_emoji = pet_in_slot.get('emoji', '๐Ÿพ') + pet_previews += f'
{pet_emoji} {pet_name}
' + active_count += 1 + else: + pet_previews += '
Empty
' + status_text = f"{active_count}/6 pets" if active_count > 0 else "Empty team" + else: + # For Teams 2 and 3, use saved configuration data + if config['team_data']: + for pos in range(1, 7): + if str(pos) in config['team_data'] and config['team_data'][str(pos)]: + pet_info = config['team_data'][str(pos)] + pet_emoji = pet_info.get('emoji', '๐Ÿพ') + pet_previews += f'
{pet_emoji} {pet_info["name"]}
' + else: + pet_previews += '
Empty
' + else: + pet_previews = '
Empty
' * 6 + status_text = f"{config['pet_count']}/6 pets" if config['pet_count'] > 0 else "Empty team" + + active_class = "active" if config['slot'] == 1 else "" # Default to team 1 as active + + team_cards_html += f''' +
+
+

{config['name']}

+ {status_text} +
+
+ {pet_previews} +
+ +
+ ''' + + # Replace placeholder with actual team cards + team_builder_content = team_builder_content.replace('', team_cards_html) + # Get the unified template html_content = self.get_page_template(f"Team Builder - {nickname}", team_builder_content, "") @@ -6742,6 +6615,764 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): self.end_headers() self.wfile.write(html_content.encode()) + def serve_test_teambuilder(self, nickname): + """Serve the test team builder interface with simplified team management""" + from urllib.parse import unquote + nickname = unquote(nickname) + + print(f"DEBUG: serve_test_teambuilder called with nickname: {nickname}") + + try: + from src.database import Database + database = Database() + + # Get event loop + try: + loop = asyncio.get_running_loop() + except RuntimeError: + loop = asyncio.new_event_loop() + asyncio.set_event_loop(loop) + + player_data = loop.run_until_complete(self.fetch_player_data(database, nickname)) + + if player_data is None: + self.serve_player_not_found(nickname) + return + + pets = player_data['pets'] + if not pets: + self.serve_test_teambuilder_no_pets(nickname) + return + + self.serve_test_teambuilder_interface(nickname, pets) + + except Exception as e: + print(f"Error loading test team builder for {nickname}: {e}") + self.serve_player_error(nickname, f"Error loading test team builder: {str(e)}") + + def serve_test_teambuilder_no_pets(self, nickname): + """Show message when player has no pets using unified template""" + content = f""" +
+

๐Ÿพ Test Team Builder

+

Simplified team management (Test Version)

+
+ +
+

๐Ÿพ No Pets Found

+

{nickname}, you need to catch some pets before using the team builder!

+

Head to the IRC channel and use !explore to find wild pets!

+ โ† Back to Profile +
+ """ + + html_content = self.get_page_template(f"Test Team Builder - {nickname}", content, "") + + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(html_content.encode()) + + def serve_test_teambuilder_interface(self, nickname, pets): + """Serve the simplified test team builder interface""" + # Get team configurations for this player + import asyncio + + try: + from src.database import Database + database = Database() + + # Get event loop + try: + loop = asyncio.get_running_loop() + except RuntimeError: + loop = asyncio.new_event_loop() + asyncio.set_event_loop(loop) + + print(f"Debug: Getting player data for {nickname}") + # Get player info + player = loop.run_until_complete(database.get_player(nickname)) + if not player: + self.serve_player_error(nickname, "Player not found") + return + + print(f"Debug: Player found: {player}") + # Get team configurations + team_configs = loop.run_until_complete(database.get_player_team_configurations(player['id'])) + print(f"Debug: Team configs: {team_configs}") + + # Create the simplified interface + print(f"Debug: Creating content with {len(pets)} pets") + # TEMPORARY: Use simple content to test + content = f""" +

Test Team Builder - {nickname}

+

Found {len(pets)} pets and {len(team_configs)} team configs

+

First pet: {pets[0]['nickname'] if pets else 'No pets'}

+ """ + print("Debug: Content created successfully") + + html_content = self.get_page_template(f"Test Team Builder - {nickname}", content, "") + + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(html_content.encode()) + + except Exception as e: + print(f"Error in serve_test_teambuilder_interface: {e}") + self.serve_player_error(nickname, f"Error loading interface: {str(e)}") + + def create_test_teambuilder_content(self, nickname, pets, team_configs): + """Create the simplified test team builder HTML content""" + import json + + # Pre-process pets data for JavaScript + pets_data = [] + for pet in pets: + pets_data.append({ + 'id': pet['id'], + 'name': pet['nickname'], + 'level': pet['level'], + 'type_primary': pet['type1'], + 'rarity': 1 + }) + pets_json = json.dumps(pets_data) + + # Build team cards for each configuration + team_cards_html = "" + for config in team_configs: + pet_previews = "" + if config['team_data']: + for pos in range(1, 7): + if str(pos) in config['team_data'] and config['team_data'][str(pos)]: + pet_info = config['team_data'][str(pos)] + pet_previews += f'
{pet_info["name"]}
' + else: + pet_previews += '
Empty
' + else: + pet_previews = '
No pets assigned
' + + status_text = f"{config['pet_count']}/6 pets" if config['pet_count'] > 0 else "Empty team" + + team_cards_html += f''' +
+
+

{config['name']}

+ {status_text} +
+
+ {pet_previews} +
+
+ +
+
+ ''' + + return f''' + + +
+
+

๐Ÿพ Test Team Builder

+

Choose a team to edit, make changes, and save with PIN verification

+
+ +
+

Choose Team to Edit

+

Select one of your 3 teams to edit. Each team can have up to 6 pets.

+ +
+ {team_cards_html} +
+
+ +
+
+

Editing Team 1

+ +
+ +
+ +
+ +
+ +
+ +
+

๐Ÿ” PIN Verification Required

+

A 6-digit PIN has been sent to you via IRC private message.

+

Enter the PIN below to confirm your team changes:

+ + +
+
+
+
+ + + ''' + def handle_team_save(self, nickname): """Handle team save request and generate PIN""" try: @@ -6774,9 +7405,18 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): print(f"Error in handle_team_save: {e}") self.send_json_response({"success": False, "error": "Internal server error"}, 500) - async def _handle_team_save_async(self, nickname, team_data): + async def _handle_team_save_async(self, nickname, save_data): """Async handler for team save""" try: + # Extract team data and slot from new structure + if isinstance(save_data, dict) and 'teamData' in save_data: + team_data = save_data['teamData'] + team_slot = save_data.get('teamSlot', 1) # Default to slot 1 + else: + # Backwards compatibility - old format + team_data = save_data + team_slot = 1 + # Get player player = await self.database.get_player(nickname) if not player: @@ -6787,11 +7427,15 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): if not validation["valid"]: return {"success": False, "error": validation["error"]} - # Create pending team change with PIN + # Create pending team change with PIN (include team slot info) import json + change_data = { + 'teamData': team_data, + 'teamSlot': team_slot + } result = await self.database.create_pending_team_change( player["id"], - json.dumps(team_data) + json.dumps(change_data) ) if result["success"]: @@ -7083,6 +7727,2205 @@ class PetBotRequestHandler(BaseHTTPRequestHandler): except Exception as e: print(f"Error in _handle_team_config_rename_async: {e}") return {"success": False, "error": str(e)} + + def handle_team_config_apply(self, nickname, slot): + """Handle applying team configuration to active team""" + try: + # Validate slot number + try: + slot_num = int(slot) + if slot_num < 1 or slot_num > 3: + self.send_json_response({"success": False, "error": "Slot must be 1, 2, or 3"}, 400) + return + except ValueError: + self.send_json_response({"success": False, "error": "Invalid slot number"}, 400) + return + + # Run async operations + import asyncio + result = asyncio.run(self._handle_team_config_apply_async(nickname, slot_num)) + + if result["success"]: + self.send_json_response(result, 200) + else: + self.send_json_response(result, 404 if "not found" in result.get("error", "") else 400) + + except Exception as e: + print(f"Error in handle_team_config_apply: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _handle_team_config_apply_async(self, nickname, slot_num): + """Async handler for team configuration apply""" + try: + # Get player + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Apply the team configuration + result = await self.database.apply_team_configuration(player["id"], slot_num) + return result + + except Exception as e: + print(f"Error in _handle_team_config_apply_async: {e}") + return {"success": False, "error": str(e)} + + def handle_test_team_save(self, nickname): + """Handle test team builder save request and generate PIN""" + try: + # Get POST data + content_length = int(self.headers.get('Content-Length', 0)) + if content_length == 0: + self.send_json_response({"success": False, "error": "No data provided"}, 400) + return + + post_data = self.rfile.read(content_length).decode('utf-8') + + try: + import json + data = json.loads(post_data) + team_slot = data.get('team_slot') + team_data = data.get('team_data', {}) + + if not team_slot or team_slot not in [1, 2, 3]: + self.send_json_response({"success": False, "error": "Invalid team slot"}, 400) + return + + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid JSON data"}, 400) + return + + # Run async operations + import asyncio + result = asyncio.run(self._handle_test_team_save_async(nickname, team_slot, team_data)) + + if result["success"]: + self.send_json_response(result, 200) + else: + self.send_json_response(result, 400) + + except Exception as e: + print(f"Error in handle_test_team_save: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _handle_test_team_save_async(self, nickname, team_slot, team_data): + """Async handler for test team save""" + try: + # Get player + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Generate PIN and store pending change + import json + team_json = json.dumps(team_data) + config_name = f"Team {team_slot}" + + result = await self.database.save_team_configuration( + player["id"], team_slot, config_name, team_json + ) + + if result: + # Generate PIN for verification (using existing PIN system) + pin_result = await self.database.create_team_change_pin( + player["id"], team_json + ) + + if pin_result["success"]: + # Send PIN to IRC + if hasattr(self.server, 'bot') and self.server.bot: + self.server.bot.send_private_message( + nickname, + f"๐Ÿ” Team {team_slot} Save PIN: {pin_result['pin_code']} (expires in 10 minutes)" + ) + + return {"success": True, "message": "PIN sent to IRC"} + else: + return {"success": False, "error": "Failed to generate PIN"} + else: + return {"success": False, "error": "Failed to save team configuration"} + + except Exception as e: + print(f"Error in _handle_test_team_save_async: {e}") + return {"success": False, "error": str(e)} + + def handle_test_team_verify(self, nickname): + """Handle test team builder PIN verification""" + try: + # Get PIN from request body + content_length = int(self.headers['Content-Length']) + post_data = self.rfile.read(content_length) + + import json + try: + data = json.loads(post_data.decode('utf-8')) + pin_code = data.get('pin', '').strip() + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid request data"}, 400) + return + + if not pin_code or len(pin_code) != 6: + self.send_json_response({"success": False, "error": "PIN must be 6 digits"}, 400) + return + + # Run async verification + import asyncio + result = asyncio.run(self._handle_test_team_verify_async(nickname, pin_code)) + + if result["success"]: + self.send_json_response(result, 200) + else: + self.send_json_response(result, 400) + + except Exception as e: + print(f"Error in handle_test_team_verify: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _handle_test_team_verify_async(self, nickname, pin_code): + """Async handler for test team PIN verification""" + try: + # Get player + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Verify PIN + result = await self.database.verify_team_change_pin(player["id"], pin_code) + + if result["success"]: + return {"success": True, "message": "Team configuration saved successfully!"} + else: + return {"success": False, "error": result.get("error", "Invalid PIN")} + + except Exception as e: + print(f"Error in _handle_test_team_verify_async: {e}") + return {"success": False, "error": str(e)} + + def handle_pet_rename_request(self, nickname): + """Handle pet rename request and generate PIN""" + try: + # Get POST data + content_length = int(self.headers.get('Content-Length', 0)) + if content_length == 0: + self.send_json_response({"success": False, "error": "No data provided"}, 400) + return + + post_data = self.rfile.read(content_length).decode('utf-8') + + try: + import json + data = json.loads(post_data) + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid JSON data"}, 400) + return + + # Run async operations + import asyncio + try: + result = asyncio.run(self._handle_pet_rename_request_async(nickname, data)) + except Exception as async_error: + print(f"Async error in pet rename: {async_error}") + self.send_json_response({"success": False, "error": f"Async error: {str(async_error)}"}, 500) + return + + if result["success"]: + self.send_json_response(result, 200) + else: + self.send_json_response(result, 400) + + except Exception as e: + print(f"Error in handle_pet_rename_request: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _handle_pet_rename_request_async(self, nickname, data): + """Async handler for pet rename request""" + try: + # Get player + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Validate required fields + if "pet_id" not in data or "new_nickname" not in data: + return {"success": False, "error": "Missing pet_id or new_nickname"} + + pet_id = data["pet_id"] + new_nickname = data["new_nickname"] + + # Request pet rename with PIN + result = await self.database.request_pet_rename(player["id"], pet_id, new_nickname) + + if result["success"]: + # Send PIN via IRC + self.send_pet_rename_pin_via_irc(nickname, result["pin"]) + return { + "success": True, + "message": f"PIN sent to {nickname} via IRC. Check your messages!" + } + else: + return result + + except Exception as e: + print(f"Error in _handle_pet_rename_request_async: {e}") + return {"success": False, "error": str(e)} + + def handle_pet_rename_verify(self, nickname): + """Handle PIN verification for pet rename""" + try: + # Get POST data + content_length = int(self.headers.get('Content-Length', 0)) + if content_length == 0: + self.send_json_response({"success": False, "error": "No data provided"}, 400) + return + + post_data = self.rfile.read(content_length).decode('utf-8') + + try: + import json + data = json.loads(post_data) + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid JSON data"}, 400) + return + + # Run async operations + import asyncio + result = asyncio.run(self._handle_pet_rename_verify_async(nickname, data)) + + if result["success"]: + self.send_json_response(result, 200) + else: + self.send_json_response(result, 400) + + except Exception as e: + print(f"Error in handle_pet_rename_verify: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _handle_pet_rename_verify_async(self, nickname, data): + """Async handler for pet rename PIN verification""" + try: + # Get player + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Validate required field + if "pin" not in data: + return {"success": False, "error": "Missing PIN"} + + pin_code = data["pin"] + + # Verify PIN and apply pet rename + result = await self.database.verify_pet_rename(player["id"], pin_code) + + if result["success"]: + return { + "success": True, + "message": f"Pet renamed to '{result['new_nickname']}' successfully!", + "new_nickname": result["new_nickname"] + } + else: + return result + + except Exception as e: + print(f"Error in _handle_pet_rename_verify_async: {e}") + return {"success": False, "error": str(e)} + + def send_pet_rename_pin_via_irc(self, nickname, pin_code): + """Send pet rename PIN to player via IRC private message""" + print(f"๐Ÿ” Pet rename PIN for {nickname}: {pin_code}") + + # Try to send via IRC bot if available + if self.bot and hasattr(self.bot, 'send_message_sync'): + try: + # Send PIN via private message using sync wrapper + self.bot.send_message_sync(nickname, f"๐Ÿ” Pet Rename PIN: {pin_code}") + self.bot.send_message_sync(nickname, f"๐Ÿ’ก Enter this PIN on the web page to confirm your pet rename. PIN expires in 15 seconds.") + print(f"โœ… Pet rename PIN sent to {nickname} via IRC") + except Exception as e: + print(f"โŒ Failed to send pet rename PIN via IRC: {e}") + else: + print(f"โŒ No IRC bot available to send pet rename PIN to {nickname}") + print(f"๐Ÿ’ก Manual pet rename PIN for {nickname}: {pin_code}") + + def serve_admin_login(self): + """Serve the admin login page""" + import sys + sys.path.append('.') + from config import ADMIN_USER + + content = """ +
+

๐Ÿ” Admin Control Panel

+

Authorized access only

+
+ +
+

Authentication Required

+

This area is restricted to bot administrators.

+ +
+
+ + +
+ + + +
+ A PIN will be sent to your IRC private messages +
+
+ + + +
+
+ + + + + """ + + html = self.get_page_template("Admin Login", content, "") + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(html.encode()) + + def handle_admin_auth(self): + """Handle admin authentication request and generate PIN""" + try: + # Get POST data + content_length = int(self.headers.get('Content-Length', 0)) + if content_length == 0: + self.send_json_response({"success": False, "error": "No data provided"}, 400) + return + + post_data = self.rfile.read(content_length).decode('utf-8') + + try: + import json + data = json.loads(post_data) + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid JSON data"}, 400) + return + + nickname = data.get('nickname', '').strip() + + # Verify admin user + import sys + sys.path.append('.') + from config import ADMIN_USER + + if nickname.lower() != ADMIN_USER.lower(): + self.send_json_response({"success": False, "error": "Access denied"}, 403) + return + + # Generate PIN + import random + pin = ''.join([str(random.randint(0, 9)) for _ in range(6)]) + + # Store PIN with expiration (15 minutes) + import time + expiry = time.time() + (15 * 60) + + # Store in database for verification + import asyncio + result = asyncio.run(self._store_admin_pin_async(nickname, pin, expiry)) + + if result: + # Send PIN via IRC + self.send_admin_pin_via_irc(nickname, pin) + self.send_json_response({"success": True, "message": "PIN sent via IRC"}) + else: + self.send_json_response({"success": False, "error": "Failed to generate PIN"}, 500) + + except Exception as e: + print(f"Error in handle_admin_auth: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _store_admin_pin_async(self, nickname, pin, expiry): + """Store admin PIN in database""" + try: + import aiosqlite + # Create temporary admin_pins table if it doesn't exist + async with aiosqlite.connect(self.database.db_path) as db: + # Create table if it doesn't exist + await db.execute(""" + CREATE TABLE IF NOT EXISTS admin_pins ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + nickname TEXT NOT NULL, + pin_code TEXT NOT NULL, + expires_at REAL NOT NULL, + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + ) + """) + + # Insert admin PIN + await db.execute(""" + INSERT INTO admin_pins (nickname, pin_code, expires_at) + VALUES (?, ?, ?) + """, (nickname, pin, expiry)) + await db.commit() + return True + except Exception as e: + print(f"Error storing admin PIN: {e}") + return False + + def handle_admin_verify(self): + """Handle admin PIN verification""" + try: + # Get POST data + content_length = int(self.headers.get('Content-Length', 0)) + if content_length == 0: + self.send_json_response({"success": False, "error": "No data provided"}, 400) + return + + post_data = self.rfile.read(content_length).decode('utf-8') + + try: + import json + data = json.loads(post_data) + except json.JSONDecodeError: + self.send_json_response({"success": False, "error": "Invalid JSON data"}, 400) + return + + nickname = data.get('nickname', '').strip() + pin = data.get('pin', '').strip() + + # Verify PIN + import asyncio + result = asyncio.run(self._verify_admin_pin_async(nickname, pin)) + + if result: + # Create session token + import hashlib + import time + session_token = hashlib.sha256(f"{nickname}:{pin}:{time.time()}".encode()).hexdigest() + + # Store session + self.admin_sessions[session_token] = { + 'nickname': nickname, + 'expires': time.time() + (60 * 60) # 1 hour session + } + + # Set cookie + self.send_response(200) + self.send_header('Content-type', 'application/json') + self.send_header('Set-Cookie', f'admin_session={session_token}; Path=/admin; HttpOnly') + self.end_headers() + self.wfile.write(json.dumps({"success": True}).encode()) + else: + self.send_json_response({"success": False, "error": "Invalid or expired PIN"}, 401) + + except Exception as e: + print(f"Error in handle_admin_verify: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _verify_admin_pin_async(self, nickname, pin): + """Verify admin PIN from database""" + try: + import aiosqlite + import time + current_time = time.time() + + # Check for valid PIN + async with aiosqlite.connect(self.database.db_path) as db: + cursor = await db.execute(""" + SELECT pin_code FROM admin_pins + WHERE nickname = ? AND pin_code = ? AND expires_at > ? + """, (nickname, pin, current_time)) + result = await cursor.fetchone() + + if result: + # Delete used PIN + await db.execute(""" + DELETE FROM admin_pins + WHERE nickname = ? AND pin_code = ? + """, (nickname, pin)) + await db.commit() + return True + + return False + except Exception as e: + print(f"Error verifying admin PIN: {e}") + return False + + def send_admin_pin_via_irc(self, nickname, pin_code): + """Send admin PIN to user via IRC private message""" + print(f"๐Ÿ” Admin PIN for {nickname}: {pin_code}") + + # Try to send via IRC bot if available + if self.bot and hasattr(self.bot, 'send_message_sync'): + try: + # Send PIN via private message + self.bot.send_message_sync(nickname, f"๐Ÿ” Admin Panel PIN: {pin_code}") + self.bot.send_message_sync(nickname, f"โš ๏ธ This PIN expires in 15 minutes. Do not share it with anyone!") + self.bot.send_message_sync(nickname, f"๐Ÿ’ก Enter this PIN at the admin login page to access the control panel.") + print(f"โœ… Admin PIN sent to {nickname} via IRC") + except Exception as e: + print(f"โŒ Failed to send admin PIN via IRC: {e}") + else: + print(f"โŒ No IRC bot available to send admin PIN to {nickname}") + print(f"๐Ÿ’ก Manual admin PIN for {nickname}: {pin_code}") + + def check_admin_session(self): + """Check if user has valid admin session""" + # Get cookie + cookie_header = self.headers.get('Cookie', '') + session_token = None + + for cookie in cookie_header.split(';'): + if cookie.strip().startswith('admin_session='): + session_token = cookie.strip()[14:] + break + + if not session_token: + return None + + # Check if session is valid + import time + session = self.admin_sessions.get(session_token) + + if session and session['expires'] > time.time(): + # Extend session + session['expires'] = time.time() + (60 * 60) + return session['nickname'] + + # Invalid or expired session + if session_token in self.admin_sessions: + del self.admin_sessions[session_token] + + return None + + def serve_admin_dashboard(self): + """Serve the admin dashboard page""" + # Check admin session + admin_user = self.check_admin_session() + if not admin_user: + # Redirect to login + self.send_response(302) + self.send_header('Location', '/admin') + self.end_headers() + return + + # Get system statistics + import asyncio + stats = asyncio.run(self._get_admin_stats_async()) + + content = f""" +
+

๐ŸŽฎ Admin Control Panel

+

Welcome, {admin_user}!

+
+ + +
+

๐Ÿ“Š System Statistics

+
+
+

๐Ÿ‘ฅ Total Players

+
{stats['total_players']}
+
+
+

๐Ÿพ Total Pets

+
{stats['total_pets']}
+
+
+

โš”๏ธ Active Battles

+
{stats['active_battles']}
+
+
+

๐Ÿ’พ Database Size

+
{stats['db_size']}
+
+
+
+ + +
+

๐Ÿ‘ฅ Player Management

+
+

Search Player

+ + +
+
+
+ + +
+

๐Ÿ”ง System Controls

+
+
+

Database Management

+ + +
+
+ +
+

IRC Management

+ + + +
+
+
+
+ + +
+

๐ŸŒ Game Management

+
+
+

Weather Control

+ + + +
+
+ +
+

Rate Limiting

+ + + +
+
+
+
+ + + + + """ + + html = self.get_page_template("Admin Dashboard", content, "") + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(html.encode()) + + def _get_location_options(self, locations): + """Generate HTML options for locations""" + options = "" + for location in locations: + options += f'' + return options + + async def _get_admin_stats_async(self): + """Get admin dashboard statistics""" + import aiosqlite + import os + + stats = { + 'total_players': 0, + 'total_pets': 0, + 'active_battles': 0, + 'db_size': '0 MB', + 'total_achievements': 0, + 'total_badges': 0, + 'total_items': 0, + 'locations': [] + } + + try: + async with aiosqlite.connect(self.database.db_path) as db: + # Get player count + cursor = await db.execute("SELECT COUNT(*) FROM players") + stats['total_players'] = (await cursor.fetchone())[0] + + # Get pet count + cursor = await db.execute("SELECT COUNT(*) FROM pets") + stats['total_pets'] = (await cursor.fetchone())[0] + + # Get active battles (if table exists) + try: + cursor = await db.execute("SELECT COUNT(*) FROM active_battles") + stats['active_battles'] = (await cursor.fetchone())[0] + except: + stats['active_battles'] = 0 + + # Get achievement count + try: + cursor = await db.execute("SELECT COUNT(*) FROM player_achievements") + stats['total_achievements'] = (await cursor.fetchone())[0] + except: + stats['total_achievements'] = 0 + + # Get badge count (if table exists) + try: + cursor = await db.execute("SELECT COUNT(*) FROM player_badges") + stats['total_badges'] = (await cursor.fetchone())[0] + except: + stats['total_badges'] = 0 + + # Get item count (if table exists) + try: + cursor = await db.execute("SELECT COUNT(*) FROM player_items") + stats['total_items'] = (await cursor.fetchone())[0] + except: + stats['total_items'] = 0 + + # Get locations + cursor = await db.execute("SELECT id, name FROM locations ORDER BY name") + locations = await cursor.fetchall() + stats['locations'] = [{'id': loc[0], 'name': loc[1]} for loc in locations] + + # Get database size + if os.path.exists(self.database.db_path): + size_bytes = os.path.getsize(self.database.db_path) + stats['db_size'] = f"{size_bytes / 1024 / 1024:.2f} MB" + + except Exception as e: + print(f"Error getting admin stats: {e}") + + return stats + + def handle_admin_api(self, endpoint): + """Handle admin API requests""" + print(f"Admin API request: {endpoint}") + + # Check admin session + admin_user = self.check_admin_session() + if not admin_user: + print(f"Unauthorized admin API request for endpoint: {endpoint}") + self.send_json_response({"success": False, "error": "Unauthorized"}, 401) + return + + print(f"Authorized admin API request from {admin_user} for endpoint: {endpoint}") + + # Get POST data if it's a POST request + content_length = int(self.headers.get('Content-Length', 0)) + post_data = {} + + if content_length > 0 and self.command == 'POST': + try: + import json + post_data = json.loads(self.rfile.read(content_length).decode('utf-8')) + except: + pass + + # Parse query parameters for GET requests + query_params = {} + if '?' in endpoint: + endpoint, query_string = endpoint.split('?', 1) + for param in query_string.split('&'): + if '=' in param: + key, value = param.split('=', 1) + query_params[key] = value + + # Route to appropriate handler + if endpoint.startswith('player/'): + # Handle player endpoints - support both info and updates + player_path = endpoint[7:] # Remove 'player/' prefix + + if player_path.endswith('/update'): + # Player update endpoint + player_name = player_path[:-7] # Remove '/update' suffix + if self.command == 'POST': + self.handle_admin_player_update(player_name, post_data) + else: + self.send_json_response({"success": False, "error": "Method not allowed"}, 405) + else: + # Player info endpoint + self.handle_admin_player_get(player_path) + elif endpoint == 'backup': + if self.command == 'POST': + self.handle_admin_backup_create() + else: + self.send_json_response({"success": False, "error": "Method not allowed"}, 405) + elif endpoint == 'backups': + self.handle_admin_backups_list() + elif endpoint == 'broadcast': + self.handle_admin_broadcast(post_data) + elif endpoint == 'irc-status': + print(f"IRC status endpoint hit!") + # Add a simple test first + try: + print(f"Calling handle_admin_irc_status...") + self.handle_admin_irc_status() + print(f"handle_admin_irc_status completed") + except Exception as e: + print(f"Exception in IRC status handler: {e}") + import traceback + traceback.print_exc() + # Send a simple fallback response + self.send_json_response({ + "success": False, + "error": "IRC status handler failed", + "details": str(e) + }, 500) + elif endpoint == 'weather': + self.handle_admin_weather_set(post_data) + elif endpoint == 'rate-stats': + self.handle_admin_rate_stats(query_params) + elif endpoint == 'rate-reset': + self.handle_admin_rate_reset(post_data) + elif endpoint == 'test': + # Simple test endpoint + import datetime + self.send_json_response({ + "success": True, + "message": "Test endpoint working", + "timestamp": str(datetime.datetime.now()) + }) + else: + self.send_json_response({"success": False, "error": "Unknown endpoint"}, 404) + + def handle_admin_player_search(self, data): + """Search for a player""" + nickname = data.get('nickname', '').strip() + if not nickname: + self.send_json_response({"success": False, "error": "No nickname provided"}) + return + + import asyncio + player = asyncio.run(self.database.get_player(nickname)) + + if player: + # Get additional stats + pet_count = asyncio.run(self._get_player_pet_count_async(player['id'])) + + self.send_json_response({ + "success": True, + "player": { + "nickname": player['nickname'], + "level": player['level'], + "money": player['money'], + "pet_count": pet_count, + "experience": player['experience'] + } + }) + else: + self.send_json_response({"success": False, "error": "Player not found"}) + + async def _get_player_pet_count_async(self, player_id): + """Get player's pet count""" + import aiosqlite + async with aiosqlite.connect(self.database.db_path) as db: + cursor = await db.execute("SELECT COUNT(*) FROM pets WHERE player_id = ?", (player_id,)) + return (await cursor.fetchone())[0] + + def handle_admin_backup_create(self): + """Create a database backup""" + if self.bot and hasattr(self.bot, 'backup_manager'): + import asyncio + result = asyncio.run(self.bot.backup_manager.create_backup("manual", "Admin web interface")) + + if result['success']: + self.send_json_response({ + "success": True, + "message": f"Backup created: {result['filename']}" + }) + else: + self.send_json_response({ + "success": False, + "error": result.get('error', 'Backup failed') + }) + else: + self.send_json_response({ + "success": False, + "error": "Backup system not available" + }) + + def handle_admin_weather_set(self, data): + """Set weather for a location""" + location = data.get('location', '').strip() + weather = data.get('weather', '').strip() + + if not location or not weather: + self.send_json_response({"success": False, "error": "Missing location or weather"}) + return + + # Execute weather change using database directly + try: + import asyncio + result = asyncio.run(self._set_weather_for_location_async(location, weather)) + + if result.get("success"): + self.send_json_response({ + "success": True, + "message": result.get("message", f"Weather set to {weather} in {location}") + }) + else: + self.send_json_response({ + "success": False, + "error": result.get("error", "Failed to set weather") + }) + + except Exception as e: + print(f"Error setting weather: {e}") + self.send_json_response({ + "success": False, + "error": f"Weather system error: {str(e)}" + }) + + async def _set_weather_for_location_async(self, location, weather): + """Async helper to set weather for location""" + try: + import json + import datetime + import random + + # Load weather patterns + try: + with open("config/weather_patterns.json", "r") as f: + weather_data = json.load(f) + except FileNotFoundError: + return { + "success": False, + "error": "Weather configuration file not found" + } + + # Validate weather type + weather_types = list(weather_data["weather_types"].keys()) + if weather not in weather_types: + return { + "success": False, + "error": f"Invalid weather type. Valid types: {', '.join(weather_types)}" + } + + weather_config = weather_data["weather_types"][weather] + + # Calculate duration (3 hours = 180 minutes) + duration_range = weather_config.get("duration_minutes", [90, 180]) + duration = random.randint(duration_range[0], duration_range[1]) + + end_time = datetime.datetime.now() + datetime.timedelta(minutes=duration) + + # Set weather for the location + result = await self.database.set_weather_for_location( + location, weather, end_time.isoformat(), + weather_config.get("spawn_modifier", 1.0), + ",".join(weather_config.get("affected_types", [])) + ) + + if result.get("success"): + # Announce weather change if it actually changed and bot is available + if result.get("changed") and self.bot and hasattr(self.bot, 'game_engine'): + await self.bot.game_engine.announce_weather_change( + location, result.get("previous_weather"), weather, "web" + ) + + return { + "success": True, + "message": f"Weather set to {weather} in {location} for {duration} minutes" + } + else: + return { + "success": False, + "error": result.get("error", "Failed to set weather") + } + + except Exception as e: + print(f"Error in _set_weather_for_location_async: {e}") + return { + "success": False, + "error": str(e) + } + + def handle_admin_announce(self, data): + """Send announcement to IRC""" + message = data.get('message', '').strip() + + if not message: + self.send_json_response({"success": False, "error": "No message provided"}) + return + + if self.bot and hasattr(self.bot, 'send_message_sync'): + try: + # Send to main channel + self.bot.send_message_sync("#petz", f"๐Ÿ“ข ANNOUNCEMENT: {message}") + self.send_json_response({"success": True, "message": "Announcement sent"}) + except Exception as e: + self.send_json_response({"success": False, "error": str(e)}) + else: + self.send_json_response({"success": False, "error": "IRC not available"}) + + def handle_admin_monitor(self, monitor_type): + """Get monitoring data""" + # TODO: Implement real monitoring data + self.send_json_response({ + "success": True, + "type": monitor_type, + "data": [] + }) + + def handle_admin_player_get(self, nickname): + """Get player information""" + if not nickname: + self.send_json_response({"success": False, "error": "No nickname provided"}, 400) + return + + try: + import asyncio + result = asyncio.run(self._get_player_info_async(nickname)) + + if result["success"]: + self.send_json_response(result) + else: + self.send_json_response(result, 404) + + except Exception as e: + print(f"Error in handle_admin_player_get: {e}") + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + def handle_admin_player_update(self, nickname, data): + """Update player information""" + if not nickname: + self.send_json_response({"success": False, "error": "No nickname provided"}, 400) + return + + if not data: + self.send_json_response({"success": False, "error": "No update data provided"}, 400) + return + + try: + import asyncio + result = asyncio.run(self._update_player_async(nickname, data)) + + if result["success"]: + self.send_json_response(result) + else: + self.send_json_response(result, 400) + + except Exception as e: + print(f"Error in handle_admin_player_update: {e}") + import traceback + traceback.print_exc() + self.send_json_response({"success": False, "error": "Internal server error"}, 500) + + async def _update_player_async(self, nickname, data): + """Update player information asynchronously""" + try: + # Validate input data + allowed_fields = ['level', 'experience', 'money'] + updates = {} + + for field in allowed_fields: + if field in data: + value = data[field] + + # Validate each field + if field == 'level': + if not isinstance(value, int) or value < 1 or value > 100: + return {"success": False, "error": "Level must be between 1 and 100"} + updates[field] = value + elif field == 'experience': + if not isinstance(value, int) or value < 0: + return {"success": False, "error": "Experience cannot be negative"} + updates[field] = value + elif field == 'money': + if not isinstance(value, int) or value < 0: + return {"success": False, "error": "Money cannot be negative"} + updates[field] = value + + if not updates: + return {"success": False, "error": "No valid fields to update"} + + # Check if player exists + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Update player data + success = await self.database.update_player_admin(player["id"], updates) + + if success: + return { + "success": True, + "message": f"Updated {', '.join(updates.keys())} for player {nickname}", + "updated_fields": list(updates.keys()) + } + else: + return {"success": False, "error": "Failed to update player data"} + + except Exception as e: + print(f"Error updating player: {e}") + return {"success": False, "error": str(e)} + + async def _get_player_info_async(self, nickname): + """Get player information asynchronously""" + try: + player = await self.database.get_player(nickname) + if not player: + return {"success": False, "error": "Player not found"} + + # Get additional stats + pets = await self.database.get_player_pets(player["id"]) + + # Get location name if current_location_id is set + location_name = "Unknown" + if player.get("current_location_id"): + try: + location_data = await self.database.get_location_by_id(player["current_location_id"]) + if location_data: + location_name = location_data["name"] + else: + location_name = f"Location ID {player['current_location_id']}" + except Exception as loc_error: + print(f"Error resolving location: {loc_error}") + location_name = f"Location ID {player['current_location_id']}" + + # Get team composition + team_info = await self.database.get_team_composition(player["id"]) + + return { + "success": True, + "player": { + "nickname": player["nickname"], + "level": player["level"], + "experience": player["experience"], + "money": player["money"], + "current_location": location_name, + "pet_count": len(pets), + "active_pets": team_info.get("active_pets", 0), + "total_pets": team_info.get("total_pets", 0) + } + } + except Exception as e: + print(f"Error getting player info: {e}") + return {"success": False, "error": str(e)} + + def handle_admin_backups_list(self): + """List available backups""" + try: + import os + backup_dir = "backups" + + if not os.path.exists(backup_dir): + self.send_json_response({ + "success": True, + "backups": [] + }) + return + + backups = [] + for filename in os.listdir(backup_dir): + if filename.endswith('.gz'): + filepath = os.path.join(backup_dir, filename) + size = os.path.getsize(filepath) + # Convert size to human readable format + if size < 1024: + size_str = f"{size} B" + elif size < 1024 * 1024: + size_str = f"{size / 1024:.1f} KB" + else: + size_str = f"{size / (1024 * 1024):.1f} MB" + + backups.append({ + "name": filename, + "size": size_str + }) + + # Sort by name (newest first) + backups.sort(key=lambda x: x['name'], reverse=True) + + self.send_json_response({ + "success": True, + "backups": backups[:10] # Show only last 10 backups + }) + + except Exception as e: + print(f"Error listing backups: {e}") + self.send_json_response({"success": False, "error": str(e)}, 500) + + def handle_admin_broadcast(self, data): + """Send IRC broadcast message""" + message = data.get('message', '').strip() + if not message: + self.send_json_response({"success": False, "error": "No message provided"}, 400) + return + + try: + # Send to IRC channel via bot + if self.bot and hasattr(self.bot, 'send_message_sync'): + from config import IRC_CONFIG + channel = IRC_CONFIG.get("channel", "#petz") + self.bot.send_message_sync(channel, f"๐Ÿ“ข Admin Announcement: {message}") + + self.send_json_response({ + "success": True, + "message": "Broadcast sent successfully" + }) + else: + self.send_json_response({"success": False, "error": "IRC bot not available"}, 500) + + except Exception as e: + print(f"Error sending broadcast: {e}") + self.send_json_response({"success": False, "error": str(e)}, 500) + + def handle_admin_irc_status(self): + """Get comprehensive IRC connection status and activity""" + try: + print(f"IRC status request - checking bot availability...") + bot = self.bot + print(f"Bot instance: {bot}") + + if bot and hasattr(bot, 'connection_manager') and bot.connection_manager: + connection_manager = bot.connection_manager + print(f"Connection manager: {connection_manager}") + + if hasattr(connection_manager, 'get_connection_stats'): + try: + stats = connection_manager.get_connection_stats() + print(f"Got comprehensive connection stats") + + # Return comprehensive IRC status + response_data = { + "success": True, + "irc_status": stats + } + print(f"Sending comprehensive IRC response") + self.send_json_response(response_data) + + except Exception as stats_error: + print(f"Error getting connection stats: {stats_error}") + import traceback + traceback.print_exc() + self.send_json_response({ + "success": False, + "error": f"Failed to get connection stats: {str(stats_error)}" + }, 500) + else: + print("Connection manager has no get_connection_stats method") + self.send_json_response({ + "success": False, + "error": "Connection manager missing get_connection_stats method" + }, 500) + else: + print("No bot instance or connection manager available") + self.send_json_response({ + "success": True, + "irc_status": { + "connected": False, + "state": "disconnected", + "error": "Bot instance or connection manager not available" + } + }) + + except Exception as e: + print(f"Error getting IRC status: {e}") + import traceback + traceback.print_exc() + try: + self.send_json_response({"success": False, "error": str(e)}, 500) + except Exception as json_error: + print(f"Failed to send JSON error response: {json_error}") + # Send a basic HTTP error response if JSON fails + self.send_error(500, f"IRC status error: {str(e)}") + + def handle_admin_rate_stats(self, query_params): + """Get rate limiting statistics""" + try: + username = query_params.get('user', '').strip() + + if self.bot and hasattr(self.bot, 'rate_limiter') and self.bot.rate_limiter: + if username: + # Get stats for specific user + user_stats = self.bot.rate_limiter.get_user_stats(username) + self.send_json_response({ + "success": True, + "stats": { + "violations": user_stats.get("violations", 0), + "banned": user_stats.get("banned", False), + "last_violation": user_stats.get("last_violation", "Never") + } + }) + else: + # Get global stats + global_stats = self.bot.rate_limiter.get_global_stats() + self.send_json_response({ + "success": True, + "stats": { + "total_users": global_stats.get("total_users", 0), + "active_bans": global_stats.get("active_bans", 0), + "total_violations": global_stats.get("total_violations", 0) + } + }) + else: + self.send_json_response({ + "success": True, + "stats": { + "total_users": 0, + "active_bans": 0, + "total_violations": 0 + } + }) + + except Exception as e: + print(f"Error getting rate stats: {e}") + self.send_json_response({"success": False, "error": str(e)}, 500) + + def handle_admin_rate_reset(self, data): + """Reset rate limiting for user or globally""" + try: + username = data.get('user', '').strip() if data.get('user') else None + + if self.bot and hasattr(self.bot, 'rate_limiter') and self.bot.rate_limiter: + if username: + # Reset for specific user + success = self.bot.rate_limiter.reset_user(username) + if success: + self.send_json_response({ + "success": True, + "message": f"Rate limits reset for user: {username}" + }) + else: + self.send_json_response({"success": False, "error": f"User {username} not found"}, 404) + else: + # Global reset + self.bot.rate_limiter.reset_all() + self.send_json_response({ + "success": True, + "message": "All rate limits reset successfully" + }) + else: + self.send_json_response({"success": False, "error": "Rate limiter not available"}, 500) + + except Exception as e: + print(f"Error resetting rate limits: {e}") + self.send_json_response({"success": False, "error": str(e)}, 500) class PetBotWebServer: