base.jee.api.cassandra.GetTemplates 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 java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import base.KeyValue;
import base.Query;
import base.jee.Constants;
import base.security.PermissionException;
import base.security.User;
import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;
import com.datastax.driver.core.Session;

public class GetTemplates extends Query {

	private CassandraAPI c;
	private User user;

	public GetTemplates(CassandraAPI c, User user) throws PermissionException {
		this.c = c;
		this.user = user;

		if(user != null && !user.isAuthenticated()) {
			throw new PermissionException(this.getClass().getSimpleName(), user, "Users must be authenticated to view template options.", Constants.TEMPLATE_MANAGE_ROLE);
		}

	}

	public List execute() throws IOException {
		List results = new LinkedList<>();

		Session s = c.getCassandraSession();
		ResultSet r = s.execute("select name from template");
		for(Row rows : r) {
			results.add(new KeyValue(rows.getString("name"), ""));
		}

		Collections.sort(results);
		return results;
	}

	@Override
	public String getJsonParameters() {
		return "{" +
				"}";
	}

	@Override
	public Query newWithParameters(Map parameters) throws IOException, PermissionException {
		User user = (User)parameters.get("user");

		if(!user.hasRole(Constants.TEMPLATE_MANAGE_ROLE)) {
			throw new PermissionException(getClass().getSimpleName(), user, "You do not have permission to manage templates.", Constants.TEMPLATE_MANAGE_ROLE);
		}

		return new GetTemplates((CassandraAPI)parameters.get("api"), (User)parameters.get("user"));
	}

}