add invalid YAML format exception for loadYaml

SaptarshiSarkar12 · Feb 24, 2024 · fafb0af · fafb0af
1 parent 68dcaf1
commit fafb0af
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CLI/src/main/java/main/Drifty_CLI.java b/CLI/src/main/java/main/Drifty_CLI.java
@@ -4,6 +4,7 @@
 import cli.utils.MessageBroker;
 import cli.init.Environment;
 import org.yaml.snakeyaml.Yaml;
+import org.yaml.snakeyaml.error.YAMLException;
 import preferences.AppSettings;
 import properties.MessageType;
 import properties.Program;
@@ -262,6 +263,8 @@ private static Map<String, List<String>> loadYamlData() {
                 data = new HashMap<>();
             }
             data.computeIfAbsent("links", k -> new ArrayList<>()); // Ensure 'links' list exists
+        } catch (YAMLException e) {
+            messageBroker.msgLogError("Invalid YAML format in file: " + e.getMessage());
         } catch (IOException e) {
             messageBroker.msgLogError("Error reading YAML file: " + e.getMessage());
         }