Add the error handling for missing/invalid file/URL input

1 files changed, 5 insertions, 0 deletions

diff --git a/CLI/ruby/lib/game_of_life/generators/input.rb b/CLI/ruby/lib/game_of_life/generators/input.rb
index c0d96c1..3082e70 100644
--- a/CLI/ruby/lib/game_of_life/generators/input.rb
+++ b/CLI/ruby/lib/game_of_life/generators/input.rb
@@ -22,6 +22,11 @@ module GameOfLife
           # where any alpha numeric character is considered a living cell (true)
           raw.map! { |row| row.chars.map { |char| char.match?(/[[:alnum:]]/) } }
           populate(parsed_input: raw, width: options["width"].to_i, height: options["height"].to_i)
+        rescue OpenURI::HTTPError, SocketError
+          raise GameOfLife::Error, "URL isn't avaialable, please check it again and make sure the URL is correct"
+        rescue Errno::ENOENT
+          raise GameOfLife::Error,
+                "File isn't avaialable, please check it again and make sure the path to the input file is correct"
         end
 
         # Populate the {GameOfLife::Universe} with reference to the parsed input 2D array (True/False)