package com.provectus.kafka.ui.service.masking.policies;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ContainerNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import java.util.List;

class Remove extends MaskingPolicy {

  Remove(List<String> fieldNames) {
    super(fieldNames);
  }

  @Override
  public String applyToString(String str) {
    return "null";
  }

  @Override
  public ContainerNode<?> applyToJsonContainer(ContainerNode<?> node) {
    return (ContainerNode<?>) removeFields(node);
  }

  private JsonNode removeFields(JsonNode node) {
    if (node.isObject()) {
      ObjectNode obj = ((ObjectNode) node).objectNode();
      node.fields().forEachRemaining(f -> {
        String fieldName = f.getKey();
        JsonNode fieldVal = f.getValue();
        if (!fieldShouldBeMasked(fieldName)) {
          obj.set(fieldName, removeFields(fieldVal));
        }
      });
      return obj;
    } else if (node.isArray()) {
      var arr = ((ArrayNode) node).arrayNode(node.size());
      node.elements().forEachRemaining(e -> arr.add(removeFields(e)));
      return arr;
    }
    return node;
  }
}