info.bliki.wiki.template.Padright 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 {{padright: ... }}, which pads
 * a string with a character to the specified width. See: Magic words
 * - Formatting
 *
 */
public class Padright extends AbstractTemplateFunction {
    public final static ITemplateFunction CONST = new Padright();

    @Override
    public String parseFunction(List list, IWikiModel model, char[] src, int beginIndex, int endIndex, boolean isSubst) {
        if (list.size() == 1) {
            return isSubst ? list.get(0) : parseTrim(list.get(0), model);
        }
        if (list.size() > 1) {
            String arg0 = isSubst ? list.get(0) : parseTrim(list.get(0), model);
            String arg1 = isSubst ? list.get(1) : parseTrim(list.get(1), model);
            String arg2 = "0"; // default value
            if (list.size() > 2) {
                arg2 = isSubst ? list.get(2) : parseTrim(list.get(2), model);
            }
            int arg1Int = 0;
            try {
                arg1Int = Integer.parseInt(arg1);
            } catch (NumberFormatException e) {
                return arg0;
            }
            if (arg1Int < 0) {
                return null;
            }
            if (arg0.length() >= arg1Int) {
                return arg0.substring(0, arg1Int);
            }
            if (arg2.length() != 1) {
                return null;
            }
            arg1Int = arg1Int - arg0.length();
            char arg2Char = arg2.charAt(0);
            StringBuilder buffer = new StringBuilder(arg1Int + arg0.length());
            buffer.append(arg0);
            for (int i = 0; i < arg1Int; i++) {
                buffer.append(arg2Char);
            }
            return buffer.toString();
        }
        return null;
    }
}