NotEnoughItems/src/main/java/codechicken/nei/SearchTextFormatter.java

package codechicken.nei;

import java.util.List;
import java.util.StringJoiner;

import net.minecraft.util.EnumChatFormatting;

import codechicken.nei.FormattedTextField.TextFormatter;
import codechicken.nei.SearchTokenParser.SearchToken;
import codechicken.nei.search.SearchExpressionFormatVisitor;
import codechicken.nei.search.SearchExpressionUtils;

public class SearchTextFormatter implements TextFormatter {

    protected final SearchTokenParser searchParser;

    public SearchTextFormatter(SearchTokenParser searchParser) {
        this.searchParser = searchParser;
    }

    public String format(String text) {
        final int patternMode = NEIClientConfig.getIntSetting("inventory.search.patternMode");
        if (patternMode == 3) {
            final int spaceModeEnabled = NEIClientConfig.getIntSetting("inventory.search.spaceMode");
            if (spaceModeEnabled == 1) {
                text = SearchTokenParser.SPACE_PATTERN.matcher(text).replaceAll("\\\\ ");
            }
            final SearchExpressionFormatVisitor visitor = new SearchExpressionFormatVisitor(searchParser);
            return SearchExpressionUtils.visitSearchExpression(text, visitor);
        } else {
            final String[] parts = (text + "| ").split("\\|");
            StringJoiner formattedText = new StringJoiner(SearchExpressionUtils.HIGHLIGHTS.OR + "|");

            for (int i = 0; i < parts.length - 1; i++) {
                final String filterText = parts[i];
                final List<SearchToken> tokens = searchParser.splitSearchText(filterText);
                StringBuilder formattedPart = new StringBuilder();
                int startIndex = 0;

                for (SearchToken token : tokens) {
                    formattedPart.append(filterText.substring(startIndex, token.start));
                    EnumChatFormatting tokenColor = EnumChatFormatting.RESET;

                    if (token.firstChar != null) {
                        tokenColor = searchParser.getProvider(token.firstChar).getHighlightedColor();
                    }

                    if (token.ignore) {
                        formattedPart.append(SearchExpressionUtils.HIGHLIGHTS.NEGATE + "-");
                    }

                    if (token.firstChar != null) {
                        formattedPart.append(tokenColor + String.valueOf(token.firstChar));
                    }

                    if (token.quotes) {
                        formattedPart.append(SearchExpressionUtils.HIGHLIGHTS.QUOTED + "\"");
                    }

                    if (!token.rawText.isEmpty()) {
                        formattedPart.append(tokenColor + token.rawText);
                    }

                    if (token.quotes) {
                        formattedPart.append(SearchExpressionUtils.HIGHLIGHTS.QUOTED + "\"");
                    }

                    startIndex = token.end;
                }

                formattedPart.append(filterText.substring(startIndex, filterText.length()));
                formattedText.add(formattedPart);
            }

            return formattedText.toString();
        }
    }
}