VarInLinkExpandLinkRefTransformer.java

package com.aperture_software.glados_wiki.markdown;

import com.aperture_software.markdown4j.LinkRefTransformer;

import java.util.HashMap;
import java.util.Map;

/**
 * Created by jhyun on 13. 12. 25.
 */
public class VarInLinkExpandLinkRefTransformer implements LinkRefTransformer {

    private Map<String, String> vars = new HashMap<String, String>();

    public VarInLinkExpandLinkRefTransformer() {
    }

    public VarInLinkExpandLinkRefTransformer(Map<String, String> vars) {
        this.vars = vars;
    }

    public Map<String, String> getVars() {
        return vars;
    }

    public void setVars(Map<String, String> vars) {
        this.vars = vars;
    }

    @Override
    public String transform(String linkRef) {
        String s = new String(linkRef);
        //
        for (String k : vars.keySet()) {
            s = s.replaceAll("\\$\\{" + k + "\\}", vars.get(k));
        }
        //
        return s;
    }
}