1 files changed, 30 insertions, 0 deletions
diff --git a/gamemode/client/cl_inventory.lua b/gamemode/client/cl_inventory.lua
index 94690ac..ab6f196 100644
--- a/gamemode/client/cl_inventory.lua
+++ b/gamemode/client/cl_inventory.lua
@@ -25,6 +25,13 @@ local function createPanel()
   frame:SetTitle( "Inventory" )
   frame:MakePopup()
   frame:SetPos(scrx-invxsize,0)
+  frame:SetKeyboardInputEnabled(true)
+  frame:ShowCloseButton(false)
+  frame.OnKeyCodePressed = function(self, key)
+    if(key == KEY_Q) then
+      frame:Close()
+    end
+  end
 
   local tabsheet = vgui.Create("DPropertySheet", frame)
   tabsheet:Dock(FILL)
@@ -41,6 +48,7 @@ local function createPanel()
   end
 
   --Inventory
+
   local layout = vgui.Create( "DTileLayout", invtab )
   layout:SetBaseSize( 64 ) -- Tile size
   layout:Dock( FILL )
@@ -48,6 +56,9 @@ local function createPanel()
   layout:MakeDroppable( "unique_name" ) -- Allows us to rearrange children
   PrintTable(Resources)
   for k, v in SortedPairs( Resources ) do
+    if(v == 0) then
+      continue
+    end
     local selection = vgui.Create("DImageButton")
     if(GMS.Resources[k] == nil) then --This resource is not registered!
       selection:SetImage("vgui/avatar_default")
@@ -77,13 +88,32 @@ local function createPanel()
     end
     layout:Add( selection )
   end
+
+  return frame
 end
 
 local invpanel = nil
 function GM:OnSpawnMenuOpen()
   print("Spawn menu hooked correctly")
+
   if(invpanel == nil) then
     invpanel = createPanel()
+    return
+  end
+  print("Not nil")
+  if(! invpanel:IsValid()) then
+    invpanel = createPanel()
+    return
+  end
+end
+
+function GM:ReloadSpawnMenu()
+  if(!invpanel:IsValid()) then
+    return
+  end
+  if(invpanel != nil) then
+    invpanel:Close()
+    invpanel = createPanel()
   end
 end