Task/WP-724: Mutation Hooks: Compress Files

client/src/components/DataFiles/tests/DataFiles.test.jsx

@@ -19,6 +19,10 @@ describe('DataFiles', () => {
           compress: '',
         },
       },
+      allocations: {
+        portal_alloc: 'TACC-ACI',
+        active: [{ projectId: 'active-project' }],
+      },
       systems: systemsFixture,
       files: filesFixture,
       pushKeys: {

client/src/hooks/datafiles/mutations/useCompress.ts

@@ -0,0 +1,156 @@
+import { useMutation } from '@tanstack/react-query';
+import { useSelector, useDispatch, shallowEqual } from 'react-redux';
+import { getCompressParams } from 'utils/getCompressParams';
+import { apiClient } from 'utils/apiClient';
+import { TTapisFile, TPortalSystem } from 'utils/types';
+import { TJobBody, TJobPostResponse } from './useSubmitJob';
+
+async function submitJobUtil(body: TJobBody) {
+  const res = await apiClient.post<TJobPostResponse>(
+    `/api/workspace/jobs`,
+    body
+  );
+  return res.data.response;
+}
+
+function useCompress() {
+  const dispatch = useDispatch();
+  const status = useSelector(
+    (state: any) => state.files.operationStatus.compress,
+    shallowEqual
+  );
+
+  const setStatus = (newStatus: any) => {
+    dispatch({
+      type: 'DATA_FILES_SET_OPERATION_STATUS',
+      payload: { status: newStatus, operation: 'compress' },
+    });
+  };
+
+  const compressErrorAction = (errorMessage: any) => {
+    return {
+      type: 'DATA_FILES_SET_OPERATION_STATUS',
+      payload: {
+        status: { type: 'ERROR', message: errorMessage },
+        operation: 'compress',
+      },
+    };
+  };
+
+  const compressApp = useSelector(
+    (state: any) => state.workbench.config.compressApp
+  );
+
+  const defaultAllocation = useSelector(
+    (state: any) =>
+      state.allocations.portal_alloc || state.allocations.active[0].projectName
+  );
+
+  const systems = useSelector(
+    (state: any) => state.systems.storage.configuration
+  );
+
+  const { mutateAsync } = useMutation({ mutationFn: submitJobUtil });
+
+  const compress = ({
+    scheme,
+    files,
+    filename,
+    compressionType,
+  }: {
+    scheme: string;
+    files: TTapisFile[];
+    filename: string;
+    compressionType: string;
+  }) => {
+    dispatch({
+      type: 'DATA_FILES_SET_OPERATION_STATUS',
+      payload: { status: 'RUNNING', operation: 'compress' },
+    });
+
+    let defaultPrivateSystem: TPortalSystem | undefined;
+
+    if (files[0].scheme === 'private' && files[0].api === 'tapis') {
+      defaultPrivateSystem === null;
+    }
+
+    if (scheme !== 'private' && scheme !== 'projects') {
+      defaultPrivateSystem = systems.find((s: any) => s.default);
+
+      if (!defaultPrivateSystem) {
+        throw new Error('Folder downloads are unavailable in this portal', {
+          cause: 'compressError',
+        });
+      }
+    }
+
+    const params = getCompressParams(
+      files,
+      filename,
+      compressionType,
+      compressApp,
+      defaultAllocation,
+      defaultPrivateSystem
+    );
+
+    return mutateAsync(
+      {
+        job: params,
+      },
+      {
+        onSuccess: (response: any) => {
+          // If the execution system requires pushing keys, then
+          // bring up the modal and retry the compress action
+          if (response.execSys) {
+            dispatch({
+              type: 'SYSTEMS_TOGGLE_MODAL',
+              payload: {
+                operation: 'pushKeys',
+                props: {
+                  system: response.execSys,
+                  onCancel: compressErrorAction('An error has occurred'),
+                },
+              },
+            });
+          } else if (response.status === 'PENDING') {
+            dispatch({
+              type: 'DATA_FILES_SET_OPERATION_STATUS',
+              payload: { status: { type: 'SUCCESS' }, operation: 'compress' },
+            });
+            dispatch({
+              type: 'ADD_TOAST',
+              payload: {
+                message: 'Compress job submitted.',
+              },
+            });
+            dispatch({
+              type: 'DATA_FILES_SET_OPERATION_STATUS',
+              payload: { operation: 'compress', status: {} },
+            });
+            dispatch({
+              type: 'DATA_FILES_TOGGLE_MODAL',
+              payload: { operation: 'compress', props: {} },
+            });
+          }
+        },
+        onError: (response) => {
+          const errorMessage =
+            response.cause === 'compressError'
+              ? response.message
+              : 'An error has occurred.';
+          dispatch({
+            type: 'DATA_FILES_SET_OPERATION_STATUS',
+            payload: {
+              status: { type: 'ERROR', message: errorMessage },
+              operation: 'compress',
+            },
+          });
+        },
+      }
+    );
+  };
+
+  return { compress, status, setStatus };
+}
+
+export default useCompress;

client/src/hooks/datafiles/mutations/useSubmitJob.ts

@@ -0,0 +1,57 @@
+import {
+  TTapisSystem,
+  TAppFileInput,
+  TTapisJob,
+  TJobArgSpecs,
+  TJobKeyValuePair,
+} from 'utils/types';
+
+export type TJobPostOperations = 'resubmitJob' | 'cancelJob' | 'submitJob';
+
+export type TParameterSetSubmit = {
+  appArgs?: TJobArgSpecs;
+  containerArgs?: TJobArgSpecs;
+  schedulerOptions?: TJobArgSpecs;
+  envVariables?: TJobKeyValuePair[];
+};
+
+export type TConfigurationValues = {
+  execSystemId?: string;
+  execSystemLogicalQueue?: string;
+  maxMinutes?: number;
+  nodeCount?: number;
+  coresPerNode?: number;
+  allocation?: string;
+  memoryMB?: number;
+};
+
+export type TOutputValues = {
+  name: string;
+  archiveSystemId?: string;
+  archiveSystemDir?: string;
+};
+
+export interface TJobSubmit extends TConfigurationValues, TOutputValues {
+  archiveOnAppError?: boolean;
+  appId: string;
+  fileInputs?: TAppFileInput[];
+  parameterSet?: TParameterSetSubmit;
+}
+
+export type TJobBody = {
+  operation?: TJobPostOperations;
+  uuid?: string;
+  job: TJobSubmit;
+  licenseType?: string;
+  isInteractive?: boolean;
+  execSystemId?: string;
+};
+
+export interface IJobPostResponse extends TTapisJob {
+  execSys?: TTapisSystem;
+}
+
+export type TJobPostResponse = {
+  response: IJobPostResponse;
+  status: number;
+};

client/src/hooks/datafiles/useDataFilesAllocations.js

@@ -0,0 +1,30 @@
+const dataFilesAllocations = {
+  hosts: {
+    'ls6.tacc.utexas.edu': ['TACC-ACI'],
+    'data.tacc.utexas.edu': ['TACC-ACI'],
+    'ranch.tacc.utexas.edu': ['TACC-ACI'],
+    'stampede2.tacc.utexas.edu': ['TACC-ACI'],
+    'maverick2.tacc.utexas.edu': ['TACC-ACI'],
+    'frontera.tacc.utexas.edu': ['TACC-ACI'],
+  },
+  portal_alloc: 'TACC-ACI',
+  active: [
+    {
+      title: 'TACC-ACI',
+      projectId: 9192,
+      projectName: 'TACC-ACI',
+      systems: [
+        {
+          name: 'ls6',
+          host: 'ls6.tacc.utexas.edu',
+        },
+        {
+          name: 'frontera',
+          host: 'frontera.tacc.utexas.edu',
+        },
+      ],
+    },
+  ],
+};
+
+export default dataFilesAllocations;