D-Net

package eu.dnetlib.goldoa.service.dao;

import eu.dnetlib.goldoa.domain.AccountAction;

import eu.dnetlib.goldoa.domain.Affiliation;

import eu.dnetlib.goldoa.domain.Budget;

import eu.dnetlib.goldoa.domain.Organization;

import eu.dnetlib.goldoa.domain.UserRole;

import eu.dnetlib.goldoa.domain.UserRolePK;

import eu.dnetlib.goldoa.domain.Project;

import eu.dnetlib.goldoa.domain.ProjectCoordinator;

import eu.dnetlib.goldoa.domain.Publisher;

import eu.dnetlib.goldoa.domain.Role;

import eu.dnetlib.goldoa.domain.User;

import eu.dnetlib.goldoa.service.UserManagerImpl;

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

import org.hibernate.Criteria;

import org.hibernate.Query;

import org.hibernate.criterion.Order;

import org.hibernate.criterion.Restrictions;

import org.springframework.beans.factory.annotation.Autowired;

/*import org.springframework.jdbc.core.BatchPreparedStatementSetter;

import org.springframework.jdbc.core.JdbcTemplate;

import org.springframework.jdbc.core.PreparedStatementSetter;

import org.springframework.jdbc.core.RowMapper;*/

import org.springframework.stereotype.Repository;

import org.springframework.transaction.annotation.Transactional;

import javax.sql.DataSource;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Timestamp;

import java.sql.Types;

import java.util.ArrayList;

import java.util.Date;

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.ilike("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 (User) 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);

	}

	/*public void deleteAffiliations(final String email) {

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

		Affiliation af = (Affiliation) cr.add(Restrictions.eq("email",email)).list().get(0);

		getSession().delete(af);

	}*/

	/*public void deletePersonRoles(final String email) {

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

		UserRole ur = (UserRole) cr.add(Restrictions.eq("email",email)).list().get(0);

		getSession().delete(ur);

	}*/

	//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 saveUserRoles(final String email, final List<Role> roles) {

		User user = getUserByEmail(email);

		user.setRole(roles);

		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(int i=0;i<users.size();i++)

			users.add(rs.get(i).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 role_id) {

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

		User user = getUserByEmail(email);

		Role role = getRole(role_id);

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

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

		ur.setApproved(true);

		getSession().update(ur);

	}

	public void rejectUserRole(String email, String roleId) {

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

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

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

		getSession().delete(ur);

	}

	@SuppressWarnings("unchecked")

	public List<User> getUsers() {

		return createEntityCriteria().list();

	}

	private User readPerson(ResultSet rs) throws SQLException {

		return null;

//		Person person = new Person();

//

//		person.setId(rs.getString("id"));

//		person.setEmail(rs.getString("email"));

//		person.setName(rs.getString("firstname"));

//		person.setLastname(rs.getString("lastname"));

//		person.setInitials(rs.getString("initials"));

//		person.setTelephone(rs.getString("telephone"));

//		person.setActive(rs.getBoolean("active"));

//		person.setOrcidId(rs.getString("orcidid"));

//		person.setPassword(rs.getString("password"));

//		person.setSource(rs.getString("source"));

//		person.setPublisher(new Publisher(rs.getString("publisher")));

//

//		person.setCoordinatedProjects(new ArrayList<Project>());

//		for (String projectId : (String[]) rs.getArray("projects").getArray()) {

//			person.getCoordinatedProjects().add(new Project(projectId));

//		}

//

//		person.setAffiliations(new ArrayList<Affiliation>());

//		for (String affString : (String[]) rs.getArray("affiliations").getArray()) {

//			if (affString != null) {

//				String[] parts = affString.split("\\|\\|");

//				String org = parts[0];

//				String dep = parts[1];

//				Date start = parts[2] == "0" ? null : new Date(Long.parseLong(parts[2]));

//				Date end = parts[3] == "0" ? null : new Date(Long.parseLong(parts[3]));

//

//				Affiliation affiliation = new Affiliation();

//

//				affiliation.setPerson(person);

//				affiliation.setOrganization(new Organization(org));

//				affiliation.setDepartment(dep);

//				affiliation.setStart(start);

//				affiliation.setEnd(end);

//

//				person.getAffiliations().add(affiliation);

//			}

//		}

//

//		person.setRoles(new ArrayList<PersonRole>());

//		for (String rString : (String[]) rs.getArray("roles").getArray()) {

//			if (rString != null) {

//				String[] parts = rString.split("\\|\\|");

//				String role = parts[0];

//				boolean approved = Boolean.parseBoolean(parts[1]);

//

//				person.getRoles().add(new PersonRole(person, getRole(role), approved));

//			}

//		}

//

//		return person;

	}

}