////////////////////////////////////////////////////////////////////////////////
//    Scorched3D (c) 2000-2011, 2026
//
//    This file is part of Scorched3D.
//
//    Scorched3D is free software; you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation; either version 2 of the License, or
//    (at your option) any later version.
//
//    Scorched3D is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License along
//    with this program; if not, write to the Free Software Foundation, Inc.,
//    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
////////////////////////////////////////////////////////////////////////////////

#ifndef __INCLUDE_TanketAccessories_hpp_INCLUDE__
#define __INCLUDE_TanketAccessories_hpp_INCLUDE__

#include <weapons/AccessoryPart.hpp>
#include <tanket/TanketWeapon.hpp>
#include <tanket/TanketAutoDefense.hpp>
#include <tanket/TanketBatteries.hpp>
#include <lang/LangString.hpp>

class Tanket;
class ScorchedContext;

class TanketAccessories
{
public:
	TanketAccessories( ScorchedContext& context );
	virtual ~TanketAccessories();

	void setTanket( Tanket* tanket );

	void newMatch();
	void add( Accessory* accessory, int count, bool check = true );
	void rm( Accessory* accessory, int count );
	void clearAccessories();

	bool canUse( Accessory* accessory );

	bool                     accessoryAllowed( Accessory* accessory, int count );
	int                      getAccessoryCount( Accessory* accessory );
	void                     getAllAccessories( std::list< Accessory* >& result );
	std::list< Accessory* >& getAllAccessoriesByType( AccessoryPart::AccessoryType type );
	std::list< Accessory* >& getAllAccessoriesByGroup( const char* groupName );

	// clang-format off
	// uncrustify off
	TanketWeapon&      getWeapons()     { return tankWeapon_; }
	TanketAutoDefense& getAutoDefense() { return tankAuto_; }
	TanketBatteries&   getBatteries()   { return tankBatteries_; }
	// uncrustify on
	// clang-format on

	LangString getAccessoryCountString( Accessory* accessory );
	LangString getAccessoryAndCountString( Accessory* accessory );

	// Serialize the tank accessories
	bool writeMessage( NamedNetBuffer& buffer, bool writeAccessories );
	bool readMessage( NetBufferReader& reader );

protected:
	ScorchedContext&  context_;
	TanketWeapon      tankWeapon_;
	TanketAutoDefense tankAuto_;
	TanketBatteries   tankBatteries_;
	Tanket*           tanket_;

	typedef std::list< Accessory* >                          AccessoryList;
	std::map< Accessory*, int >                              accessories_;
	std::map< std::string, AccessoryList* >                  accessoryGroups_;
	std::map< AccessoryPart::AccessoryType, AccessoryList* > accessoryTypes_;

	void changed();
	void add_( Accessory* accessory, int count, bool check );
};

#endif  // __INCLUDE_TanketAccessories_hpp_INCLUDE__
