D-Net

package eu.dnetlib.goldoa.service.dao;

import eu.dnetlib.goldoa.domain.AccountAction;

import eu.dnetlib.goldoa.domain.UserRole;

import eu.dnetlib.goldoa.domain.Role;

import eu.dnetlib.goldoa.domain.User;

import org.apache.commons.codec.digest.DigestUtils;

import org.hibernate.Criteria;

import org.hibernate.Hibernate;

import org.hibernate.criterion.Restrictions;

import org.hibernate.type.StringType;

import org.springframework.stereotype.Repository;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Timestamp;

import java.util.ArrayList;

import java.util.List;

/**

 * Created by antleb on 3/13/15.

 */

@Repository

public class UserDAO extends AbstractDao<String, User> {

    /**

     * Returns the person with the given id. Objects of different type are placeholders containing only their id.

     * Objects representing relations are fully initialized, with placeholders for other main entities.

     *

     * @param userId

     * @return

     */

    public User getUserById(String userId) {

        return (User) createEntityCriteria().add(Restrictions.eq("id", userId)).list().get(0);

    }

    /**

     * Returns the person with the given email. Objects of different type are placeholders containing only their id.

     * Objects representing relations are fully initialized, with placeholders for other main entities.

     *

     * @param email

     * @return

     */

    @SuppressWarnings("unchecked")

    public User getUserByEmail(String email) {

        List<User> users = createEntityCriteria().add(Restrictions.eq("email", email)).list();

        if (users.size() == 0)

            return null;

        return users.get(0);

    }

    public void saveToken(final String email, final String token, final int tokenTTL) {

        AccountAction acc_action = new AccountAction();

        acc_action.setUser(getUserByEmail(email));

        acc_action.setType("activation");

        acc_action.setToken(token);

        acc_action.setDate(new Timestamp(System.currentTimeMillis()));

        acc_action.setExpires(new Timestamp(System.currentTimeMillis() + 1000L * 60L * 60L * 24L * tokenTTL));

        getSession().persist(acc_action);

    }

    public boolean verifyLogin(String email, String password) {

        List users = createEntityCriteria().add(Restrictions.eq("email", email)).list();

        //		.add(Restrictions.eq("password", DigestUtils.md5Hex(password))).list();

        if (users.size() != 0)

            return true;

        return false;

    }

    @SuppressWarnings("unchecked")

    public boolean verifyToken(final String email, final String token) {

        Criteria cr = getSession().createCriteria(AccountAction.class);

        List<AccountAction> ac = cr.add(Restrictions.eq("email", email))

                .add(Restrictions.eq("token", token)).list();

        if (ac.size() > 0)

            return true;

        return false;

    }

    public void activateUser(final String email) {

        User user = (User) createEntityCriteria().add(Restrictions.eq("email", email)).list().get(0);

        user.setActive(true);

        getSession().update(user);

    }

    public void deleteToken(final String email, final String token) {

        Criteria cr = getSession().createCriteria(AccountAction.class);

        AccountAction ac = (AccountAction) cr.add(Restrictions.eq("email", email))

                .add(Restrictions.eq("token", token))

                .add(Restrictions.eq("type", "activation")).list().get(0);

        getSession().delete(ac);

    }

    @SuppressWarnings("unchecked")

    public boolean updatePassword(final String newPassword, final String email) {

        List<User> rs = createEntityCriteria().add(Restrictions.eq("email", email)).list();

        if (rs.get(0) != null) {

            User user = rs.get(0);

            user.setPassword(DigestUtils.md5Hex(newPassword));

            getSession().update(user);

            return true;

        }

        return false;

    }

    @SuppressWarnings("unchecked")

    public List<Role> getRoles() {

        Criteria cr = getSession().createCriteria(Role.class);

        return cr.list();

    }

    public void saveUser(final User user) {

        for (UserRole ur : user.getRoles()) {

            Role r = ur.getPk().getRole();

            getSession().persist(r);

        }

        persist(user);

    }

    //TODO MOVE TO PROJECT DAO

	/*public void removeProjectCoordinators(final String email) {

		Criteria cr = getSession().createCriteria(ProjectCoordinator.class);

		ProjectCoordinator pc = (ProjectCoordinator) cr.add(Restrictions.eq("email",email)).list().get(0);

		getSession().delete(pc);

	}*/

	/*public void saveProjectCoordinators(final String email, final List<ProjectCoordinator> projectCoordinators) {

		User user = (User) createEntityCriteria().add(Restrictions.eq("email", email)).list().get(0);

		user.setProjectCoordinators(projectCoordinators);

		persist(user);

	}*/

	/*public void saveAffiliations(final User user, final List<Affiliation> affiliations) {

		user.setAffiliations(affiliations);

		persist(user);

	}*/

    @SuppressWarnings("unchecked")

    public List<User> getUsersByRole(String role_id) {

        Criteria cr = getSession().createCriteria(Role.class);

        Role role = (Role) cr.add(Restrictions.eq("id", role_id)).list().get(0);

        cr = getSession().createCriteria(UserRole.class);

        List<UserRole> rs = cr.add(Restrictions.eq("approved", true))

                .add(Restrictions.eq("pk.role", role)).list();

        List<User> users = new ArrayList<>();

        for (UserRole ur : rs)

            users.add(ur.getPk().getUser());

        return users;

    }

    public List<User> getModerators() {

        return getUsersByRole("moderator");

    }

    public List<User> getAccountingOfficers() {

        return getUsersByRole("accounting");

    }

    public Role getRole(final String roleId) {

        Criteria cr = getSession().createCriteria(Role.class);

        cr.add(Restrictions.eq("id", roleId));

        return (Role) cr.list().get(0);

    }

    public void deactivateUser(String email) {

        User user = (User) createEntityCriteria().add(Restrictions.eq("email", email)).list().get(0);

        user.setActive(false);

        getSession().update(user);

    }

    public void acceptUserRole(String email, String roleId) {

        changeRoleApprovedStatus(email, roleId, true);

    }

    public void rejectUserRole(String email, String roleId) {

        changeRoleApprovedStatus(email, roleId, false);

    }

    public void changeRoleApprovedStatus(String email, String roleId, boolean status) {

        Criteria cr = getSession().createCriteria(UserRole.class);

        User user = getUserByEmail(email);

        Role role = getRole(roleId);

        UserRole ur = (UserRole) cr.add(Restrictions.eq("pk.user", user))

                .add(Restrictions.eq("pk.role", role)).list().get(0);

        ur.setApproved(status);

        getSession().update(ur);

    }

    @SuppressWarnings("unchecked")

    public List<User> getUsers() {

        return createEntityCriteria().list();

    }

}