Add action to deploymentadd-action-to-deployment-backup

2 files changed, 8 insertions, 1 deletions

diff --git a/app/models/concerns/deployable.rb b/app/models/concerns/deployable.rb
index ba3abf28ab0..d85d624fff3 100644
--- a/app/models/concerns/deployable.rb
+++ b/app/models/concerns/deployable.rb
@@ -21,7 +21,8 @@ module Deployable
         sha: sha,
         user: user,
         deployable: self,
-        on_stop: on_stop).tap do |_|
+        on_stop: on_stop,
+        action: environment_action).tap do |_|
         self.reload # Reload relationships
       end
     end
diff --git a/app/models/deployment.rb b/app/models/deployment.rb
index eecb927ad01..9cea601f4bf 100644
--- a/app/models/deployment.rb
+++ b/app/models/deployment.rb
@@ -47,6 +47,12 @@ class Deployment < ActiveRecord::Base
     end
   end
 
+  enum_with_nil action: {
+    unknown_action: nil,
+    start: 1,
+    stop: 2
+  }
+
   enum status: {
     created: 0,
     running: 1,