additional n8n functions

websiteFunctions/dockerviews.py

@@ -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)})) 

websiteFunctions/static/websiteFunctions/DockerContainers.js

@@ -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,38 +236,55 @@ app.controller('ListDockersitecontainer', function ($scope, $http) {
             }
         };
 
-        $http.post(url, data, config).then(ListInitialData, cantLoadInitialData);
+        $http.post(url, data, config).then(
+            function(response) {
+                $scope.cyberpanelLoading = true;
+                $('#cyberpanelLoading').hide();
 
-        function ListInitialData(response) {
+                if (response.data.status === 1) {
-            $scope.cyberpanelLoading = true;
+                    new PNotify({
-            $('#cyberpanelLoading').hide();
+                        title: 'Success!',
-
+                        text: 'Container ' + action + ' successful.',
-            if (response.data.status === 1) {
+                        type: 'success'
-                new PNotify({
+                    });
-                    title: 'Success!',
+                    
-                    text: 'Container ' + action + ' successful.',
+                    // Update container status after action
-                    type: 'success'
+                    $scope.selectedContainer.status = action === 'stop' ? 'stopped' : 'running';
-                });
+                    
-                // Refresh container info after action
+                    // Refresh container info
-                $scope.Lunchcontainer($scope.selectedContainer.id);
+                    $scope.Lunchcontainer($scope.selectedContainer.id);
-            } else {
+                } else {
+                    new PNotify({
+                        title: 'Operation Failed!',
+                        text: response.data.error_message || 'An unknown error occurred.',
+                        type: 'error'
+                    });
+                }
+            },
+            function(error) {
+                $scope.cyberpanelLoading = true;
+                $('#cyberpanelLoading').hide();
+                
                 new PNotify({
                     title: 'Operation Failed!',
-                    text: response.data.error_message,
+                    text: 'Connection disrupted or server error occurred.',
                     type: 'error'
                 });
+                
+                console.error("Error during container action:", error);
             }
-        }
+        );
+    };
 
-        function cantLoadInitialData(response) {
+    // Update the container selection when actions are triggered
-            $scope.cyberpanelLoading = true;
+    $scope.setSelectedContainer = function(container) {
-            $('#cyberpanelLoading').hide();
+        $scope.selectedContainer = container;
-            new PNotify({
+    };
-                title: 'Operation Failed!',
+
-                text: 'Connection disrupted, refresh the page.',
+    // Update the button click handlers to set selected container
-                type: 'error'
+    $scope.handleAction = function(action, container) {
-            });
+        $scope.setSelectedContainer(container);
-        }
+        $scope.cAction(action);
     };
 
     $scope.openSettings = function(container) {

websiteFunctions/templates/websiteFunctions/DockerSiteHome.html

@@ -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">

websiteFunctions/urls.py

@@ -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'),

websiteFunctions/views.py

@@ -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']