pl.edu.icm.unity.engine.scripts.MainGroovyExecutor Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2017 Bixbit - Krzysztof Benedyczak All rights reserved.
 * See LICENCE.txt file for licensing information.
 */
package pl.edu.icm.unity.engine.scripts;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;

import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import eu.unicore.util.configuration.ConfigurationException;
import groovy.lang.Binding;
import pl.edu.icm.unity.MessageSource;
import pl.edu.icm.unity.base.event.PersistableEvent;
import pl.edu.icm.unity.base.utils.Log;
import pl.edu.icm.unity.engine.api.AttributeClassManagement;
import pl.edu.icm.unity.engine.api.AttributeTypeManagement;
import pl.edu.icm.unity.engine.api.AttributesManagement;
import pl.edu.icm.unity.engine.api.AuthenticatorManagement;
import pl.edu.icm.unity.engine.api.BulkProcessingManagement;
import pl.edu.icm.unity.engine.api.CredentialManagement;
import pl.edu.icm.unity.engine.api.CredentialRequirementManagement;
import pl.edu.icm.unity.engine.api.EndpointManagement;
import pl.edu.icm.unity.engine.api.EnquiryManagement;
import pl.edu.icm.unity.engine.api.EntityCredentialManagement;
import pl.edu.icm.unity.engine.api.EntityManagement;
import pl.edu.icm.unity.engine.api.GroupsManagement;
import pl.edu.icm.unity.engine.api.IdentityTypesManagement;
import pl.edu.icm.unity.engine.api.InvitationManagement;
import pl.edu.icm.unity.engine.api.MessageTemplateManagement;
import pl.edu.icm.unity.engine.api.NotificationsManagement;
import pl.edu.icm.unity.engine.api.PreferencesManagement;
import pl.edu.icm.unity.engine.api.RealmsManagement;
import pl.edu.icm.unity.engine.api.RegistrationsManagement;
import pl.edu.icm.unity.engine.api.TranslationProfileManagement;
import pl.edu.icm.unity.engine.api.UserImportManagement;
import pl.edu.icm.unity.engine.api.attributes.AttributeTypeSupport;
import pl.edu.icm.unity.engine.api.config.UnityServerConfiguration;
import pl.edu.icm.unity.engine.api.event.EventCategory;
import pl.edu.icm.unity.engine.api.identity.IdentityTypeSupport;
import pl.edu.icm.unity.engine.api.initializers.ScriptConfiguration;
import pl.edu.icm.unity.engine.api.initializers.ScriptType;
import pl.edu.icm.unity.engine.api.session.SessionManagement;
import pl.edu.icm.unity.engine.api.token.TokensManagement;
import pl.edu.icm.unity.engine.api.translation.form.RegistrationFormTranslationActionGenerator;
import pl.edu.icm.unity.engine.api.utils.GroupDelegationConfigGenerator;

/**
 * Executes GROOVY scripts given by user in
 * {@link UnityServerConfiguration#SCRIPTS} configuration.
 *
 * @author Roman Krysinski ([email protected])
 */
@Component
public class MainGroovyExecutor
{
	private static final Logger LOG = Log.getLogger(Log.U_SERVER_SCRIPT, MainGroovyExecutor.class);
	
	@Autowired
	private MessageSource unityMessageSource;
	@Autowired
	private UnityServerConfiguration config;
	@Autowired
	private AttributeTypeSupport attributeTypeSupport;
	@Autowired
	private IdentityTypeSupport identityTypeSupport;
	
	@Autowired
	@Qualifier("insecure")
	private BulkProcessingManagement bulkProcessingManagement;
	@Autowired
	@Qualifier("insecure")
	private PreferencesManagement preferencesManagement;
	@Autowired
	@Qualifier("insecure")
	private UserImportManagement userImportManagement;
	@Autowired
	@Qualifier("insecure")
	private IdentityTypesManagement identityTypesManagement;
	@Autowired
	@Qualifier("insecure")
	private AttributeClassManagement attributeClassManagement;
	@Autowired
	@Qualifier("insecure")
	private AttributesManagement attributesManagement;
	@Autowired
	@Qualifier("insecure")
	private AttributeTypeManagement attributeTypeManagement;
	@Autowired
	@Qualifier("insecure")
	private AuthenticatorManagement authenticatorManagement;
	@Autowired
	@Qualifier("insecure")
	private CredentialManagement credentialManagement;
	@Autowired
	@Qualifier("insecure")
	private CredentialRequirementManagement credentialRequirementManagement;
	@Autowired
	@Qualifier("insecure")
	private EndpointManagement endpointManagement;
	@Autowired
	@Qualifier("insecure")
	private EnquiryManagement enquiryManagement;
	@Autowired
	@Qualifier("insecure")
	private EntityCredentialManagement entityCredentialManagement;
	@Autowired
	@Qualifier("insecure")
	private EntityManagement entityManagement;
	@Autowired
	@Qualifier("insecure")
	private GroupsManagement groupsManagement;
	@Autowired
	@Qualifier("insecure")
	private InvitationManagement invitationManagement;
	@Autowired
	@Qualifier("insecure")
	private MessageTemplateManagement messageTemplateManagement;
	@Autowired
	@Qualifier("insecure")
	private NotificationsManagement notificationsManagement;
	@Autowired
	@Qualifier("insecure")
	private RealmsManagement realmsManagement;
	@Autowired
	@Qualifier("insecure")
	private RegistrationsManagement registrationsManagement;
	@Autowired
	@Qualifier("insecure")
	private TranslationProfileManagement translationProfileManagement;
	@Autowired
	@Qualifier("insecure")
	private GroupDelegationConfigGenerator groupDelegationConfigGenerator;
	@Autowired
	private SessionManagement sessionManagement;
	@Autowired
	private RegistrationFormTranslationActionGenerator regTranslationActionGenerator;
	@Autowired
	private TokensManagement tokensManagement;
	@Autowired
	private ApplicationContext applicationContext;
	
	public void run(ScriptConfiguration conf, PersistableEvent event)
	{
		if (conf == null || conf.getType() != ScriptType.groovy)
			throw new IllegalArgumentException(
					"conf must not be null and must be of " + 
							ScriptType.groovy + " type");
		Reader scriptReader = getFileReader(conf.getFileLocation());
		try
		{
			GroovyRunner.run(conf.getTrigger(), conf.getFileLocation(), scriptReader, 
				getBinding(event));
		} finally
		{
			try
			{
				scriptReader.close();
			} catch (IOException e)
			{
				LOG.error("Problem closing the stream used to read Groovy script", e);
			}
		}
	}

	private Reader getFileReader(String location)
	{
		try
		{
			InputStream is = location.startsWith("classpath:") ?
					applicationContext.getResource(location).getInputStream() :
					new FileInputStream(location);
			return new InputStreamReader(is);
		} catch (IOException e)
		{
			throw new ConfigurationException("Error loading script " + location, e);
		}
	}

	Binding getBinding(PersistableEvent event)
	{
		Binding binding = new Binding();
		binding.setVariable("event", event.getTrigger());
		binding.setVariable("context", event.getContents());
		binding.setVariable("config", config);
		binding.setVariable("attributeClassManagement", attributeClassManagement);
		binding.setVariable("attributesManagement", attributesManagement);
		binding.setVariable("attributeTypeManagement", attributeTypeManagement);
		binding.setVariable("authenticatorManagement", authenticatorManagement);
		binding.setVariable("bulkProcessingManagement", bulkProcessingManagement);
		binding.setVariable("credentialManagement", credentialManagement);
		binding.setVariable("credentialRequirementManagement", credentialRequirementManagement);
		binding.setVariable("endpointManagement", endpointManagement);
		binding.setVariable("enquiryManagement", enquiryManagement);
		binding.setVariable("entityCredentialManagement", entityCredentialManagement);
		binding.setVariable("entityManagement", entityManagement);
		binding.setVariable("groupsManagement", groupsManagement);
		binding.setVariable("identityTypesManagement", identityTypesManagement);
		binding.setVariable("invitationManagement", invitationManagement);
		binding.setVariable("messageTemplateManagement", messageTemplateManagement);
		binding.setVariable("notificationsManagement", notificationsManagement);
		binding.setVariable("preferencesManagement", preferencesManagement);
		binding.setVariable("realmsManagement", realmsManagement);
		binding.setVariable("registrationsManagement", registrationsManagement);
		binding.setVariable("translationProfileManagement", translationProfileManagement);
		binding.setVariable("userImportManagement", userImportManagement);
		binding.setVariable("msgSrc", unityMessageSource);
		binding.setVariable("attributeTypeSupport", attributeTypeSupport);
		binding.setVariable("identityTypeSupport", identityTypeSupport);
		binding.setVariable("groupDelegationConfigGenerator", groupDelegationConfigGenerator);
		binding.setVariable("sessionManagement", sessionManagement);
		binding.setVariable("regTranslationActionGenerator", regTranslationActionGenerator);
		binding.setVariable("tokensManagement", tokensManagement);
		binding.setVariable("applicationContext", applicationContext);
		boolean coldStart = false;
		if (event.getTrigger().equals(EventCategory.POST_INIT.toString()) || 
				event.getTrigger().equals(EventCategory.PRE_INIT.toString()))
			coldStart = Boolean.valueOf(event.getContents());
		binding.setVariable("isColdStart", coldStart);
		binding.setVariable("log", LOG);
		return binding;
	}
}