mirror of
https://github.com/jhd3197/ServerKit.git
synced 2026-11-09 05:24:49 +09:00
Juan Denis
d2d87ecc1d
Introduce a full-featured workflow automation system: add WorkflowEngine service to execute workflows, background scheduler to trigger cron workflows, and croniter dependency. Extend Workflow model with automation fields (is_active, trigger_type, trigger_config, last_run_at, last_status) and add WorkflowExecution and WorkflowLog models. Add API endpoints to start executions, list executions, fetch execution details and logs, and persist execution records/logs. Implement frontend components for workflow nodes, config panels, and an execution history UI (multiple new React components) and wire minor builder UI imports. Note: database migrations are required to add new columns and tables.
257 lines
7.8 KiB
Python
257 lines
7.8 KiB
Python
import json
|
|
from flask import Blueprint, request, jsonify
|
|
from flask_jwt_extended import jwt_required, get_jwt_identity
|
|
from app import db
|
|
from app.models import Workflow, User, WorkflowExecution, WorkflowLog
|
|
|
|
workflows_bp = Blueprint('workflows', __name__)
|
|
|
|
|
|
@workflows_bp.route('', methods=['GET'])
|
|
@jwt_required()
|
|
def get_workflows():
|
|
"""Get all workflows for the current user."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
|
|
if user.role == 'admin':
|
|
workflows = Workflow.query.all()
|
|
else:
|
|
workflows = Workflow.query.filter_by(user_id=current_user_id).all()
|
|
|
|
return jsonify({
|
|
'workflows': [w.to_dict() for w in workflows]
|
|
})
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>', methods=['GET'])
|
|
@jwt_required()
|
|
def get_workflow(workflow_id):
|
|
"""Get a single workflow by ID."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
workflow = Workflow.query.get(workflow_id)
|
|
|
|
if not workflow:
|
|
return jsonify({'error': 'Workflow not found'}), 404
|
|
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
return jsonify(workflow.to_dict())
|
|
|
|
|
|
@workflows_bp.route('', methods=['POST'])
|
|
@jwt_required()
|
|
def create_workflow():
|
|
"""Create a new workflow."""
|
|
current_user_id = get_jwt_identity()
|
|
data = request.get_json()
|
|
|
|
if not data:
|
|
return jsonify({'error': 'No data provided'}), 400
|
|
|
|
name = data.get('name')
|
|
if not name:
|
|
return jsonify({'error': 'Name is required'}), 400
|
|
|
|
workflow = Workflow(
|
|
name=name,
|
|
description=data.get('description', ''),
|
|
nodes=json.dumps(data.get('nodes', [])),
|
|
edges=json.dumps(data.get('edges', [])),
|
|
viewport=json.dumps(data.get('viewport')) if data.get('viewport') else None,
|
|
is_active=data.get('is_active', False),
|
|
trigger_type=data.get('trigger_type', 'manual'),
|
|
trigger_config=json.dumps(data.get('trigger_config', {})),
|
|
user_id=current_user_id
|
|
)
|
|
|
|
db.session.add(workflow)
|
|
db.session.commit()
|
|
|
|
return jsonify({
|
|
'message': 'Workflow created successfully',
|
|
'workflow': workflow.to_dict()
|
|
}), 201
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>', methods=['PUT'])
|
|
@jwt_required()
|
|
def update_workflow(workflow_id):
|
|
"""Update an existing workflow."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
workflow = Workflow.query.get(workflow_id)
|
|
|
|
if not workflow:
|
|
return jsonify({'error': 'Workflow not found'}), 404
|
|
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
data = request.get_json()
|
|
if not data:
|
|
return jsonify({'error': 'No data provided'}), 400
|
|
|
|
if 'name' in data:
|
|
workflow.name = data['name']
|
|
if 'description' in data:
|
|
workflow.description = data['description']
|
|
if 'nodes' in data:
|
|
workflow.nodes = json.dumps(data['nodes'])
|
|
if 'edges' in data:
|
|
workflow.edges = json.dumps(data['edges'])
|
|
if 'viewport' in data:
|
|
workflow.viewport = json.dumps(data['viewport']) if data['viewport'] else None
|
|
|
|
# Automation fields
|
|
if 'is_active' in data:
|
|
workflow.is_active = data['is_active']
|
|
if 'trigger_type' in data:
|
|
workflow.trigger_type = data['trigger_type']
|
|
if 'trigger_config' in data:
|
|
workflow.trigger_config = json.dumps(data['trigger_config'])
|
|
|
|
db.session.commit()
|
|
|
|
return jsonify({
|
|
'message': 'Workflow updated successfully',
|
|
'workflow': workflow.to_dict()
|
|
})
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>', methods=['DELETE'])
|
|
@jwt_required()
|
|
def delete_workflow(workflow_id):
|
|
"""Delete a workflow."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
workflow = Workflow.query.get(workflow_id)
|
|
|
|
if not workflow:
|
|
return jsonify({'error': 'Workflow not found'}), 404
|
|
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
db.session.delete(workflow)
|
|
db.session.commit()
|
|
|
|
return jsonify({'message': 'Workflow deleted successfully'})
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>/execute', methods=['POST'])
|
|
@jwt_required()
|
|
def execute_workflow(workflow_id):
|
|
"""Trigger a workflow execution manually."""
|
|
from app.services.workflow_engine import WorkflowEngine
|
|
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
workflow = Workflow.query.get(workflow_id)
|
|
|
|
if not workflow:
|
|
return jsonify({'error': 'Workflow not found'}), 404
|
|
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
data = request.get_json() or {}
|
|
context = data.get('context', {})
|
|
|
|
execution_id = WorkflowEngine.execute_workflow(
|
|
workflow_id=workflow_id,
|
|
trigger_type='manual',
|
|
context=context
|
|
)
|
|
|
|
return jsonify({
|
|
'message': 'Workflow execution started',
|
|
'execution_id': execution_id
|
|
})
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>/executions', methods=['GET'])
|
|
@jwt_required()
|
|
def get_workflow_executions(workflow_id):
|
|
"""Get execution history for a workflow."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
workflow = Workflow.query.get(workflow_id)
|
|
|
|
if not workflow:
|
|
return jsonify({'error': 'Workflow not found'}), 404
|
|
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
executions = WorkflowExecution.query.filter_by(workflow_id=workflow_id).order_by(WorkflowExecution.started_at.desc()).all()
|
|
|
|
return jsonify({
|
|
'executions': [e.to_dict() for e in executions]
|
|
})
|
|
|
|
|
|
@workflows_bp.route('/executions/<int:execution_id>', methods=['GET'])
|
|
@jwt_required()
|
|
def get_execution_details(execution_id):
|
|
"""Get details for a specific execution."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
execution = WorkflowExecution.query.get(execution_id)
|
|
|
|
if not execution:
|
|
return jsonify({'error': 'Execution not found'}), 404
|
|
|
|
workflow = execution.workflow
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
return jsonify(execution.to_dict())
|
|
|
|
|
|
@workflows_bp.route('/executions/<int:execution_id>/logs', methods=['GET'])
|
|
@jwt_required()
|
|
def get_execution_logs(execution_id):
|
|
"""Get logs for a specific execution."""
|
|
current_user_id = get_jwt_identity()
|
|
user = User.query.get(current_user_id)
|
|
execution = WorkflowExecution.query.get(execution_id)
|
|
|
|
if not execution:
|
|
return jsonify({'error': 'Execution not found'}), 404
|
|
|
|
workflow = execution.workflow
|
|
if user.role != 'admin' and workflow.user_id != current_user_id:
|
|
return jsonify({'error': 'Access denied'}), 403
|
|
|
|
logs = WorkflowLog.query.filter_by(execution_id=execution_id).order_by(WorkflowLog.timestamp.asc()).all()
|
|
|
|
return jsonify({
|
|
'logs': [l.to_dict() for l in logs]
|
|
})
|
|
|
|
|
|
@workflows_bp.route('/<int:workflow_id>/deploy', methods=['POST'])
|
|
@jwt_required()
|
|
def deploy_workflow(workflow_id):
|
|
"""
|
|
Deploy all resources from a workflow.
|
|
"""
|
|
from app.services.workflow_service import WorkflowService
|
|
|
|
current_user_id = get_jwt_identity()
|
|
|
|
result = WorkflowService.deploy_workflow(workflow_id, current_user_id)
|
|
|
|
if result.get('success'):
|
|
return jsonify(result), 200
|
|
elif result.get('error') == 'Workflow not found':
|
|
return jsonify(result), 404
|
|
elif result.get('error') == 'Access denied':
|
|
return jsonify(result), 403
|
|
else:
|
|
# Partial success or errors - return 200 with error details
|
|
return jsonify(result), 200
|