From a9564e046e338922b7b1cc1704999107e27247ff Mon Sep 17 00:00:00 2001
From: Thomas Bernard <thomashbernard@gmail.com>
Date: Tue, 28 May 2024 11:22:08 +0200
Subject: [PATCH] Refactor category handling

---
 frontend/src/components/ResultPage.vue        | 41 ++++---------------
 .../src/utils/firearms-utils/arme-alarme.ts   |  2 +-
 .../utils/firearms-utils/autre-pistolet.ts    |  2 +-
 .../epaule-a-levier-sous-garde.ts             |  4 +-
 .../utils/firearms-utils/epaule-a-pompe.ts    |  5 ++-
 .../epaule-a-un-coup-par-canon.ts             |  4 +-
 .../utils/firearms-utils/epaule-a-verrou.ts   |  4 +-
 .../epaule-semi-auto-militaire-milieu-20e.ts  |  2 +-
 .../epaule-semi-auto-style-chasse.ts          |  4 +-
 .../firearms-utils/epaule_mecanisme_ancien.ts |  2 +-
 frontend/src/utils/firearms-utils/index.ts    |  8 ++++
 .../pistolet-mecanisme-ancien.ts              |  2 +-
 .../pistolet-semi-auto-moderne.ts             |  2 +-
 frontend/src/utils/firearms-utils/revolver.ts |  4 +-
 .../semi-auto-style-militaire-autre.ts        |  2 +-
 15 files changed, 40 insertions(+), 48 deletions(-)

diff --git a/frontend/src/components/ResultPage.vue b/frontend/src/components/ResultPage.vue
index 458644b3..2d4433e0 100644
--- a/frontend/src/components/ResultPage.vue
+++ b/frontend/src/components/ResultPage.vue
@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import { ref, computed, watchEffect } from 'vue'
+import { ref, computed } from 'vue'
 import axios from 'axios'
 import SnackbarAlert from '@/components/SnackbarAlert.vue'
 import { TYPOLOGIES, MEASURED_GUNS_TYPOLOGIES } from '@/utils/firearms-utils/index'
@@ -7,38 +7,15 @@ import { isUserUsingCrosscall } from '@/utils/isUserUsingCrosscall'
 import { useSnackbarStore } from '@/stores/snackbar'
 import { useStepsStore } from '@/stores/steps'
 import { useResultStore } from '@/stores/result'
-import { useRouter, useRoute } from 'vue-router'
 import { getMentionsFromCategories } from '@/utils/mentions'
 
 const { setMessage } = useSnackbarStore()
 const stepsStore = useStepsStore()
 const resultStore = useResultStore()
-const router = useRouter()
-const route = useRoute()
-
-function getCategoryFromTypologyAndMeasures (typology: string, gunLength: number, gunBarrelLength: number) {
-  if (gunLength !== null && gunBarrelLength !== null) {
-    switch (typology) {
-      case 'epaule_a_pompe':
-        if (gunLength > 75 && gunBarrelLength > 55) { return 'C' } else { return 'B' }
-      case 'epaule_a_un_coup_par_canon':
-        if (gunLength > 75 && gunBarrelLength > 40) { return 'C' } else { return 'B' }
-      case 'epaule_a_levier_sous_garde': case 'epaule_a_verrou': case 'epaule_semi_auto_style_chasse':
-        if (gunLength < 75 || gunBarrelLength < 40) { return 'B' }
-        if (gunLength > 75 && gunBarrelLength > 55) { return 'C' }
-    }
-  }
-  return TYPOLOGIES[typology]?.category
-}
-
-watchEffect(() => {
-  if (!resultStore.img) router.push({ name: 'StartPage' })
-})
 
 const confidence = computed(() => resultStore.confidence)
 const confidenceLevel = computed(() => resultStore.confidenceLevel)
 const img = computed(() => resultStore.img)
-const imgUrl = computed(() => resultStore.imgUrl)
 
 const typology = computed(() => resultStore.typology)
 
@@ -60,7 +37,7 @@ const category = computed(() => {
   } else if (typology.value === 'revolver') {
     return TYPOLOGIES[typology.value]?.categoryWithoutSecuring
   } else {
-    return getCategoryFromTypologyAndMeasures(typology.value, resultStore.gunLength, resultStore.gunBarrelLength)
+    return TYPOLOGIES[typology.value]?.getCategory(resultStore.gunLength, resultStore.gunBarrelLength)
   }
 })
 
@@ -68,7 +45,7 @@ const disclaimer = computed(() => TYPOLOGIES[typology.value] && Object.hasOwn(TY
 
 function sendFeedback (isCorrect: boolean) {
   const json = {
-    image_url: imgUrl.value,
+    image_url: resultStore.imgUrl,
     feedback: isCorrect,
     confidence: confidence.value,
     label: typology.value,
@@ -98,7 +75,7 @@ function sendFeedback (isCorrect: boolean) {
   <div class="result-frame -mx-8 py-5 px-8">
     <div class="result">
       <h2
-        v-if="route.name === 'IdentificationTypologyResult' && isDummyTypology === true"
+        v-if="$route.name === 'IdentificationTypologyResult' && isDummyTypology === true"
         class="typology-title bg-white py-4"
       >
         Typologie de l'arme
@@ -167,7 +144,7 @@ function sendFeedback (isCorrect: boolean) {
               >
                 {{ label }}
               </h3>
-              <template v-if="confidenceLevel !== 'low' && (route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)">
+              <template v-if="confidenceLevel !== 'low' && ($route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)">
                 <h3
                   class="fr-alert__title"
                   data-testid="arm-category"
@@ -180,14 +157,14 @@ function sendFeedback (isCorrect: boolean) {
               </template>
             </div>
             <div
-              v-if="disclaimer && confidenceLevel !== 'low' && (route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)"
+              v-if="disclaimer && confidenceLevel !== 'low' && ($route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)"
               class="fr-alert fr-alert--warning"
             >
               <p v-html="disclaimer" />
             </div>
             <MissingCardAlert v-if="MEASURED_GUNS_TYPOLOGIES.includes(typology) && isCardDetected === false && isDummy == false" />
             <div
-              v-if="confidenceLevel !== 'low' && (route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)"
+              v-if="confidenceLevel !== 'low' && ($route.name !== 'IdentificationTypologyResult' || isDummyTypology !== true)"
               class="fr-callout mt-3"
             >
               <p class="fr-callout__text">
@@ -201,7 +178,7 @@ function sendFeedback (isCorrect: boolean) {
               </p>
             </div>
             <div
-              v-if="confidenceLevel !== 'low' && route.name === 'IdentificationTypologyResult' && isDummyTypology"
+              v-if="confidenceLevel !== 'low' && $route.name === 'IdentificationTypologyResult' && isDummyTypology"
               class="fr-alert fr-alert--warning"
             >
               <p>
@@ -222,7 +199,7 @@ function sendFeedback (isCorrect: boolean) {
       <SnackbarAlert class="text-center" />
     </div>
     <div
-      v-if="confidenceLevel !== 'low' && route.name !== 'IdentificationFinalResult'"
+      v-if="confidenceLevel !== 'low' && $route.name !== 'IdentificationFinalResult'"
       :aria-disabled="isFeedbackDone"
       class="feedback"
     >
diff --git a/frontend/src/utils/firearms-utils/arme-alarme.ts b/frontend/src/utils/firearms-utils/arme-alarme.ts
index 3da2d84d..b8856c61 100644
--- a/frontend/src/utils/firearms-utils/arme-alarme.ts
+++ b/frontend/src/utils/firearms-utils/arme-alarme.ts
@@ -28,7 +28,7 @@ import pistoletAlarmeImg13Zoom from '@/assets/guide-identification/photos/arme_a
 
 export const arme_alarme = {
   displayLabel: 'Arme d\'alarme',
-  category: 'D',
+  getCategory: () => 'D',
   options: [
     {
       label: 'ZORAKI R2',
diff --git a/frontend/src/utils/firearms-utils/autre-pistolet.ts b/frontend/src/utils/firearms-utils/autre-pistolet.ts
index f9331b22..89465ba0 100644
--- a/frontend/src/utils/firearms-utils/autre-pistolet.ts
+++ b/frontend/src/utils/firearms-utils/autre-pistolet.ts
@@ -16,7 +16,7 @@ Autre pistolet
 export const autre_pistolet = {
   typologie: 'autre_pistolet',
   displayLabel: 'Pistolet divers',
-  category: 'A, B ou D',
+  getCategory: () => 'A, B ou D',
   isDummyTypology: false,
   options_text: '',
   options: {
diff --git a/frontend/src/utils/firearms-utils/epaule-a-levier-sous-garde.ts b/frontend/src/utils/firearms-utils/epaule-a-levier-sous-garde.ts
index 7922f7eb..85aac5be 100644
--- a/frontend/src/utils/firearms-utils/epaule-a-levier-sous-garde.ts
+++ b/frontend/src/utils/firearms-utils/epaule-a-levier-sous-garde.ts
@@ -2,7 +2,7 @@ import levierSousGardeMagasinAvantImg from '@/assets/guide-mise-en-securite/phot
 import levierSousGardePasMagasinImg from '@/assets/guide-mise-en-securite/photos/epaule_a_levier_sous_garde/epaule_levier_ss_garde_magasin_fixe.jpg'
 import levierSousGardeMagasinAvantVideo from '@/assets/guide-mise-en-securite/videos/epaule_a_levier_sous_garde/epaule_levier_ss_garde_magasin_avant.mp4'
 import levierSousGardePasMagasinVideo from '@/assets/guide-mise-en-securite/videos/epaule_a_levier_sous_garde/epaule_levier_ss_garde_magasin_fixe.mp4'
-import { getEpaulLevierVerrouDisclaimer } from '@/utils/firearms-utils/index'
+import { getEpaulLevierVerrouDisclaimer, getCommonCategory } from '@/utils/firearms-utils/index'
 /*
 Armes d’épaule à levier de sous-garde
   Magasin avant
@@ -10,7 +10,7 @@ Armes d’épaule à levier de sous-garde
 */
 export const epaule_a_levier_sous_garde = {
   displayLabel: "Arme d'épaule à levier de sous-garde",
-  category: 'B ou C',
+  getCategory: getCommonCategory,
   isDummyTypology: false,
   options_text: 'En manipulant l’arme avec précaution dans une <span class="font-bold">direction sans risque</span>, inspectez l’<span class="font-bold">extrémité du tube sous le canon</span>. Sélectionnez ce que vous voyez :',
   options: {
diff --git a/frontend/src/utils/firearms-utils/epaule-a-pompe.ts b/frontend/src/utils/firearms-utils/epaule-a-pompe.ts
index e8dee088..10501001 100644
--- a/frontend/src/utils/firearms-utils/epaule-a-pompe.ts
+++ b/frontend/src/utils/firearms-utils/epaule-a-pompe.ts
@@ -2,10 +2,13 @@ import epaulePompeSimpleVideo from '@/assets/guide-mise-en-securite/videos/epaul
 // import epaulePompeCompliqueVideo from '@/assets/guide-mise-en-securite/videos/epaule_a_pompe/epaule_pompe_complique.mp4'
 /*
 3 - Armes d’épaule à pompe
+
 */
 export const epaule_a_pompe = {
   displayLabel: "Arme d'épaule à pompe",
-  category: 'B ou C',
+  getCategory: (gunLength: number, gunBarrelLength: number) => {
+    if (gunLength && gunBarrelLength) { return gunLength > 75 && gunBarrelLength > 55 ? 'C' : 'B' } else { return 'B ou C' }
+  },
   isDummyTypology: false,
   options_text: '',
   text_steps: {
diff --git a/frontend/src/utils/firearms-utils/epaule-a-un-coup-par-canon.ts b/frontend/src/utils/firearms-utils/epaule-a-un-coup-par-canon.ts
index 5f09bdba..492f7f79 100644
--- a/frontend/src/utils/firearms-utils/epaule-a-un-coup-par-canon.ts
+++ b/frontend/src/utils/firearms-utils/epaule-a-un-coup-par-canon.ts
@@ -12,7 +12,9 @@ Armes d’épaule à un coup par canon
   */
 export const epaule_a_un_coup_par_canon = {
   displayLabel: "Arme d'épaule à un coup par canon",
-  category: 'B ou C',
+  getCategory: (gunLength: number, gunBarrelLength: number) => {
+    if (gunLength && gunBarrelLength) { return (gunLength > 75 && gunBarrelLength > 40) ? 'C' : 'B' } else { return 'B ou C' }
+  },
   isDummyTypology: false,
   options_text: 'Sélectionnez ce que vous voyez sur la <span class="font-bold">zone centrale de l’arme</span>',
   options: {
diff --git a/frontend/src/utils/firearms-utils/epaule-a-verrou.ts b/frontend/src/utils/firearms-utils/epaule-a-verrou.ts
index b97dd5a9..0f3f8151 100644
--- a/frontend/src/utils/firearms-utils/epaule-a-verrou.ts
+++ b/frontend/src/utils/firearms-utils/epaule-a-verrou.ts
@@ -1,14 +1,14 @@
 import epauleAVerrouCartridges from '@/assets/guide-identification/photos/epaule_a_verrou/epaule_a_verrou_chargeur_cartouche.jpg'
 import epauleAVerrouBalls from '@/assets/guide-identification/photos/epaule_a_verrou/epaule_a_verrou_chargeur_bille.jpg'
 import epauleAVerrouVideo from '@/assets/guide-mise-en-securite/videos/epaule_a_verrou/epaule_verrou_chargeur-video.mp4'
-import { getEpaulLevierVerrouDisclaimer } from '@/utils/firearms-utils/index'
+import { getEpaulLevierVerrouDisclaimer, getCommonCategory } from '@/utils/firearms-utils/index'
 
 /*
 Armes d’épaule à verrou
 */
 export const epaule_a_verrou = {
   displayLabel: 'Arme d’épaule à verrou',
-  category: 'B ou C',
+  getCategory: getCommonCategory,
   isDummyTypology: true,
   text_steps: {
     1: 'Observer l’arme en l’orientant dans une <span class="font-bold">direction sans risque</span>, en manipulant avec précaution',
diff --git a/frontend/src/utils/firearms-utils/epaule-semi-auto-militaire-milieu-20e.ts b/frontend/src/utils/firearms-utils/epaule-semi-auto-militaire-milieu-20e.ts
index b997a2c6..479cd8d7 100644
--- a/frontend/src/utils/firearms-utils/epaule-semi-auto-militaire-milieu-20e.ts
+++ b/frontend/src/utils/firearms-utils/epaule-semi-auto-militaire-milieu-20e.ts
@@ -3,7 +3,7 @@ Armes d’épaule semi-automatiques ou automatiques type militaire milieu 20e
 */
 export const epaule_semi_auto_style_militaire_milieu_20e = {
   displayLabel: 'Arme d’épaule semi-automatique ou automatique',
-  category: 'A ou B',
+  getCategory: () => 'A ou B',
   isDummyTypology: false,
   getDisclaimer: () => "<strong>Catégorie A</strong> si à l'origine l’arme était à <strong>répétition automatique</strong> puis a été <strong>transformée</strong> en arme <strong>semi automatique</strong>, ou si l’arme possède <strong>une crosse rétractable / pliable</strong> et qu’en configuration la plus courte elle <strong>mesure moins de 60 cm</strong>.",
 } as const
diff --git a/frontend/src/utils/firearms-utils/epaule-semi-auto-style-chasse.ts b/frontend/src/utils/firearms-utils/epaule-semi-auto-style-chasse.ts
index 0a38593c..81f850f7 100644
--- a/frontend/src/utils/firearms-utils/epaule-semi-auto-style-chasse.ts
+++ b/frontend/src/utils/firearms-utils/epaule-semi-auto-style-chasse.ts
@@ -1,3 +1,5 @@
+import { getCommonCategory } from '@/utils/firearms-utils/index'
+
 const DISCLAIMERS = {
   short: '<strong>Catégorie B</strong> si la <strong>capacité > à 3 munitions</strong>, ou si le <strong>chargeur est amovible</strong>.',
   long: '<ul><li><strong>Catégorie B</strong> si la <strong>capacité > à 3 munitions</strong>, ou si le <strong>canon est lisse</strong>, ou si le <strong>chargeur est amovible</strong>.</li><li><strong>Catégorie C</strong> si le <strong>canon est rayé</strong>.</li></ul>',
@@ -8,7 +10,7 @@ Armes d’épaule semi-automatiques ou automatiques type chasse
 */
 export const epaule_semi_auto_style_chasse = {
   displayLabel: 'Arme d’épaule semi-automatique',
-  category: 'B ou C',
+  getCategory: getCommonCategory,
   isDummyTypology: false,
   getDisclaimer: (category: string, isCardDetected: boolean) => {
     if (['B ou C', 'C'].includes(category)) {
diff --git a/frontend/src/utils/firearms-utils/epaule_mecanisme_ancien.ts b/frontend/src/utils/firearms-utils/epaule_mecanisme_ancien.ts
index 2dc588b6..a9b96088 100644
--- a/frontend/src/utils/firearms-utils/epaule_mecanisme_ancien.ts
+++ b/frontend/src/utils/firearms-utils/epaule_mecanisme_ancien.ts
@@ -3,6 +3,6 @@ Armes d’épaule à mécanisme ancien
 */
 export const epaule_mecanisme_ancien = {
   displayLabel: "Arme d'épaule à mécanisme ancien",
-  category: 'D',
+  getCategory: () => 'D',
   isDummyTypology: false,
 } as const
diff --git a/frontend/src/utils/firearms-utils/index.ts b/frontend/src/utils/firearms-utils/index.ts
index 594ea655..e6e483a3 100644
--- a/frontend/src/utils/firearms-utils/index.ts
+++ b/frontend/src/utils/firearms-utils/index.ts
@@ -100,3 +100,11 @@ export function getEpaulLevierVerrouDisclaimer (category: string, isCardDetected
     return '<ul><li><strong>Catégorie B</strong> : si la capacité > 11 munitions ou si le <strong>canon est lisse</strong></li><li><strong>Catégorie C</strong> : si la capacité < 11 munitions et le <strong>canon est rayé</strong></li></ul>'
   }
 }
+
+export function getCommonCategory (gunLength: number, gunBarrelLength: number) {
+  if (gunLength && gunBarrelLength) {
+    if (gunLength < 75 || gunBarrelLength < 40) { return 'B' }
+    if (gunLength > 75 && gunBarrelLength > 55) { return 'C' }
+  }
+  return 'B ou C'
+}
diff --git a/frontend/src/utils/firearms-utils/pistolet-mecanisme-ancien.ts b/frontend/src/utils/firearms-utils/pistolet-mecanisme-ancien.ts
index 5367d09a..bbd1e49b 100644
--- a/frontend/src/utils/firearms-utils/pistolet-mecanisme-ancien.ts
+++ b/frontend/src/utils/firearms-utils/pistolet-mecanisme-ancien.ts
@@ -3,6 +3,6 @@ Pistolet à mécanisme ancien
 */
 export const pistolet_mecanisme_ancien = {
   displayLabel: 'Pistolet à mécanisme ancien',
-  category: 'D',
+  getCategory: () => 'D',
   isDummyTypology: false,
 } as const
diff --git a/frontend/src/utils/firearms-utils/pistolet-semi-auto-moderne.ts b/frontend/src/utils/firearms-utils/pistolet-semi-auto-moderne.ts
index cd5778f0..68145045 100644
--- a/frontend/src/utils/firearms-utils/pistolet-semi-auto-moderne.ts
+++ b/frontend/src/utils/firearms-utils/pistolet-semi-auto-moderne.ts
@@ -11,7 +11,7 @@ Pistolets semi-auto modernes
 */
 export const pistolet_semi_auto_moderne = {
   displayLabel: 'Pistolet semi-automatique',
-  category: 'B',
+  getCategory: () => 'B',
   isDummyTypology: true,
   pistolet_semi_auto_moderne_text_option: 'Sélectionner ce que vous voyez sur votre arme : <span class="font-bold">bouton à proximité du pontet du côté gauche de la poignée</span>, OU <span class="font-bold">bouton sur le talon</span> de la crosse.',
   textOptions: '',
diff --git a/frontend/src/utils/firearms-utils/revolver.ts b/frontend/src/utils/firearms-utils/revolver.ts
index 4739a134..a2f89a77 100644
--- a/frontend/src/utils/firearms-utils/revolver.ts
+++ b/frontend/src/utils/firearms-utils/revolver.ts
@@ -29,7 +29,7 @@ Revolvers
 */
 export const revolver = {
   displayLabel: 'Revolver',
-  category: 'B',
+  getCategory: () => 'B',
   categoryWithoutSecuring: 'B ou D',
   isDummyTypology: true,
   options_step_1_text: 'En maintenant l’arme dans une  <span class="font-bold">direction sécurisée</span> , sélectionnez ce que vous voyez.',
@@ -170,6 +170,6 @@ export const revolver = {
 
 export const revolver_black_powder = {
   displayLabel: 'Revolver à poudre noire',
-  category: 'D',
+  getCategory: () => 'D',
   isDummyTypology: false,
 } as const
diff --git a/frontend/src/utils/firearms-utils/semi-auto-style-militaire-autre.ts b/frontend/src/utils/firearms-utils/semi-auto-style-militaire-autre.ts
index 3a24a0d6..2728a3d6 100644
--- a/frontend/src/utils/firearms-utils/semi-auto-style-militaire-autre.ts
+++ b/frontend/src/utils/firearms-utils/semi-auto-style-militaire-autre.ts
@@ -20,7 +20,7 @@ Armes d’épaule semi-automatiques ou automatiques type militaire moderne
 */
 export const semi_auto_style_militaire_autre = {
   displayLabel: 'Arme semi-automatique ou automatique',
-  category: 'A ou B',
+  getCategory: () => 'A ou B',
   isDummyTypology: true,
   options_text: 'Observez la position du chargeur et sélectionnez ce que vous voyez :',
   options: {