Files
UpdateServer/app.py
T

502 lines
16 KiB
Python
Raw Normal View History

2026-03-11 19:43:33 +00:00
#!/usr/bin/env python3
"""
Update Server for Bots
Serves version information to bots checking for updates.
"""
2026-03-13 13:22:46 +00:00
from flask import Flask, jsonify, request, render_template_string
from flask_cors import CORS
2026-03-11 19:43:33 +00:00
import json
import os
from datetime import datetime
2026-03-13 13:22:46 +00:00
import threading
2026-03-11 19:43:33 +00:00
# Load version data from JSON file
VERSIONS_FILE = 'versions.json'
LOG_FILE = 'update_checks.log'
2026-03-13 13:22:46 +00:00
# Create API app
api_app = Flask(__name__, static_folder=None)
CORS(api_app) # Enable CORS for cross-origin requests from admin GUI
# Create Admin app
admin_app = Flask(__name__)
2026-03-11 19:43:33 +00:00
def load_versions():
"""Load version info from file."""
if os.path.exists(VERSIONS_FILE):
try:
with open(VERSIONS_FILE, 'r') as f:
return json.load(f)
except Exception as e:
print(f"Error loading versions.json: {e}")
return {}
return {}
2026-03-13 13:22:46 +00:00
def save_versions(versions):
"""Save version info to file."""
try:
with open(VERSIONS_FILE, 'w') as f:
json.dump(versions, f, indent=2)
return True
except Exception as e:
print(f"Error saving versions.json: {e}")
return False
2026-03-11 19:43:33 +00:00
def log_check(bot_name):
"""Log update check requests."""
try:
with open(LOG_FILE, 'a') as f:
f.write(f"{datetime.now().isoformat()} - {bot_name}\n")
except Exception as e:
print(f"Error logging check: {e}")
2026-03-13 13:22:46 +00:00
# ====== API ROUTES (Port 5555) ======
@api_app.route('/api/version/<bot_name>', methods=['GET'])
2026-03-11 19:43:33 +00:00
def get_version(bot_name):
"""Get latest version for a specific bot."""
log_check(bot_name)
versions = load_versions()
if bot_name in versions:
return jsonify(versions[bot_name])
return jsonify({"error": "Bot not found"}), 404
2026-03-13 13:22:46 +00:00
@api_app.route('/api/versions', methods=['GET'])
2026-03-11 19:43:33 +00:00
def get_all_versions():
"""Get all bot versions."""
return jsonify(load_versions())
2026-03-13 13:22:46 +00:00
@api_app.route('/health', methods=['GET'])
2026-03-11 19:43:33 +00:00
def health():
"""Health check endpoint."""
return jsonify({"status": "ok"})
2026-03-13 13:22:46 +00:00
@api_app.route('/', methods=['GET'])
2026-03-11 19:43:33 +00:00
def index():
"""Index page with server info."""
versions = load_versions()
bot_count = len(versions)
return jsonify({
"service": "Update Server",
"status": "running",
"bots_tracked": bot_count,
"endpoints": {
"/health": "Health check",
"/api/versions": "Get all bot versions",
"/api/version/<bot_name>": "Get specific bot version"
}
})
2026-03-13 13:22:46 +00:00
@api_app.errorhandler(404)
def api_not_found(error):
2026-03-11 19:43:33 +00:00
"""Handle 404 errors."""
return jsonify({"error": "Endpoint not found"}), 404
2026-03-13 13:22:46 +00:00
@api_app.errorhandler(500)
def api_internal_error(error):
2026-03-11 19:43:33 +00:00
"""Handle 500 errors."""
return jsonify({"error": "Internal server error"}), 500
2026-03-13 13:22:46 +00:00
# ====== ADMIN ROUTES (Port 5566) ======
ADMIN_HTML = """
<!DOCTYPE html>
<html>
<head>
<title>Update Server Admin</title>
<link rel="icon" href="data:image/svg+xml,<svg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 100 100'><text y='0.9em' font-size='90'>🤖</text></svg>">
<style>
* { margin: 0; padding: 0; box-sizing: border-box; }
body {
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
min-height: 100vh;
padding: 20px;
}
.container {
max-width: 1000px;
margin: 0 auto;
}
.header {
text-align: center;
color: white;
margin-bottom: 30px;
}
.header h1 {
font-size: 2.5em;
margin-bottom: 10px;
}
.content {
display: grid;
grid-template-columns: 1fr 1fr;
gap: 20px;
}
.card {
background: white;
border-radius: 8px;
padding: 25px;
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.2);
}
.card h2 {
margin-bottom: 20px;
color: #333;
font-size: 1.5em;
border-bottom: 2px solid #667eea;
padding-bottom: 10px;
}
.form-group {
margin-bottom: 15px;
}
label {
display: block;
margin-bottom: 5px;
color: #555;
font-weight: 500;
}
input, textarea {
width: 100%;
padding: 10px;
border: 1px solid #ddd;
border-radius: 4px;
font-family: inherit;
font-size: 14px;
}
input:focus, textarea:focus {
outline: none;
border-color: #667eea;
box-shadow: 0 0 0 3px rgba(102, 126, 234, 0.1);
}
textarea {
resize: vertical;
min-height: 80px;
}
button {
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
color: white;
padding: 10px 20px;
border: none;
border-radius: 4px;
cursor: pointer;
font-weight: 600;
font-size: 14px;
transition: transform 0.2s;
}
button:hover {
transform: translateY(-2px);
box-shadow: 0 5px 15px rgba(102, 126, 234, 0.3);
}
.bot-list {
list-style: none;
}
.bot-item {
background: #f8f9fa;
padding: 15px;
margin-bottom: 10px;
border-radius: 4px;
border-left: 4px solid #667eea;
display: flex;
justify-content: space-between;
align-items: center;
}
.bot-info {
flex: 1;
}
.bot-name {
font-weight: 600;
color: #333;
margin-bottom: 5px;
}
.bot-version {
color: #666;
font-size: 13px;
}
.bot-actions {
display: flex;
gap: 10px;
}
.btn-small {
padding: 6px 12px;
font-size: 12px;
margin: 0;
}
.btn-danger {
background: #dc3545;
}
.btn-danger:hover {
box-shadow: 0 5px 15px rgba(220, 53, 69, 0.3);
}
.message {
padding: 12px;
border-radius: 4px;
margin-bottom: 15px;
display: none;
}
.message.success {
display: block;
background: #d4edda;
color: #155724;
border: 1px solid #c3e6cb;
}
.message.error {
display: block;
background: #f8d7da;
color: #721c24;
border: 1px solid #f5c6cb;
}
@media (max-width: 768px) {
.content {
grid-template-columns: 1fr;
}
.header h1 {
font-size: 1.8em;
}
}
</style>
</head>
<body>
<div class="container">
<div class="header">
<h1>🤖 Update Server Admin</h1>
<p>Manage bot versions easily</p>
</div>
<div class="content">
<!-- Add/Edit Bot Form -->
<div class="card">
<h2>Add/Edit Version</h2>
<div id="formMessage" class="message"></div>
<form id="versionForm">
<div class="form-group">
<label for="botName">Bot Name *</label>
<input type="text" id="botName" name="botName" placeholder="e.g., TestPostsBot" required>
</div>
<div class="form-group">
<label for="version">Version *</label>
<input type="text" id="version" name="version" placeholder="e.g., 0.2" required>
</div>
<div class="form-group">
<label for="changelogUrl">Changelog URL</label>
<input type="url" id="changelogUrl" name="changelogUrl" placeholder="https://example.com/releases">
</div>
<button type="submit">Save Version</button>
</form>
</div>
<!-- Bot List -->
<div class="card">
<h2>Current Versions</h2>
<ul id="botList" class="bot-list">
<li style="text-align: center; color: #999; padding: 20px;">Loading...</li>
</ul>
</div>
</div>
</div>
<script>
const API_URL = window.location.protocol + '//' + window.location.hostname + ':5555';
// Load bot versions on page load
function loadBots() {
fetch(API_URL + '/api/versions')
.then(r => r.json())
.then(data => {
const list = document.getElementById('botList');
if (Object.keys(data).length === 0) {
list.innerHTML = '<li style="text-align: center; color: #999; padding: 20px;">No bots configured yet</li>';
return;
}
list.innerHTML = Object.entries(data).map(([name, info]) => `
<li class="bot-item">
<div class="bot-info">
<div class="bot-name">${name}</div>
<div class="bot-version">Version: ${info.version}</div>
${info.changelog_url ? `<div class="bot-version">Changelog: ${info.changelog_url}</div>` : ''}
</div>
<div class="bot-actions">
<button class="btn-small" onclick="editBot('${name}')">Edit</button>
<button class="btn-small btn-danger" onclick="deleteBot('${name}')">Delete</button>
</div>
</li>
`).join('');
});
}
function editBot(name) {
fetch(API_URL + '/api/version/' + name)
.then(r => r.json())
.then(data => {
document.getElementById('botName').value = name;
document.getElementById('version').value = data.version;
document.getElementById('changelogUrl').value = data.changelog_url || '';
document.getElementById('botName').disabled = true;
document.getElementById('versionForm').offsetTop && document.getElementById('versionForm').scrollIntoView({behavior: 'smooth'});
});
}
function deleteBot(name) {
if (confirm('Are you sure you want to delete ' + name + '?')) {
fetch('/admin/api/delete', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ bot_name: name })
})
.then(r => r.json())
.then(data => {
showMessage(data.message || 'Deleted', data.success ? 'success' : 'error');
loadBots();
})
.catch(e => showMessage('Error: ' + e, 'error'));
}
}
function showMessage(msg, type) {
const el = document.getElementById('formMessage');
el.textContent = msg;
el.className = 'message ' + type;
setTimeout(() => el.className = 'message', 5000);
}
document.getElementById('versionForm').addEventListener('submit', function(e) {
e.preventDefault();
const data = {
bot_name: document.getElementById('botName').value,
version: document.getElementById('version').value,
changelog_url: document.getElementById('changelogUrl').value
};
fetch('/admin/api/add', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(data)
})
.then(r => r.json())
.then(data => {
showMessage(data.message || 'Saved successfully!', data.success ? 'success' : 'error');
if (data.success) {
document.getElementById('versionForm').reset();
document.getElementById('botName').disabled = false;
loadBots();
}
})
.catch(e => showMessage('Error: ' + e, 'error'));
});
// Load bots on page load
loadBots();
// Refresh every 10 seconds
setInterval(loadBots, 10000);
</script>
</body>
</html>
"""
@admin_app.route('/', methods=['GET'])
def admin_index():
"""Admin dashboard."""
return render_template_string(ADMIN_HTML)
@admin_app.route('/admin/api/add', methods=['POST'])
def admin_add_version():
"""Add or update a bot version."""
try:
data = request.get_json()
bot_name = data.get('bot_name', '').strip()
version = data.get('version', '').strip()
changelog_url = data.get('changelog_url', '').strip()
if not bot_name or not version:
return jsonify({"success": False, "message": "Bot name and version are required"}), 400
versions = load_versions()
versions[bot_name] = {
"version": version,
"changelog_url": changelog_url if changelog_url else versions.get(bot_name, {}).get('changelog_url', '')
}
if save_versions(versions):
return jsonify({"success": True, "message": f"Version saved for {bot_name}"})
else:
return jsonify({"success": False, "message": "Failed to save version"}), 500
except Exception as e:
return jsonify({"success": False, "message": f"Error: {str(e)}"}), 500
@admin_app.route('/admin/api/delete', methods=['POST'])
def admin_delete_version():
"""Delete a bot version."""
try:
data = request.get_json()
bot_name = data.get('bot_name', '').strip()
if not bot_name:
return jsonify({"success": False, "message": "Bot name is required"}), 400
versions = load_versions()
if bot_name in versions:
del versions[bot_name]
if save_versions(versions):
return jsonify({"success": True, "message": f"{bot_name} deleted"})
return jsonify({"success": False, "message": "Bot not found"}), 404
except Exception as e:
return jsonify({"success": False, "message": f"Error: {str(e)}"}), 500
@admin_app.errorhandler(404)
def admin_not_found(error):
"""Handle 404 errors."""
return jsonify({"error": "Endpoint not found"}), 404
@admin_app.errorhandler(500)
def admin_internal_error(error):
"""Handle 500 errors."""
return jsonify({"error": "Internal server error"}), 500
# ====== RUN BOTH APPS ======
def run_api():
"""Run API on port 5555."""
print("[API] Starting on port 5555...")
api_app.run(host='0.0.0.0', port=5555, debug=False)
def run_admin():
"""Run Admin on port 5566."""
print("[ADMIN] Starting on port 5566...")
admin_app.run(host='0.0.0.0', port=5566, debug=False)
2026-03-11 19:43:33 +00:00
if __name__ == '__main__':
2026-03-13 13:22:46 +00:00
versions_count = len(load_versions())
print(f"[STARTUP] Loaded {versions_count} bots from versions.json")
# Run both apps in separate threads
api_thread = threading.Thread(target=run_api, daemon=True)
admin_thread = threading.Thread(target=run_admin, daemon=True)
api_thread.start()
admin_thread.start()
# Keep main thread alive
try:
api_thread.join()
except KeyboardInterrupt:
print("\n[SHUTDOWN] Stopping servers...")