feature(app, shared-data): module setup deck config assignment (#18864)

Author: TamarZanzouri

app/src/organisms/ModuleCard/index.tsx

@@ -139,7 +139,6 @@ export const ModuleCard = (props: ModuleCardProps): JSX.Element | null => {
   const [showSetupWizard, setShowSetupWizard] = useState(false)
   const [showFWBanner, setShowFWBanner] = useState(true)
   const [showCalModal, setShowCalModal] = useState(false)
-
   const [targetProps, tooltipProps] = useHoverTooltip()
 
   const runStatus = useCurrentRunStatus()

app/src/organisms/ModuleWizardFlows/SelectLocation.tsx

@@ -13,11 +13,17 @@ import {
 } from '@opentrons/components'
 import { useUpdateDeckConfigurationMutation } from '@opentrons/react-api-client'
 import {
+  FAKE_FIXTURE_IDS,
   FLEX_ROBOT_TYPE,
+  FLEX_STACKER_FIXTURES,
+  getAAForModuleFixture,
+  getCutoutConfigReplacmentForModule,
   getCutoutFixturesForModuleModel,
   getDeckDefFromRobotType,
   getFixtureIdByCutoutIdFromModuleAnchorCutoutId,
   getModuleDisplayName,
+  getReplacementFixtureForFixtureRemoval,
+  replaceFixtureToFakeFixtureAndTransformCutoutFixturesToAA,
   SINGLE_CENTER_CUTOUTS,
   SINGLE_CENTER_SLOT_FIXTURE,
   SINGLE_LEFT_SLOT_FIXTURE,
@@ -46,7 +52,7 @@ export const BODY_STYLE = css`
     line-height: 1.75rem;
   }
 `
-interface SelectLocationProps extends ModuleSetupWizardStepProps {
+export interface SelectLocationProps extends ModuleSetupWizardStepProps {
   deckConfig: DeckConfiguration
   createMaintenanceRun: CreateMaintenanceRunType
   isLoadedInRun: boolean
@@ -66,6 +72,11 @@ export function SelectLocation(props: SelectLocationProps): JSX.Element {
     attachedModule,
     deckConfig
   )
+
+  const deckConfigWithAA = replaceFixtureToFakeFixtureAndTransformCutoutFixturesToAA(
+    deckConfig
+  )
+
   const { t } = useTranslation('module_wizard_flows')
   const moduleName = getModuleDisplayName(attachedModule.moduleModel)
   const handleOnClick = (): void => {
@@ -90,18 +101,22 @@ export function SelectLocation(props: SelectLocationProps): JSX.Element {
     (acc, { mayMountTo }) => [...acc, ...mayMountTo],
     []
   )
-  const editableCutoutIds = deckConfig.reduce<CutoutId[]>(
+
+  const editableCutoutIds = deckConfigWithAA.reduce<CutoutId[]>(
     (acc, { cutoutId, cutoutFixtureId, opentronsModuleSerialNumber }) => {
       const isCurrentConfiguration =
         Object.values(configuredFixtureIdByCutoutId).includes(
           cutoutFixtureId
         ) && attachedModule.serialNumber === opentronsModuleSerialNumber
       if (
         // in run setup, module calibration only available when module location is already correctly configured
-        !isLoadedInRun &&
-        mayMountToCutoutIds.includes(cutoutId) &&
-        (isCurrentConfiguration ||
-          SINGLE_SLOT_FIXTURES.includes(cutoutFixtureId))
+        (!isLoadedInRun &&
+          mayMountToCutoutIds.includes(cutoutId) &&
+          (isCurrentConfiguration ||
+            SINGLE_SLOT_FIXTURES.includes(cutoutFixtureId) ||
+            // fake fixtures include mag block next to an empty staging slot and a waste chute next to an empty staging slot
+            FAKE_FIXTURE_IDS.includes(cutoutFixtureId))) ||
+        FLEX_STACKER_FIXTURES.includes(cutoutFixtureId)
       ) {
         return [...acc, cutoutId]
       }
@@ -116,32 +131,43 @@ export function SelectLocation(props: SelectLocationProps): JSX.Element {
       moduleFixtures
     )
     if (!isEqual(selectedFixtureIdByCutoutIds, configuredFixtureIdByCutoutId)) {
-      updateDeckConfiguration(
-        deckConfig.map(cc => {
-          if (cc.cutoutId in configuredFixtureIdByCutoutId) {
-            let replacementFixtureId: CutoutFixtureId = SINGLE_LEFT_SLOT_FIXTURE
-            if (SINGLE_CENTER_CUTOUTS.includes(cc.cutoutId)) {
-              replacementFixtureId = SINGLE_CENTER_SLOT_FIXTURE
-            } else if (SINGLE_RIGHT_CUTOUTS.includes(cc.cutoutId)) {
-              replacementFixtureId = SINGLE_RIGHT_SLOT_FIXTURE
-            }
+      const updatedDeckConfig = deckConfig.map(cc => {
+        if (cc.cutoutId in configuredFixtureIdByCutoutId) {
+          if (SINGLE_CENTER_CUTOUTS.includes(cc.cutoutId)) {
             return {
               ...cc,
-              cutoutFixtureId: replacementFixtureId,
+              cutoutFixtureId: SINGLE_CENTER_SLOT_FIXTURE,
               opentronsModuleSerialNumber: undefined,
             }
-          } else if (cc.cutoutId in selectedFixtureIdByCutoutIds) {
+          } else if (SINGLE_RIGHT_CUTOUTS.includes(cc.cutoutId)) {
             return {
               ...cc,
-              cutoutFixtureId:
-                selectedFixtureIdByCutoutIds[cc.cutoutId] ?? cc.cutoutFixtureId,
-              opentronsModuleSerialNumber: attachedModule.serialNumber,
+              cutoutFixtureId: SINGLE_RIGHT_SLOT_FIXTURE,
+              opentronsModuleSerialNumber: undefined,
             }
-          } else {
-            return cc
           }
-        })
-      )
+          return {
+            ...cc,
+            cutoutFixtureId: SINGLE_LEFT_SLOT_FIXTURE,
+            opentronsModuleSerialNumber: undefined,
+          }
+        } else if (cc.cutoutId in selectedFixtureIdByCutoutIds) {
+          const fixtureReplacement = getCutoutConfigReplacmentForModule(
+            anchorCutoutId,
+            selectedFixtureIdByCutoutIds[cc.cutoutId] ?? cc.cutoutFixtureId,
+            attachedModule.moduleModel,
+            deckConfig
+          )
+          return {
+            ...cc,
+            cutoutFixtureId: fixtureReplacement,
+            opentronsModuleSerialNumber: attachedModule.serialNumber,
+          }
+        } else {
+          return cc
+        }
+      })
+      updateDeckConfiguration(updatedDeckConfig)
     }
   }
 
@@ -153,15 +179,25 @@ export function SelectLocation(props: SelectLocationProps): JSX.Element {
     updateDeckConfiguration(
       deckConfig.map(cc => {
         if (cc.cutoutId in removedFixtureIdByCutoutIds) {
-          let replacementFixtureId: CutoutFixtureId = SINGLE_LEFT_SLOT_FIXTURE
-          if (SINGLE_CENTER_CUTOUTS.includes(cc.cutoutId)) {
-            replacementFixtureId = SINGLE_CENTER_SLOT_FIXTURE
-          } else if (SINGLE_RIGHT_CUTOUTS.includes(cc.cutoutId)) {
-            replacementFixtureId = SINGLE_RIGHT_SLOT_FIXTURE
-          }
+          const fixtureInPlace = deckConfigWithAA.find(
+            dc => dc.cutoutId === anchorCutoutId
+          )
+          const removedDefaultFixture = removedFixtureIdByCutoutIds[
+            cc.cutoutId
+          ] as CutoutFixtureId // we know there is a match by the condition
+          const aa = getAAForModuleFixture(
+            anchorCutoutId,
+            removedDefaultFixture,
+            attachedModule.moduleModel
+          )
+          const replacment = getReplacementFixtureForFixtureRemoval(
+            fixtureInPlace?.cutoutFixtureId ?? removedDefaultFixture,
+            anchorCutoutId,
+            aa
+          )
           return {
             ...cc,
-            cutoutFixtureId: replacementFixtureId,
+            cutoutFixtureId: replacment,
             opentronsModuleSerialNumber: undefined,
           }
         } else {
@@ -180,6 +216,7 @@ export function SelectLocation(props: SelectLocationProps): JSX.Element {
           handleClickAdd={handleAddFixture}
           handleClickRemove={handleRemoveFixture}
           editableCutoutIds={editableCutoutIds}
+          moduleModel={attachedModule.moduleModel}
           selectedCutoutId={
             deckConfig.find(
               ({ cutoutId, opentronsModuleSerialNumber }) =>