When --no-prepare is selected, actually update fly.toml

lib/generators/dockerfile_generator.rb

@@ -282,14 +282,12 @@ def generate_app
       fly_attach_consul
     end
 
-    if fly_processes # therefore File.exist?('fly.toml')
+    if File.exist?('fly.toml') && (fly_processes || !options.prepare)
       if File.stat("fly.toml").size > 0
         template "fly.toml.erb", "fly.toml"
       else
-        toml = fly_make_processes(fly_processes)
-        if toml != IO.read("fly.toml")
-          File.write "fly.toml", toml
-        end
+        toml = fly_make_toml
+        File.write "fly.toml", toml if toml != ""
       end
     end
 
@@ -1042,19 +1040,31 @@ def fly_attach_consul
     system "#{flyctl} consul attach"
   end
 
-  def fly_make_processes(list)
+  def fly_make_toml
     toml = File.read("fly.toml")
 
-    if toml.include? "[processes]"
-      toml.sub!(/\[processes\].*?(\n\n|\n?\z)/m, "[processes]\n" +
-        list.map { |name, cmd| "  #{name} = #{cmd.inspect}" }.join("\n") + '\1')
-    else
-      toml += "\n[processes]\n" +
-        list.map { |name, cmd| "  #{name} = #{cmd.inspect}\n" }.join
+    list = fly_processes
+    if list
+      if toml.include? "[processes]"
+        toml.sub!(/\[processes\].*?(\n\n|\n?\z)/m, "[processes]\n" +
+          list.map { |name, cmd| "  #{name} = #{cmd.inspect}" }.join("\n") + '\1')
+      else
+        toml += "\n[processes]\n" +
+          list.map { |name, cmd| "  #{name} = #{cmd.inspect}\n" }.join
+
+        app = list.has_key?("app") ? "app" : list.keys.first
 
-      app = list.has_key?("app") ? "app" : list.keys.first
+        toml.sub! "[http_service]\n", "\\0  processes = [#{app.inspect}]\n"
+      end
+    end
 
-      toml.sub! "[http_service]\n", "\\0  processes = [#{app.inspect}]\n"
+    if options.prepare == false
+      deploy = "[deploy]\n  release_command = #{dbprep_command.inspect}\n\n"
+      if toml.include? "[deploy]"
+        toml.sub! /\[deploy\].*?(\n\n|\n?\z)/m, deploy
+      else
+        toml += deploy
+      end
     end
 
     toml

lib/generators/templates/fly.toml.erb

@@ -1 +1 @@
-<%= fly_make_processes(fly_processes) -%>
+<%= fly_make_toml -%>

test/results/no_prep/fly.toml

@@ -0,0 +1,3 @@
+[deploy]
+  release_command = "./bin/rails db:prepare"
+

test/test_no_prep.rb

@@ -6,8 +6,13 @@ class TestNoPrep < TestBase
   @rails_options = "--minimal"
   @generate_options = "--no-prepare"
 
+  def app_setup
+    FileUtils.touch "fly.toml"
+  end
+
   def test_no_prep
     check_dockerfile
     check_entrypoint
+    check_toml
   end
 end