From f1fdfd7100a7d299c34c837545119d48fe817cce Mon Sep 17 00:00:00 2001
From: Pierre-Olivier Mercier <nemunaire@nemunai.re>
Date: Fri, 15 Mar 2024 12:51:19 +0100
Subject: [PATCH] Handle travaux

---
 api/traffic.go | 49 +++++++++++++++++++++++++++----------------------
 1 file changed, 27 insertions(+), 22 deletions(-)

diff --git a/api/traffic.go b/api/traffic.go
index 4c402bb..bfec889 100644
--- a/api/traffic.go
+++ b/api/traffic.go
@@ -118,30 +118,35 @@ func declareTrafficRoutes(router *gin.RouterGroup) {
 				}
 			}
 
-			icr := effectGradation(disruption.Severity.Effect)
-			if icr > slug {
-				slug = icr
-				switch disruption.Severity.Effect {
-				case navitia.JourneyStatusUnknownEffect:
-					pgt.Slug = "critique"
-				case navitia.EffectNoService:
-					pgt.Slug = "critique"
-				case navitia.JourneyStatusReducedService:
-					pgt.Slug = "alerte"
-				case navitia.JourneyStatusSignificantDelay:
-					pgt.Slug = "alerte"
-				case navitia.JourneyStatusDetour:
-					pgt.Slug = "normal_trav"
-				case navitia.JourneyStatusAdditionalService:
-					pgt.Slug = "normal_trav"
-				case navitia.JourneyStatusOtherEffect:
-					pgt.Slug = "normal_trav"
-				case navitia.JourneyStatusStopMoved:
-					pgt.Slug = "normal_trav"
-				default:
-					pgt.Slug = "normal"
+			if disruption.Cause == "travaux" {
+				pgt.Slug = "normal_trav"
+			} else {
+				icr := effectGradation(disruption.Severity.Effect)
+				if icr > slug {
+					slug = icr
+					switch disruption.Severity.Effect {
+					case navitia.JourneyStatusUnknownEffect:
+						pgt.Slug = "critique"
+					case navitia.EffectNoService:
+						pgt.Slug = "critique"
+					case navitia.JourneyStatusReducedService:
+						pgt.Slug = "alerte"
+					case navitia.JourneyStatusSignificantDelay:
+						pgt.Slug = "alerte"
+					case navitia.JourneyStatusDetour:
+						pgt.Slug = "alerte"
+					case navitia.JourneyStatusAdditionalService:
+						pgt.Slug = "alerte"
+					case navitia.JourneyStatusOtherEffect:
+						pgt.Slug = "alerte"
+					case navitia.JourneyStatusStopMoved:
+						pgt.Slug = "alerte"
+					default:
+						pgt.Slug = "normal"
+					}
 				}
 			}
+
 		}
 
 		if pgt.Title == "" {