/*
 * Copyright 2014 astamuse company,Ltd.
 * 
 * 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 com.astamuse.asta4d.util.i18n.pattern;

import java.util.Locale;
import java.util.MissingResourceException;
import java.util.ResourceBundle;

import com.astamuse.asta4d.Configuration;
import com.astamuse.asta4d.util.i18n.LocalizeUtil;

/**
 * We allow row splitted messages. For a given key, if there is no corresponding message, we will try to search keys as "key#1", "key#2"...
 * then combine them as a single message.
 * 
 * @author e-ryu
 * 
 */
public class JDKResourceBundleMessagePatternRetriever implements MessagePatternRetriever {

    private ResourceBundleFactory resourceBundleFactory = new CharsetResourceBundleFactory();

    protected String[] resourceNames = new String[0];

    public ResourceBundleFactory getResourceBundleFactory() {
        return resourceBundleFactory;
    }

    public void setResourceBundleFactory(ResourceBundleFactory resourceBundleFactory) {
        this.resourceBundleFactory = resourceBundleFactory;
    }

    public void setResourceNames(String... resourceNames) {
        this.resourceNames = resourceNames.clone();
    }

    @Override
    public String retrieve(Locale locale, String key) {
        String pattern = null;
        for (String resourceName : resourceNames) {
            try {
                ResourceBundle resourceBundle = getResourceBundle(resourceName, locale);
                pattern = retrieveResourceFromBundle(resourceBundle, key);
                // if the pattern is not found, MissingResourceException will be
                // thrown and the break will be skipped
                break;
            } catch (MissingResourceException e) {
                //
            }
        }
        return pattern;
    }

    protected ResourceBundle getResourceBundle(String resourceName, Locale locale) {
        Configuration config = Configuration.getConfiguration();

        if (!config.isCacheEnable()) {
            ResourceBundle.clearCache();
        }

        return resourceBundleFactory.retrieveResourceBundle(resourceName, LocalizeUtil.defaultWhenNull(locale));
    }

    /**
     * In this method, we allow row splitted message. For a given key, if there is no corresponding message, we will try to search keys as
     * "key#1", "key#2"... then combine them as a single message.
     * 
     * @param bundle
     * @param key
     * @throws MissingResourceException
     *             if no object for the given key can be found
     * @return
     */
    protected String retrieveResourceFromBundle(ResourceBundle bundle, String key) throws MissingResourceException {
        String res = null;
        try {
            return bundle.getString(key);
        } catch (MissingResourceException e) {
            // check if there is a splitted message
            try {
                res = bundle.getString(key + "#1");
            } catch (MissingResourceException ex) {
                // we throw the original exception
                throw e;
            }

            StringBuilder sb = new StringBuilder(res.length() * 3);
            sb.append(res);
            try {
                for (int row = 2;/* loop for ever */; row++) {
                    sb.append(bundle.getString(key + "#" + row));
                }
            } catch (MissingResourceException ex) {
                return sb.toString();
            }

        }
    }
}