package org.eclipse.aether.transport.http;
   
   /*
    * 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.
   */
  
  import java.io.Closeable;
  import java.util.Arrays;
  import java.util.Iterator;
  import java.util.Map;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  
  import org.apache.http.HttpHost;
  import org.apache.http.config.RegistryBuilder;
  import org.apache.http.conn.HttpClientConnectionManager;
  import org.apache.http.conn.socket.ConnectionSocketFactory;
  import org.apache.http.conn.socket.PlainConnectionSocketFactory;
  import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
  import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
  import org.eclipse.aether.RepositoryCache;
  import org.eclipse.aether.RepositorySystemSession;
  import org.eclipse.aether.util.ConfigUtils;
  
  import javax.net.ssl.HostnameVerifier;
  import javax.net.ssl.SSLSocketFactory;
  
  /**
   * Container for HTTP-related state that can be shared across incarnations of the transporter to optimize the
   * communication with servers.
   */
  final class GlobalState
      implements Closeable
  {
  
      static class CompoundKey
      {
  
          private final Object[] keys;
  
          CompoundKey( Object... keys )
          {
              this.keys = keys;
          }
  
          @Override
          public boolean equals( Object obj )
          {
              if ( this == obj )
              {
                  return true;
              }
              if ( obj == null || !getClass().equals( obj.getClass() ) )
              {
                  return false;
              }
              CompoundKey that = (CompoundKey) obj;
              return Arrays.equals( keys, that.keys );
          }
  
          @Override
          public int hashCode()
          {
              int hash = 17;
              hash = hash * 31 + Arrays.hashCode( keys );
              return hash;
          }
  
          @Override
          public String toString()
          {
              return Arrays.toString( keys );
          }
      }
  
      private static final String KEY = GlobalState.class.getName();
  
      private static final String CONFIG_PROP_CACHE_STATE = "aether.connector.http.cacheState";
  
      private final ConcurrentMap<SslConfig, HttpClientConnectionManager> connectionManagers;
  
      private final ConcurrentMap<CompoundKey, Object> userTokens;
  
      private final ConcurrentMap<HttpHost, AuthSchemePool> authSchemePools;
 
     private final ConcurrentMap<CompoundKey, Boolean> expectContinues;
 
     public static GlobalState get( RepositorySystemSession session )
     {
         GlobalState cache;
         RepositoryCache repoCache = session.getCache();
         if ( repoCache == null || !ConfigUtils.getBoolean( session, true, CONFIG_PROP_CACHE_STATE ) )
         {
             cache = null;
         }
         else
         {
             Object tmp = repoCache.get( session, KEY );
             if ( tmp instanceof GlobalState )
             {
                 cache = (GlobalState) tmp;
             }
             else
             {
                 synchronized ( GlobalState.class )
                 {
                     tmp = repoCache.get( session, KEY );
                     if ( tmp instanceof GlobalState )
                     {
                         cache = (GlobalState) tmp;
                     }
                     else
                     {
                         cache = new GlobalState();
                         repoCache.put( session, KEY, cache );
                     }
                 }
             }
         }
         return cache;
     }
 
     private GlobalState()
     {
         connectionManagers = new ConcurrentHashMap<>();
         userTokens = new ConcurrentHashMap<>();
         authSchemePools = new ConcurrentHashMap<>();
         expectContinues = new ConcurrentHashMap<>();
     }
 
     @Override
     public void close()
     {
         for ( Iterator<Map.Entry<SslConfig, HttpClientConnectionManager>> it = connectionManagers.entrySet().iterator();
               it.hasNext(); )
         {
             HttpClientConnectionManager connMgr = it.next().getValue();
             it.remove();
             connMgr.shutdown();
         }
     }
 
     public HttpClientConnectionManager getConnectionManager( SslConfig config )
     {
         HttpClientConnectionManager manager = connectionManagers.get( config );
         if ( manager == null )
         {
             HttpClientConnectionManager connMgr = newConnectionManager( config );
             manager = connectionManagers.putIfAbsent( config, connMgr );
             if ( manager != null )
             {
                 connMgr.shutdown();
             }
             else
             {
                 manager = connMgr;
             }
         }
         return manager;
     }
 
     @SuppressWarnings( "checkstyle:magicnumber" )
     public static HttpClientConnectionManager newConnectionManager( SslConfig sslConfig )
     {
         RegistryBuilder<ConnectionSocketFactory> registryBuilder = RegistryBuilder.<ConnectionSocketFactory>create()
                 .register( "http", PlainConnectionSocketFactory.getSocketFactory() );
 
         if ( sslConfig == null )
         {
             registryBuilder.register( "https", SSLConnectionSocketFactory.getSystemSocketFactory() );
         }
         else
         {
             SSLSocketFactory sslSocketFactory = ( sslConfig.context != null )
                     ? sslConfig.context.getSocketFactory() : (SSLSocketFactory) SSLSocketFactory.getDefault();
 
             HostnameVerifier hostnameVerifier = ( sslConfig.verifier != null )
                     ? sslConfig.verifier : SSLConnectionSocketFactory.getDefaultHostnameVerifier();
 
             registryBuilder.register( "https", new SSLConnectionSocketFactory(
                     sslSocketFactory, sslConfig.protocols, sslConfig.cipherSuites, hostnameVerifier ) );
         }
         PoolingHttpClientConnectionManager connMgr = new PoolingHttpClientConnectionManager( registryBuilder.build() );
         connMgr.setMaxTotal( 100 );
         connMgr.setDefaultMaxPerRoute( 50 );
         return connMgr;
     }
 
     public Object getUserToken( CompoundKey key )
     {
         return userTokens.get( key );
     }
 
     public void setUserToken( CompoundKey key, Object userToken )
     {
         if ( userToken != null )
         {
             userTokens.put( key, userToken );
         }
         else
         {
             userTokens.remove( key );
         }
     }
 
     public ConcurrentMap<HttpHost, AuthSchemePool> getAuthSchemePools()
     {
         return authSchemePools;
     }
 
     public Boolean getExpectContinue( CompoundKey key )
     {
         return expectContinues.get( key );
     }
 
     public void setExpectContinue( CompoundKey key, boolean enabled )
     {
         expectContinues.put( key, enabled );
     }
 
 }