1 files changed, 13 insertions, 3 deletions
diff --git a/lib/gitlab_access_status.rb b/lib/gitlab_access_status.rb
index 7fb88be..4b9d32a 100644
--- a/lib/gitlab_access_status.rb
+++ b/lib/gitlab_access_status.rb
@@ -1,20 +1,30 @@
 require 'json'
 
 class GitAccessStatus
-  attr_reader :message, :repository_path
+  attr_reader :message, :repository_path, :memory_status, :memory_message
 
-  def initialize(status, message, repository_path)
+  def initialize(status, message, repository_path, memory = nil)
     @status = status
     @message = message
     @repository_path = repository_path
+
+    if memory
+      @memory_status = memory["status"]
+      @memory_message = memory["message"]
+    end
   end
 
   def self.create_from_json(json)
     values = JSON.parse(json)
-    self.new(values["status"], values["message"], values["repository_path"])
+    self.new(values["status"], values["message"],
+             values["repository_path"], values["memory"])
   end
 
   def allowed?
     @status
   end
+
+  def memory_limit?
+    @memory_status
+  end
 end