mirror of
https://github.com/bunny-lab-io/Borealis.git
synced 2025-07-27 08:28:28 -06:00
- Added Sidebar Menu
- Removed Workflows from Menu Bar - Implemented Node Drawing Logic - Misc Adjustments
This commit is contained in:
113
Data/server.py
113
Data/server.py
@ -1,9 +1,5 @@
|
||||
from flask import Flask, send_from_directory, jsonify, request, abort
|
||||
from flask import Flask, send_from_directory
|
||||
import os
|
||||
import importlib
|
||||
import inspect
|
||||
import uuid
|
||||
from OdenGraphQt import BaseNode
|
||||
|
||||
# Determine the absolute path for the React build folder
|
||||
build_folder = os.path.join(os.getcwd(), "web-interface", "build")
|
||||
@ -12,41 +8,6 @@ if not os.path.exists(build_folder):
|
||||
|
||||
app = Flask(__name__, static_folder=build_folder, static_url_path="/")
|
||||
|
||||
# Directory where nodes are stored
|
||||
NODES_PACKAGE = "Nodes"
|
||||
|
||||
# In-memory workflow storage
|
||||
workflow_data = {
|
||||
"nodes": [],
|
||||
"edges": [] # Store connections separately
|
||||
}
|
||||
|
||||
def import_nodes_from_folder(package_name):
|
||||
"""Dynamically import node classes from the given package and list them."""
|
||||
nodes_by_category = {}
|
||||
package = importlib.import_module(package_name)
|
||||
package_path = package.__path__[0]
|
||||
|
||||
for root, _, files in os.walk(package_path):
|
||||
rel_path = os.path.relpath(root, package_path).replace(os.sep, ".")
|
||||
module_prefix = f"{package_name}.{rel_path}" if rel_path != "." else package_name
|
||||
category_name = os.path.basename(root)
|
||||
|
||||
for file in files:
|
||||
if file.endswith(".py") and file != "__init__.py":
|
||||
module_name = f"{module_prefix}.{file[:-3]}"
|
||||
try:
|
||||
module = importlib.import_module(module_name)
|
||||
for name, obj in inspect.getmembers(module, inspect.isclass):
|
||||
if issubclass(obj, BaseNode) and obj.__module__ == module.__name__:
|
||||
if category_name not in nodes_by_category:
|
||||
nodes_by_category[category_name] = []
|
||||
nodes_by_category[category_name].append(obj.NODE_NAME)
|
||||
except Exception as e:
|
||||
print(f"Failed to import {module_name}: {e}")
|
||||
|
||||
return nodes_by_category
|
||||
|
||||
@app.route("/")
|
||||
def serve_frontend():
|
||||
"""Serve the React app."""
|
||||
@ -55,77 +16,5 @@ def serve_frontend():
|
||||
return send_from_directory(app.static_folder, "index.html")
|
||||
return "<h1>Borealis React App Code Not Found</h1><p>Please re-deploy Borealis Workflow Automation Tool</p>", 404
|
||||
|
||||
@app.route("/api/nodes", methods=["GET"])
|
||||
def get_available_nodes():
|
||||
"""Return available node types."""
|
||||
nodes = import_nodes_from_folder(NODES_PACKAGE)
|
||||
return jsonify(nodes)
|
||||
|
||||
@app.route("/api/workflow", methods=["GET", "POST"])
|
||||
def handle_workflow():
|
||||
"""Retrieve or update the workflow."""
|
||||
global workflow_data
|
||||
if request.method == "GET":
|
||||
return jsonify(workflow_data)
|
||||
elif request.method == "POST":
|
||||
data = request.get_json()
|
||||
if not data:
|
||||
abort(400, "Invalid workflow data")
|
||||
workflow_data = data
|
||||
return jsonify({"status": "success", "workflow": workflow_data})
|
||||
|
||||
@app.route("/api/node", methods=["POST"])
|
||||
def create_node():
|
||||
"""Create a new node with a unique UUID."""
|
||||
data = request.get_json()
|
||||
if not data or "nodeType" not in data:
|
||||
abort(400, "Invalid node data")
|
||||
|
||||
node_id = str(uuid.uuid4()) # Generate a unique ID
|
||||
node = {
|
||||
"id": node_id,
|
||||
"type": data["nodeType"],
|
||||
"position": data.get("position", {"x": 100, "y": 100}),
|
||||
"properties": data.get("properties", {})
|
||||
}
|
||||
workflow_data["nodes"].append(node)
|
||||
return jsonify({"status": "success", "node": node})
|
||||
|
||||
@app.route("/api/node/<string:node_id>", methods=["PUT", "DELETE"])
|
||||
def modify_node(node_id):
|
||||
"""Update or delete a node."""
|
||||
global workflow_data
|
||||
if request.method == "PUT":
|
||||
data = request.get_json()
|
||||
for node in workflow_data["nodes"]:
|
||||
if node["id"] == node_id:
|
||||
node["position"] = data.get("position", node["position"])
|
||||
node["properties"] = data.get("properties", node["properties"])
|
||||
return jsonify({"status": "success", "node": node})
|
||||
abort(404, "Node not found")
|
||||
|
||||
elif request.method == "DELETE":
|
||||
workflow_data["nodes"] = [n for n in workflow_data["nodes"] if n["id"] != node_id]
|
||||
return jsonify({"status": "success", "deletedNode": node_id})
|
||||
|
||||
@app.route("/api/edge", methods=["POST"])
|
||||
def create_edge():
|
||||
"""Create a new connection (edge) between nodes."""
|
||||
data = request.get_json()
|
||||
if not data or "source" not in data or "target" not in data:
|
||||
abort(400, "Invalid edge data")
|
||||
|
||||
edge_id = str(uuid.uuid4())
|
||||
edge = {"id": edge_id, "source": data["source"], "target": data["target"]}
|
||||
workflow_data["edges"].append(edge)
|
||||
return jsonify({"status": "success", "edge": edge})
|
||||
|
||||
@app.route("/api/edge/<string:edge_id>", methods=["DELETE"])
|
||||
def delete_edge(edge_id):
|
||||
"""Delete an edge by ID."""
|
||||
global workflow_data
|
||||
workflow_data["edges"] = [e for e in workflow_data["edges"] if e["id"] != edge_id]
|
||||
return jsonify({"status": "success", "deletedEdge": edge_id})
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(host="0.0.0.0", port=5000, debug=False)
|
||||
|
Reference in New Issue
Block a user