diff --git a/frontend/src/App.jsx b/frontend/src/App.jsx
index 11b4d3b..e7af31e 100644
--- a/frontend/src/App.jsx
+++ b/frontend/src/App.jsx
@@ -275,7 +275,12 @@ function App() {
case "IrishRailStation":
objectTitle = item.trainStationDesc + " Train Station";
popupContent = (
-
+
);
markerText = item.trainStationCode + " " + item.trainStationDesc;
diff --git a/frontend/src/components/TrainStationPopup.jsx b/frontend/src/components/TrainStationPopup.jsx
index 67e11dc..2225ab8 100644
--- a/frontend/src/components/TrainStationPopup.jsx
+++ b/frontend/src/components/TrainStationPopup.jsx
@@ -1,7 +1,14 @@
import React, { useState, useRef } from "react";
import { useMap } from "react-leaflet";
-const TrainStationPopup = ({ item, objectTitle }) => {
+const TrainStationPopup = ({ item, objectTitle, toggleFavourite, favourites }) => {
+ const [isFavourite, setIsFavourite] = useState(favourites.IrishRailStation?.includes(item.trainStationCode));
+
+ const handleToggleFavourite = () => {
+ toggleFavourite("IrishRailStation", item.trainStationCode);
+ setIsFavourite((prev) => !prev);
+ };
+
const [trainInfo, setTrainInfo] = useState("");
const map = useMap(); // Access the Leaflet map instance
@@ -62,7 +69,15 @@ const TrainStationPopup = ({ item, objectTitle }) => {
return (
-
{objectTitle}
+
+
{objectTitle}
+
+
- Train Station Name: {item.trainStationDesc}
- Train Station ID: {item.trainStationID}
@@ -83,7 +98,7 @@ const TrainStationPopup = ({ item, objectTitle }) => {
Load incoming trains