org.jspringbot.keyword.i18n.I18nExpressionHandler Maven / Gradle / Ivy

Go to download
package org.jspringbot.keyword.i18n;

import org.apache.commons.lang.StringUtils;
import org.jspringbot.keyword.expression.plugin.AbstractSimpleExpressionHandler;
import org.jspringbot.syntax.HighlightRobotLogger;

public class I18nExpressionHandler extends AbstractSimpleExpressionHandler {
    public static final HighlightRobotLogger LOG = HighlightRobotLogger.getLogger(I18nExpressionHandler.class);

    public static final String EXPRESSION_PREFIX = "i18n";

    private I18nHelper helper;

    public I18nExpressionHandler(I18nHelper helper) {
        this.helper = helper;
    }

    @Override
    public String getPrefix() {
        return EXPRESSION_PREFIX;
    }

    @Override
    protected Object evaluateInternal(String expression, String[] tokens) {
        LOG.keywordAppender().appendProperty("Expression Handler", "jSpringBot i18n");

        if(tokens.length == 1) {
            return helper.getMessage(tokens[0]);
        }

        if(tokens.length == 2) {
            if(StringUtils.equalsIgnoreCase(tokens[0], "locale")) {
                if(StringUtils.equalsIgnoreCase(tokens[1], "language")) {
                    return helper.getLanguage();
                } else if(StringUtils.equalsIgnoreCase(tokens[1], "country")) {
                    return helper.getCountry();
                } else if(StringUtils.equalsIgnoreCase(tokens[1], "displayCountry")) {
                    return helper.getDisplayCountry();
                } else if(StringUtils.equalsIgnoreCase(tokens[1], "displayLanguage")) {
                    return helper.getDisplayLanguage();
                }

                throw new IllegalArgumentException(String.format("Syntax error for i18 locale property '%s' unknown.", tokens[1]));
            }

            return helper.getMessage(tokens[0], tokens[1]);
        }

        throw new IllegalArgumentException(String.format("Syntax error for i18 expression '%s'.", expression));
    }
}