lib/souls/generate/mutation.rb in souls-0.28.5 vs lib/souls/generate/mutation.rb in souls-0.28.6

@@ -28,11 +28,11 @@
       @relation_params = []
       File.open(file_path, "a") do |new_line|
         File.open(path, "r") do |f|
           f.each_line.with_index do |line, _i|
             if @on
-              if line.include?("end") || line.include?("t.index")
+              if line.include?("t.index") || line.strip == "end"
                 if @user_exist
                   new_line.write(<<-TEXT)
 
     def resolve **args
       args[:user_id] = context[:user].id
@@ -122,10 +122,10 @@
       @relation_params = []
       File.open(file_path, "a") do |new_line|
         File.open(path, "r") do |f|
           f.each_line.with_index do |line, _i|
             if @on
-              if line.include?("end") || line.include?("t.index")
+              if line.include?("t.index") || line.strip == "end"
                 if @user_exist
                   new_line.write(<<-TEXT)
 
     def resolve **args
       args[:user_id] = context[:user].id