info.rmapproject.auth.dao.ApiKeyDaoImpl.java Source code

Introduction

Here is the source code for info.rmapproject.auth.dao.ApiKeyDaoImpl.java

Source

/*******************************************************************************
 * Copyright 2018 Johns Hopkins University
 *
 * 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.
 *
 * This software was produced as part of the RMap Project (http://rmap-project.info),
 * The RMap Project was funded by the Alfred P. Sloan Foundation and is a 
 * collaboration between Data Conservancy, Portico, and IEEE.
 *******************************************************************************/
package info.rmapproject.auth.dao;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;

import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.query.Query;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import info.rmapproject.auth.exception.RMapAuthException;
import info.rmapproject.auth.model.ApiKey;

/**
 * Implementation of ApiKeyDao used to interact with data in the ApiKey table.
 *
 * @author khanson
 */

@Repository("apiKeyDao")
public class ApiKeyDaoImpl implements ApiKeyDao {

    /** The Constant logger. */
    private static final Logger LOG = LoggerFactory.getLogger(ApiKeyDaoImpl.class);

    /** Data base session factory instance. */
    @Autowired
    private SessionFactory sessionFactory;

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#addApiKey(ApiKey)
     */
    @Override
    public int addApiKey(ApiKey apiKey) throws RMapAuthException {
        Session session = this.sessionFactory.getCurrentSession();
        session.persist(apiKey);
        LOG.info("API key saved successfully, API key={}", apiKey);
        return apiKey.getApiKeyId();
    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#updateApiKey(ApiKey)
     */
    @Override
    public void updateApiKey(ApiKey apiKey) throws RMapAuthException {
        Session session = this.sessionFactory.getCurrentSession();
        session.update(apiKey);
        LOG.info("API key updated successfully, API Key={}", apiKey);
    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#getApiKeyById(int)
     */
    @Override
    public ApiKey getApiKeyById(int apiKeyId) throws RMapAuthException {
        Session session = this.sessionFactory.getCurrentSession();
        ApiKey apiKey = (ApiKey) session.load(ApiKey.class, apiKeyId);
        LOG.info("User record loaded successfully, User details={}", apiKey);
        return apiKey;
    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#getApiKeyByKeySecret(String,String)
     */
    @Override
    @SuppressWarnings("unchecked")
    public ApiKey getApiKeyByKeySecret(String accessKey, String secret) throws RMapAuthException {
        Session session = this.sessionFactory.getCurrentSession();
        Query<ApiKey> query = session.createQuery("from ApiKey where accessKey=:accessKey and secret=:secret");
        query.setParameter("accessKey", accessKey);
        query.setParameter("secret", secret);
        List<ApiKey> apiKeys = query.list();
        if (apiKeys != null && !apiKeys.isEmpty()) {
            LOG.info("Api key list loaded successfully");
            return apiKeys.get(0);
        } else {
            return null;
        }
    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#getApiKeyByKeySecret(String)
     */
    @Override
    @SuppressWarnings("unchecked")
    public ApiKey getApiKeyByKeyUri(String keyUri) throws RMapAuthException {
        Session session = this.sessionFactory.getCurrentSession();
        Query<ApiKey> query = session.createQuery("from ApiKey where keyUri=:keyUri");
        query.setParameter("keyUri", keyUri);
        List<ApiKey> apiKeys = query.list();
        if (apiKeys != null && !apiKeys.isEmpty()) {
            LOG.info("Api key list loaded successfully");
            return apiKeys.get(0);
        } else {
            return null;
        }
    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#getAgentUriByKeySecret(String,String)
     */
    @Override
    @SuppressWarnings("unchecked")
    public URI getAgentUriByKeySecret(String accessKey, String secret) throws RMapAuthException {
        URI agentUri = null;
        Session session = this.sessionFactory.getCurrentSession();
        Query<String> query = session.createNativeQuery(
                "select distinct rmapAgentUri from ApiKeys " + "inner join Users on ApiKeys.userId = Users.userId "
                        + "where accessKey=:accessKey and secret=:secret");
        query.setParameter("accessKey", accessKey);
        query.setParameter("secret", secret);
        List<String> rmapAgentUris = query.list();
        if (rmapAgentUris != null && !rmapAgentUris.isEmpty()) {
            LOG.info("User list loaded successfully");
            String agentId = rmapAgentUris.get(0);
            try {
                agentUri = new URI(agentId);
            } catch (URISyntaxException ex) {
                throw new RMapAuthException(ex);
            }
        }
        return agentUri;

    }

    /* (non-Javadoc)
     * @see info.rmapproject.auth.dao.ApiKeyDao#listApiKeyByUser(int)
     */
    @Override
    @SuppressWarnings("unchecked")
    public List<ApiKey> listApiKeyByUser(int userId) {
        Session session = this.sessionFactory.getCurrentSession();
        Query<ApiKey> query = session.createQuery("from ApiKey where userId=:userId");
        query.setParameter("userId", userId);
        List<ApiKey> apiKeys = query.list();
        LOG.info("Api key list loaded successfully");
        return apiKeys;
    }

}