Arcturus-Community/src/main/java/com/eu/habbo/messages/outgoing/inventory/InventoryBadgesComposer.java

package com.eu.habbo.messages.outgoing.inventory;

import com.eu.habbo.habbohotel.users.Habbo;
import com.eu.habbo.habbohotel.users.HabboBadge;
import com.eu.habbo.messages.ServerMessage;
import com.eu.habbo.messages.outgoing.MessageComposer;
import com.eu.habbo.messages.outgoing.Outgoing;
import gnu.trove.set.hash.THashSet;

public class InventoryBadgesComposer extends MessageComposer {
    private final Habbo habbo;

    public InventoryBadgesComposer(Habbo habbo) {
        this.habbo = habbo;
    }

    @Override
    public ServerMessage compose() {
        if (this.habbo == null)
            return null;

        THashSet<HabboBadge> equippedBadges = new THashSet<>();

        this.response.init(Outgoing.InventoryBadgesComposer);

        this.response.appendInt(this.habbo.getInventory().getBadgesComponent().getBadges().size());
        for (HabboBadge badge : this.habbo.getInventory().getBadgesComponent().getBadges()) {
            this.response.appendInt(badge.getId());
            this.response.appendString(badge.getCode());

            if (badge.getSlot() > 0)
                equippedBadges.add(badge);
        }

        this.response.appendInt(equippedBadges.size());

        for (HabboBadge badge : equippedBadges) {
            this.response.appendInt(badge.getSlot());
            this.response.appendString(badge.getCode());
        }

        return this.response;
    }
}