Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/items/NewUserGift.java

package com.eu.habbo.habbohotel.items;

import com.eu.habbo.Emulator;
import com.eu.habbo.habbohotel.users.Habbo;
import com.eu.habbo.habbohotel.users.HabboItem;
import com.eu.habbo.messages.ISerialize;
import com.eu.habbo.messages.ServerMessage;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

public class NewUserGift implements ISerialize
{
    private final int id;
    private final Type type;
    private final String imageUrl;
    private Map<String, String> items = new HashMap<>();

    public NewUserGift(ResultSet set) throws SQLException
    {
        this.id = set.getInt("id");
        this.type = Type.valueOf(set.getString("type").toUpperCase());
        this.imageUrl = set.getString("image");
        this.items.put(this.type == Type.ROOM ? "" : set.getString("value"), this.type == Type.ROOM ? set.getString("value") : "");
    }
    public NewUserGift(int id, Type type, String imageUrl, Map<String, String> items)
    {
        this.id = id;
        this.imageUrl = imageUrl;
        this.type = type;
        this.items = items;
    }

    @Override
    public void serialize(ServerMessage message)
    {
        message.appendString(this.imageUrl);
        message.appendInt(this.items.size());
        for (Map.Entry<String, String> entry : this.items.entrySet())
        {
            message.appendString(entry.getKey()); //Item Name
            message.appendString(entry.getValue()); //Extra Info
        }
    }

    public void give(Habbo habbo)
    {
        if (this.type == Type.ITEM)
        {
            for (Map.Entry<String, String> set : this.items.entrySet())
            {
                Item item = Emulator.getGameEnvironment().getItemManager().getItem(set.getKey());

                if (item != null)
                {
                    HabboItem createdItem = Emulator.getGameEnvironment().getItemManager().createItem(habbo.getHabboInfo().getId(), item, 0, 0, "");

                    if (createdItem != null)
                    {
                        habbo.addFurniture(createdItem);
                    }
                }
            }
        }
        else if (this.type == Type.ROOM)
        {
            //TODO Give room
        }
    }

    public int getId()
    {
        return this.id;
    }

    public Type getType()
    {
        return this.type;
    }

    public String getImageUrl()
    {
        return this.imageUrl;
    }

    public Map<String, String> getItems()
    {
        return this.items;
    }

    public enum Type
    {
        ITEM,
        ROOM
    }
}