Handling scratch files which have a size of 0.

dchelimsky · Jun 12, 2009 · 67ea48f · 67ea48f
1 parent 550ebe5
commit 67ea48f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/generators/saikuro.rb b/lib/generators/saikuro.rb
@@ -106,7 +106,7 @@ def initialize(path)
 
   def self.is_valid_text_file?(path)
     File.open(path, "r") do |f|
-      unless f.readline.match /--/
+      if f.eof? || !f.readline.match /--/
         return false
       else
         return true
@@ -125,8 +125,8 @@ def to_h
 
   def get_elements
     begin
-      while ( line = @file_handle.readline) do
-			  element ||= nil
+      while (line = @file_handle.readline) do
+        return [] if line.blank?
         if line.match /START/
 				  unless element.nil?
 					  @elements << element