//
   //  ========================================================================
   //  Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd.
   //  ------------------------------------------------------------------------
   //  All rights reserved. This program and the accompanying materials
   //  are made available under the terms of the Eclipse Public License v1.0
   //  and Apache License v2.0 which accompanies this distribution.
   //
   //      The Eclipse Public License is available at
  //      http://www.eclipse.org/legal/epl-v10.html
  //
  //      The Apache License v2.0 is available at
  //      http://www.opensource.org/licenses/apache2.0.php
  //
  //  You may elect to redistribute this code under either of these licenses.
  //  ========================================================================
  //
  
  package org.eclipse.jetty.start;
  
  import static org.eclipse.jetty.start.UsageException.ERR_BAD_ARG;
  
  import java.io.IOException;
  import java.io.OutputStream;
  import java.util.ArrayList;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  import java.util.Properties;
  import java.util.Stack;
  import java.util.TreeMap;
  import java.util.regex.Matcher;
  import java.util.regex.Pattern;
  
  import org.eclipse.jetty.start.Props.Prop;
  
  /**
   * Management of Properties.
   * <p>
   * This is larger in scope than the standard {@link java.util.Properties}, as it will also handle tracking the origin of each property, if it was overridden,
   * and also allowing for <code>${property}</code> expansion.
   */
  public final class Props implements Iterable<Prop>
  {
      private static final Pattern __propertyPattern = Pattern.compile("(?<=[^$]|^)\\$\\{([^}]*)\\}");
      
      public static class Prop
      {
          public String key;
          public String value;
          public String origin;
          public Prop overrides;
  
          public Prop(String key, String value, String origin)
          {
              this.key = key;
              this.value = value;
              this.origin = origin;
          }
  
          public Prop(String key, String value, String origin, Prop overrides)
          {
              this(key,value,origin);
              this.overrides = overrides;
          }
  
          @Override
          public String toString()
          {
              StringBuilder builder = new StringBuilder();
              builder.append("Prop [key=");
              builder.append(key);
              builder.append(", value=");
              builder.append(value);
              builder.append(", origin=");
              builder.append(origin);
              builder.append(", overrides=");
              builder.append(overrides);
              builder.append("]");
              return builder.toString();
          }
      }
  
      public static final String ORIGIN_SYSPROP = "<system-property>";
      
      public static String getValue(String arg)
      {
          int idx = arg.indexOf('=');
          if (idx == (-1))
          {
              throw new UsageException(ERR_BAD_ARG,"Argument is missing a required value: %s",arg);
          }
          String value = arg.substring(idx + 1).trim();
          if (value.length() <= 0)
          {
              throw new UsageException(ERR_BAD_ARG,"Argument is missing a required value: %s",arg);
          }
          return value;
      }
 
     public static List<String> getValues(String arg)
     {
         String v = getValue(arg);
         ArrayList<String> l = new ArrayList<>();
         for (String s : v.split(","))
         {
             if (s != null)
             {
                 s = s.trim();
                 if (s.length() > 0)
                 {
                     l.add(s);
                 }
             }
         }
         return l;
     }
 
     private Map<String, Prop> props = new TreeMap<>(String.CASE_INSENSITIVE_ORDER);
     private List<String> sysPropTracking = new ArrayList<>();
 
     public void addAll(Props other)
     {
         this.props.putAll(other.props);
         this.sysPropTracking.addAll(other.sysPropTracking);
     }
     
     /**
      * Add a potential argument as a property.
      * <p>
      * If arg is not a property, ignore it.
      * @param arg the argument to parse for a potential property
      * @param source the source for this argument (to track origin of property from)
      * @return true if the property was added, false if the property wasn't added
      */
     public boolean addPossibleProperty(String arg, String source)
     {
         // Start property (syntax similar to System property)
         if (arg.startsWith("-D"))
         {
             String[] assign = arg.substring(2).split("=",2);
             switch (assign.length)
             {
                 case 2:
                     setSystemProperty(assign[0],assign[1]);
                     setProperty(assign[0],assign[1],source);
                     return true;
                 case 1:
                     setSystemProperty(assign[0],"");
                     setProperty(assign[0],"",source);
                     return true;
                 default:
                     return false;
             }
         }
 
         // Is this a raw property declaration?
         int idx = arg.indexOf('=');
         if (idx >= 0)
         {
             String key = arg.substring(0,idx);
             String value = arg.substring(idx + 1);
 
             setProperty(key,value,source);
             return true;
         }
 
         // All other strings are ignored
         return false;
     }
 
     public String cleanReference(String property)
     {
         String name = property.trim();
         if (name.startsWith("${") && name.endsWith("}"))
         {
             name = name.substring(2,name.length() - 1);
         }
         return name.trim();
     }
 
     public boolean containsKey(String key)
     {
         return props.containsKey(key);
     }
 
     public String expand(String str)
     {
         return expand(str,new Stack<String>());
     }
 
     public String expand(String str, Stack<String> seenStack)
     {
         if (str == null)
         {
             return str;
         }
 
         if (str.indexOf("${") < 0)
         {
             // Contains no potential expressions.
             return str;
         }
 
         Matcher mat = __propertyPattern.matcher(str);
         StringBuilder expanded = new StringBuilder();
         int offset = 0;
         String property;
         String value;
 
         while (mat.find(offset))
         {
             property = mat.group(1);
 
             // Loop detection
             if (seenStack.contains(property))
             {
                 StringBuilder err = new StringBuilder();
                 err.append("Property expansion loop detected: ");
                 int idx = seenStack.lastIndexOf(property);
                 for (int i = idx; i < seenStack.size(); i++)
                 {
                     err.append(seenStack.get(i));
                     err.append(" -> ");
                 }
                 err.append(property);
                 throw new PropsException(err.toString());
             }
 
             seenStack.push(property);
 
             // find property name
             expanded.append(str.subSequence(offset,mat.start()));
             // get property value
             value = getString(property);
             if (value == null)
             {
                 StartLog.trace("Unable to expand: %s",property);
                 expanded.append(mat.group());
             }
             else
             {
                 // recursively expand
                 value = expand(value,seenStack);
                 expanded.append(value);
             }
             // update offset
             offset = mat.end();
         }
 
         // leftover
         expanded.append(str.substring(offset));
 
         // special case for "$$"
         if (expanded.indexOf("$$") >= 0)
         {
             return expanded.toString().replaceAll("\\$\\$","\\$");
         }
 
         return expanded.toString();
     }
 
     public Prop getProp(String key)
     {
         return getProp(key,true);
     }
 
     public Prop getProp(String key, boolean searchSystemProps)
     {
         Prop prop = props.get(key);
         if ((prop == null) && searchSystemProps)
         {
             // try system property
             prop = getSystemProperty(key);
         }
         return prop;
     }
 
     public String getString(String key)
     {
         if (key == null)
         {
             throw new PropsException("Cannot get value for null key");
         }
 
         String name = cleanReference(key);
 
         if (name.length() == 0)
         {
             throw new PropsException("Cannot get value for empty key");
         }
 
         Prop prop = getProp(name);
         if (prop == null)
         {
             return null;
         }
         return prop.value;
     }
 
     public String getString(String key, String defVal)
     {
         String val = getString(key);
         if (val == null)
         {
             return defVal;
         }
         return val;
     }
 
     private Prop getSystemProperty(String key)
     {
         String value = System.getProperty(key);
         if (value == null)
         {
             return null;
         }
         return new Prop(key,value,ORIGIN_SYSPROP);
     }
 
     public static boolean hasPropertyKey(String name)
     {
         return __propertyPattern.matcher(name).find();
     }
 
     @Override
     public Iterator<Prop> iterator()
     {
         return props.values().iterator();
     }
 
     public void reset()
     {
         props.clear();
     }
 
     public void setProperty(Prop prop)
     {
         props.put(prop.key,prop);
     }
 
     public void setProperty(String key, String value, String origin)
     {
         Prop prop = props.get(key);
         if (prop == null)
         {
             prop = new Prop(key,value,origin);
         }
         else
         {
             prop = new Prop(key,value,origin,prop);
         }
         props.put(key,prop);
     }
 
     public int size()
     {
         return props.size();
     }
 
     public void store(OutputStream stream, String comments) throws IOException
     {
         Properties props = new Properties();
         // add all Props as normal properties, with expansion performed.
         for (Prop prop : this)
         {
             props.setProperty(prop.key,expand(prop.value));
         }
         // write normal properties file
         props.store(stream,comments);
     }
 
     public void setSystemProperty(String key, String value)
     {
         System.setProperty(key,value);
         sysPropTracking.add(key);
     }
 }