Add latest changes from gitlab-org/gitlab@13-0-stable-ee

1 files changed, 31 insertions, 0 deletions

diff --git a/app/models/namespace/root_storage_size.rb b/app/models/namespace/root_storage_size.rb
new file mode 100644
index 00000000000..d61917e468e
--- /dev/null
+++ b/app/models/namespace/root_storage_size.rb
@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+class Namespace::RootStorageSize
+  def initialize(root_namespace)
+    @root_namespace = root_namespace
+  end
+
+  def above_size_limit?
+    return false if limit == 0
+
+    usage_ratio > 1
+  end
+
+  def usage_ratio
+    return 0 if limit == 0
+
+    current_size.to_f / limit.to_f
+  end
+
+  def current_size
+    @current_size ||= root_namespace.root_storage_statistics&.storage_size
+  end
+
+  def limit
+    @limit ||= Gitlab::CurrentSettings.namespace_storage_size_limit.megabytes
+  end
+
+  private
+
+  attr_reader :root_namespace
+end