From ffe7b2b129729f84fdaf6527ca939223f52d237d Mon Sep 17 00:00:00 2001
From: WickedJack99 <wickedjack1@proton.me>
Date: Sat, 6 Apr 2024 22:13:55 +0200
Subject: [PATCH] Added null check.

---
 .../FileReaderWriter/FileReaderWriter.java    | 38 ++++++++++---------
 1 file changed, 20 insertions(+), 18 deletions(-)

diff --git a/src/main/java/vslab1/src/FileReaderWriter/FileReaderWriter.java b/src/main/java/vslab1/src/FileReaderWriter/FileReaderWriter.java
index 9a45a90..c543a0e 100644
--- a/src/main/java/vslab1/src/FileReaderWriter/FileReaderWriter.java
+++ b/src/main/java/vslab1/src/FileReaderWriter/FileReaderWriter.java
@@ -245,25 +245,27 @@ public class FileReaderWriter {
                             peerObject.put("onlineStatus", "unknown");
                         }
 
-                        peerToUpdate.filesMap().forEach((fileName, filePath) -> {
-                            boolean fileExists = false;
-                            
-                            for (Object file : peerObject.getJSONArray("files")) {
-                                JSONObject fileObject = (JSONObject)file;
-                                if (fileObject.getString("fileName") == fileName) {
-                                    fileObject.put("filePath", filePath);
-                                    fileExists = true;
-                                    break;
+                        if (peerToUpdate.filesMap() != null) {
+                            peerToUpdate.filesMap().forEach((fileName, filePath) -> {
+                                boolean fileExists = false;
+                                
+                                for (Object file : peerObject.getJSONArray("files")) {
+                                    JSONObject fileObject = (JSONObject)file;
+                                    if (fileObject.getString("fileName") == fileName) {
+                                        fileObject.put("filePath", filePath);
+                                        fileExists = true;
+                                        break;
+                                    }
                                 }
-                            }
-        
-                            if (fileExists == false) {
-                                JSONObject fileInfo = new JSONObject();
-                                fileInfo.put("fileName", fileName);
-                                fileInfo.put("filePath", filePath);
-                                peerObject.append("files", fileInfo);
-                            }
-                        });
+            
+                                if (fileExists == false) {
+                                    JSONObject fileInfo = new JSONObject();
+                                    fileInfo.put("fileName", fileName);
+                                    fileInfo.put("filePath", filePath);
+                                    peerObject.append("files", fileInfo);
+                                }
+                            });
+                        }
                     }
                 });