From 83bd1e86ea0d04bbfad0e4873909f1effeff6b4f Mon Sep 17 00:00:00 2001
From: Spythere <spythere@protonmail.com>
Date: Sun, 22 May 2022 17:43:19 +0200
Subject: [PATCH] Mniejsze poprawki

---
 src/components/JournalView/JournalTimetables.vue | 3 ++-
 src/constants/storeConstants.ts                  | 4 +++-
 src/store/index.ts                               | 6 ++++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/components/JournalView/JournalTimetables.vue b/src/components/JournalView/JournalTimetables.vue
index ebc6b76..33a8c81 100644
--- a/src/components/JournalView/JournalTimetables.vue
+++ b/src/components/JournalView/JournalTimetables.vue
@@ -158,7 +158,8 @@ import { URLs } from '@/scripts/utils/apiURLs';
 import { journalTimetableFilters } from '@/data/journalFilters';
 import { JournalFilterType } from '@/scripts/enums/JournalFilterType';
 
-const PROD_MODE = true;
+const DEV_MODE = true;
+const PROD_MODE = !DEV_MODE || process.env.NODE_ENV === "production";
 
 const API_URL = PROD_MODE ? `${URLs.stacjownikAPI}/api/getTimetables` : 'http://localhost:3001/api/getTimetables';
 
diff --git a/src/constants/storeConstants.ts b/src/constants/storeConstants.ts
index 64a93b4..59dd736 100644
--- a/src/constants/storeConstants.ts
+++ b/src/constants/storeConstants.ts
@@ -24,5 +24,7 @@ export const GETTERS = {
     swdrDataStatus: "swdrDataStatus",
     trainsDataStatus: "trainsDataStatus",
 
-    currentRegion: "currentRegion"
+    currentRegion: "currentRegion",
+
+    webSocket: "webSocket"
 }
\ No newline at end of file
diff --git a/src/store/index.ts b/src/store/index.ts
index bb78270..8bd0f1e 100644
--- a/src/store/index.ts
+++ b/src/store/index.ts
@@ -17,7 +17,7 @@ import { getLocoURL, getScheduledTrain, getStatusID, getStatusTimestamp, parseSp
 import { URLs } from '@/scripts/utils/apiURLs';
 import ScheduledTrain from '@/scripts/interfaces/ScheduledTrain';
 import StationRoutes from '@/scripts/interfaces/StationRoutes';
-import { io } from 'socket.io-client';
+import { io, Socket } from 'socket.io-client';
 import { APIData, State, StationJSONData } from './types';
 
 const connectToDevAPI = false;
@@ -66,7 +66,9 @@ export const store = createStore<State>({
     trainsDataStatus: (state): DataStatus => state.trainsDataStatus,
     dataStatus: (state): DataStatus => state.dataConnectionStatus,
 
-    currentRegion: (state): { id: string; value: string } => state.region
+    currentRegion: (state): { id: string; value: string } => state.region,
+
+    webSocket: (state): Socket | undefined => state.webSocket
   },
 
   actions: {