net.dv8tion.jda.internal.handle.GuildStickersUpdateHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of JDA Show documentation
Java wrapper for the popular chat & VOIP service: Discord https://discord.com
The newest version!
/*
 * Copyright 2015 Austin Keener, Michael Ritter, Florian Spieß, and the JDA contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.dv8tion.jda.internal.handle;

import gnu.trove.map.TLongObjectMap;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.sticker.GuildSticker;
import net.dv8tion.jda.api.events.sticker.GuildStickerAddedEvent;
import net.dv8tion.jda.api.events.sticker.GuildStickerRemovedEvent;
import net.dv8tion.jda.api.events.sticker.update.GuildStickerUpdateAvailableEvent;
import net.dv8tion.jda.api.events.sticker.update.GuildStickerUpdateDescriptionEvent;
import net.dv8tion.jda.api.events.sticker.update.GuildStickerUpdateNameEvent;
import net.dv8tion.jda.api.events.sticker.update.GuildStickerUpdateTagsEvent;
import net.dv8tion.jda.api.utils.cache.CacheFlag;
import net.dv8tion.jda.api.utils.data.DataArray;
import net.dv8tion.jda.api.utils.data.DataObject;
import net.dv8tion.jda.internal.JDAImpl;
import net.dv8tion.jda.internal.entities.EntityBuilder;
import net.dv8tion.jda.internal.entities.GuildImpl;
import net.dv8tion.jda.internal.entities.sticker.GuildStickerImpl;
import net.dv8tion.jda.internal.utils.Helpers;
import net.dv8tion.jda.internal.utils.UnlockHook;
import net.dv8tion.jda.internal.utils.cache.SnowflakeCacheViewImpl;
import org.apache.commons.collections4.CollectionUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

public class GuildStickersUpdateHandler extends SocketHandler
{
    public GuildStickersUpdateHandler(JDAImpl api)
    {
        super(api);
    }

    @Override
    protected Long handleInternally(DataObject content)
    {
        if (!getJDA().isCacheFlagSet(CacheFlag.STICKER))
            return null;
        final long guildId = content.getLong("guild_id");
        if (getJDA().getGuildSetupController().isLocked(guildId))
            return guildId;

        GuildImpl guild = (GuildImpl) getJDA().getGuildById(guildId);
        if (guild == null)
        {
            getJDA().getEventCache().cache(EventCache.Type.GUILD, guildId, responseNumber, allContent, this::handle);
            return null;
        }

        DataArray array = content.getArray("stickers");
        List oldStickers, newStickers;
        SnowflakeCacheViewImpl stickersView = guild.getStickersView();
        EntityBuilder builder = api.getEntityBuilder();
        try (UnlockHook hook = stickersView.writeLock())
        {
            TLongObjectMap stickersMap = stickersView.getMap();
            oldStickers = new ArrayList<>(stickersMap.valueCollection()); //snapshot of sticker cache
            newStickers = new ArrayList<>();
            for (int i = 0; i < array.length(); i++)
            {
                DataObject current = array.getObject(i);
                final long stickerId = current.getLong("id");
                GuildStickerImpl sticker = (GuildStickerImpl) stickersMap.get(stickerId);
                GuildStickerImpl oldSticker = null;

                if (sticker == null)
                {
                    sticker = (GuildStickerImpl) builder.createRichSticker(current);
                    newStickers.add(sticker);
                }
                else
                {
                    // sticker is in our cache which is why we don't want to remove it in cleanup later
                    oldStickers.remove(sticker);
                    oldSticker = sticker.copy();
                }

                sticker.setName(current.getString("name"));
                sticker.setAvailable(current.getBoolean("available"));
                sticker.setDescription(current.getString("description", ""));
                sticker.setTags(Helpers.setOf(current.getString("tags").split(",\\s*")));

                // finally, update the sticker
                stickersMap.put(sticker.getIdLong(), sticker);
                // check for updated fields and fire events
                handleReplace(guild, oldSticker, sticker);
            }
            for (GuildSticker e : oldStickers)
                stickersMap.remove(e.getIdLong());
        }
        //cleanup old stickers that don't exist anymore
        for (GuildSticker e : oldStickers)
        {
            getJDA().handleEvent(
                new GuildStickerRemovedEvent(
                    getJDA(), responseNumber,
                    guild, e));
        }

        for (GuildSticker e : newStickers)
        {
            getJDA().handleEvent(
                new GuildStickerAddedEvent(
                    getJDA(), responseNumber,
                    guild, e));
        }

        return null;
    }

    private void handleReplace(Guild guild, GuildStickerImpl oldSticker, GuildStickerImpl newSticker)
    {
        if (oldSticker == null || newSticker == null) return;

        if (!Objects.equals(oldSticker.getName(), newSticker.getName()))
        {
            getJDA().handleEvent(
                new GuildStickerUpdateNameEvent(
                    getJDA(), responseNumber,
                    guild, newSticker, oldSticker.getName()));
        }

        if (!Objects.equals(oldSticker.getDescription(), newSticker.getDescription()))
        {
            getJDA().handleEvent(
                new GuildStickerUpdateDescriptionEvent(
                    getJDA(), responseNumber,
                    guild, newSticker, oldSticker.getDescription()));
        }

        if (oldSticker.isAvailable() != newSticker.isAvailable())
        {
            getJDA().handleEvent(
                new GuildStickerUpdateAvailableEvent(
                    getJDA(), responseNumber,
                    guild, newSticker, oldSticker.isAvailable()));
        }

        if (!CollectionUtils.isEqualCollection(oldSticker.getTags(), newSticker.getTags()))
        {
            getJDA().handleEvent(
                new GuildStickerUpdateTagsEvent(
                    getJDA(), responseNumber,
                    guild, newSticker, oldSticker.getTags()));
        }

    }
}