leap.oauth2.server.store.DefaultJdbcAuthzClientStore Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of leap-oauth2-server Show documentation
There is a newer version: 0.7.13b
/*
 * Copyright 2015 the original author or authors.
 *
 * 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 leap.oauth2.server.store;

import leap.core.AppConfig;
import leap.lang.Strings;
import leap.lang.path.AntPathPattern;
import leap.oauth2.server.client.AuthzClient;
import leap.oauth2.server.client.AuthzClientStore;
import leap.oauth2.server.client.SimpleAuthzClient;
import leap.oauth2.server.entity.AuthzClientEntity;
import leap.orm.OrmMetadata;
import leap.orm.command.CreateEntityCommand;
import leap.orm.dao.Dao;
import leap.orm.dmo.Dmo;
import leap.orm.sql.SqlCommand;

public class DefaultJdbcAuthzClientStore extends AbstractJdbcAuthzStore implements AuthzClientStore {

    public static final String LOAD_CLIENT_SQL_KEY = "oauth2.as.loadClient";

    protected SqlCommand loadClientCommand;

    @Override
    public AuthzClient loadClient(String clientId) {
    	AuthzClientEntity entity=loadAuthzClientEntity(clientId);

        return null == entity ? null : createAuthzClientFromEntity(entity);
    }

    protected AuthzClientEntity loadAuthzClientEntity(String clientId){
    	AuthzClientEntity entity;

        if(null == loadClientCommand) {
            entity = dao.findOrNull(AuthzClientEntity.class, clientId);
        }else{
            entity = dao.createQuery(AuthzClientEntity.class, loadClientCommand).firstOrNull();
        }
        return entity;
    }

    protected AuthzClient createAuthzClientFromEntity(AuthzClientEntity entity) {
        SimpleAuthzClient client = new SimpleAuthzClient();

        client.setId(entity.getId());
        client.setSecret(entity.getSecret());
        client.setRedirectUri(entity.getRedirectUri());
        client.setLogoutUri(entity.getLogoutUri());

        if(!Strings.isEmpty(entity.getRedirectUriPattern())) {
            client.setRedirectUriPattern(new AntPathPattern(entity.getRedirectUriPattern()));
        }

        if(!Strings.isEmpty(entity.getLogoutUriPattern())) {
            client.setLogoutUriPattern(new AntPathPattern(entity.getLogoutUriPattern()));
        }

        client.setAccessTokenExpires(entity.getAccessTokenExpires());
        client.setRefreshTokenExpires(entity.getRefreshTokenExpires());
        client.setAllowAuthorizationCode(entity.getAllowAuthorizationCode());
        client.setAllowRefreshToken(entity.getAllowRefreshToken());
        client.setAllowLoginToken(entity.getAllowLoginToken());
        client.setGrantedScope(entity.getGrantedScope());
        client.setEnabled(entity.isEnabled());

        return client;
    }

    @Override
    protected void init(AppConfig config) {
        createEntityMapping(dmo, config.isDebug());
        resolveSqlCommands(dao, dao.getOrmContext().getMetadata());
    }

    protected void createEntityMapping(Dmo dmo, boolean debug) {
        OrmMetadata md = dmo.getOrmContext().getMetadata();

        if(null == md.tryGetEntityMapping(AuthzClientEntity.class)) {
            CreateEntityCommand cmd = dmo.cmdCreateEntity(AuthzClientEntity.class);
            cmd.setUpgradeTable(debug);
            cmd.execute();
        }
    }

    protected void resolveSqlCommands(Dao dao, OrmMetadata md) {
        loadClientCommand = md.tryGetSqlCommand(LOAD_CLIENT_SQL_KEY);
    }
}