diff --git a/components/ActionSheetComponents/EachTrainInfo.js b/components/ActionSheetComponents/EachTrainInfo.js index 4c81861..becdb0e 100644 --- a/components/ActionSheetComponents/EachTrainInfo.js +++ b/components/ActionSheetComponents/EachTrainInfo.js @@ -40,7 +40,6 @@ export const EachTrainInfo = (props) => { } = props.payload; const [trainData, setTrainData] = useState([]); const [currentPosition, setCurrentPosition] = useState([]); - const [trainPositionSwitch, setTrainPositionSwitch] = useState("false"); const { currentTrain } = useCurrentTrain(); @@ -81,22 +80,22 @@ export const EachTrainInfo = (props) => { const mainTrainStationPosition = trainData.findIndex( (d) => d.split(",")[0] == stationInfo ); - const relationMain = - mainTrainStationPosition == 0 - ? "head" - : mainTrainStationPosition == trainData.length - 1 - ? "tail" - : "middle"; + + const relationMain = (() => { + if (mainTrainStationPosition == 0) return "head"; + if (mainTrainStationPosition == trainData.length - 1) return "tail"; + return "middle"; + })(); const subTrainStationPosition = showNearTrain.findIndex( (d) => d.split(",")[0] == stationInfo ); - const relationSub = - subTrainStationPosition == 0 - ? "head" - : subTrainStationPosition == showNearTrain.length - 1 - ? "tail" - : "middle"; + const relationSub = (() => { + if (subTrainStationPosition == 0) return "head"; + if (subTrainStationPosition == showNearTrain.length - 1) return "tail"; + return "middle"; + })(); + switch (relationMain) { case "head": if (relationSub == "head") {