prerna.reactor.export.AsTaskReactor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.reactor.export;

import java.util.HashSet;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Vector;

import prerna.auth.User;
import prerna.auth.utils.AbstractSecurityUtils;
import prerna.auth.utils.SecurityQueryUtils;
import prerna.reactor.AbstractReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.sablecc2.om.task.ConstantDataTask;

public class AsTaskReactor extends AbstractReactor {

	/**
	 * This class is responsible for collecting the first element from a task and returning it as a noun
	 */
	
	public AsTaskReactor() {
		this.keysToGet = new String[]{ReactorKeysEnum.VALUE.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		User user = this.insight.getUser();
		// throw error is user doesn't have rights to export data
		if(AbstractSecurityUtils.adminSetExporter() && !SecurityQueryUtils.userIsExporter(user)) {
			AbstractReactor.throwUserNotExporterError();
		}
		NounMetadata inputValue = getInputValue();
		List dataValues = new Vector();
		dataValues.add(new Object[]{inputValue.getValue()});
		
		ConstantDataTask task = new ConstantDataTask();
		Map returnData = new Hashtable();
		returnData.put("values", dataValues);
		returnData.put("headers", new String[]{"constant"});
		task.setOutputData(returnData);
		
		return new NounMetadata(task, PixelDataType.FORMATTED_DATA_SET, PixelOperationType.TASK_DATA);
	}
	
	private NounMetadata getInputValue() {
		GenRowStruct grs = this.store.getNoun(this.keysToGet[0]);
		if(grs != null && !grs.isEmpty()) {
			return grs.getNoun(0);
		}
		
		Set inKeys = new HashSet(this.store.getNounKeys());
		inKeys.remove("all");
		inKeys.remove(this.keysToGet[0]);
		for(String k : inKeys) {
			grs = this.store.getNoun(k);
			if(grs != null && !grs.isEmpty()) {
				return grs.getNoun(0);
			}
		}
		
		if(this.curRow != null && !this.curRow.isEmpty()) {
			return this.curRow.getNoun(0);
		}
		
		return null;
	}

}