net.dv8tion.jda.api.interactions.commands.localization.LocalizationMap 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.api.interactions.commands.localization;

import net.dv8tion.jda.api.interactions.DiscordLocale;
import net.dv8tion.jda.api.utils.data.DataObject;
import net.dv8tion.jda.api.utils.data.SerializableData;
import net.dv8tion.jda.internal.utils.Checks;
import net.dv8tion.jda.internal.utils.JDALogger;
import org.slf4j.Logger;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Consumer;

/**
 * Class which contains a mapping from {@link DiscordLocale} to a translated String, similar to a {@code Map}.
 * 
This is used for command, option, and choice localization.
 */
public class LocalizationMap implements SerializableData
{
    public static final Logger LOG = JDALogger.getLog(LocalizationMap.class);

    protected final Map map = new HashMap<>();
    private final Consumer checkConsumer;

    public LocalizationMap(@Nonnull Consumer checkConsumer)
    {
        this.checkConsumer = checkConsumer;
    }

    private void putTranslation(DiscordLocale locale, String translation)
    {
        Checks.check(locale != DiscordLocale.UNKNOWN, "Cannot put an 'UNKNOWN' DiscordLocale");
        this.map.put(locale, translation);
    }

    @Nonnull
    @Override
    public DataObject toData()
    {
        final DataObject data = DataObject.empty();
        map.forEach((locale, localizedString) -> data.put(locale.getLocale(), localizedString));
        return data;
    }

    /**
     * Sets the given localized string to be used for the specified locale.
     *
     * @param  locale
     *         The locale on which to apply the localized string
     *
     * @param  localizedString
     *         The localized string to use
     *
     * @throws IllegalArgumentException
     *         
     *             If the locale is null
     *             If the localized string is null
     *             If the locale is {@link DiscordLocale#UNKNOWN}
     *             If the localized string does not pass the corresponding attribute check
     *         
     */
    public void setTranslation(@Nonnull DiscordLocale locale, @Nonnull String localizedString)
    {
        Checks.notNull(locale, "Locale");
        Checks.notNull(localizedString, "Localized string");

        checkConsumer.accept(localizedString);
        putTranslation(locale, localizedString);
    }

    /**
     * Adds all the translations from the supplied map into this LocalizationMap.
     *
     * @param  map
     *         The map containing the localized strings
     *
     * @throws IllegalArgumentException
     *         
     *             If the map is null
     *             If the map contains an {@link DiscordLocale#UNKNOWN} key
     *             If the map contains a localized string which does not pass the corresponding attribute check
     *         
     */
    public void setTranslations(@Nonnull Map map)
    {
        Checks.notNull(map, "Map");

        map.forEach((discordLocale, localizedString) ->
        {
            checkConsumer.accept(localizedString);
            putTranslation(discordLocale, localizedString);
        });
    }

    /**
     * Gets the localized string for the specified {@link DiscordLocale}.
     *
     * @param  locale
     *         The locale from which to get the localized string
     *
     * @return Possibly-null localized string
     */
    @Nullable
    public String get(@Nonnull DiscordLocale locale)
    {
        Checks.notNull(locale, "Locale");

        return map.get(locale);
    }

    /**
     * Gets the unmodifiable map representing this LocalizationMap.
     * 
The changes on this LocalizationMap will be reflected on the returned map.
     *
     * @return The unmodifiable map of this LocalizationMap
     */
    @Nonnull
    public Map toMap()
    {
        return Collections.unmodifiableMap(map);
    }
}