MessageManager.java

package com.supaham.commons.bukkit.language;

import com.supaham.commons.placeholders.Placeholder;
import com.supaham.commons.placeholders.PlaceholderData;
import com.supaham.commons.placeholders.PlaceholderSet;

import org.jetbrains.annotations.NotNull;

import java.util.HashMap;
import java.util.Map;

import javax.annotation.Nullable;

/**
 * Represents a {@link Message} manager.
 *
 * @since 0.1
 */
public class MessageManager {

  private Map<String, Message> messages = new HashMap<>();

  private final Map<Character, Theme> themes;

  private PlaceholderSet<? extends Placeholder> placeholderSet;

  public MessageManager() {
    this(null);
  }

  public MessageManager(Map<Character, Theme> themes) {
    this.themes = themes == null ? new HashMap<Character, Theme>() : themes;
  }

  public String parseMessage(@NotNull final String message, @Nullable Map<Object, Object> locals) {
    final StringBuilder buffer = new StringBuilder(message.length() + 10);
    char previousChar = ' ';
    for (int i = 0; i < message.length() - 1; i++) {
      final char currentChar = message.charAt(i);
      if (currentChar == Theme.THEME_MARKER && previousChar != Theme.THEME_ESCAPE_CHAR) {
        Theme theme = themes.get(message.charAt(i + 1));
        if (theme != null) {
          buffer.append(theme.getTheme());
          i++;
        } else {
          buffer.append(currentChar);
        }
      } else {
        buffer.append(currentChar);
      }
      previousChar = currentChar;
    }
    String result = "";
    if (!message.isEmpty()) {
      buffer.append(message.charAt(message.length() - 1));
      result = buffer.toString();

      if (this.placeholderSet != null && locals != null) {
        PlaceholderData data = PlaceholderData.builder().input(result).put(locals).build();
        result = this.placeholderSet.apply(data);
      }
    }
    return result;
  }

  /**
   * Gets a {@link Message} by node.
   *
   * @param node node of the {@link Message} to get
   *
   * @return the {@link Message} object
   */
  public Message getMessage(@NotNull String node) {
    return messages.get(node);
  }

  /**
   * Adds a {@link Message} to this {@link MessageManager}.
   *
   * @param message message to add
   */
  public void addMessage(@NotNull Message message) {
    this.messages.put(message.getNode(), message);
  }

  /**
   * Gets a {@link Theme} by {@link Character}.
   *
   * @param character character of the theme to get
   *
   * @return the {@link Theme} belonging to the {@code character}
   */
  public Theme getTheme(@NotNull Character character) {
    return this.themes.get(character);
  }

  /**
   * Adds a {@link Theme} to this {@link MessageManager}.
   *
   * @param theme theme to add
   *
   * @return old theme that was replaced
   */
  public Theme addTheme(@NotNull Theme theme) {
    return this.themes.put(theme.getCode(), theme);
  }

  public PlaceholderSet<? extends Placeholder> getPlaceholderSet() {
    return placeholderSet;
  }

  public void setPlaceholderSet(@Nullable PlaceholderSet<? extends Placeholder> set) {
    this.placeholderSet = set;
  }

  public Map<String, Message> getMessages() {
    return messages;
  }
}