/*
    * ====================================================================
    * 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.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  
  package org.apache.hc.core5.testing.nio;
  
  import java.util.Iterator;
  
  import org.apache.hc.core5.http.Header;
  import org.apache.hc.core5.http.HttpConnection;
  import org.apache.hc.core5.http.HttpRequest;
  import org.apache.hc.core5.http.HttpResponse;
  import org.apache.hc.core5.http.impl.Http1StreamListener;
  import org.apache.hc.core5.http.message.RequestLine;
  import org.apache.hc.core5.http.message.StatusLine;
  import org.apache.hc.core5.testing.classic.LoggingSupport;
  import org.slf4j.LoggerFactory;
  import org.slf4j.Logger;
  
  public class LoggingHttp1StreamListener implements Http1StreamListener {
  
      enum Type { CLIENT, SERVER }
  
      public final static LoggingHttp1StreamListeneristener.html#LoggingHttp1StreamListener">LoggingHttp1StreamListener INSTANCE_CLIENT = new LoggingHttp1StreamListener(Type.CLIENT);
      public final static LoggingHttp1StreamListeneristener.html#LoggingHttp1StreamListener">LoggingHttp1StreamListener INSTANCE_SERVER = new LoggingHttp1StreamListener(Type.SERVER);
  
      private final Logger connLog = LoggerFactory.getLogger("org.apache.hc.core5.http.connection");
      private final Logger headerLog = LoggerFactory.getLogger("org.apache.hc.core5.http.headers");
      private final String requestDirection;
      private final String responseDirection;
  
      private LoggingHttp1StreamListener(final Type type) {
          this.requestDirection = type == Type.CLIENT ? " >> " : " << ";
          this.responseDirection = type == Type.CLIENT ? " << " : " >> ";
      }
  
      @Override
      public void onRequestHead(final HttpConnection connection, final HttpRequest request) {
          if (headerLog.isDebugEnabled()) {
              final String idRequestDirection = LoggingSupport.getId(connection) + requestDirection;
              headerLog.debug("{}{}", idRequestDirection, new RequestLine(request));
              for (final Iterator<Header> it = request.headerIterator(); it.hasNext(); ) {
                  headerLog.debug("{}{}", idRequestDirection, it.next());
              }
          }
      }
  
      @Override
      public void onResponseHead(final HttpConnection connection, final HttpResponse response) {
          if (headerLog.isDebugEnabled()) {
              final String id = LoggingSupport.getId(connection);
              headerLog.debug("{}{}{}", id, responseDirection, new StatusLine(response));
              for (final Iterator<Header> it = response.headerIterator(); it.hasNext(); ) {
                  headerLog.debug("{}{}{}", id, responseDirection, it.next());
              }
          }
      }
  
      @Override
      public void onExchangeComplete(final HttpConnection connection, final boolean keepAlive) {
          if (connLog.isDebugEnabled()) {
              if (keepAlive) {
                  connLog.debug("{} connection is kept alive", LoggingSupport.getId(connection));
              } else {
                  connLog.debug("{} connection is not kept alive", LoggingSupport.getId(connection));
              }
          }
      }
  
  }