feat(#569): sort drugs

hpi-dhc · Jun 20, 2023 · 0a9ac25 · 0a9ac25
1 parent 6eada9d
commit 0a9ac25
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/app/lib/common/widgets/drug_list/builder.dart b/app/lib/common/widgets/drug_list/builder.dart
@@ -27,6 +27,22 @@ List<Widget> _buildDrugCards(
   if (filteredDrugs.isEmpty && noDrugsMessage != null) {
     return [errorIndicator(noDrugsMessage)];
   }
+  final sortedWarningLevels = [
+    'WarningLevel.red',
+    'WarningLevel.yellow',
+    'WarningLevel.green',
+    null,
+  ];
+  filteredDrugs.sort((drugA, drugB) {
+    final warningLevelComparison = sortedWarningLevels.indexOf(
+      drugA.userGuideline()?.annotations.warningLevel.toString())
+        .compareTo(sortedWarningLevels.indexOf(
+          drugB.userGuideline()?.annotations.warningLevel.toString()));
+    if (warningLevelComparison == 0) {
+      return drugA.name.compareTo(drugB.name);
+    }
+    return warningLevelComparison;
+  });
   return [
     SizedBox(height: 8),
     ...filteredDrugs.map((drug) => Column(children: [