libraries/shiro-core/src/main/java/org/qi4j/library/shiro/domain/passwords/PasswordRealmMixin.java

/*
 * Copyright (c) 2012, Paul Merlin. All Rights Reserved.
 *
 * Licensed 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.qi4j.library.shiro.domain.passwords;

import java.util.Set;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.SimpleAuthenticationInfo;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.authc.credential.DefaultPasswordService;
import org.apache.shiro.authc.credential.PasswordMatcher;
import org.apache.shiro.authc.credential.PasswordService;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.Authorizer;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.crypto.hash.DefaultHashService;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.realm.Realm;
import org.apache.shiro.subject.PrincipalCollection;
import org.qi4j.api.configuration.Configuration;
import org.qi4j.api.injection.scope.Structure;
import org.qi4j.api.injection.scope.This;
import org.qi4j.api.query.QueryBuilder;
import org.qi4j.api.service.ServiceActivation;
import org.qi4j.api.structure.Module;
import org.qi4j.api.unitofwork.UnitOfWork;
import org.qi4j.library.shiro.Shiro;
import org.qi4j.library.shiro.domain.permissions.RoleAssignee;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import static org.qi4j.api.query.QueryExpressions.eq;
import static org.qi4j.api.query.QueryExpressions.templateFor;

public class PasswordRealmMixin
        extends AuthorizingRealm
        implements Realm, Authorizer, PasswordService, ServiceActivation
{

    private static final Logger LOG = LoggerFactory.getLogger( Shiro.LOGGER_NAME );

    @Structure
    private Module module;

    @This
    private Configuration<PasswordRealmConfiguration> configuration;

    private final DefaultPasswordService passwordService;

    public PasswordRealmMixin()
    {
        super();
        passwordService = new DefaultPasswordService();
        PasswordMatcher matcher = new PasswordMatcher();
        matcher.setPasswordService( passwordService );
        setCredentialsMatcher( matcher );
    }

    @Override
    public void activateService()
            throws Exception
    {
        configuration.refresh();
        PasswordRealmConfiguration config = configuration.get();
        String algorithm = config.hashAlgorithmName().get();
        Integer iterations = config.hashIterationsCount().get();
        if ( algorithm != null || iterations != null ) {
            DefaultHashService hashService = ( DefaultHashService ) passwordService.getHashService();
            if ( algorithm != null ) {
                hashService.setHashAlgorithmName( algorithm );
            }
            if ( iterations != null ) {
                hashService.setHashIterations( iterations );
            }
        }
    }

    @Override
    public void passivateService()
            throws Exception
    {
    }

    @Override
    public String encryptPassword( Object plaintextPassword )
            throws IllegalArgumentException
    {
        return passwordService.encryptPassword( plaintextPassword );
    }

    @Override
    public boolean passwordsMatch( Object submittedPlaintext, String encrypted )
    {
        return passwordService.passwordsMatch( submittedPlaintext, encrypted );
    }

    @Override
    protected AuthenticationInfo doGetAuthenticationInfo( AuthenticationToken token )
            throws AuthenticationException
    {
        UnitOfWork uow = module.newUnitOfWork();
        try {

            String username = ( ( UsernamePasswordToken ) token ).getUsername();
            PasswordSecurable account = findPasswordSecurable( uow, username );
            if ( account == null ) {
                LOG.debug( "Unknown subject identifier: {}" + username );
                return null;
            }
            LOG.debug( "Found account for {}: {}", username, account );
            return new SimpleAuthenticationInfo( account.subjectIdentifier().get(), account.password().get(), getName() );

        } finally {
            uow.discard();
        }
    }

    @Override
    protected AuthorizationInfo doGetAuthorizationInfo( PrincipalCollection principals )
    {
        UnitOfWork uow = module.newUnitOfWork();
        try {

            String username = getAvailablePrincipal( principals ).toString();
            RoleAssignee roleAssignee = findRoleAssignee( uow, username );
            if ( roleAssignee == null ) {
                LOG.debug( "No authorization info for {}", username );
                return null;
            }
            LOG.debug( "Found role assignee for {}: {}", username, roleAssignee );
            Set<String> roleNames = roleAssignee.roleNames();
            Set<String> permissionStrings = roleAssignee.permissionStrings();
            LOG.debug( "Found role assignee has the following roles: {}", roleNames );
            LOG.debug( "Found role assignee has the following permissions: {}", permissionStrings );
            SimpleAuthorizationInfo atzInfo = new SimpleAuthorizationInfo( roleNames );
            atzInfo.setStringPermissions( permissionStrings );
            return atzInfo;

        } finally {
            uow.discard();
        }
    }

    private PasswordSecurable findPasswordSecurable( UnitOfWork uow, String username )
    {
        QueryBuilder<PasswordSecurable> builder = module.newQueryBuilder( PasswordSecurable.class );
        builder = builder.where( eq( templateFor( PasswordSecurable.class ).subjectIdentifier(), username ) );
        return uow.newQuery( builder ).find();
    }

    private RoleAssignee findRoleAssignee( UnitOfWork uow, String username )
    {
        QueryBuilder<RoleAssignee> builder = module.newQueryBuilder( RoleAssignee.class );
        builder = builder.where( eq( templateFor( RoleAssignee.class ).subjectIdentifier(), username ) );
        return uow.newQuery( builder ).find();
    }

}