info.bliki.wiki.template.Ifeq Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of bliki-core Show documentation

This is the core project.

The newest version!

package info.bliki.wiki.template;

import info.bliki.wiki.model.IWikiModel;

import java.util.List;

/**
 * A template parser function for {{ #ifeq: ... }} syntax. See Mediwiki's
 * Help:Extension:ParserFunctions
 *
 */
public class Ifeq extends AbstractTemplateFunction {
    public final static ITemplateFunction CONST = new Ifeq();

    @Override
    public String parseFunction(List list, IWikiModel model, char[] src, int beginIndex, int endIndex, boolean isSubst) {
        if (list.size() > 2) {
            String first = isSubst ? list.get(0) : parseTrim(list.get(0), model);
            String second = isSubst ? list.get(1) : parseTrim(list.get(1), model);
            if (equalsTypes(first, second)) {
                return isSubst ? list.get(2) : parseTrim(list.get(2), model);
            } else if (list.size() >= 4) {
                return isSubst ? list.get(3) : parseTrim(list.get(3), model);
            }
        }
        return null;
    }

    private boolean equalsTypes(String first, String second) {

        boolean result = false;
        if (first.length() == 0) {
            return second.length() == 0;
        }
        if (second.length() == 0) {
            return first.length() == 0;
        }
        if (first.charAt(0) == '+') {
            first = first.substring(1);
        }
        if (second.charAt(0) == '+') {
            second = second.substring(1);
        }

        try {
            double d1 = Double.parseDouble(first);
            double d2 = Double.parseDouble(second);
            if (d1 == d2) {
                result = true;
            }
        } catch (NumberFormatException e) {
            result = first.equals(second);
        }

        return result;
    }
}