AddRelationshipHandler.java

/**
 * Copyright 2008 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package edu.unc.lib.dl.security.wshandlers;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;
import java.util.Map;

import javax.xml.ws.handler.soap.SOAPMessageContext;

import org.apache.cxf.binding.soap.SoapFault;
import org.fcrepo.common.Constants;
import org.fcrepo.server.security.RequestCtx;
import org.fcrepo.server.security.xacml.pep.ContextHandler;
import org.fcrepo.server.security.xacml.pep.PEPException;
import org.fcrepo.server.security.xacml.pep.ws.operations.AbstractOperationHandler;
import org.fcrepo.server.security.xacml.pep.ws.operations.OperationHandlerException;
import org.fcrepo.server.security.xacml.util.LogUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import org.jboss.security.xacml.sunxacml.attr.AnyURIAttribute;
import org.jboss.security.xacml.sunxacml.attr.AttributeValue;


/**
 * @author nishen@melcoe.mq.edu.au
 */
public class AddRelationshipHandler
        extends AbstractOperationHandler {

    private static final Logger logger =
            LoggerFactory.getLogger(AddRelationshipHandler.class);

	public static final String RESOURCE_RELATIONSHIP_PREDICATE = "http://www.w3.org/1999/02/22-rdf-syntax-ns#predicate";

    public AddRelationshipHandler(ContextHandler contextHandler)
            throws PEPException {
        super(contextHandler);
    }

    @Override
    public RequestCtx handleResponse(SOAPMessageContext context)
            throws OperationHandlerException {
        return null;
    }

    @Override
    public RequestCtx handleRequest(SOAPMessageContext context)
            throws OperationHandlerException {
        logger.debug("AddRelationshipHandler/handleRequest!");

        RequestCtx req = null;

        try {
            Map<URI, AttributeValue> resAttr = getResources(context);

            Map<URI, AttributeValue> actions = new HashMap<URI, AttributeValue>();
            actions.put(Constants.ACTION.ID.getURI(),
                        Constants.ACTION.ADD_RELATIONSHIP
                                .getStringAttribute());
            actions.put(Constants.ACTION.API.getURI(),
                        Constants.ACTION.APIM.getStringAttribute());
            
            addPredicate(context, resAttr);

            req =
                    getContextHandler().buildRequest(getSubjects(context),
                                                     actions,
                                                     resAttr,
                                                     getEnvironment(context));

            String pid = resAttr.get(Constants.OBJECT.PID.getURI()).toString();
            LogUtil.statLog(getUser(context),
                            Constants.ACTION.ADD_RELATIONSHIP.uri,
                            pid,
                            null);
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
            throw new OperationHandlerException(e.getMessage(), e);
        }

        return req;
    }
    
    protected void addPredicate(SOAPMessageContext context, Map<URI, AttributeValue> resAttr) throws OperationHandlerException, URISyntaxException {
        Object oMap = null;
        String relationship = null;
        try {
            oMap = getSOAPRequestObjects(context);
            logger.debug("Retrieved SOAP Request Objects");
        } catch (SoapFault af) {
            logger.error("Error obtaining SOAP Request Objects", af);
            throw new OperationHandlerException("Error obtaining SOAP Request Objects",
                                                af);
        }
        try {
        	relationship = (String) callGetter("getRelationship",oMap);
            if (relationship != null && !"".equals(relationship)) {
            	resAttr.put(new URI(RESOURCE_RELATIONSHIP_PREDICATE), new AnyURIAttribute(new URI(relationship)));
            } else {
            	logger.error("Could not determine predicate for relationship being added.");
            }
        } catch (Exception e) {
            logger.error("Error obtaining parameters", e);
            throw new OperationHandlerException("Error obtaining parameters.",
                                                e);
        }
        logger.debug("Extracted SOAP Request Objects");
    }
}