mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-07-27 04:08:29 -06:00
123 lines
4.0 KiB
Python
123 lines
4.0 KiB
Python
from flask import Flask, request, jsonify, send_from_directory
|
|
from flask_socketio import SocketIO, emit
|
|
import time
|
|
import os
|
|
|
|
# ---------------------------------------------
|
|
# React Frontend Hosting Configuration
|
|
# ---------------------------------------------
|
|
build_folder = os.path.join(os.getcwd(), "web-interface", "build")
|
|
if not os.path.exists(build_folder):
|
|
print("WARNING: web-interface build folder not found. Please build your React app.")
|
|
|
|
app = Flask(__name__, static_folder=build_folder, static_url_path="/")
|
|
socketio = SocketIO(app, cors_allowed_origins="*") # Allow cross-origin WebSocket connections
|
|
|
|
@app.route("/")
|
|
def serve_index():
|
|
index_path = os.path.join(build_folder, "index.html")
|
|
if os.path.exists(index_path):
|
|
return send_from_directory(build_folder, "index.html")
|
|
return "<h1>Borealis React App Code Not Found</h1><p>Please re-deploy Borealis Workflow Automation Tool</p>", 404
|
|
|
|
@app.route("/<path:path>")
|
|
def serve_react_app(path):
|
|
full_path = os.path.join(build_folder, path)
|
|
if os.path.exists(full_path):
|
|
return send_from_directory(build_folder, path)
|
|
return send_from_directory(build_folder, "index.html")
|
|
|
|
# ---------------------------------------------
|
|
# Borealis Agent Management (Hybrid: API + WebSockets)
|
|
# ---------------------------------------------
|
|
registered_agents = {}
|
|
agent_configurations = {}
|
|
latest_images = {}
|
|
|
|
# API Endpoints (kept for provisioning and status)
|
|
@app.route("/api/agent/checkin", methods=["POST"])
|
|
def agent_checkin():
|
|
data = request.json
|
|
agent_id = data.get("agent_id")
|
|
hostname = data.get("hostname", "unknown")
|
|
|
|
registered_agents[agent_id] = {
|
|
"agent_id": agent_id,
|
|
"hostname": hostname,
|
|
"last_seen": time.time(),
|
|
"status": "orphaned" if agent_id not in agent_configurations else "provisioned"
|
|
}
|
|
return jsonify({"status": "ok"})
|
|
|
|
@app.route("/api/agent/provision", methods=["POST"])
|
|
def provision_agent():
|
|
data = request.json
|
|
agent_id = data.get("agent_id")
|
|
config = {
|
|
"task": data.get("task", "screenshot"),
|
|
"x": data.get("x", 100),
|
|
"y": data.get("y", 100),
|
|
"w": data.get("w", 300),
|
|
"h": data.get("h", 200),
|
|
"interval": data.get("interval", 1000),
|
|
"visible": data.get("visible", True)
|
|
}
|
|
agent_configurations[agent_id] = config
|
|
if agent_id in registered_agents:
|
|
registered_agents[agent_id]["status"] = "provisioned"
|
|
return jsonify({"status": "provisioned"})
|
|
|
|
@app.route("/api/agents")
|
|
def get_agents():
|
|
return jsonify(registered_agents)
|
|
|
|
# WebSocket Handlers
|
|
@socketio.on('connect_agent')
|
|
def handle_agent_connect(data):
|
|
agent_id = data.get('agent_id')
|
|
hostname = data.get('hostname', 'unknown')
|
|
|
|
registered_agents[agent_id] = {
|
|
"agent_id": agent_id,
|
|
"hostname": hostname,
|
|
"last_seen": time.time(),
|
|
"status": "connected"
|
|
}
|
|
|
|
print(f"Agent connected: {agent_id}")
|
|
emit('agent_connected', {'status': 'connected'})
|
|
|
|
@socketio.on('screenshot')
|
|
def handle_screenshot(data):
|
|
agent_id = data.get('agent_id')
|
|
image_base64 = data.get('image_base64')
|
|
|
|
if agent_id and image_base64:
|
|
latest_images[agent_id] = {
|
|
'image_base64': image_base64,
|
|
'timestamp': time.time()
|
|
}
|
|
|
|
# Real-time broadcast to connected dashboards
|
|
emit('new_screenshot', {
|
|
'agent_id': agent_id,
|
|
'image_base64': image_base64
|
|
}, broadcast=True)
|
|
|
|
emit('screenshot_received', {'status': 'ok'})
|
|
print(f"Screenshot received from agent: {agent_id}")
|
|
else:
|
|
emit('error', {'message': 'Invalid screenshot data'})
|
|
|
|
@socketio.on('request_config')
|
|
def handle_request_config(data):
|
|
agent_id = data.get('agent_id')
|
|
config = agent_configurations.get(agent_id, {})
|
|
|
|
emit('agent_config', config)
|
|
|
|
# ---------------------------------------------
|
|
# Server Start
|
|
# ---------------------------------------------
|
|
if __name__ == "__main__":
|
|
socketio.run(app, host="0.0.0.0", port=5000, debug=False) |