Add artifacts_expire_in method for Ci::Build

1 files changed, 12 insertions, 0 deletions

diff --git a/app/models/ci/build.rb b/app/models/ci/build.rb
index 5eb9fe5f1f5..9f66ae63a55 100644
--- a/app/models/ci/build.rb
+++ b/app/models/ci/build.rb
@@ -355,6 +355,18 @@ module Ci
       !artifacts? && artifacts_expire_at && artifacts_expire_at < Time.now
     end
 
+    def artifacts_expire_in
+      artifacts_expire_at - Time.now if artifacts_expire_at
+    end
+
+    def artifacts_expire_in=(value)
+      if value
+        self.artifacts_expire_at = Time.now + ChronicDuration.parse(value)
+      else
+        self.artifacts_expire_at = nil
+      end
+    end
+
     def keep_artifacts!
       self.update(artifacts_expire_at: nil)
     end