org.catools.common.security.CSensitiveDataMaskingManager Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of common.config Show documentation

The general configuration manager for CATools projects

There is a newer version: 1.0.1

package org.catools.common.security;

import org.apache.commons.lang3.StringUtils;

import java.util.*;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;

public class CSensitiveDataMaskingManager {
  private static final List maskList = new ArrayList<>();
  private static final Set ignoreList = new HashSet<>();
  private static final String DEFAULT_MASK = "*".repeat(6);

  public static void clear() {
    doAction(
        (maskList, ignoreList) -> {
          maskList.clear();
          ignoreList.clear();
          return true;
        });
  }

  public static void addMask(final String... masks) {
    doAction(
        (maskList, ignoreList) -> {
          for (String mask : masks) {
            if (StringUtils.isNotBlank(mask)) {
              if (!ignoreList.contains(mask) && !maskList.contains(mask)) {
                maskList.add(mask);
              }
              Collections.sort(maskList, (o1, o2) -> o2.length() - o1.length());
            }
          }
          return true;
        });
  }

  public static void ignore(final String... inputs) {
    doAction(
        (maskList, ignoreList) -> {
          for (String input : inputs) {
            if (StringUtils.isNotBlank(input)) {
              if (!ignoreList.contains(input)) {
                ignoreList.add(input);
              }
              maskList.removeIf(s -> ignoreList.contains(s));
            }
          }
          return true;
        });
  }

  public static String mask(final Object input) {
    final AtomicReference message =
        new AtomicReference<>(input == null ? StringUtils.EMPTY : input.toString());
    if (CSecurityConfigs.maskSensitiveData()) {
      return doAction(
          (maskList, ignoreList) -> {
            for (String securityMask : maskList) {
              while (message.get().contains(securityMask)) {
                message.set(message.get().replace(securityMask, DEFAULT_MASK));
              }
            }
            return message.get();
          });
    }
    return message.get();
  }

  private static synchronized  T doAction(BiFunction, Set, T> func) {
    return func.apply(maskList, ignoreList);
  }
}