BaseApplication.java

package org.batfish.representation.juniper;

import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.batfish.common.util.ComparableStructure;
import org.batfish.datamodel.IpAccessListLine;
import org.batfish.common.Warnings;

public class BaseApplication extends ComparableStructure<String>
      implements Application {

   public static class Term extends ComparableStructure<String> {

      /**
       *
       */
      private static final long serialVersionUID = 1L;

      private final IpAccessListLine _line;

      public Term(String name) {
         super(name);
         _line = new IpAccessListLine();
      }

      public void applyTo(IpAccessListLine destinationLine) {
         destinationLine.getIpProtocols().addAll(_line.getIpProtocols());
         destinationLine.getDstPorts().addAll(_line.getDstPorts());
         destinationLine.getSrcPorts().addAll(_line.getSrcPorts());
      }

      public IpAccessListLine getLine() {
         return _line;
      }

   }

   /**
    *
    */
   private static final long serialVersionUID = 1L;

   private boolean _ipv6;

   private Term _mainTerm;

   private final Map<String, Term> _terms;

   public BaseApplication(String name) {
      super(name);
      _mainTerm = new Term(getMainTermName());
      _terms = new LinkedHashMap<>();
   }

   @Override
   public void applyTo(IpAccessListLine srcLine, List<IpAccessListLine> lines,
         Warnings w) {
      Collection<Term> terms;
      if (_terms.isEmpty()) {
         terms = Collections.singletonList(_mainTerm);
      }
      else {
         terms = _terms.values();
      }
      for (Term term : terms) {
         IpAccessListLine newLine = new IpAccessListLine();
         newLine.getDstIps().addAll(srcLine.getDstIps());
         newLine.getSrcIps().addAll(srcLine.getSrcIps());
         newLine.setAction(srcLine.getAction());
         term.applyTo(newLine);
         lines.add(newLine);
      }
   }

   @Override
   public boolean getIpv6() {
      return _ipv6;
   }

   public Term getMainTerm() {
      return _mainTerm;
   }

   private String getMainTermName() {
      return "~MAIN_TERM~" + _key;
   }

   public Map<String, Term> getTerms() {
      return _terms;
   }

   public void setIpv6(boolean ipv6) {
      _ipv6 = true;
   }

}