Re-write source selection to not use the not_nil! method

1 files changed, 3 insertions, 6 deletions

diff --git a/src/flint.cr b/src/flint.cr
index 89f1d3f..02255b2 100644
--- a/src/flint.cr
+++ b/src/flint.cr
@@ -55,12 +55,9 @@ module Flint
 
     parser.parse
 
-    if read_stdin
-      source_io = STDIN
-    elsif !source_file.nil?
-      # compilier complains without the not_nil!
-      source_io = File.new(source_file.not_nil!)
-    else
+    source_io = read_stdin ? STDIN : source_file.try { |f| File.new(f) }
+
+    if source_io.nil?
       puts "ERROR: No source file chosen"
       puts parser
       exit(1)