package org.apache.archiva.security;
   
   /*
    * 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.RepositoryAdminException;
  import org.apache.archiva.admin.model.beans.ManagedRepository;
  import org.apache.archiva.admin.model.managed.ManagedRepositoryAdmin;
  import org.apache.archiva.redback.authentication.AuthenticationResult;
  import org.apache.archiva.redback.authorization.AuthorizationException;
  import org.apache.archiva.redback.role.RoleManager;
  import org.apache.archiva.redback.role.RoleManagerException;
  import org.apache.archiva.redback.system.DefaultSecuritySession;
  import org.apache.archiva.redback.system.SecuritySession;
  import org.apache.archiva.redback.system.SecuritySystem;
  import org.apache.archiva.redback.users.User;
  import org.apache.archiva.redback.users.UserManagerException;
  import org.apache.archiva.redback.users.UserNotFoundException;
  import org.apache.archiva.security.common.ArchivaRoleConstants;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.stereotype.Service;
  
  import javax.inject.Inject;
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * DefaultUserRepositories
   */
  @Service( "userRepositories" )
  public class DefaultUserRepositories
      implements UserRepositories
  {
  
      @Inject
      private SecuritySystem securitySystem;
  
      @Inject
      private RoleManager roleManager;
  
      @Inject
      private ManagedRepositoryAdmin managedRepositoryAdmin;
  
      private Logger log = LoggerFactory.getLogger( getClass() );
  
      @Override
      public List<String> getObservableRepositoryIds( String principal )
          throws PrincipalNotFoundException, AccessDeniedException, ArchivaSecurityException
      {
          String operation = ArchivaRoleConstants.OPERATION_REPOSITORY_ACCESS;
  
          return getAccessibleRepositoryIds( principal, operation );
      }
  
      @Override
      public List<String> getManagableRepositoryIds( String principal )
          throws PrincipalNotFoundException, AccessDeniedException, ArchivaSecurityException
      {
          String operation = ArchivaRoleConstants.OPERATION_REPOSITORY_UPLOAD;
  
          return getAccessibleRepositoryIds( principal, operation );
      }
  
      private List<String> getAccessibleRepositoryIds( String principal, String operation )
          throws ArchivaSecurityException, AccessDeniedException, PrincipalNotFoundException
      {
  
          List<ManagedRepository> managedRepositories = getAccessibleRepositories( principal, operation );
          List<String> repoIds = new ArrayList<>( managedRepositories.size() );
          for ( ManagedRepository managedRepository : managedRepositories )
          {
              repoIds.add( managedRepository.getId() );
          }
  
          return repoIds;
      }
  
      @Override
      public List<ManagedRepository> getAccessibleRepositories( String principal )
          throws ArchivaSecurityException, AccessDeniedException, PrincipalNotFoundException
      {
          return getAccessibleRepositories( principal, ArchivaRoleConstants.OPERATION_REPOSITORY_ACCESS );
     }
 
     @Override
     public List<ManagedRepository> getManagableRepositories(String principal) throws ArchivaSecurityException, AccessDeniedException, PrincipalNotFoundException {
         return getAccessibleRepositories( principal, ArchivaRoleConstants.OPERATION_REPOSITORY_UPLOAD );
     }
 
     private List<ManagedRepository> getAccessibleRepositories( String principal, String operation )
         throws ArchivaSecurityException, AccessDeniedException, PrincipalNotFoundException
     {
         SecuritySession securitySession = createSession( principal );
 
         List<ManagedRepository> managedRepositories = new ArrayList<>();
 
         try
         {
             List<ManagedRepository> repos = managedRepositoryAdmin.getManagedRepositories();
 
             for ( ManagedRepository repo : repos )
             {
                 try
                 {
                     String repoId = repo.getId();
                     if ( securitySystem.isAuthorized( securitySession, operation, repoId ) )
                     {
                         managedRepositories.add( repo );
                     }
                 }
                 catch ( AuthorizationException e )
                 {
                     // swallow.
 
                     log.debug( "Not authorizing '{}' for repository '{}': {}", principal, repo.getId(),
                                e.getMessage() );
 
                 }
             }
 
             return managedRepositories;
         }
         catch ( RepositoryAdminException e )
         {
             throw new ArchivaSecurityException( e.getMessage(), e );
         }
     }
 
     private SecuritySession createSession( String principal )
         throws ArchivaSecurityException, AccessDeniedException
     {
         User user;
         try
         {
             user = securitySystem.getUserManager().findUser( principal );
             if ( user == null )
             {
                 throw new ArchivaSecurityException(
                     "The security system had an internal error - please check your system logs" );
             }
         }
         catch ( UserNotFoundException e )
         {
             throw new PrincipalNotFoundException( "Unable to find principal " + principal + "", e );
         }
         catch ( UserManagerException e )
         {
             throw new ArchivaSecurityException( e.getMessage(), e );
         }
 
         if ( user.isLocked() )
         {
             throw new AccessDeniedException( "User " + principal + "(" + user.getFullName() + ") is locked." );
         }
 
         AuthenticationResult authn = new AuthenticationResult( true, principal, null );
         authn.setUser( user );
         return new DefaultSecuritySession( authn, user );
     }
 
     @Override
     public void createMissingRepositoryRoles( String repoId )
         throws ArchivaSecurityException
     {
         try
         {
             if ( !roleManager.templatedRoleExists( ArchivaRoleConstants.TEMPLATE_REPOSITORY_OBSERVER, repoId ) )
             {
                 roleManager.createTemplatedRole( ArchivaRoleConstants.TEMPLATE_REPOSITORY_OBSERVER, repoId );
             }
 
             if ( !roleManager.templatedRoleExists( ArchivaRoleConstants.TEMPLATE_REPOSITORY_MANAGER, repoId ) )
             {
                 roleManager.createTemplatedRole( ArchivaRoleConstants.TEMPLATE_REPOSITORY_MANAGER, repoId );
             }
         }
         catch ( RoleManagerException e )
         {
             throw new ArchivaSecurityException( "Unable to create roles for configured repositories: " + e.getMessage(),
                                                 e );
         }
     }
 
     @Override
     public boolean isAuthorizedToUploadArtifacts( String principal, String repoId )
         throws PrincipalNotFoundException, ArchivaSecurityException
     {
         try
         {
             SecuritySession securitySession = createSession( principal );
 
             return securitySystem.isAuthorized( securitySession, ArchivaRoleConstants.OPERATION_REPOSITORY_UPLOAD,
                                                 repoId );
 
         }
         catch ( AuthorizationException e )
         {
             throw new ArchivaSecurityException( e.getMessage(), e);
         }
     }
 
     @Override
     public boolean isAuthorizedToDeleteArtifacts( String principal, String repoId )
         throws ArchivaSecurityException
     {
         try
         {
             SecuritySession securitySession = createSession( principal );
 
             return securitySystem.isAuthorized( securitySession, ArchivaRoleConstants.OPERATION_REPOSITORY_DELETE,
                                                 repoId );
 
         }
         catch ( AuthorizationException e )
         {
             throw new ArchivaSecurityException( e.getMessage(), e);
         }
     }
 
     public SecuritySystem getSecuritySystem()
     {
         return securitySystem;
     }
 
     public void setSecuritySystem( SecuritySystem securitySystem )
     {
         this.securitySystem = securitySystem;
     }
 
     public RoleManager getRoleManager()
     {
         return roleManager;
     }
 
     public void setRoleManager( RoleManager roleManager )
     {
         this.roleManager = roleManager;
     }
 }