MessageStyleRegistryModule

package com.feeliiix.fundamental.plugin.registry;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkArgument;

import com.feeliiix.fundamental.api.message.style.MessageStyle;
import com.feeliiix.fundamental.api.message.style.MessageStyles;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Maps;
import org.spongepowered.api.registry.AdditionalCatalogRegistryModule;

import java.util.Collection;
import java.util.Map;
import java.util.Optional;

public final class MessageStyleRegistryModule implements AdditionalCatalogRegistryModule<MessageStyle> {

    private final Map<String, MessageStyle> messageStyleMap = Maps.newHashMap();

    @Override
    public void registerDefaults() {
        this.messageStyleMap.put("none", MessageStyles.NONE);
        this.messageStyleMap.put("normal", MessageStyles.NORMAL);
        this.messageStyleMap.put("success", MessageStyles.SUCCESS);
        this.messageStyleMap.put("warning", MessageStyles.WARNING);
        this.messageStyleMap.put("failure", MessageStyles.FAILURE);
    }

    @Override
    public void registerAdditionalCatalog(MessageStyle extraCatalog) {
        checkArgument(!this.messageStyleMap.containsKey(checkNotNull(extraCatalog).getId().toLowerCase()));
        this.messageStyleMap.put(extraCatalog.getId().toLowerCase(), extraCatalog);
    }

    @Override
    public Optional<MessageStyle> getById(String id) {
        return Optional.ofNullable(this.messageStyleMap.get(checkNotNull(id).toLowerCase()));
    }

    @Override
    public Collection<MessageStyle> getAll() {
        return ImmutableSet.copyOf(this.messageStyleMap.values());
    }

}