content: package org.apache.archiva.admin.repository.networkproxy;
  /*
   * 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 org.apache.archiva.admin.model.AuditInformation;
  import org.apache.archiva.admin.model.RepositoryAdminException;
  import org.apache.archiva.admin.model.beans.NetworkProxy;
  import org.apache.archiva.admin.model.networkproxy.NetworkProxyAdmin;
  import org.apache.archiva.admin.repository.AbstractRepositoryAdmin;
  import org.apache.archiva.configuration.model.Configuration;
  import org.apache.archiva.configuration.model.NetworkProxyConfiguration;
  import org.apache.archiva.metadata.model.facets.AuditEvent;
  import org.apache.archiva.repository.RemoteRepository;
  import org.apache.archiva.repository.RepositoryException;
  import org.apache.archiva.repository.RepositoryRegistry;
  import org.apache.archiva.repository.features.RemoteIndexFeature;
  import org.apache.commons.lang3.StringUtils;
  import org.springframework.stereotype.Service;
  
  import javax.inject.Inject;
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * @author Olivier Lamy
   * @since 1.4-M1
   */
  @Service( "networkProxyAdmin#default" )
  public class DefaultNetworkProxyAdmin
      extends AbstractRepositoryAdmin
      implements NetworkProxyAdmin
  {
  
      @Inject
      RepositoryRegistry repositoryRegistry;
  
      @Override
      public List<NetworkProxy> getNetworkProxies()
          throws RepositoryAdminException
      {
          List<NetworkProxy> networkProxies =
              new ArrayList<>( getArchivaConfiguration().getConfiguration().getNetworkProxies().size() );
          for ( NetworkProxyConfiguration networkProxyConfiguration : getArchivaConfiguration().getConfiguration().getNetworkProxies() )
          {
              networkProxies.add( getNetworkProxy( networkProxyConfiguration ) );
          }
          return networkProxies;
      }
  
      @Override
      public NetworkProxy getNetworkProxy( String networkProxyId )
          throws RepositoryAdminException
      {
          for ( NetworkProxy networkProxy : getNetworkProxies() )
          {
              if ( StringUtils.equals( networkProxyId, networkProxy.getId() ) )
              {
                  return networkProxy;
              }
          }
  
          return null;
      }
  
      @Override
      public void addNetworkProxy( NetworkProxy networkProxy, AuditInformation auditInformation )
          throws RepositoryAdminException
      {
          if ( networkProxy == null )
          {
              return;
          }
          if ( getNetworkProxy( networkProxy.getId() ) != null )
          {
              throw new RepositoryAdminException(
                  "cannot add NetworkProxy with id " + networkProxy.getId() + " already exist" );
          }
          Configuration configuration = getArchivaConfiguration().getConfiguration();
          configuration.addNetworkProxy( getNetworkProxyConfiguration( networkProxy ) );
  
          triggerAuditEvent( networkProxy.getId(), null, AuditEvent.ADD_NETWORK_PROXY, auditInformation );
  
          saveConfiguration( configuration );
      }
  
      @Override
      public void updateNetworkProxy( NetworkProxy networkProxy, AuditInformation auditInformation )
          throws RepositoryAdminException
      {
          if ( networkProxy == null )
          {
              return;
          }
          if ( getNetworkProxy( networkProxy.getId() ) == null )
          {
              throw new RepositoryAdminException(
                  "cannot update NetworkProxy with id " + networkProxy.getId() + " as not exist" );
          }
          Configuration configuration = getArchivaConfiguration().getConfiguration();
          NetworkProxyConfiguration networkProxyConfiguration = getNetworkProxyConfiguration( networkProxy );
          configuration.removeNetworkProxy( networkProxyConfiguration );
          configuration.addNetworkProxy( networkProxyConfiguration );
  
          triggerAuditEvent( networkProxy.getId(), null, AuditEvent.MODIFY_NETWORK_PROXY, auditInformation );
  
          saveConfiguration( configuration );
      }
  
      @Override
      public void deleteNetworkProxy( String networkProxyId, AuditInformation auditInformation )
          throws RepositoryAdminException
      {
  
          NetworkProxy networkProxy = getNetworkProxy( networkProxyId );
          if ( networkProxy == null )
          {
              throw new RepositoryAdminException(
                  "cannot delete NetworkProxy with id " + networkProxyId + " as not exist" );
          }
          for ( RemoteRepository repo : repositoryRegistry.getRemoteRepositories()) {
              if (repo.supportsFeature( RemoteIndexFeature.class )) {
                  RemoteIndexFeature rif = repo.getFeature( RemoteIndexFeature.class );
                  if (networkProxyId.equals(rif.getProxyId())) {
                      rif.setProxyId( null );
                      try
                      {
                          repositoryRegistry.putRepository( repo );
                      }
                      catch ( RepositoryException e )
                      {
                          log.error("Could not update repository {}", repo.getId(), e);
                      }
                  }
              }
          }
  
          Configuration configuration = getArchivaConfiguration().getConfiguration();
          NetworkProxyConfiguration networkProxyConfiguration = getNetworkProxyConfiguration( networkProxy );
          configuration.removeNetworkProxy( networkProxyConfiguration );
          saveConfiguration( configuration );
          triggerAuditEvent( networkProxy.getId(), null, AuditEvent.DELETE_NETWORK_PROXY, auditInformation );
      }
  
      protected NetworkProxy getNetworkProxy( NetworkProxyConfiguration networkProxyConfiguration )
      {
          return networkProxyConfiguration == null
              ? null
              : getModelMapper().map( networkProxyConfiguration, NetworkProxy.class );
      }
  
      protected NetworkProxyConfiguration getNetworkProxyConfiguration( NetworkProxy networkProxy )
      {
          return networkProxy == null
              ? null
              : getModelMapper().map( networkProxy, NetworkProxyConfiguration.class );
      }
  }
  