/*
    * Licensed to the Apache Software Foundation (ASF) under one or more
    * contributor license agreements. See the NOTICE file distributed with
    * this work for additional information regarding copyright ownership.
    * The ASF 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 org.apache.logging.log4j.core.config;
  
  import org.apache.logging.log4j.Level;
  import org.apache.logging.log4j.Logger;
  import org.apache.logging.log4j.core.Appender;
  import org.apache.logging.log4j.core.Filter;
  import org.apache.logging.log4j.core.Layout;
  import org.apache.logging.log4j.core.LogEvent;
  import org.apache.logging.log4j.core.appender.ConsoleAppender;
  import org.apache.logging.log4j.core.config.plugins.PluginAttr;
  import org.apache.logging.log4j.core.config.plugins.PluginConfiguration;
  import org.apache.logging.log4j.core.config.plugins.PluginFactory;
  import org.apache.logging.log4j.core.config.plugins.PluginManager;
  import org.apache.logging.log4j.core.config.plugins.PluginElement;
  import org.apache.logging.log4j.core.config.plugins.PluginNode;
  import org.apache.logging.log4j.core.config.plugins.PluginType;
  import org.apache.logging.log4j.core.config.plugins.PluginValue;
  import org.apache.logging.log4j.core.filter.AbstractFilterable;
  import org.apache.logging.log4j.core.helpers.NameUtil;
  import org.apache.logging.log4j.core.layout.PatternLayout;
  import org.apache.logging.log4j.core.lookup.Interpolator;
  import org.apache.logging.log4j.core.lookup.MapLookup;
  import org.apache.logging.log4j.core.lookup.StrLookup;
  import org.apache.logging.log4j.core.lookup.StrSubstitutor;
  import org.apache.logging.log4j.status.StatusLogger;
  import org.apache.logging.log4j.util.PropertiesUtil;
  
  import java.lang.annotation.Annotation;
  import java.lang.reflect.Array;
  import java.lang.reflect.Method;
  import java.lang.reflect.Modifier;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  import java.util.Map;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  import java.util.concurrent.CopyOnWriteArrayList;
  
  /**
   * The Base Configuration. Many configuration implementations will extend this class.
   */
  public class BaseConfiguration extends AbstractFilterable implements Configuration {
      /**
       * Allow subclasses access to the status logger without creating another instance.
       */
      protected static final Logger LOGGER = StatusLogger.getLogger();
  
      /**
       * The root node of the configuration.
       */
      protected Node rootNode;
  
      /**
       * The Plugin Manager.
       */
      protected PluginManager pluginManager;
  
      /**
       * Listeners for configuration changes.
       */
      protected final List<ConfigurationListener> listeners =
          new CopyOnWriteArrayList<ConfigurationListener>();
  
      /**
       * The ConfigurationMonitor that checks for configuration changes.
       */
      protected ConfigurationMonitor monitor = new DefaultConfigurationMonitor();
  
      private String name;
  
      private ConcurrentMap<String, Appender<?>> appenders = new ConcurrentHashMap<String, Appender<?>>();
  
      private ConcurrentMap<String, LoggerConfig> loggers = new ConcurrentHashMap<String, LoggerConfig>();
  
      private final StrLookup tempLookup = new Interpolator();
  
      private final StrSubstitutor subst = new StrSubstitutor(tempLookup);
  
      private LoggerConfig root = new LoggerConfig();
  
      private final boolean started = false;
  
     private final ConcurrentMap<String, Object> componentMap = new ConcurrentHashMap<String, Object>();
 
     /**
      * Constructor.
      */
     protected BaseConfiguration() {
         pluginManager = new PluginManager("Core");
         rootNode = new Node();
     }
 
     public Map<String, String> getProperties() {
         return (Map<String, String>) componentMap.get(CONTEXT_PROPERTIES);
     }
 
     /**
      * Initialize the configuration.
      */
     public void start() {
         pluginManager.collectPlugins();
         setup();
         doConfigure();
         for (final LoggerConfig logger : loggers.values()) {
             logger.startFilter();
         }
         for (final Appender appender : appenders.values()) {
             appender.start();
         }
 
         startFilter();
     }
 
     /**
      * Tear down the configuration.
      */
     public void stop() {
         for (final LoggerConfig logger : loggers.values()) {
             logger.clearAppenders();
             logger.stopFilter();
         }
         // Stop the appenders in reverse order in case they still have activity.
         final Appender[] array = appenders.values().toArray(new Appender[appenders.size()]);
         for (int i = array.length - 1; i >= 0; --i) {
             array[i].stop();
         }
         stopFilter();
     }
 
     protected void setup() {
     }
 
     public Object getComponent(final String name) {
         return componentMap.get(name);
     }
 
     public void addComponent(final String name, final Object obj) {
         componentMap.putIfAbsent(name, obj);
     }
 
     protected void doConfigure() {
         boolean setRoot = false;
         boolean setLoggers = false;
         for (final Node child : rootNode.getChildren()) {
             createConfiguration(child, null);
             if (child.getObject() == null) {
                 continue;
             }
             if (child.getName().equalsIgnoreCase("properties")) {
                 if (tempLookup == subst.getVariableResolver()) {
                     subst.setVariableResolver((StrLookup) child.getObject());
                 } else {
                     LOGGER.error("Properties declaration must be the first element in the configuration");
                 }
                 continue;
             } else if (tempLookup == subst.getVariableResolver()) {
                 final Map<String, String> map = (Map<String, String>) componentMap.get(CONTEXT_PROPERTIES);
                 final StrLookup lookup = map == null ? null : new MapLookup(map);
                 subst.setVariableResolver(new Interpolator(lookup));
             }
             if (child.getName().equalsIgnoreCase("appenders")) {
                 appenders = (ConcurrentMap<String, Appender<?>>) child.getObject();
             } else if (child.getObject() instanceof Filter) {
                 addFilter((Filter) child.getObject());
             } else if (child.getName().equalsIgnoreCase("loggers")) {
                 final Loggers l = (Loggers) child.getObject();
                 loggers = l.getMap();
                 setLoggers = true;
                 if (l.getRoot() != null) {
                     root = l.getRoot();
                     setRoot = true;
                 }
             } else {
                 LOGGER.error("Unknown object \"" + child.getName() + "\" of type " +
                     child.getObject().getClass().getName() + " is ignored");
             }
         }
 
         if (!setLoggers) {
             LOGGER.warn("No Loggers were configured, using default. Is the Loggers element missing?");
             setToDefault();
             return;
         } else if (!setRoot) {
             LOGGER.warn("No Root logger was configured, using default");
             setToDefault();
             return;
         }
 
         for (final Map.Entry<String, LoggerConfig> entry : loggers.entrySet()) {
             final LoggerConfig l = entry.getValue();
             for (final AppenderRef ref : l.getAppenderRefs()) {
                 final Appender app = appenders.get(ref.getRef());
                 if (app != null) {
                     l.addAppender(app, ref.getLevel(), ref.getFilter());
                 } else {
                     LOGGER.error("Unable to locate appender " + ref.getRef() + " for logger " + l.getName());
                 }
             }
 
         }
 
         setParents();
     }
 
     private void setToDefault() {
         setName(DefaultConfiguration.DEFAULT_NAME);
         final Layout layout = PatternLayout.createLayout("%d{HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n",
             null, null, null);
         final Appender appender = ConsoleAppender.createAppender(layout, null, "SYSTEM_OUT", "Console", "false",
             "true");
         appender.start();
         addAppender(appender);
         final LoggerConfig root = getRootLogger();
         root.addAppender(appender, null, null);
 
         final String levelName = PropertiesUtil.getProperties().getStringProperty(DefaultConfiguration.DEFAULT_LEVEL);
         final Level level = levelName != null && Level.valueOf(levelName) != null ?
             Level.valueOf(levelName) : Level.ERROR;
         root.setLevel(level);
     }
 
     protected PluginManager getPluginManager() {
         return pluginManager;
     }
 
     /**
      * Set the name of the configuration.
      * @param name The name.
      */
     public void setName(final String name) {
         this.name = name;
     }
 
     /**
      * Returns the name of the configuration.
      * @return the name of the configuration.
      */
     public String getName() {
         return name;
     }
 
     /**
      * Add a listener for changes on the configuration.
      * @param listener The ConfigurationListener to add.
      */
     public void addListener(final ConfigurationListener listener) {
         listeners.add(listener);
     }
 
     /**
      * Remove a ConfigurationListener.
      * @param listener The ConfigurationListener to remove.
      */
     public void removeListener(final ConfigurationListener listener) {
         listeners.remove(listener);
     }
 
     /**
      * Returns the Appender with the specified name.
      * @param name The name of the Appender.
      * @return the Appender with the specified name or null if the Appender cannot be located.
      */
     public Appender getAppender(final String name) {
         return appenders.get(name);
     }
 
     /**
      * Returns a Map containing all the Appenders and their name.
      * @return A Map containing each Appender's name and the Appender object.
      */
     public Map<String, Appender<?>> getAppenders() {
         return appenders;
     }
 
     /**
      * Adds an Appender to the configuration.
      * @param appender The Appender to add.
      */
     public void addAppender(final Appender appender) {
         appenders.put(appender.getName(), appender);
     }
 
     public StrSubstitutor getSubst() {
         return subst;
     }
 
     public ConfigurationMonitor getConfigurationMonitor() {
         return monitor;
     }
 
     /**
      * Associates an Appender with a LoggerConfig. This method is synchronized in case a Logger with the
      * same name is being updated at the same time.
      *
      * Note: This method is not used when configuring via configuration. It is primarily used by
      * unit tests.
      * @param logger The Logger the Appender will be associated with.
      * @param appender The Appender.
      */
     public synchronized void addLoggerAppender(final org.apache.logging.log4j.core.Logger logger,
                                                final Appender appender) {
         final String name = logger.getName();
         appenders.putIfAbsent(name, appender);
         final LoggerConfig lc = getLoggerConfig(name);
         if (lc.getName().equals(name)) {
             lc.addAppender(appender, null, null);
         } else {
             final LoggerConfig nlc = new LoggerConfig(name, lc.getLevel(), lc.isAdditive());
             nlc.addAppender(appender, null, null);
             nlc.setParent(lc);
             loggers.putIfAbsent(name, nlc);
             setParents();
             logger.getContext().updateLoggers();
         }
     }
     /**
      * Associates a Filter with a LoggerConfig. This method is synchronized in case a Logger with the
      * same name is being updated at the same time.
      *
      * Note: This method is not used when configuring via configuration. It is primarily used by
      * unit tests.
      * @param logger The Logger the Fo;ter will be associated with.
      * @param filter The Filter.
      */
     public synchronized void addLoggerFilter(final org.apache.logging.log4j.core.Logger logger, final Filter filter) {
         final String name = logger.getName();
         final LoggerConfig lc = getLoggerConfig(name);
         if (lc.getName().equals(name)) {
 
             lc.addFilter(filter);
         } else {
             final LoggerConfig nlc = new LoggerConfig(name, lc.getLevel(), lc.isAdditive());
             nlc.addFilter(filter);
             nlc.setParent(lc);
             loggers.putIfAbsent(name, nlc);
             setParents();
             logger.getContext().updateLoggers();
         }
     }
     /**
      * Marks a LoggerConfig as additive. This method is synchronized in case a Logger with the
      * same name is being updated at the same time.
      *
      * Note: This method is not used when configuring via configuration. It is primarily used by
      * unit tests.
      * @param logger The Logger the Appender will be associated with.
      * @param additive True if the LoggerConfig should be additive, false otherwise.
      */
     public synchronized void setLoggerAdditive(final org.apache.logging.log4j.core.Logger logger,
                                                final boolean additive) {
         final String name = logger.getName();
         final LoggerConfig lc = getLoggerConfig(name);
         if (lc.getName().equals(name)) {
             lc.setAdditive(additive);
         } else {
             final LoggerConfig nlc = new LoggerConfig(name, lc.getLevel(), additive);
             nlc.setParent(lc);
             loggers.putIfAbsent(name, nlc);
             setParents();
             logger.getContext().updateLoggers();
         }
     }
 
     /**
      * Remove an Appender. First removes any associations between LoggerContigs and the Appender, removes
      * the Appender from this appender list and then stops the appender. This method is synchronized in
      * case an Appender with the same name is being added during the removal.
      * @param name the name of the appender to remove.
      */
     public synchronized void removeAppender(final String name) {
         for (final LoggerConfig logger : loggers.values()) {
             logger.removeAppender(name);
         }
         final Appender app = appenders.remove(name);
 
         if (app != null) {
             app.stop();
         }
     }
 
     /**
      * Locates the appropriate LoggerConfig for a Logger name. This will remove tokens from the
      * package name as necessary or return the root LoggerConfig if no other matches were found.
      * @param name The Logger name.
      * @return The located LoggerConfig.
      */
     public LoggerConfig getLoggerConfig(final String name) {
         if (loggers.containsKey(name)) {
             return loggers.get(name);
         }
         String substr = name;
         while ((substr = NameUtil.getSubName(substr)) != null) {
             if (loggers.containsKey(substr)) {
                 return loggers.get(substr);
             }
         }
         return root;
     }
 
     /**
      * Returns the root Logger.
      * @return the root Logger.
      */
     public LoggerConfig getRootLogger() {
         return root;
     }
 
     /**
      * Returns a Map of all the LoggerConfigs.
      * @return a Map with each entry containing the name of the Logger and the LoggerConfig.
      */
     public Map<String, LoggerConfig> getLoggers() {
         return Collections.unmodifiableMap(loggers);
     }
 
     /**
      * Returns the LoggerConfig with the specified name.
      * @param name The Logger name.
      * @return The LoggerConfig or null if no match was found.
      */
     public LoggerConfig getLogger(final String name) {
         return loggers.get(name);
     }
 
     /**
      * Adding a logger cannot be done atomically so is not allowed in an active configuration. Adding
      * or removing a Logger requires creating a new configuration and then switching.
      *
      * @param name The name of the Logger.
      * @param loggerConfig The LoggerConfig.
      */
     public void addLogger(final String name, final LoggerConfig loggerConfig) {
         if (started) {
             final String msg = "Cannot add logger " + name + " to an active configuration";
             LOGGER.warn(msg);
             throw new IllegalStateException(msg);
         }
         loggers.put(name, loggerConfig);
         setParents();
     }
 
     /**
      * Removing a logger cannot be done atomically so is not allowed in an active configuration. Adding
      * or removing a Logger requires creating a new configuration and then switching.
      *
      * @param name The name of the Logger.
      */
     public void removeLogger(final String name) {
         if (started) {
             final String msg = "Cannot remove logger " + name + " in an active configuration";
             LOGGER.warn(msg);
             throw new IllegalStateException(msg);
         }
         loggers.remove(name);
         setParents();
     }
 
     public void createConfiguration(final Node node, final LogEvent event) {
         final PluginType type = node.getType();
         if (type != null && type.isDeferChildren()) {
             node.setObject(createPluginObject(type, node, event));
         } else {
             for (final Node child : node.getChildren()) {
                 createConfiguration(child, event);
             }
 
             if (type == null) {
                 if (node.getParent() != null) {
                     LOGGER.error("Unable to locate plugin for " + node.getName());
                 }
             } else {
                 node.setObject(createPluginObject(type, node, event));
             }
         }
     }
 
    /*
     * Retrieve a static public 'method to create the desired object. Every parameter
     * will be annotated to identify the appropriate attribute or element to use to
     * set the value of the parameter.
     * Parameters annotated with PluginAttr will always be set as Strings.
     * Parameters annotated with PluginElement may be Objects or arrays. Collections
     * and Maps are currently not supported, although the factory method that is called
     * can create these from an array.
     *
     * Although the happy path works, more work still needs to be done to log incorrect
     * parameters. These will generally result in unhelpful InvocationTargetExceptions.
     * @param classClass the class.
     * @return the instantiate method or null if there is none by that
     * description.
     */
     private Object createPluginObject(final PluginType type, final Node node, final LogEvent event)
     {
         final Class clazz = type.getPluginClass();
 
         if (Map.class.isAssignableFrom(clazz)) {
             try {
                 final Map<String, Object> map = (Map<String, Object>) clazz.newInstance();
                 for (final Node child : node.getChildren()) {
                     map.put(child.getName(), child.getObject());
                 }
                 return map;
             } catch (final Exception ex) {
                 LOGGER.warn("Unable to create Map for " + type.getElementName() + " of class " +
                     clazz);
             }
         }
 
         if (List.class.isAssignableFrom(clazz)) {
             try {
                 final List<Object> list = (List<Object>) clazz.newInstance();
                 for (final Node child : node.getChildren()) {
                     list.add(child.getObject());
                 }
                 return list;
             } catch (final Exception ex) {
                 LOGGER.warn("Unable to create List for " + type.getElementName() + " of class " +
                     clazz);
             }
         }
 
         Method factoryMethod = null;
 
         for (final Method method : clazz.getMethods()) {
             if (method.isAnnotationPresent(PluginFactory.class)) {
                 factoryMethod = method;
                 break;
             }
         }
         if (factoryMethod == null) {
             return null;
         }
 
         final Annotation[][] parmArray = factoryMethod.getParameterAnnotations();
         final Class[] parmClasses = factoryMethod.getParameterTypes();
         if (parmArray.length != parmClasses.length) {
             LOGGER.error("Number of parameter annotations does not equal the number of paramters");
         }
         final Object[] parms = new Object[parmClasses.length];
 
         int index = 0;
         final Map<String, String> attrs = node.getAttributes();
         final List<Node> children = node.getChildren();
         final StringBuilder sb = new StringBuilder();
         final List<Node> used = new ArrayList<Node>();
 
         /*
          * For each parameter:
          * If the parameter is an attribute store the value of the attribute in the parameter array.
          * If the parameter is an element:
          *   Determine if the required parameter is an array.
          *     If so, if a child contains the array, use it,
          *      otherwise create the array from all child nodes of the correct type.
          *     Store the array into the parameter array.
          *   If not an array, store the object in the child node into the parameter array.
          */
         for (final Annotation[] parmTypes : parmArray) {
             for (final Annotation a : parmTypes) {
                 if (sb.length() == 0) {
                     sb.append(" with params(");
                 } else {
                     sb.append(", ");
                 }
                 if (a instanceof PluginNode) {
                     parms[index] = node;
                     sb.append("Node=").append(node.getName());
                 } else if (a instanceof PluginConfiguration) {
                     parms[index] = this;
                     if (this.name != null) {
                         sb.append("Configuration(").append(name).append(")");
                     } else {
                         sb.append("Configuration");
                     }
                 } else if (a instanceof PluginValue) {
                     final String name = ((PluginValue) a).value();
                     String v = node.getValue();
                     if (v == null) {
                         v = getAttrValue("value", attrs);
                     }
                     final String value = subst.replace(event, v);
                     sb.append(name).append("=\"").append(value).append("\"");
                     parms[index] = value;
                 } else if (a instanceof PluginAttr) {
                     final String name = ((PluginAttr) a).value();
                     final String value = subst.replace(event, getAttrValue(name, attrs));
                     sb.append(name).append("=\"").append(value).append("\"");
                     parms[index] = value;
                 } else if (a instanceof PluginElement) {
                     final PluginElement elem = (PluginElement) a;
                     final String name = elem.value();
                     if (parmClasses[index].isArray()) {
                         final Class parmClass = parmClasses[index].getComponentType();
                         final List<Object> list = new ArrayList<Object>();
                         sb.append(name).append("={");
                         boolean first = true;
                         for (final Node child : children) {
                             final PluginType childType = child.getType();
                             if (elem.value().equalsIgnoreCase(childType.getElementName()) ||
                                 parmClass.isAssignableFrom(childType.getPluginClass())) {
                                 used.add(child);
                                 if (!first) {
                                     sb.append(", ");
                                 }
                                 first = false;
                                 final Object obj = child.getObject();
                                 if (obj == null) {
                                     LOGGER.error("Null object returned for " + child.getName() + " in " +
                                         node.getName());
                                     continue;
                                 }
                                 if (obj.getClass().isArray()) {
                                     printArray(sb, (Object[]) obj);
                                     parms[index] = obj;
                                     break;
                                 }
                                 sb.append(child.toString());
                                 list.add(obj);
                             }
                         }
                         sb.append("}");
                         if (parms[index] != null) {
                             break;
                         }
                         if (list.size() > 0 && !parmClass.isAssignableFrom(list.get(0).getClass())) {
                             LOGGER.error("Attempted to assign List containing class " +
                                 list.get(0).getClass().getName() + " to array of type " + parmClass +
                                 " for attribute " + name);
                             break;
                         }
                         final Object[] array = (Object[]) Array.newInstance(parmClass, list.size());
                         int i = 0;
                         for (final Object obj : list) {
                             array[i] = obj;
                             ++i;
                         }
                         parms[index] = array;
                     } else {
                         final Class parmClass = parmClasses[index];
                         boolean present = false;
                         for (final Node child : children) {
                             final PluginType childType = child.getType();
                             if (elem.value().equals(childType.getElementName()) ||
                                 parmClass.isAssignableFrom(childType.getPluginClass())) {
                                 sb.append(child.getName()).append("(").append(child.toString()).append(")");
                                 present = true;
                                 used.add(child);
                                 parms[index] = child.getObject();
                                 break;
                             }
                         }
                         if (!present) {
                             sb.append("null");
                         }
                     }
                 }
             }
             ++index;
         }
         if (sb.length() > 0) {
             sb.append(")");
         }
 
         if (attrs.size() > 0) {
             final StringBuilder eb = new StringBuilder();
             for (final String key : attrs.keySet()) {
                 if (eb.length() == 0) {
                     eb.append(node.getName());
                     eb.append(" contains ");
                     if (attrs.size() == 1) {
                         eb.append("an invalid element or attribute ");
                     } else {
                         eb.append("invalid attributes ");
                     }
                 } else {
                     eb.append(", ");
                 }
                 eb.append("\"");
                 eb.append(key);
                 eb.append("\"");
 
             }
             LOGGER.error(eb.toString());
         }
 
         if (!type.isDeferChildren() && used.size() != children.size()) {
             for (final Node child : children) {
                 if (used.contains(child)) {
                     continue;
                 }
                 final String nodeType = node.getType().getElementName();
                 final String start = nodeType.equals(node.getName()) ? node.getName() : nodeType + " " + node.getName();
                 LOGGER.error(start + " has no parameter that matches element " + child.getName());
             }
         }
 
         try {
             final int mod = factoryMethod.getModifiers();
             if (!Modifier.isStatic(mod)) {
                 LOGGER.error(factoryMethod.getName() + " method is not static on class " +
                     clazz.getName() + " for element " + node.getName());
                 return null;
             }
             LOGGER.debug("Calling {} on class {} for element {}{}", factoryMethod.getName(), clazz.getName(),
                 node.getName(), sb.toString());
             //if (parms.length > 0) {
                 return factoryMethod.invoke(null, parms);
             //}
             //return factoryMethod.invoke(null, node);
         } catch (final Exception e) {
             LOGGER.error("Unable to invoke method " + factoryMethod.getName() + " in class " +
                 clazz.getName() + " for element " + node.getName(), e);
         }
         return null;
     }
 
     private void printArray(final StringBuilder sb, final Object... array) {
         boolean first = true;
         for (final Object obj : array) {
             if (!first) {
                 sb.append(", ");
             }
             sb.append(obj.toString());
             first = false;
         }
     }
 
     private String getAttrValue(final String name, final Map<String, String> attrs) {
         for (final String key : attrs.keySet()) {
             if (key.equalsIgnoreCase(name)) {
                 final String attr = attrs.get(key);
                 attrs.remove(key);
                 return attr;
             }
         }
         return null;
     }
 
     private void setParents() {
          for (final Map.Entry<String, LoggerConfig> entry : loggers.entrySet()) {
             final LoggerConfig logger = entry.getValue();
             String name = entry.getKey();
             if (!name.equals("")) {
                 final int i = name.lastIndexOf('.');
                 if (i > 0) {
                     name = name.substring(0, i);
                     LoggerConfig parent = getLoggerConfig(name);
                     if (parent == null) {
                         parent = root;
                     }
                     logger.setParent(parent);
                 }
             }
         }
     }
 }