package speiger.src.builder.mappers;

import java.util.function.UnaryOperator;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import speiger.src.builder.misc.RegexUtil;

public class InjectMapper implements UnaryOperator<String>
{
	Pattern pattern;
	String replacement;
	String braces = "()";
	boolean removeBraces;
	
	public InjectMapper(String pattern, String replacement)
	{
		this.pattern = Pattern.compile(pattern);
		this.replacement = replacement;
	}
	
	public InjectMapper setBraceType(String s) 
	{
		if(s.length() != 2) throw new IllegalStateException("Start and End char only");
		braces = s;
		return this;
	}
	
	public InjectMapper removeBraces()
	{
		removeBraces = true;
		return this;
	}
	
	@Override
	public String apply(String t)
	{
		Matcher matcher = pattern.matcher(t);
		if(matcher.find())
		{
			StringBuffer buffer = new StringBuffer();			
			do
			{
				String text = RegexUtil.searchUntil(t, matcher.end()-1, braces.charAt(0), braces.charAt(1));
				if(!text.isEmpty())
				{
					RegexUtil.skip(matcher.appendReplacement(buffer, ""), text.length());
					buffer.append(String.format(replacement, getString(text)));
				}
            } while (matcher.find());
            matcher.appendTail(buffer);
            return buffer.toString();
		}
		return t;
	}
	
	protected String getString(String s)
	{
		return removeBraces ? s.substring(1, s.length() - 1) : s;
	}
}