ResourceBundleMessageResolver.java

package com.github.mygreen.supercsv.localization;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.MissingResourceException;
import java.util.Objects;
import java.util.Optional;
import java.util.ResourceBundle;

import com.github.mygreen.supercsv.util.ArgUtils;

/**
 * {@link ResourceBundle}を元にメッセージを解決するクラス。
 * <p>クラスパスのルートにリソース名が{@literal SuperCsvMessages}のプロパティファイルを配置していると自動的に読み込みます。</p>
 * <p>デフォルトでは、{@link ResourceBundleMessageResolver#DEFAULT_MESSAGE}に配置されているリソースファイルを読み込みます。</p>
 * 
 * @version 2.2
 * @author T.TSUCHIE
 *
 */
public class ResourceBundleMessageResolver implements MessageResolver {
    
    /**
     * デフォルトのメッセージソースのぱす
     */
    public static final String DEFAULT_MESSAGE = "com.github.mygreen.supercsv.localization.SuperCsvMessages";
    
    private final Map<ResourceBundle, List<String>> messageBundleKeys = new HashMap<ResourceBundle, List<String>>(8);
    
    private final LinkedList<ResourceBundle> messageBundles = new LinkedList<ResourceBundle>();
    
    /**
     * メッセージリソースのパスを指定して、インスタンスを作成します。
     * @param baseName メッセージリソースのパス。
     * @param appendUserResource クラスパスのルートにあるユーザ定義のメッセージソースも読み込むかどうか指定します。
     *      引数baseNameの値が {@literal sample.SampleMessages}のとき、クラスパスのルート上にある「SampleMessages」を読み込みます。
     * @throws NullPointerException baseName is null.
     * @throws IllegalArgumentException baseName is empty.
     */
    public ResourceBundleMessageResolver(final String baseName, final boolean appendUserResource) {
        ArgUtils.notEmpty(baseName, "baseName");
        
        addResourceBundle(ResourceBundle.getBundle(baseName, new EncodingControl("UTF-8")));
        
        // ユーザ定義のリソースを読み込む
        if(appendUserResource) {
            // リソース名の切り出し
            final int index = baseName.lastIndexOf(".");
            if(index > 0) {
                final String userName = baseName.substring(index+1);
                try {
                    addResourceBundle(ResourceBundle.getBundle(userName, new EncodingControl("UTF-8")));
                } catch(Throwable e) { }
            }
        }
        
    }
    
    /**
     * デフォルトのコンストラクタ。
     * <p>デフォルトのメッセージソース{@link #DEFAULT_MESSAGE}が自動的に読み込まれます。</p>
     * 
     */
    public ResourceBundleMessageResolver() {
        this(DEFAULT_MESSAGE, true);
    }
    
    /**
     * 独自のメッセージソースを指定してインスタンスを作成する。
     * <p>デフォルトのメッセージソース{@link #DEFAULT_MESSAGE}が自動的に読み込まれます。</p>
     * @param resourceBundle 独自のメッセージメース
     * @throws NullPointerException resourceBundle is null.
     */
    public ResourceBundleMessageResolver(final ResourceBundle resourceBundle) {
        this();
        Objects.requireNonNull(resourceBundle, "resourceBundle should no be null");
        addResourceBundle(resourceBundle);
    }
    
    /**
     * {@inheritDoc}
     */
    public Optional<String> getMessage(final String code) {
        for(final ResourceBundle bundle : messageBundles) {
            final List<String> keys = messageBundleKeys.get(bundle);
            if(keys.contains(code)) {
                try {
                    return Optional.of(bundle.getString(code));
                } catch(MissingResourceException e) {
                    return Optional.empty();
                }
            }
        }
        
        return Optional.empty();
    }
    
    /**
     * メッセージソースを追加します。
     * @param resourceBundle 追加するメッセージソース。
     * @return 既に追加済みの場合はfalseを返します。
     * @throws NullPointerException resourceBundle is null.
     */
    public final boolean addResourceBundle(final ResourceBundle resourceBundle) {
        Objects.requireNonNull(resourceBundle, "resourceBundle should not be null.");
        
        if(messageBundles.contains(resourceBundle)) {
            return false;
        }
        
        messageBundles.addFirst(resourceBundle);
        final List<String> keys = new ArrayList<String>();
        
        for(final Enumeration<String> keysEnum = resourceBundle.getKeys(); keysEnum.hasMoreElements();) {
            keys.add(keysEnum.nextElement());
        }
        
        messageBundleKeys.put(resourceBundle, keys);
        
        return true;
    }
    
   /**
     * メッセージソースを削除する。
     * @param resourceBundle 削除対象のメッセージソース
     * @return 登録されているメッセージソースがある場合はtrueを返します。
     * @throws NullPointerException resourceBundle is null.
     */
    public boolean removeResourceBundle(final ResourceBundle resourceBundle) {
        Objects.requireNonNull(resourceBundle, "resourceBundle should not be null.");
        
        if(!messageBundles.contains(resourceBundle)) {
            return false;
        }
        
        messageBundles.remove(resourceBundle);
        return true;
    }
}