com.squarespace.template.MapFormat.java Source code

Introduction

Here is the source code for com.squarespace.template.MapFormat.java

Source

/**
 * Copyright (c) 2014 SQUARESPACE, Inc.
 *
 * 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.squarespace.template;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.lang3.StringEscapeUtils;

/**
 * Support class for formatting error messages flexibly using a Map.
 */
public class MapFormat {

    private static final Pattern KEY_ESCAPE = Pattern.compile("%\\([a-zA-Z][a-zA-Z0-9]*\\)");

    private static final String EMPTY_STRING = "";

    private List<String> keys = new ArrayList<>();

    private String format;

    private String nullPlaceholder = EMPTY_STRING;

    public MapFormat(String raw) {
        init(raw);
    }

    public MapFormat(String raw, String nullPlaceholder) {
        this.nullPlaceholder = nullPlaceholder;
        init(raw);
    }

    private void init(String raw) {
        StringBuffer buf = new StringBuffer();
        Matcher matcher = KEY_ESCAPE.matcher(raw);
        int mark = 0;
        int length = raw.length();
        while (matcher.find()) {
            int start0 = matcher.start();
            if (mark < start0) {
                buf.append(raw.substring(mark, start0));
            }
            buf.append(raw.charAt(start0));
            keys.add(raw.substring(start0 + 2, matcher.end() - 1));
            mark = matcher.end();
        }
        if (mark < length) {
            buf.append(raw.substring(mark, length));
        }
        this.format = buf.toString();
    }

    public String getFormat() {
        return format;
    }

    public String apply(Map<String, Object> params) {
        List<Object> values = new ArrayList<>(params.size());
        for (String key : keys) {
            Object obj = params.get(key);
            if (obj == null) {
                obj = nullPlaceholder;
            }
            values.add(obj);
        }
        return StringEscapeUtils.escapeJava(String.format(format, values.toArray()));
    }
}