/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package gov.hhs.fha.nhinc.gateway.policyenginetransformation;

import gov.hhs.fha.nhinc.common.nhinccommonadapter.CheckPolicyRequestType;
import gov.hhs.fha.nhinc.common.eventcommon.SubjectReidentificationEventType;
import gov.hhs.fha.nhinc.common.eventcommon.SubjectReidentificationMessageType;
import gov.hhs.fha.nhinc.gateway.policyenginetransformation.Helpers.ActionHelper;
import gov.hhs.fha.nhinc.gateway.policyenginetransformation.Helpers.AttributeHelper;
import gov.hhs.fha.nhinc.gateway.policyenginetransformation.Helpers.Constants;
import gov.hhs.fha.nhinc.gateway.policyenginetransformation.Helpers.InboundOutboundChecker;
import gov.hhs.fha.nhinc.gateway.policyenginetransformation.Helpers.SubjectHelper;
import javax.xml.bind.JAXBElement;
import oasis.names.tc.xacml._2_0.context.schema.os.RequestType;
import oasis.names.tc.xacml._2_0.context.schema.os.ResourceType;
import oasis.names.tc.xacml._2_0.context.schema.os.SubjectType;
import org.hl7.v3.II;
import org.hl7.v3.PRPAIN201309UV;
import org.hl7.v3.PRPAMT201307UVParameterList;
import org.hl7.v3.PRPAMT201307UVPatientIdentifier;
import org.hl7.v3.PRPAMT201307UVQueryByParameter;

/**
 *
 * @author svalluripalli
 */
public class SubjectReidentificationTransformHelper {

    private static final String ActionInValue = "SubjectDiscoveryReidentificationIn";
    private static final String ActionOutValue = "SubjectDiscoveryReidentificationOut";
    private static final String PatientAssigningAuthorityAttributeId = Constants.AssigningAuthorityAttributeId;
    private static final String PatientIdAttributeId = Constants.ResourceIdAttributeId;

    public static CheckPolicyRequestType transformSubjectReidentificationToCheckPolicy(SubjectReidentificationEventType event) {
        CheckPolicyRequestType genericPolicyRequest = new CheckPolicyRequestType();
        SubjectReidentificationMessageType message = event.getMessage();
        PRPAIN201309UV subjectReidentification = message.getPRPAIN201309UV();
        RequestType request = new RequestType();
        if (event.getDirection() !=  null) {
            System.out.println("event.getDirection() is " + event.getDirection());
        }
        if (InboundOutboundChecker.IsInbound(event.getDirection())) {
            request.setAction(ActionHelper.actionFactory(ActionInValue));
        }
        if (InboundOutboundChecker.IsOutbound(event.getDirection())) {
            request.setAction(ActionHelper.actionFactory(ActionOutValue));
        }
        System.out.println("Action is >>>" + request.getAction().getAttribute().get(0).getAttributeId() + "<<<");
        SubjectType subject = SubjectHelper.subjectFactoryReident(event.getSendingHomeCommunity(), event.getMessage().getAssertion());
        request.getSubject().add(subject);
        II ii = extractPatientIdentifier(subjectReidentification);
        if (ii != null) {
            ResourceType resource = new ResourceType();
            resource.getAttribute().add(AttributeHelper.attributeFactory(PatientAssigningAuthorityAttributeId, Constants.DataTypeString, ii.getRoot()));
            resource.getAttribute().add(AttributeHelper.attributeFactory(PatientIdAttributeId, Constants.DataTypeString, ii.getExtension()));
            request.getResource().add(resource);
        }
        genericPolicyRequest.setRequest(request);
        return genericPolicyRequest;
    }

    /**
     * Extract a Patient from 201309
     * @param message
     * @return II (PatientId root and extension)
     */
    private static II extractPatientIdentifier(PRPAIN201309UV message) {
        II ii = null;
        PRPAMT201307UVQueryByParameter queryByParameter = null;
        PRPAMT201307UVParameterList parameterList = null;
        PRPAMT201307UVPatientIdentifier patientIdentifier = null;
        if ((message != null) && (message.getControlActProcess() != null)) {
            JAXBElement<PRPAMT201307UVQueryByParameter> queryByParamterElement = message.getControlActProcess().getQueryByParameter();
            if (null != queryByParamterElement) {
                queryByParameter = queryByParamterElement.getValue();
                if (null != queryByParameter) {
                    parameterList = queryByParameter.getParameterList();
                    if (null != parameterList) {
                        patientIdentifier = (parameterList.getPatientIdentifier() != null) ? parameterList.getPatientIdentifier().get(0) : null;
                        if (null != patientIdentifier) {
                            if (null != patientIdentifier.getValue() && patientIdentifier.getValue().size() >= 1) {
                                ii = patientIdentifier.getValue().get(0);
                                System.out.println(">>>>> patientIdentifier " + ii.getExtension() + " " + ii.getRoot());
                            }
                        }
                    }
                }
            }
        }
        return ii;
    }
}