content: /*
   * Copyright 2021-2024 Aklivity Inc.
   *
   * Aklivity licenses this file to you under the Apache License,
   * version 2.0 (the "License"); you may not use this file except in compliance
   * with the License. You may obtain a copy of the License at:
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
   * License for the specific language governing permissions and limitations
   * under the License.
   */
  package io.aklivity.zilla.runtime.binding.tcp.internal.config;
  
  import java.net.InetAddress;
  import java.net.InetSocketAddress;
  import java.util.Arrays;
  import java.util.regex.Matcher;
  import java.util.regex.Pattern;
  
  import org.agrona.collections.IntHashSet;
  
  import io.aklivity.zilla.runtime.binding.tcp.config.TcpConditionConfig;
  import io.aklivity.zilla.runtime.binding.tcp.internal.util.Cidr;
  
  public final class TcpConditionMatcher
  {
      public final Cidr cidr;
      public final Matcher authority;
      public final IntHashSet ports;
  
      public TcpConditionMatcher(
          TcpConditionConfig condition)
      {
          this.cidr = condition.cidr != null ? new Cidr(condition.cidr) : null;
          this.authority = condition.authority != null ? asMatcher(condition.authority) : null;
          this.ports = condition.ports != null ? asIntHashSet(condition.ports) : null;
      }
  
      public boolean matches(
          InetSocketAddress remote)
      {
          return matchesCidr(remote.getAddress()) &&
                  matchesAuthority(remote.getAddress()) &&
                  matchesPort(remote.getPort());
      }
  
      private boolean matchesCidr(
          InetAddress remote)
      {
          return cidr == null || cidr.matches(remote);
      }
  
      private boolean matchesAuthority(
          InetAddress remote)
      {
          return authority == null || authority.reset(remote.getHostName()).matches();
      }
  
      private boolean matchesPort(
          int port)
      {
          return ports == null || ports.contains(port);
      }
  
      private static Matcher asMatcher(
          String wildcard)
      {
          return Pattern.compile(wildcard.replace(".", "\\.").replace("*", ".*")).matcher("");
      }
  
      private static IntHashSet asIntHashSet(
          int[] ports)
      {
          IntHashSet set = new IntHashSet(ports.length);
          Arrays.stream(ports).forEach(set::add);
          return set;
      }
  }
  