Untitled

package com.amazon.grcs.icas.text.analysis.core.textformatter;

import com.amazon.grcs.icas.text.analysis.model.PatternReplacement;
import lombok.NonNull;

import java.util.List;
import java.util.Set;

public class AttrAwarePatternTextCleaner extends PatternBasedTextCleaner implements AttrAwareCleaner {

    @NonNull
    private final Set<String> excludedAttrs;

    public AttrAwarePatternTextCleaner(List<PatternReplacement> patternReplacements, @NonNull Set<String> excludedAttrs) {
        super(patternReplacements);
        this.excludedAttrs = excludedAttrs;
    }

    @Override
    public String clean(String text, String attr) {
        if (excludedAttrs.contains(attr)) {
            return text;
        }
        return super.clean(text);
    }

}