From ee7c5539f38c5e66d06610d457efe983196372e2 Mon Sep 17 00:00:00 2001
From: Kamil Trzcinski <ayufan@ayufan.eu>
Date: Fri, 10 Jun 2016 14:25:54 +0200
Subject: Add artifacts_expire_in method for Ci::Build

---
 app/models/ci/build.rb | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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
-- 
cgit v1.2.1