package com.atlassian.sal.core.net;
   
   import com.atlassian.sal.api.user.UserManager;
   import com.atlassian.sal.core.net.auth.TrustedTokenScheme;
   import com.atlassian.sal.core.trusted.CertificateFactory;
   
   import org.apache.http.HttpHost;
   import org.apache.http.auth.AuthScope;
   import org.apache.http.auth.UsernamePasswordCredentials;
  import org.apache.http.client.protocol.HttpClientContext;
  import org.apache.http.impl.client.CloseableHttpClient;
  
  /**
   * HttpClient implementation of Request interface
   */
  public class HttpClientTrustedRequest extends HttpClientRequest
  {
  
      private final UserManager userManager;
      private final CertificateFactory certificateFactory;
  
      public HttpClientTrustedRequest(final UserManager userManager, final CertificateFactory certificateFactory, final CloseableHttpClient httpClient, final HttpClientContext httpClientContext, final MethodType initialMethodType, final String initialUrl)
      {
          super(httpClient, httpClientContext, initialMethodType, initialUrl);
          this.userManager = userManager;
          this.certificateFactory = certificateFactory;
      }
  
      public HttpClientTrustedRequest addTrustedTokenAuthentication(final String hostname)
      {
          return addTrustedTokenAuthentication(hostname, userManager.getRemoteUsername());
      }
  
      public HttpClientTrustedRequest addTrustedTokenAuthentication(final String hostname, final String username)
      {
          httpClientContext.getCredentialsProvider().setCredentials(
                  new AuthScope(hostname, AuthScope.ANY_PORT),
                  new UsernamePasswordCredentials(username, null));
          httpClientContext.getAuthCache().put(new HttpHost(hostname), new TrustedTokenScheme(certificateFactory));
          return this;
      }
  }