Merge branch 'master' into add-git-username-env-variable

litmuschaos · Oct 9, 2024 · c86b8c9 · c86b8c9
2 parents 7b6b9b0 + c8e145c
commit c86b8c9
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 9 deletions.
diff --git a/chaoscenter/web/src/views/ExperimentBuilderTemplateSelection/blankCanvasTemplate.ts b/chaoscenter/web/src/views/ExperimentBuilderTemplateSelection/blankCanvasTemplate.ts
@@ -10,7 +10,12 @@ export default function blankCanvasTemplate(
 
   switch (infrastructureType) {
     case InfrastructureType.KUBERNETES:
-      return kubernetesBlankCanvasTemplate(experimentName, experiment?.chaosInfrastructure?.namespace);
+      return kubernetesBlankCanvasTemplate(
+        experimentName, 
+        experiment?.chaosInfrastructure?.namespace,
+        undefined, 
+        experiment?.imageRegistry,
+      );
   }
 }
 

diff --git a/chaoscenter/web/src/views/ExperimentDashboardV2/ExperimentDashboardV2Table.tsx b/chaoscenter/web/src/views/ExperimentDashboardV2/ExperimentDashboardV2Table.tsx
@@ -215,10 +215,10 @@ const ExperimentDashboardV2Table = ({
                 );
               case ExperimentRunStatus.QUEUED:
                 return (
-                  <RunExperimentButton
-                    buttonProps={{ disabled: true }}
+                  <StopExperimentButton
                     experimentID={data.experimentID}
                     refetchExperiments={refetchExperiments}
+                    infrastructureType={InfrastructureType.KUBERNETES}
                   />
                 );
               default:

diff --git a/chaoscenter/web/src/views/ExperimentDashboardV2/ExperimentDashboardV2TableMenu.tsx b/chaoscenter/web/src/views/ExperimentDashboardV2/ExperimentDashboardV2TableMenu.tsx
@@ -33,7 +33,9 @@ export const MenuCell = ({
     onError: error => showError(error.message)
   });
 
-  const lastExperimentRunStatus = data.recentExecutions[0]?.experimentRunStatus;
+  const isDeleteButtonEnabled = !data.recentExecutions.some(
+    execution => execution?.experimentRunStatus === ExperimentRunStatus.RUNNING
+  );
 
   // <!-- confirmation dialog boxes -->
   const confirmationDialogProps = {
@@ -108,10 +110,7 @@ export const MenuCell = ({
             icon="main-trash"
             text={getString('deleteExperiment')}
             onClick={openDeleteDialog}
-            disabled={
-              lastExperimentRunStatus === ExperimentRunStatus.RUNNING ||
-              lastExperimentRunStatus === ExperimentRunStatus.QUEUED
-            }
+            disabled={isDeleteButtonEnabled == false}
             permission={PermissionGroup.OWNER}
           />
         </Menu>

diff --git a/chaoscenter/web/src/views/StudioOverview/StudioOverview.tsx b/chaoscenter/web/src/views/StudioOverview/StudioOverview.tsx
@@ -23,6 +23,8 @@ import { ChaosInfrastructureReferenceFieldProps, StudioErrorState, StudioTabs }
 import experimentYamlService from 'services/experiment';
 import KubernetesChaosInfrastructureReferenceFieldController from '@controllers/KubernetesChaosInfrastructureReferenceField';
 import { InfrastructureType } from '@api/entities';
+import { getImageRegistry } from '@api/core/ImageRegistry'; 
+import { getScope } from '@utils'; 
 import css from './StudioOverview.module.scss';
 
 interface StudioOverviewViewProps {
@@ -52,6 +54,20 @@ export default function StudioOverviewView({
 
   const [currentExperiment, setCurrentExperiment] = React.useState<ExperimentMetadata | undefined>();
 
+  const scope = getScope();
+
+   // Fetch the image registry data using Apollo's useQuery hook
+   const { data: getImageRegistryData, loading: imageRegistryLoading } = getImageRegistry({
+    projectID: scope.projectID,
+  });
+
+  const imageRegistry = getImageRegistryData?.getImageRegistry?{
+      name: getImageRegistryData.getImageRegistry.imageRegistryInfo.imageRegistryName,
+      repo: getImageRegistryData.getImageRegistry.imageRegistryInfo.imageRepoName,
+      secret: getImageRegistryData.getImageRegistry.imageRegistryInfo.secretName,
+  }
+  : undefined;
+
   React.useEffect(() => {
     experimentHandler?.getExperiment(experimentKey).then(experiment => {
       delete experiment?.manifest;
@@ -85,6 +101,9 @@ export default function StudioOverviewView({
           })
         })}
         onSubmit={values => {
+
+          values.imageRegistry = imageRegistry
+
           if (values.chaosInfrastructure.namespace === undefined) {
             delete values.chaosInfrastructure.namespace;
           }
@@ -144,7 +163,13 @@ export default function StudioOverviewView({
                       text={getString('cancel')}
                       onClick={openDiscardDialog}
                     />
-                    <Button type="submit" intent="primary" text={getString('next')} rightIcon="chevron-right" />
+                    <Button
+                      type="submit" 
+                      intent="primary" 
+                      text={getString('next')} 
+                      rightIcon="chevron-right" 
+                      disabled={imageRegistryLoading}
+                    />
                   </Layout.Horizontal>
                 </Form>
               </Layout.Vertical>