additional n8n functions

This commit is contained in:
usmannasir
2025-04-11 16:04:45 +05:00
parent 9c29ec8599
commit 54e793cfab
5 changed files with 164 additions and 33 deletions

View File

@@ -0,0 +1,107 @@
import json
import docker
from django.http import HttpResponse
from .models import DockerSites
from .website import ACLManager
from django.shortcuts import redirect
from plogical.CyberCPLogFileWriter import CyberCPLogFileWriter as logging
def loadLoginPage(request):
return redirect('/login')
class DockerManager:
def __init__(self):
self.client = docker.from_env()
def get_container(self, container_id):
try:
return self.client.containers.get(container_id)
except docker.errors.NotFound:
return None
except Exception as e:
logging.writeToFile(f"Error getting container {container_id}: {str(e)}")
return None
def require_login(view_func):
def wrapper(request, *args, **kwargs):
try:
userID = request.session['userID']
return view_func(request, userID, *args, **kwargs)
except KeyError:
return redirect(loadLoginPage)
return wrapper
@require_login
def startContainer(request, userID):
try:
data = json.loads(request.body)
container_id = data.get('container_id')
site_name = data.get('name')
# Verify ownership
docker_site = DockerSites.objects.get(SiteName=site_name)
if not ACLManager.checkOwnership(docker_site.admin.domain, userID):
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Unauthorized access'}))
# Get and start container
docker_manager = DockerManager()
container = docker_manager.get_container(container_id)
if not container:
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Container not found'}))
container.start()
return HttpResponse(json.dumps({'status': 1}))
except Exception as e:
return HttpResponse(json.dumps({'status': 0, 'error_message': str(e)}))
@require_login
def stopContainer(request, userID):
try:
data = json.loads(request.body)
container_id = data.get('container_id')
site_name = data.get('name')
# Verify ownership
docker_site = DockerSites.objects.get(SiteName=site_name)
if not ACLManager.checkOwnership(docker_site.admin.domain, userID):
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Unauthorized access'}))
# Get and stop container
docker_manager = DockerManager()
container = docker_manager.get_container(container_id)
if not container:
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Container not found'}))
container.stop()
return HttpResponse(json.dumps({'status': 1}))
except Exception as e:
return HttpResponse(json.dumps({'status': 0, 'error_message': str(e)}))
@require_login
def restartContainer(request, userID):
try:
data = json.loads(request.body)
container_id = data.get('container_id')
site_name = data.get('name')
# Verify ownership
docker_site = DockerSites.objects.get(SiteName=site_name)
if not ACLManager.checkOwnership(docker_site.admin.domain, userID):
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Unauthorized access'}))
# Get and restart container
docker_manager = DockerManager()
container = docker_manager.get_container(container_id)
if not container:
return HttpResponse(json.dumps({'status': 0, 'error_message': 'Container not found'}))
container.restart()
return HttpResponse(json.dumps({'status': 1}))
except Exception as e:
return HttpResponse(json.dumps({'status': 0, 'error_message': str(e)}))

View File

@@ -209,24 +209,25 @@ app.controller('ListDockersitecontainer', function ($scope, $http) {
$scope.cyberpanelLoading = false;
$('#cyberpanelLoading').show();
var url;
var url = "/docker/";
switch(action) {
case 'start':
url = "/docker/StartContainerAPP";
url += "startContainer";
break;
case 'stop':
url = "/docker/StopContainerAPP";
url += "stopContainer";
break;
case 'restart':
url = "/docker/RestartContainerAPP";
url += "restartContainer";
break;
default:
console.error("Unknown action:", action);
return;
}
var data = {
'name': $('#sitename').html(),
'id': $scope.selectedContainer.id
'container_id': $scope.selectedContainer.id
};
var config = {
@@ -235,9 +236,8 @@ app.controller('ListDockersitecontainer', function ($scope, $http) {
}
};
$http.post(url, data, config).then(ListInitialData, cantLoadInitialData);
function ListInitialData(response) {
$http.post(url, data, config).then(
function(response) {
$scope.cyberpanelLoading = true;
$('#cyberpanelLoading').hide();
@@ -247,26 +247,44 @@ app.controller('ListDockersitecontainer', function ($scope, $http) {
text: 'Container ' + action + ' successful.',
type: 'success'
});
// Refresh container info after action
// Update container status after action
$scope.selectedContainer.status = action === 'stop' ? 'stopped' : 'running';
// Refresh container info
$scope.Lunchcontainer($scope.selectedContainer.id);
} else {
new PNotify({
title: 'Operation Failed!',
text: response.data.error_message,
text: response.data.error_message || 'An unknown error occurred.',
type: 'error'
});
}
}
function cantLoadInitialData(response) {
},
function(error) {
$scope.cyberpanelLoading = true;
$('#cyberpanelLoading').hide();
new PNotify({
title: 'Operation Failed!',
text: 'Connection disrupted, refresh the page.',
text: 'Connection disrupted or server error occurred.',
type: 'error'
});
console.error("Error during container action:", error);
}
);
};
// Update the container selection when actions are triggered
$scope.setSelectedContainer = function(container) {
$scope.selectedContainer = container;
};
// Update the button click handlers to set selected container
$scope.handleAction = function(action, container) {
$scope.setSelectedContainer(container);
$scope.cAction(action);
};
$scope.openSettings = function(container) {

View File

@@ -403,13 +403,13 @@
<h2 id="domainNamePage" class="d-flex justify-content-between align-items-center">
<span>{% trans "Currently managing: " %} {$ web.name $}</span>
<div class="btn-group" role="group" aria-label="Container Actions">
<button class="btn btn-success btn-sm" ng-click="cAction('start')" ng-if="web.status !== 'running'" title="Start Container">
<button class="btn btn-success btn-sm" ng-click="handleAction('start', web)" ng-if="web.status !== 'running'" title="Start Container">
<i class="fa fa-play"></i> Start
</button>
<button class="btn btn-warning btn-sm" ng-click="cAction('restart')" ng-if="web.status === 'running'" title="Restart Container">
<button class="btn btn-warning btn-sm" ng-click="handleAction('restart', web)" ng-if="web.status === 'running'" title="Restart Container">
<i class="fa fa-refresh"></i> Restart
</button>
<button class="btn btn-danger btn-sm" ng-click="cAction('stop')" ng-if="web.status === 'running'" title="Stop Container">
<button class="btn btn-danger btn-sm" ng-click="handleAction('stop', web)" ng-if="web.status === 'running'" title="Stop Container">
<i class="fa fa-stop"></i> Stop
</button>
<button class="btn btn-primary btn-sm" ng-click="openSettings(web)" title="Container Settings">

View File

@@ -179,6 +179,11 @@ urlpatterns = [
path('ListDockerSites', views.ListDockerSites, name='ListDockerSites'),
path('fetchDockersite', views.fetchDockersite, name='fetchDockersite'),
# Docker Container Actions
path('docker/startContainer', views.startContainer, name='startContainer'),
path('docker/stopContainer', views.stopContainer, name='stopContainer'),
path('docker/restartContainer', views.restartContainer, name='restartContainer'),
# SSH Configs
path('getSSHConfigs', views.getSSHConfigs, name='getSSHConfigs'),
path('deleteSSHKey', views.deleteSSHKey, name='deleteSSHKey'),

View File

@@ -14,6 +14,7 @@ from websiteFunctions.models import wpplugins
from websiteFunctions.website import WebsiteManager
from websiteFunctions.pluginManager import pluginManager
from django.views.decorators.csrf import csrf_exempt
from .dockerviews import startContainer, stopContainer, restartContainer
def loadWebsitesHome(request):
val = request.session['userID']