diff --git a/frontend/src/App.jsx b/frontend/src/App.jsx
index a35e043..ff696b9 100644
--- a/frontend/src/App.jsx
+++ b/frontend/src/App.jsx
@@ -176,90 +176,16 @@ function App() {
switch (item.objectType) {
case "IrishRailTrain":
objectTitle = "Irish Rail Train: " + item.trainCode;
- let trainType;
- switch (item.trainType) {
- case "M":
- trainType = "Mainline";
- icon = "mainline";
- break;
- case "S":
- trainType = "Suburban";
- icon = "suburban";
- break;
- case "D":
- trainType = "DART";
- icon = "dart";
- break;
- default:
- trainType = "Unknown";
- }
+ icon = item.trainTypeFull + item.trainPunctualityStatus;
- let trainStatus;
- switch (item.trainStatus) {
- case "R":
- trainStatus = "Running";
- break;
-
- case "T":
- trainStatus = "Terminated";
- break;
-
- case "N":
- trainStatus = "Not yet running";
- break;
-
- default:
- trainStatus = "Unknown";
- }
-
- const splitMessage = item.trainPublicMessage.split("\\n");
- const match = splitMessage[1].match(/(-?\d+)\s+mins\s+late/);
- const punctuality = match ? parseInt(match[1], 10) : NaN;
- let latenessMessage;
- let punctualityStr;
-
- if (punctuality < 0) {
- punctualityStr = "early";
- } else if (punctuality === 0) {
- punctualityStr = "On time";
- } else if (punctuality > 0) {
- punctualityStr = "late";
- } else {
- punctualityStr = "N/A";
- }
-
- if (punctualityStr === "early") {
- latenessMessage = -punctuality + " minute" + (punctuality === -1 ? "" : "s") + " early";
- icon += "OnTime";
- }
- else if (punctualityStr === "On time") {
- latenessMessage = punctualityStr;
- icon += "OnTime";
- }
- else if (punctualityStr === "late") {
- latenessMessage = punctuality + " minute" + (punctuality === 1 ? "" : "s") + " late";
-
- if (trainStatus === "Running") {
- icon += "Late";
- }
- else {
- icon += "NotRunning";
- }
- }
- else {
- latenessMessage = "On time";
- icon += "NotRunning";
+ if (item.trainStatusFull == "Terminated" || item.trainStatusFull == "Not yet running") {
+ icon = item.trainTypeFull + "NotRunning";
}
popupContent = (