base.jee.api.cassandra.SearchPeople Maven / Gradle / Ivy

Go to download
/**
 * Creative commons Attribution-NonCommercial license.
 *
 * http://creativecommons.org/licenses/by-nc/2.5/au/deed.en_GB
 *
 * NO WARRANTY IS GIVEN OR IMPLIED, USE AT YOUR OWN RISK.
 */
package base.jee.api.cassandra;

import java.io.IOException;

import base.Query;
import base.text.TagsToArray;
import com.datastax.driver.core.Row;
import com.datastax.driver.core.PreparedStatement;

import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import base.jee.Constants;
import base.json.Json;
import base.security.PermissionException;
import base.security.Person;
import base.security.User;
import com.datastax.driver.core.Session;

import static base.jee.api.cassandra.util.Log.log;

public class SearchPeople extends Query {

	private CassandraAPI api;
	private User user;
	private String keywords;

	public SearchPeople() {
	}

	public SearchPeople(CassandraAPI api, User user, String keywords) throws PermissionException {

		if(api == null) {
			throw new IllegalArgumentException("Invalid parameter: api");
		}
		if(user == null || !user.isAuthenticated()) {
			throw new PermissionException(getClass().getSimpleName(), user, "Requires authenticated user.", Constants.PERSON_MANAGE_ROLE);
		}

		this.api = api;
		this.user = user;
		this.keywords = keywords;
	}

	@Override
	public Query newWithParameters(Map parameters) throws PermissionException {
		return new SearchPeople(
				(CassandraAPI)parameters.get("api"),
				(User)parameters.get("user"),
				(String)parameters.get("keywords"));
	}

	public List execute() throws IOException {
		List results = new LinkedList<>();
		Session s = api.getCassandraSession();

		if(!user.hasRole(Constants.PERSON_MANAGE_ROLE)) {
			log(s, "WARN", user, "Permission denied invoking: " + SearchPeople.class.getSimpleName() + " " + getJsonParameters());
			throw new PermissionException(this.getClass().getSimpleName(), user, "Only person managers may search people.", Constants.PERSON_MANAGE_ROLE);
		}

		PreparedStatement q = s.prepare(
				"select uuid, first_name, last_name, email, last_auth, created, updated, expiry "+
				"from person ");
		String[] parts = TagsToArray.tagsToArray(keywords);
		for(Row r : s.execute(q.bind())) {
			boolean match = false;
			for(String part : parts) {
				if(r.getString(1).toLowerCase().contains(part.toLowerCase())) {
					match = true;
					break;
				}
				if(r.getString(2).toLowerCase().contains(part.toLowerCase())) {
					match = true;
					break;
				}
				if(r.getString(3).toLowerCase().contains(part.toLowerCase())) {
					match = true;
					break;
				}
			}
			if(match) {
				Person p = new Person();
				p.setUuid(r.getUUID(0));
				p.setFirstName(r.getString(1));
				p.setLastName(r.getString(2));
				p.setEmail(r.getString(3));
				if(r.getLong(4) != 0) {
					p.setLastAuth(new Date(r.getLong(4)));
				}
				p.setCreated(new Date(r.getLong(5)));
				p.setUpdated(new Date(r.getLong(6)));
				if(r.getLong(7) > 0) {
					p.setExpiry(new Date(r.getLong(7)));
				}
				results.add(p);
			}
		}

		Collections.sort(results, new PersonComparator());

		return results;
	}

	@Override
	public String getJsonParameters() {
		return "{" +
				"\"person\":" + user.getPersonUuid() + "," +
				"\"keywords\":\"" + Json.escape(keywords) + "\"" +
				"}";
	}

	class PersonComparator implements Comparator {

		@Override
		public int compare(Person o1, Person o2) {
			int c = o1.getDisplayName().compareTo(o2.getDisplayName());

			if(c != 0) {
				return c;
			}

			return o1.getEmail().compareTo(o2.getEmail());
		}
	}

}