/*
    * 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.yaml;
  
  import java.io.IOException;
  
  import org.apache.logging.log4j.core.LoggerContext;
  import org.apache.logging.log4j.core.config.Configuration;
  import org.apache.logging.log4j.core.config.ConfigurationSource;
  import org.apache.logging.log4j.core.config.json.JsonConfiguration;
  
  import com.fasterxml.jackson.core.JsonParser;
  import com.fasterxml.jackson.databind.ObjectMapper;
  import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
  
  public class YamlConfiguration extends JsonConfiguration {
  
      public YamlConfiguration(final LoggerContext loggerContext, final ConfigurationSource configSource) {
          super(loggerContext, configSource);
      }
  
      @Override
      protected ObjectMapper getObjectMapper() {
          return new ObjectMapper(new YAMLFactory()).configure(JsonParser.Feature.ALLOW_COMMENTS, true);
      }
  
      @Override
      public Configuration reconfigure() {
          try {
              final ConfigurationSource source = getConfigurationSource().resetInputStream();
              if (source == null) {
                  return null;
              }
              return new YamlConfiguration(getLoggerContext(), source);
          } catch (final IOException ex) {
              LOGGER.error("Cannot locate file {}", getConfigurationSource(), ex);
          }
          return null;
      }
  }