package edu.northwestern.bioinformatics.studycalendar.restlets;

import gov.nih.nci.cabig.ctms.domain.DomainObject;
import org.restlet.data.MediaType;
import org.restlet.data.Method;
import org.restlet.data.Status;
import org.restlet.representation.Representation;
import org.restlet.representation.Variant;
import org.restlet.resource.ResourceException;

import java.io.IOException;

/**
 * Base class for resources which are backed by a single domain object and
 * which support GET and PUT.
 *
 * @author Rhett Sutphin
 */
public abstract class AbstractStorableDomainObjectResource<D extends DomainObject> extends AbstractDomainObjectResource<D> {
    @Override
    public void doInit() {
        super.doInit();
        getAllowedMethods().add(Method.PUT);
    }

    @Override
    @SuppressWarnings({"unchecked"})
    public Representation put(Representation entity, Variant variant) throws ResourceException {
        if (entity.getMediaType().includes(MediaType.TEXT_XML)) {
            validateEntity(entity);
            D read;
            try {
                read = getXmlSerializer().readDocument(entity.getStream());
                store(read);
            } catch (IOException e) {
                log.warn("PUT failed with IOException", e);
                throw new ResourceException(e);
            }
            getResponse().setEntity(createXmlRepresentation(read));
            if (getRequestedObject() == null) {
                getResponse().setStatus(Status.SUCCESS_CREATED);
            } else {
                getResponse().setStatus(Status.SUCCESS_OK);
            }

            return null;
        } else {
            throw new ResourceException(Status.CLIENT_ERROR_BAD_REQUEST);
        }
    }

    public abstract void store(D instance) throws ResourceException;
    protected void validateEntity(Representation entity) throws ResourceException {}
}

@@ -1673,16 +1673,33 @@
 ception(
+%0A                
 Status.C
@@ -1721,16 +1721,70 @@
 _REQUEST
+, %22Unsupported content type: %22 + entity.getMediaType()
 );%0A     

Improve error message. Noted with #1439, #1443.