base.jee.api.cassandra.InvalidateToken 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 com.datastax.driver.core.Row;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.PreparedStatement;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import base.KeyValue;
import base.Query;
import base.json.Json;
import base.security.User;

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

/**
 */
public class InvalidateToken extends Query {

	private CassandraAPI api;
	private String token;
	private String ip;

	public InvalidateToken() {
	}

	public InvalidateToken(CassandraAPI api, String token, String ip) {
		this.api = api;
		this.token = token;
		this.ip = ip;

		if(api == null) {
			throw new IllegalArgumentException("Invalid parameter: api");
		}
		if(token == null) {
			throw new IllegalArgumentException("Invalid parameter: token");
		}
	}

	@Override
	public Query newWithParameters(Map parameters) {
		return new InvalidateToken(
				(CassandraAPI)parameters.get("api"),
				(String)parameters.get("token"),
				((User)parameters.get("user")).getIp());
	}

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

		PreparedStatement q = s.prepare("select person_uuid from session_token where uid = ?");
		for(Row r : s.execute(q.bind(token))) {
			log(s, "INFO", User.userWithUuidAndIp(r.getUUID(0), ip), "Signout.");
			q = s.prepare("delete from session_token where uid = ?");
			s.execute(q.bind(token));
			results.add(new KeyValue("success", "ok"));
			return results;
		}

		log(s, "FINER", User.userWithIp(ip), "Request to invalidate " + token + " did not correspond to an active session.");
		results.add(new KeyValue("success", "ok"));
		return results;
	}

	@Override
	public String getJsonParameters() {
		return "{" +
				"\"token\":\"" + token + "\"," +
				"\"ip\":\"" + Json.escape(ip) + "\"" +
				"}";
	}


}