include( 'toolmenu.lua' )
include( 'contextmenu.lua' )
include( 'CreationMenu.lua' )

local PANEL = {}

/*---------------------------------------------------------
   Name: Paint
---------------------------------------------------------*/
function PANEL:Init()

	self.ToolMenu = vgui.Create( "ToolMenu", self )
	self.ToolMenu:Dock( RIGHT );
	self.ToolMenu:DockMargin( 0, 20, 3, 10 )
	
	self.CreateMenu = vgui.Create( "CreationMenu", self )
	self.CreateMenu:Dock( FILL );
	self.CreateMenu:DockMargin( 3, 20, 3, 10 )
	
	self.m_bHangOpen = false
	
	self:SetMouseInputEnabled( true )
	
	self.ToolToggle = vgui.Create( "DImageButton", self )
	self.ToolToggle:SetMaterial( "gui/spawnmenu_toggle" )
	self.ToolToggle:SetSize( 16, 16 );
	self.ToolToggle.DoClick = function()
	
		self.ToolMenu:SetVisible( !self.ToolMenu:IsVisible() );
		self:InvalidateLayout()
		
		if ( self.ToolMenu:IsVisible() ) then
			self.ToolToggle:SetMaterial( "gui/spawnmenu_toggle" )
		else
			self.ToolToggle:SetMaterial( "gui/spawnmenu_toggle_back" )
		end
	
	end

end

/*---------------------------------------------------------
   Name: OnClick
---------------------------------------------------------*/
function PANEL:OnMousePressed()
	
	self:Close()
	
end


/*---------------------------------------------------------
   Name: HangOpen
---------------------------------------------------------*/
function PANEL:HangOpen( bHang )
	self.m_bHangOpen = bHang
end

/*---------------------------------------------------------
   Name: HangingOpen
---------------------------------------------------------*/
function PANEL:HangingOpen()
	return self.m_bHangOpen
end

/*---------------------------------------------------------
   Name: Paint
---------------------------------------------------------*/
function PANEL:Open()

	RestoreCursorPosition()

	self.m_bHangOpen = false
	
	// If the context menu is open, try to close it..
	if ( g_ContextMenu:IsVisible() ) then 
		g_ContextMenu:Close( true )
	end
	
	if ( self:IsVisible() ) then return end
	
	CloseDermaMenus()
	
	self:MakePopup()
	self:SetVisible( true )
	self:SetKeyboardInputEnabled( false )
	self:SetMouseInputEnabled( true )
	self:SetAlpha( 255 )

	achievements.SpawnMenuOpen()

end

/*---------------------------------------------------------
   Name: Paint
---------------------------------------------------------*/
function PANEL:Close( bSkipAnim )

	if ( self.m_bHangOpen ) then 
		self.m_bHangOpen = false
		return
	end
	
	RememberCursorPosition()
	
	CloseDermaMenus()

	self:SetKeyboardInputEnabled( false )
	self:SetMouseInputEnabled( false )
	self:SetVisible( false )

end

/*---------------------------------------------------------
   Name: PerformLayout
---------------------------------------------------------*/
function PANEL:PerformLayout()

	self:SetSize( ScrW(), ScrH() )
	self:SetPos( 0, 0 )
	
	self.ToolToggle:AlignRight( 2 )
	self.ToolToggle:AlignTop( 2 )	

end

/*---------------------------------------------------------
   Name: StartKeyFocus
---------------------------------------------------------*/
function PANEL:StartKeyFocus( pPanel )

	self.m_pKeyFocus = pPanel
	self:SetKeyboardInputEnabled( true )
	self:HangOpen( true )
	
	g_ContextMenu:StartKeyFocus( pPanel )
	
end

/*---------------------------------------------------------
   Name: EndKeyFocus
---------------------------------------------------------*/
function PANEL:EndKeyFocus( pPanel )

	if ( self.m_pKeyFocus != pPanel ) then return end
	self:SetKeyboardInputEnabled( false )
	
	g_ContextMenu:EndKeyFocus( pPanel )

end

vgui.Register( "SpawnMenu", PANEL, "EditablePanel" )


/*---------------------------------------------------------
   Called to create the spawn menu..
---------------------------------------------------------*/
local function CreateSpawnMenu()

	// If we have an old spawn menu remove it.
	if ( g_SpawnMenu ) then
	
		g_SpawnMenu:Remove()
		g_SpawnMenu = nil
	
	end
	
	// Start Fresh
	spawnmenu.ClearToolMenus()
	
	// Add defaults for the gamemode. In sandbox these defaults
	// are the Main/Postprocessing/Options tabs.
	// They're added first in sandbox so they're always first
	hook.Call( "AddGamemodeToolMenuTabs", GAMEMODE )

	// Use this hook to add your custom tools
	// This ensures that the default tabs are always
	// first.
	hook.Call( "AddToolMenuTabs", GAMEMODE )
	
	// Use this hook to add your custom tools
	// We add the gamemode tool menu categories first
	// to ensure they're always at the top.
	hook.Call( "AddGamemodeToolMenuCategories", GAMEMODE )
	hook.Call( "AddToolMenuCategories", GAMEMODE )
	
	// Add the tabs to the tool menu before trying
	// to populate them with tools.
	hook.Call( "PopulateToolMenu", GAMEMODE )

	g_SpawnMenu = vgui.Create( "SpawnMenu" )
	g_SpawnMenu:SetVisible( false )
	
	CreateContextMenu()

	hook.Call( "PostReloadToolsMenu", GAMEMODE )

end

function GM:OnSpawnMenuOpen()

	// Let the gamemode decide whether we should open or not..
	if ( !hook.Call( "SpawnMenuOpen", GAMEMODE ) ) then return end

	if (g_SpawnMenu) then g_SpawnMenu:Open() end
	
end

function GM:OnSpawnMenuClose()

	if (g_SpawnMenu) then g_SpawnMenu:Close() end 
	
end


// Development command to reload the spawnmenu (usually just use gamemode_reload)
concommand.Add( "spawnmenu_reload", function() if (g_SpawnMenu) then g_SpawnMenu:Remove(); end CreateSpawnMenu() end )

// Hook to create the spawnmenu at the appropriate time (when all sents and sweps are loaded)
hook.Add( "OnGamemodeLoaded", "CreateSpawnMenu", CreateSpawnMenu )


/*---------------------------------------------------------
   Name: HOOK SpawnMenuKeyboardFocusOn
		Called when text entry needs keyboard focus
---------------------------------------------------------*/
local function SpawnMenuKeyboardFocusOn( pnl )

	if ( !ValidPanel( g_SpawnMenu ) && !ValidPanel( g_ContextMenu ) ) then return end
	if ( IsValid( pnl ) && !pnl:HasParent( g_SpawnMenu ) && !pnl:HasParent( g_ContextMenu ) ) then return end
	
	g_SpawnMenu:StartKeyFocus( pnl )

end

hook.Add( "OnTextEntryGetFocus", "SpawnMenuKeyboardFocusOn", SpawnMenuKeyboardFocusOn )


/*---------------------------------------------------------
   Name: HOOK SpawnMenuKeyboardFocusOff
		Called when text entry stops needing keyboard focus
---------------------------------------------------------*/
local function SpawnMenuKeyboardFocusOff( pnl )

	if ( !ValidPanel( g_SpawnMenu ) && !ValidPanel( g_ContextMenu ) ) then return end
	if ( IsValid( pnl ) && !pnl:HasParent( g_SpawnMenu ) && !pnl:HasParent( g_ContextMenu ) ) then return end
	
	g_SpawnMenu:EndKeyFocus( pnl )

end

hook.Add( "OnTextEntryLoseFocus", "SpawnMenuKeyboardFocusOff", SpawnMenuKeyboardFocusOff )

/*---------------------------------------------------------
   Name: HOOK SpawnMenuOpenGUIMousePressed
		Don't do context screen clicking if spawnmenu is open
---------------------------------------------------------*/
local function SpawnMenuOpenGUIMousePressed()

	if ( !ValidPanel( g_SpawnMenu ) ) then return end
	if ( !g_SpawnMenu:IsVisible() ) then return end
	
	return true

end

hook.Add( "GUIMousePressed", "SpawnMenuOpenGUIMousePressed", SpawnMenuOpenGUIMousePressed )

/*---------------------------------------------------------
   Name: HOOK SpawnMenuOpenGUIMousePressed
		Close spawnmenu if it's open
---------------------------------------------------------*/
local function SpawnMenuOpenGUIMouseReleased()

	if ( !ValidPanel( g_SpawnMenu ) ) then return end
	if ( !g_SpawnMenu:IsVisible() ) then return end
	
	g_SpawnMenu:Close()
	
	return true

end

hook.Add( "GUIMouseReleased", "SpawnMenuOpenGUIMouseReleased", SpawnMenuOpenGUIMouseReleased )