mercury-1.0.0-alpha-2/mercury-transport/mercury-transport-http/src/main/java/org/apache/maven/mercury/spi/http/client/DestinationRealmResolver.java

/**
 * 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.maven.mercury.spi.http.client;

import java.io.IOException;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import org.apache.maven.mercury.logging.IMercuryLogger;
import org.apache.maven.mercury.logging.MercuryLoggerManager;
import org.apache.maven.mercury.transport.api.Server;
import org.mortbay.jetty.client.Address;
import org.mortbay.jetty.client.HttpDestination;
import org.mortbay.jetty.client.security.Realm;
import org.mortbay.jetty.client.security.RealmResolver;

/**
 * DestinationRealmResolver
 *
 * Resolve a security realm based on the Server instance
 * which matches an async Destination.
 */
public class DestinationRealmResolver implements RealmResolver
{
    private static final IMercuryLogger _log = MercuryLoggerManager.getLogger(DestinationRealmResolver.class);
    
    protected Set<Server> _servers = new HashSet<Server>();
    
    
    
    public DestinationRealmResolver (Set<Server> servers)
    {
        _servers.addAll(servers);
    }

    public Realm getRealm(String name, HttpDestination dest, String pathSpec)
            throws IOException
    {
       Address address = dest.getAddress();
       boolean secure = dest.isSecure();

       if( _log.isDebugEnabled() )
       {
         _log.debug("Dest "+address.getHost()+":"+address.getPort()+"(secure="+secure+")" );
         _log.debug("Server list: "+_servers );
         
       }

       //get a username and password appropriate for the destination. Usernames and passwords are
       //associated with Server Credential objects.
       Server server = null;
       Iterator<Server> itor = _servers.iterator();
       while (server==null && itor.hasNext())
       {
           Server s = itor.next();
           if (s.getURL() != null)
           {
               String protocol = s.getURL().getProtocol();
               String host = s.getURL().getHost();
               int port = s.getURL().getPort();
               if( port == -1 )
               {
                 port = "https".equalsIgnoreCase( protocol ) ? 443 : 80;
               }

               if( _log.isDebugEnabled() )
                 _log.debug("Trying dest "+address.getHost()+":"+address.getPort()+"(secure="+dest.isSecure()
                     +") against server "+protocol+"://"+host+":"+port );

               if (((secure && "https".equalsIgnoreCase(protocol)) || (!secure && "http".equalsIgnoreCase(protocol)))
                   &&
                   (address.getPort() == port))
               {
                   if (address.getHost().equalsIgnoreCase(host) || address.getHost().equalsIgnoreCase(host))
                   {
                       server = s;
                       if (_log.isDebugEnabled())
                           _log.debug("Matched server "+address.getHost()+":"+address.getPort());
                   }
               }
           }
       }

       if (server == null || server.getServerCredentials() == null)
       {
           if (_log.isDebugEnabled())
               _log.debug("No server matching "+address.getHost()+":"+address.getPort()+" or no credentials");
           return null;
       }
       
       if (server.getServerCredentials().isCertificate())
           throw new UnsupportedOperationException("Certificate not supported");
       
       final Server realmServer = server;
       return new Realm ()
       {
           public String getCredentials()
           {
               return realmServer.getServerCredentials().getPass();
           }

           public String getId()
           {
               return realmServer.getId();
           }

           public String getPrincipal()
           {
               return realmServer.getServerCredentials().getUser();
           }

       };
    }

}