From 797339d4436cf4f715cf0a2cc9283d070cd2cc92 Mon Sep 17 00:00:00 2001
From: monsieurtanuki <fabrice_fontaine@hotmail.com>
Date: Fri, 12 Nov 2021 18:04:56 +0100
Subject: [PATCH] feature_2/#191 - added eco-score and nutri-score categories
 (ProductImprovement) (#288)

Impacted file:
* `Product.dart`
---
 lib/model/Product.dart | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/lib/model/Product.dart b/lib/model/Product.dart
index 1a37848c61..e99ccad08a 100644
--- a/lib/model/Product.dart
+++ b/lib/model/Product.dart
@@ -60,6 +60,30 @@ enum ProductImprovement {
   OBSOLETE_NUTRITION_IMAGE,
 }
 
+/// Category: what would this [ProductImprovement] help compute?
+enum ProductImprovementCategory {
+  /// Eco-Score category
+  ECO_SCORE,
+
+  /// Nutri-Score category
+  NUTRI_SCORE,
+}
+
+extension ProductImprovementExtension on ProductImprovement {
+  ProductImprovementCategory getCategory() {
+    switch (this) {
+      case ProductImprovement.ORIGINS_TO_BE_COMPLETED:
+        return ProductImprovementCategory.ECO_SCORE;
+      case ProductImprovement.CATEGORIES_BUT_NO_NUTRISCORE:
+      case ProductImprovement.ADD_NUTRITION_FACTS:
+      case ProductImprovement.ADD_CATEGORY:
+      case ProductImprovement.ADD_NUTRITION_FACTS_AND_CATEGORY:
+      case ProductImprovement.OBSOLETE_NUTRITION_IMAGE:
+        return ProductImprovementCategory.NUTRI_SCORE;
+    }
+  }
+}
+
 /// This class contains most of the data about a specific product.
 ///
 /// Please read the language mechanics explanation if you intend to display