src/main/java/net/glowstone/io/entity/FireworkStore.java

package net.glowstone.io.entity;

import java.util.UUID;
import net.glowstone.entity.passive.GlowFirework;
import net.glowstone.io.nbt.NbtSerialization;
import net.glowstone.util.nbt.CompoundTag;
import org.bukkit.Location;
import org.bukkit.entity.EntityType;

public class FireworkStore extends EntityStore<GlowFirework> {

    public FireworkStore() {
        super(GlowFirework.class, EntityType.FIREWORK);
    }

    @Override
    public GlowFirework createEntity(Location location, CompoundTag compound) {
        return new GlowFirework(location, null, null, null);
    }

    @Override
    public void load(GlowFirework entity, CompoundTag tag) {
        super.load(entity, tag);

        if (tag.isInt("Life")) {
            entity.setTicksLived(tag.getInt("Life"));
        }
        if (tag.isInt("LifeTime")) {
            entity.setLifeTime(tag.getInt("LifeTime"));
        }
        if (tag.isCompound("FireworksItem")) {
            entity.setFireworkItem(NbtSerialization.readItem(tag.getCompound("FireworksItem")));
        }
        if (tag.isLong("SpawningEntityMost") && tag.isLong("SpawningEntityLeast")) {
            UUID uuid = new UUID(tag.getLong("SpawningEntityMost"), tag.getLong("SpawningEntityLeast"));
            entity.setSpawningEntity(uuid);
        }
    }

    @Override
    public void save(GlowFirework entity, CompoundTag tag) {
        super.save(entity, tag);

        tag.putInt("Life", entity.getTicksLived());
        tag.putInt("LifeTime", entity.getLifeTime());
        CompoundTag fireworkItem = NbtSerialization.writeItem(entity.getFireworkItem(), -1);
        tag.putCompound("FireworksItem", fireworkItem);

        tag.putLong("SpawningEntityMost", entity.getSpawningEntity().getMostSignificantBits());
        tag.putLong("SpawningEntityLeast", entity.getSpawningEntity().getLeastSignificantBits());
    }
}