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.conn.ClientConnectionManager;
  import org.apache.http.conn.scheme.PlainSocketFactory;
  import org.apache.http.conn.scheme.Scheme;
  import org.apache.http.conn.scheme.SchemeRegistry;
  import org.apache.http.impl.conn.PoolingClientConnectionManager;
  import org.eclipse.aether.RepositoryCache;
  import org.eclipse.aether.RepositorySystemSession;
  import org.eclipse.aether.util.ConfigUtils;
  
  /**
   * 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;
  
          public 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, ClientConnectionManager> 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<SslConfig, ClientConnectionManager>();
         userTokens = new ConcurrentHashMap<CompoundKey, Object>();
         authSchemePools = new ConcurrentHashMap<HttpHost, AuthSchemePool>();
         expectContinues = new ConcurrentHashMap<CompoundKey, Boolean>();
     }
 
     public void close()
     {
         for ( Iterator<Map.Entry<SslConfig, ClientConnectionManager>> it = connectionManagers.entrySet().iterator(); it.hasNext(); )
         {
             ClientConnectionManager connMgr = it.next().getValue();
             it.remove();
             connMgr.shutdown();
         }
     }
 
     public ClientConnectionManager getConnectionManager( SslConfig config )
     {
         ClientConnectionManager manager = connectionManagers.get( config );
         if ( manager == null )
         {
             ClientConnectionManager connMgr = newConnectionManager( config );
             manager = connectionManagers.putIfAbsent( config, connMgr );
             if ( manager != null )
             {
                 connMgr.shutdown();
             }
             else
             {
                 manager = connMgr;
             }
         }
         return manager;
     }
 
     public static ClientConnectionManager newConnectionManager( SslConfig sslConfig )
     {
         SchemeRegistry schemeReg = new SchemeRegistry();
         schemeReg.register( new Scheme( "http", 80, new PlainSocketFactory() ) );
         schemeReg.register( new Scheme( "https", 443, new SslSocketFactory( sslConfig ) ) );
 
         PoolingClientConnectionManager connMgr = new PoolingClientConnectionManager( schemeReg );
         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 );
     }
 
 }