D-Net

package eu.dnetlib.enabling.is.hib.utils;

import java.util.Map.Entry;

import org.apache.commons.lang.StringUtils;

import org.springframework.beans.factory.annotation.Required;

import com.google.gson.Gson;

import eu.dnetlib.enabling.is.dao.DnetSimpleDao;

import eu.dnetlib.enabling.is.dao.DnetSimpleDaoFactory;

import eu.dnetlib.enabling.is.hib.objects.HibBlackboardMessage;

import eu.dnetlib.enabling.is.hib.objects.HibDatastructure;

import eu.dnetlib.enabling.is.hib.objects.HibDnetResourceProperty;

import eu.dnetlib.enabling.is.hib.objects.HibDnetResourcePropertyId;

import eu.dnetlib.enabling.is.hib.objects.HibResourceKind;

import eu.dnetlib.enabling.is.hib.objects.HibResourceType;

import eu.dnetlib.enabling.is.hib.objects.HibService;

import eu.dnetlib.enabling.is.hib.objects.HibServiceProtocol;

import eu.dnetlib.enabling.is.hib.objects.HibServiceProtocolId;

import eu.dnetlib.enabling.is.hib.objects.HibSubscription;

import eu.dnetlib.rmi.objects.is.BlackboardMessage;

import eu.dnetlib.rmi.objects.is.DnetDataStructure;

import eu.dnetlib.rmi.objects.is.DnetResourceFormat;

import eu.dnetlib.rmi.objects.is.DnetResourceKind;

import eu.dnetlib.rmi.objects.is.DnetResourceType;

import eu.dnetlib.rmi.objects.is.DnetService;

import eu.dnetlib.rmi.objects.is.Subcription;

import eu.dnetlib.rmi.soap.exceptions.InformationServiceException;

public class HibObjectHelper {

	private DnetSimpleDaoFactory daoFactory;

	public HibService createHibService(final DnetService service) throws InformationServiceException {

		HibResourceType type = daoFactory.getDao(HibResourceType.class).find(service.getName());

		if (type == null) {

			type = createHibResourceType(service.getName(), DnetResourceKind.SERVICE, DnetResourceFormat.SERVICE);

		}

		final HibService hib = new HibService(service.getId(), type, service.isValid());

		for (Entry<String, String> e : service.getProtocols().entrySet()) {

			hib.getProtocols().add(new HibServiceProtocol(new HibServiceProtocolId(hib, e.getKey()), e.getValue()));

		}

		for (Entry<String, String> e : service.getProperties().entrySet()) {

			hib.getProperties().add(new HibDnetResourceProperty(new HibDnetResourcePropertyId(hib, e.getKey()), e.getValue()));

		}

		for (BlackboardMessage b : service.getBlackboard()) {

			hib.getBlackboard().add(new HibBlackboardMessage(hib, b));

		}

		daoFactory.getDao(HibService.class).save(hib);

		return hib;

	}

	public HibDatastructure createHibDnetDatastructure(final DnetDataStructure ds) throws InformationServiceException {

		final HibResourceType type = daoFactory.getDao(HibResourceType.class).find(ds.getType());

		if (type == null) { throw new InformationServiceException("Resource type not registered: " + ds.getType()); }

		final HibDatastructure hib = new HibDatastructure();

		hib.setId(ds.getId());

		hib.setCode(ds.getCode());

		hib.setName(ds.getName());

		hib.setDescription(ds.getDescription());

		hib.setType(type);

		hib.setContent(ds.getContent());

		hib.setDate(ds.getDate());

		hib.setValid(ds.isValid());

		if (StringUtils.isNotBlank(ds.getServiceId())) {

			final HibService service = daoFactory.getDao(HibService.class).find(ds.getServiceId());

			if (service != null) {

				hib.setService(service);

			} else {

				throw new InformationServiceException("Missing service: " + ds.getServiceId());

			}

		}

		daoFactory.getDao(HibDatastructure.class).save(hib);

		return hib;

	}

	public HibResourceType createHibDnetResourceType(final DnetResourceType type) throws InformationServiceException {

		return createHibResourceType(type.getType(), type.getKind(), type.getFormat());

	}

	public void createHibBlackbooardMessage(final String serviceId, final BlackboardMessage message) throws InformationServiceException {

		final HibService service = daoFactory.getDao(HibService.class).find(serviceId);

		final HibBlackboardMessage hib = new HibBlackboardMessage(service, message);

		daoFactory.getDao(HibBlackboardMessage.class).save(hib);

	}

	public void createHibSubscription(final Subcription subcription) throws InformationServiceException {

		final HibService service = daoFactory.getDao(HibService.class).find(subcription.getServiceId());

		final HibSubscription hib = new HibSubscription();

		hib.setId(subcription.getId());

		hib.setOperation(subcription.getOperation());

		hib.setTable(subcription.getTable());

		hib.setCondition(new Gson().toJson(subcription.getCondition()));

		hib.setSubscriber(service);

		daoFactory.getDao(HibSubscription.class).save(hib);

	}

	public void updateHibBlackbooardMessage(final BlackboardMessage message) throws InformationServiceException {

		final DnetSimpleDao<HibBlackboardMessage> dao = daoFactory.getDao(HibBlackboardMessage.class);

		final HibBlackboardMessage hib = dao.find(message.getId());

		hib.setAction(message.getAction());

		hib.setDate(message.getDate());

		hib.setMessage(message.getJsonMessage());

		hib.setStatus(message.getStatus().toString());

		dao.save(hib);

	}

	private HibResourceType createHibResourceType(final String type,

			final DnetResourceKind kind,

			final DnetResourceFormat format) throws InformationServiceException {

		final DnetSimpleDao<HibResourceKind> daoKind = daoFactory.getDao(HibResourceKind.class);

		final HibResourceKind hibKind = daoKind.find(kind);

		final HibResourceType hibType = new HibResourceType(type, hibKind, format);

		daoFactory.getDao(HibResourceType.class).save(hibType);

		return hibType;

	}

	public DnetSimpleDaoFactory getDaoFactory() {

		return daoFactory;

	}

	@Required

	public void setDaoFactory(final DnetSimpleDaoFactory daoFactory) {

		this.daoFactory = daoFactory;

	}

}