pynode-flow 0.2.0__py3-none-any.whl

pynode/api/services.py

@@ -0,0 +1,124 @@
+"""MQTT service management API."""
+
+from flask import Blueprint, jsonify
+
+from pynode.api.helpers import _INVALID_BODY_ERROR, _get_json_body, _json_error
+
+services_bp = Blueprint('services', __name__)
+
+
+@services_bp.route('/api/services/mqtt', methods=['GET'])
+def list_mqtt_services():
+    """List all MQTT services."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+        services = mqtt_manager.list_services()
+        return jsonify({'success': True, 'services': services})
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+
+
+@services_bp.route('/api/services/mqtt', methods=['POST'])
+def create_mqtt_service():
+    """Create a new MQTT service."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+        data = _get_json_body()
+        if data is None:
+            return _json_error(_INVALID_BODY_ERROR, 400)
+
+        # Validate required fields
+        if not data.get('name'):
+            return jsonify({'success': False, 'error': 'Service name is required'}), 400
+        if not data.get('broker'):
+            return jsonify({'success': False, 'error': 'Broker address is required'}), 400
+
+        service = mqtt_manager.create_service(data)
+        return jsonify({
+            'success': True,
+            'service': {
+                'id': service.id,
+                'name': service.name,
+                'broker': service.broker,
+                'port': service.port
+            }
+        })
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+
+
+@services_bp.route('/api/services/mqtt/<service_id>', methods=['GET'])
+def get_mqtt_service(service_id):
+    """Get a specific MQTT service by ID."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+        service = mqtt_manager.get_service(service_id)
+        if not service:
+            return jsonify({'success': False, 'error': 'Service not found'}), 404
+
+        return jsonify({
+            'success': True,
+            'service': service.to_dict()
+        })
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+
+
+@services_bp.route('/api/services/mqtt/<service_id>', methods=['PUT'])
+def update_mqtt_service(service_id):
+    """Update an existing MQTT service."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+        data = _get_json_body()
+        if data is None:
+            return _json_error(_INVALID_BODY_ERROR, 400)
+
+        service = mqtt_manager.update_service(service_id, data)
+        if not service:
+            return jsonify({'success': False, 'error': 'Service not found'}), 404
+
+        return jsonify({
+            'success': True,
+            'service': service.to_dict()
+        })
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+
+
+@services_bp.route('/api/services/mqtt/<service_id>', methods=['DELETE'])
+def delete_mqtt_service(service_id):
+    """Delete an MQTT service."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+
+        if not mqtt_manager.delete_service(service_id):
+            return jsonify({
+                'success': False,
+                'error': 'Service not found or still in use'
+            }), 400
+
+        return jsonify({'success': True})
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+
+
+@services_bp.route('/api/services/mqtt/<service_id>/test', methods=['POST'])
+def test_mqtt_service(service_id):
+    """Test connection to an MQTT service."""
+    try:
+        from pynode.nodes.MQTTNode.mqtt_service import mqtt_manager
+
+        service = mqtt_manager.get_service(service_id)
+        if not service:
+            return jsonify({'success': False, 'error': 'Service not found'}), 404
+
+        # Try to connect
+        connected = service.connect()
+
+        return jsonify({
+            'success': True,
+            'connected': connected,
+            'status': 'connected' if connected else 'failed'
+        })
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500

pynode/api/sse.py

@@ -0,0 +1,58 @@
+"""Server-Sent Events debug stream.
+
+The broadcast worker itself lives on the WorkflowManager
+(``manager.start_debug_broadcast()`` / ``manager.stop_debug_broadcast()``) so
+each app instance has its own worker thread and client queues.
+"""
+
+import json
+import logging
+import queue
+
+from flask import Blueprint, Response, stream_with_context
+
+from pynode.api.helpers import _get_manager
+
+logger = logging.getLogger(__name__)
+
+sse_bp = Blueprint('sse', __name__)
+
+
+@sse_bp.route('/api/debug/stream')
+def debug_stream():
+    """Server-Sent Events stream for debug messages."""
+    manager = _get_manager()
+
+    def generate():
+        # Create a queue for this client
+        q = queue.Queue(maxsize=100)
+        client_id = id(q)
+        with manager.clients_lock:
+            manager.debug_message_queues[client_id] = q
+
+        # Start broadcast thread if not running
+        manager.start_debug_broadcast()
+
+        try:
+            yield 'data: {"type": "connected"}\n\n'
+
+            while True:
+                try:
+                    # Wait for data from broadcast thread
+                    data = q.get(timeout=1.0)
+                    yield f'data: {json.dumps(data)}\n\n'
+                except queue.Empty:
+                    # Send keepalive
+                    yield 'data: {"type": "keepalive"}\n\n'
+
+        except GeneratorExit:
+            # Client disconnected
+            with manager.clients_lock:
+                manager.debug_message_queues.pop(client_id, None)
+        except Exception as e:
+            # Log error and close connection
+            logger.error(f"SSE Error: {e}")
+            with manager.clients_lock:
+                manager.debug_message_queues.pop(client_id, None)
+
+    return Response(stream_with_context(generate()), mimetype='text/event-stream')

pynode/api/uploads.py

@@ -0,0 +1,59 @@
+"""File upload API."""
+
+import os
+
+from flask import Blueprint, jsonify, request
+
+from pynode.api.helpers import _get_manager
+
+uploads_bp = Blueprint('uploads', __name__)
+
+# File upload configuration: uploads may only land in these subdirectories
+# of the manager's upload_base_dir. Anything else is rejected.
+ALLOWED_UPLOAD_SUBDIRS = ('models', 'uploads')
+
+
+@uploads_bp.route('/api/upload/file', methods=['POST'])
+def upload_file():
+    """Upload a file (model, video, etc.) and save to the server."""
+    try:
+        if 'file' not in request.files:
+            return jsonify({'success': False, 'error': 'No file provided'}), 400
+
+        file = request.files['file']
+        if not file.filename or file.filename == '':
+            return jsonify({'success': False, 'error': 'No file selected'}), 400
+
+        # Determine upload directory from optional 'directory' field, default to models.
+        # Only allowlisted subdirectory names are accepted (no separators, no
+        # traversal) and the resolved path must stay inside the upload base dir.
+        upload_subdir = request.form.get('directory', 'models')
+        normalized_subdir = os.path.normpath(upload_subdir).replace('\\', '/')
+        if normalized_subdir not in ALLOWED_UPLOAD_SUBDIRS:
+            return jsonify({
+                'success': False,
+                'error': f"Invalid upload directory. Allowed: {', '.join(ALLOWED_UPLOAD_SUBDIRS)}"
+            }), 400
+
+        base_dir = os.path.realpath(_get_manager().upload_base_dir)
+        upload_dir = os.path.realpath(os.path.join(base_dir, normalized_subdir))
+        if upload_dir != base_dir and not upload_dir.startswith(base_dir + os.sep):
+            return jsonify({
+                'success': False,
+                'error': 'Invalid upload directory: outside allowed base'
+            }), 400
+        os.makedirs(upload_dir, exist_ok=True)
+
+        # Save the file
+        filename = os.path.basename(file.filename)
+        file_path = os.path.join(upload_dir, filename)
+        file.save(file_path)
+
+        return jsonify({
+            'success': True,
+            'model_path': file_path,
+            'file_path': file_path,
+            'filename': filename
+        })
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500