From 08d7dbbe4eb2e88d93f803f387438faef68e601d Mon Sep 17 00:00:00 2001
From: Yann Stepienik <azukaar@gmail.com>
Date: Mon, 8 May 2023 12:38:27 +0100
Subject: [PATCH] [release] v0.4.0-unstable6

---
 client/src/index.css                          |  4 ++
 client/src/pages/servapps/actionBar.jsx       | 12 ++++--
 .../src/pages/servapps/containers/network.jsx | 43 +++++++++++++------
 .../src/pages/servapps/containers/setup.jsx   |  3 +-
 .../src/pages/servapps/containers/volumes.jsx | 13 +++---
 client/src/pages/servapps/networks.jsx        |  3 ++
 client/src/pages/servapps/servapps.jsx        |  2 +-
 client/src/pages/servapps/volumes.jsx         |  3 ++
 package.json                                  |  2 +-
 src/docker/api_managecont.go                  |  8 ++++
 src/docker/api_networks.go                    |  7 +++
 src/docker/api_secureContainer.go             |  7 +++
 src/docker/api_updateContainer.go             |  7 +++
 src/index.go                                  |  8 ++--
 14 files changed, 93 insertions(+), 29 deletions(-)

diff --git a/client/src/index.css b/client/src/index.css
index a2edbe1..4ddf7cd 100644
--- a/client/src/index.css
+++ b/client/src/index.css
@@ -65,4 +65,8 @@
 
 .darken {
   filter: brightness(0.5);
+}
+
+.MuiAlert-icon {
+  align-items: center;
 }
\ No newline at end of file
diff --git a/client/src/pages/servapps/actionBar.jsx b/client/src/pages/servapps/actionBar.jsx
index a213e35..adedec6 100644
--- a/client/src/pages/servapps/actionBar.jsx
+++ b/client/src/pages/servapps/actionBar.jsx
@@ -1,6 +1,6 @@
 import React from 'react';
 import { IconButton, Tooltip } from '@mui/material';
-import { CloseSquareOutlined, DeleteOutlined, PauseCircleOutlined, PlaySquareOutlined, ReloadOutlined, RollbackOutlined, StopOutlined, UpCircleOutlined } from '@ant-design/icons';
+import { CheckCircleOutlined, CloseSquareOutlined, DeleteOutlined, PauseCircleOutlined, PlaySquareOutlined, ReloadOutlined, RollbackOutlined, StopOutlined, UpCircleOutlined } from '@ant-design/icons';
 import * as API from '../../api';
 
 const GetActions = ({
@@ -9,6 +9,8 @@ const GetActions = ({
   refreshServeApps,
   setIsUpdatingId
 }) => {
+  const [confirmDelete, setConfirmDelete] = React.useState(false);
+
   const doTo = (action) => {
     setIsUpdatingId(Id, true);
     API.docker.manageContainer(Id, action).then((res) => {
@@ -76,8 +78,12 @@ const GetActions = ({
     {
       t: 'Delete',
       if: ['exited', 'created'],
-      e: <IconButton onClick={() => {doTo('remove')}} color="error" size='large'>
-        <DeleteOutlined />
+      e: <IconButton onClick={() => {
+        if(confirmDelete) doTo('remove')
+        else setConfirmDelete(true);
+      }} color="error" size='large'>
+        {confirmDelete ?  <CheckCircleOutlined />
+        : <DeleteOutlined />}
       </IconButton>
     }
   ];
diff --git a/client/src/pages/servapps/containers/network.jsx b/client/src/pages/servapps/containers/network.jsx
index 4d78cab..1b85e44 100644
--- a/client/src/pages/servapps/containers/network.jsx
+++ b/client/src/pages/servapps/containers/network.jsx
@@ -1,6 +1,6 @@
 import React from 'react';
 import { Formik } from 'formik';
-import { Button, Stack, Grid, MenuItem, TextField, IconButton, FormHelperText, CircularProgress, useTheme } from '@mui/material';
+import { Button, Stack, Grid, MenuItem, TextField, IconButton, FormHelperText, CircularProgress, useTheme, Alert } from '@mui/material';
 import MainCard from '../../../components/MainCard';
 import { CosmosCheckbox, CosmosFormDivider, CosmosInputText, CosmosSelect }
   from '../../config/users/formShortcuts';
@@ -12,13 +12,6 @@ import { NetworksColumns } from '../networks';
 import NewNetworkButton from '../createNetwork';
 
 const NetworkContainerSetup = ({ config, containerInfo, refresh }) => {
-  const restartPolicies = [
-    ['no', 'No Restart'],
-    ['always', 'Always Restart'],
-    ['on-failure', 'Restart On Failure'],
-    ['unless-stopped', 'Restart Unless Stopped'],
-  ];
-
   const [networks, setNetworks] = React.useState([]);
   const theme = useTheme();
   const isDark = theme.palette.mode === 'dark';
@@ -40,13 +33,13 @@ const NetworkContainerSetup = ({ config, containerInfo, refresh }) => {
 
   const connect = (network) => {
     setNetworks(null);
-    return API.docker.attachNetwork(containerInfo.Id, network).then(() => {
+    return API.docker.attachNetwork(containerInfo.Name.replace('/', ''), network).then(() => {
       refreshAll();
     });
   }
 
   const disconnect = (network) => {
-    return API.docker.detachNetwork(containerInfo.Id, network).then(() => {
+    return API.docker.detachNetwork(containerInfo.Name.replace('/', ''), network).then(() => {
       refreshAll();
     });
   }
@@ -94,18 +87,20 @@ const NetworkContainerSetup = ({ config, containerInfo, refresh }) => {
             .then((res) => {
               setStatus({ success: true });
               setSubmitting(false);
+              refresh && refresh();
             }
             ).catch((err) => {
               setStatus({ success: false });
               setErrors({ submit: err.message });
               setSubmitting(false);
+              refresh && refresh();
             });
         }}
       >
         {(formik) => (
           <form noValidate onSubmit={formik.handleSubmit}>
             <Stack spacing={2}>
-              <MainCard title={'Exposed Ports'}>
+              <MainCard title={'Ports'}>
                 <Grid container spacing={4}>
                   <Grid item xs={12}>
                     {formik.values.ports.map((port, idx) => (
@@ -209,7 +204,30 @@ const NetworkContainerSetup = ({ config, containerInfo, refresh }) => {
                   </Grid>
                 </Grid>
               </MainCard>
-              <MainCard title={'Connected Networks'}>
+              <MainCard title={'Networks'}>
+                <Stack spacing={2}>
+                {networks && <Stack spacing={2}>
+                  {Object.keys(containerInfo.NetworkSettings.Networks).map((networkName) => {
+                    const network = networks.find((n) => n.Name === networkName);
+                    if (!network) {
+                      return <Alert severity="error">
+                        You are connected to a network that has been removed: <strong>{networkName}</strong>. 
+                        Either re-create it or
+                        <Button
+                          style={{ marginLeft: '10px' }}
+                          variant="outlined"
+                          color="primary"
+                          onClick={() => {
+                            disconnect(networkName);
+                          }}
+                        >
+                          Disconnect It
+                        </Button>
+                      </Alert>
+                    }
+                  })}
+                </Stack>}
+
                 {networks && <PrettyTableView
                   data={networks}
                   buttons={[
@@ -252,6 +270,7 @@ const NetworkContainerSetup = ({ config, containerInfo, refresh }) => {
                   </center>
                 </div>
                 )}
+                </Stack>
               </MainCard>
             </Stack>
           </form>
diff --git a/client/src/pages/servapps/containers/setup.jsx b/client/src/pages/servapps/containers/setup.jsx
index 876fe1e..6f5784a 100644
--- a/client/src/pages/servapps/containers/setup.jsx
+++ b/client/src/pages/servapps/containers/setup.jsx
@@ -8,7 +8,7 @@ import { DeleteOutlined, PlusCircleOutlined } from '@ant-design/icons';
 import * as API from '../../../api';
 import { LoadingButton } from '@mui/lab';
 
-const DockerContainerSetup = ({config, containerInfo}) => {
+const DockerContainerSetup = ({config, containerInfo, refresh}) => {
   const restartPolicies = [
     ['no', 'No Restart'],
     ['always', 'Always Restart'],
@@ -66,6 +66,7 @@ const DockerContainerSetup = ({config, containerInfo}) => {
             .then((res) => {
               setStatus({ success: true });
               setSubmitting(false);
+              refresh && refresh();
             }
           ).catch((err) => {
             setStatus({ success: false });
diff --git a/client/src/pages/servapps/containers/volumes.jsx b/client/src/pages/servapps/containers/volumes.jsx
index d5c40eb..ea74609 100644
--- a/client/src/pages/servapps/containers/volumes.jsx
+++ b/client/src/pages/servapps/containers/volumes.jsx
@@ -24,16 +24,16 @@ const VolumeContainerSetup = ({ config, containerInfo, refresh }) => {
   const isDark = theme.palette.mode === 'dark';
 
   React.useEffect(() => {
-    API.docker.networkList().then((res) => {
-      setVolumes(res.data);
+    API.docker.volumeList().then((res) => {
+      setVolumes(res.data.Volumes);
     });
   }, []);
 
   const refreshAll = () => {
     setVolumes(null);
     refresh().then(() => {
-      API.docker.networkList().then((res) => {
-        setVolumes(res.data);
+      API.docker.volumeList().then((res) => {
+        setVolumes(res.data.Volumes);
       });
     });
   };
@@ -75,11 +75,13 @@ const VolumeContainerSetup = ({ config, containerInfo, refresh }) => {
             .then((res) => {
               setStatus({ success: true });
               setSubmitting(false);
+              refresh && refresh();
             }
             ).catch((err) => {
               setStatus({ success: false });
               setErrors({ submit: err.message });
               setSubmitting(false);
+              refresh && refresh();
             });
         }}
       >
@@ -193,7 +195,6 @@ const VolumeContainerSetup = ({ config, containerInfo, refresh }) => {
                           {
                             title: '',
                             field: (r) => {
-                              console.log(r);
                               return (<Button
                                 variant="outlined"
                                 color="primary"
@@ -208,7 +209,7 @@ const VolumeContainerSetup = ({ config, containerInfo, refresh }) => {
                           }
                         ]}
                       />}
-                      {!volumes && (<div style={{ height: '550px' }}>
+                      {!volumes && (<div style={{ height: '100px' }}>
                         <center>
                           <br />
                           <CircularProgress />
diff --git a/client/src/pages/servapps/networks.jsx b/client/src/pages/servapps/networks.jsx
index 8c01191..c7b448f 100644
--- a/client/src/pages/servapps/networks.jsx
+++ b/client/src/pages/servapps/networks.jsx
@@ -106,6 +106,9 @@ const NetworkManagementList = () => {
                         API.docker.networkDelete(r.Id).then(() => {
                           refresh();
                           setIsLoading(false);
+                        }).catch(() => {
+                          setIsLoading(false);
+                          refresh();
                         });
                       } else {
                         setTryDelete(r.Id);
diff --git a/client/src/pages/servapps/servapps.jsx b/client/src/pages/servapps/servapps.jsx
index 32fad95..220bb4b 100644
--- a/client/src/pages/servapps/servapps.jsx
+++ b/client/src/pages/servapps/servapps.jsx
@@ -152,7 +152,7 @@ const ServeApps = () => {
 
       <Grid2 container spacing={2}>
         {serveApps && serveApps.filter(app => search.length < 2 || app.Names[0].toLowerCase().includes(search.toLowerCase())).map((app) => {
-          return <Grid2 style={gridAnim} xs={12} sm={6} md={6} lg={6} xl={4}>
+          return <Grid2 style={gridAnim} xs={12} sm={6} md={6} lg={6} xl={4} key={app.Id} item>
             <Item>
             <Stack justifyContent='space-around' direction="column" spacing={2} padding={2} divider={<Divider orientation="horizontal" flexItem />}>
               <Stack direction="column" spacing={0} alignItems="flex-start">
diff --git a/client/src/pages/servapps/volumes.jsx b/client/src/pages/servapps/volumes.jsx
index 8aef397..5161aa9 100644
--- a/client/src/pages/servapps/volumes.jsx
+++ b/client/src/pages/servapps/volumes.jsx
@@ -99,6 +99,9 @@ const VolumeManagementList = () => {
                                             API.docker.volumeDelete(r.Name).then(() => {
                                                 refresh();
                                                 setIsLoading(false);
+                                            }).catch(() => {
+                                                setIsLoading(false);
+                                                refresh();
                                             });
                                           } else {
                                             setTryDelete(r.Name);
diff --git a/package.json b/package.json
index e5118de..dcf6804 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "cosmos-server",
-  "version": "0.4.0-unstable5",
+  "version": "0.4.0-unstable6",
   "description": "",
   "main": "test-server.js",
   "bugs": {
diff --git a/src/docker/api_managecont.go b/src/docker/api_managecont.go
index d03c0e8..d963a74 100644
--- a/src/docker/api_managecont.go
+++ b/src/docker/api_managecont.go
@@ -25,10 +25,18 @@ func ManageContainerRoute(w http.ResponseWriter, req *http.Request) {
 		return
 	}
 
+
+
 	vars := mux.Vars(req)
 	containerName := utils.Sanitize(vars["containerId"])
 	// stop, start, restart, kill, remove, pause, unpause, recreate
 	action := utils.Sanitize(vars["action"])
+	
+	if os.Getenv("HOSTNAME") != "" && (action == "recreate" || action == "update") && containerName == os.Getenv("HOSTNAME") {
+		utils.Error("ManageContainer - cannot recreate/update self", nil)
+		utils.HTTPError(w, "Cannot recreate/update self", http.StatusBadRequest, "DS003")
+		return
+	}
 
 	utils.Log("ManageContainer " + containerName)
 
diff --git a/src/docker/api_networks.go b/src/docker/api_networks.go
index d2cb556..e254760 100644
--- a/src/docker/api_networks.go
+++ b/src/docker/api_networks.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"net/http"
+	"os"
 
 	"github.com/gorilla/mux"
 	"github.com/azukaar/cosmos-server/src/utils"
@@ -157,6 +158,12 @@ func DetachNetwork(w http.ResponseWriter, req *http.Request) {
 			return
 		}
 
+		if os.Getenv("HOSTNAME") != "" && networkID == "bridge" && containerID == os.Getenv("HOSTNAME") {
+			utils.Error("DetachNetwork - Cannot disconnect self from bridge", nil)
+			utils.HTTPError(w, "Cannot disconnect self from bridge", http.StatusBadRequest, "DS003")
+			return
+		}
+
 		err := DockerClient.NetworkDisconnect(context.Background(), networkID, containerID, true)
 		if err != nil {
 			utils.Error("DetachNetwork: Error while detaching network", err)
diff --git a/src/docker/api_secureContainer.go b/src/docker/api_secureContainer.go
index 6328204..b654bd8 100644
--- a/src/docker/api_secureContainer.go
+++ b/src/docker/api_secureContainer.go
@@ -3,6 +3,7 @@ package docker
 import (
 	"net/http"
 	"encoding/json"
+	"os"
 
 	"github.com/azukaar/cosmos-server/src/utils" 
 	
@@ -18,6 +19,12 @@ func SecureContainerRoute(w http.ResponseWriter, req *http.Request) {
 	containerName := utils.Sanitize(vars["containerId"])
 	status := utils.Sanitize(vars["status"])
 	
+	if os.Getenv("HOSTNAME") != "" && containerName == os.Getenv("HOSTNAME") {
+		utils.Error("SecureContainerRoute - Cannot force secure self", nil)
+		utils.HTTPError(w, "Cannot force secure self", http.StatusBadRequest, "DS003")
+		return
+	}
+	
 	if(req.Method == "GET") {
 		container, err := DockerClient.ContainerInspect(DockerContext, containerName)
 		if err != nil {
diff --git a/src/docker/api_updateContainer.go b/src/docker/api_updateContainer.go
index bb13796..3056146 100644
--- a/src/docker/api_updateContainer.go
+++ b/src/docker/api_updateContainer.go
@@ -3,6 +3,7 @@ package docker
 import (
 	"encoding/json"
 	"net/http"
+	"os"
 
 	"github.com/azukaar/cosmos-server/src/utils"
 	containerType "github.com/docker/docker/api/types/container"
@@ -37,6 +38,12 @@ func UpdateContainerRoute(w http.ResponseWriter, req *http.Request) {
 
 		vars := mux.Vars(req)
 		containerName := utils.Sanitize(vars["containerId"])
+		
+		if os.Getenv("HOSTNAME") != "" && containerName == os.Getenv("HOSTNAME") {
+			utils.Error("SecureContainerRoute - Cannot update self", nil)
+			utils.HTTPError(w, "Cannot update self", http.StatusBadRequest, "DS003")
+			return
+		}
 
 		container, err := DockerClient.ContainerInspect(DockerContext, containerName)
 		if err != nil {
diff --git a/src/index.go b/src/index.go
index 8fdd6f0..758a212 100644
--- a/src/index.go
+++ b/src/index.go
@@ -25,11 +25,9 @@ func main() {
 	docker.BootstrapAllContainersFromTags()
 
 	version, err := docker.DockerClient.ServerVersion(context.Background())
-	if err != nil {
-			panic(err)
+	if err == nil {
+		utils.Log("Docker API version: " + version.APIVersion)
 	}
-
-	utils.Log("Docker API version: " + version.APIVersion)
-
+	
 	StartServer()
 }