Package org.dspace.app.dav

Source Code of org.dspace.app.dav.DAVCollection$CountedInputStream

/**
* The contents of this file are subject to the license and copyright
* detailed in the LICENSE and NOTICE files at the root of the source
* tree and available online at
*
* http://www.dspace.org/license/
*/
package org.dspace.app.dav;

import java.io.File;
import java.io.FileOutputStream;
import java.io.FilterInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Vector;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.log4j.Logger;
import org.dspace.authorize.AuthorizeException;
import org.dspace.content.Bitstream;
import org.dspace.content.BitstreamFormat;
import org.dspace.content.Collection;
import org.dspace.content.Community;
import org.dspace.content.DSpaceObject;
import org.dspace.content.Item;
import org.dspace.content.ItemIterator;
import org.dspace.content.crosswalk.CrosswalkException;
import org.dspace.content.packager.PackageException;
import org.dspace.content.packager.PackageIngester;
import org.dspace.content.packager.PackageParameters;
import org.dspace.core.ConfigurationManager;
import org.dspace.core.Context;
import org.dspace.core.PluginManager;
import org.dspace.core.Utils;
import org.dspace.handle.HandleManager;
import org.dspace.workflow.WorkflowItem;
import org.dspace.workflow.WorkflowManager;
import org.jdom.Element;


/**
* This defines the behavior of DSpace "resources" in the WebDAV interface; it
* maps DAV operations onto DSpace object.s
*/
class DAVCollection extends DAVDSpaceObject
{
   
    /** log4j category. */
    private static Logger log = Logger.getLogger(DAVCollection.class);

    /** The collection. */
    private Collection collection = null;

    /** The temporary upload directory. */
    private static String tempDirectory = (ConfigurationManager.getProperty("upload.temp.dir") != null)
                ? ConfigurationManager.getProperty("upload.temp.dir") : System.getProperty("java.io.tmpdir");

    /** The Constant short_descriptionProperty. */
    private static final Element short_descriptionProperty = new Element(
            "short_description", DAV.NS_DSPACE);

    /** The Constant introductory_textProperty. */
    private static final Element introductory_textProperty = new Element(
            "introductory_text", DAV.NS_DSPACE);

    /** The Constant side_bar_textProperty. */
    private static final Element side_bar_textProperty = new Element(
            "side_bar_text", DAV.NS_DSPACE);

    /** The Constant copyright_textProperty. */
    private static final Element copyright_textProperty = new Element(
            "copyright_text", DAV.NS_DSPACE);

    /** The Constant provenance_descriptionProperty. */
    private static final Element provenance_descriptionProperty = new Element(
            "provenance_description", DAV.NS_DSPACE);

    /** The Constant default_licenseProperty. */
    private static final Element default_licenseProperty = new Element(
            "default_license", DAV.NS_DSPACE);

    /** The Constant logoProperty. */
    private static final Element logoProperty = new Element("logo",
            DAV.NS_DSPACE);

    /** The all props. */
    private static List<Element> allProps = new ArrayList<Element>(commonProps);
    static
    {
        allProps.add(logoProperty);
        allProps.add(short_descriptionProperty);
        allProps.add(introductory_textProperty);
        allProps.add(side_bar_textProperty);
        allProps.add(copyright_textProperty);
        allProps.add(default_licenseProperty);
        allProps.add(provenance_descriptionProperty);
        allProps.add(handleProperty);
    }

    /**
     * Instantiates a new DAV collection.
     *
     * @param context the context
     * @param request the request
     * @param response the response
     * @param pathElt the path elt
     * @param collection the collection
     */
    protected DAVCollection(Context context, HttpServletRequest request,
            HttpServletResponse response, String pathElt[],
            Collection collection)
    {
        super(context, request, response, pathElt, collection);
        this.collection = collection;
        this.type = TYPE_COLLECTION;
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#getAllProperties()
     */
    @Override
    protected List<Element> getAllProperties()
    {
        return allProps;
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#children()
     */
    @Override
    protected DAVResource[] children() throws SQLException
    {
        Vector result = new Vector();
        ItemIterator ii = this.collection.getItems();
        try
        {
            while (ii.hasNext())
            {
                Item item = ii.next();
                result.add(new DAVItem(this.context, this.request, this.response,
                        makeChildPath(item), item));
            }
        }
        finally
        {
            if (ii != null)
            {
                ii.close();
            }
        }
       
        return (DAVResource[]) result.toArray(new DAVResource[result.size()]);
    }

    /**
     * Gets the collection.
     *
     * @return the DSpace Collection object represented by this resource.
     */
    protected Collection getCollection()
    {
        return this.collection;
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVDSpaceObject#propfindInternal(org.jdom.Element)
     */
    @Override
    protected Element propfindInternal(Element property) throws SQLException,
            AuthorizeException, IOException, DAVStatusException
    {
        String value = null;

        // displayname - title or handle.
        if (elementsEqualIsh(property, displaynameProperty))
        {
            value = getObjectMetadata("name");
            if (value == null)
            {
                value = this.collection.getHandle();
            }
        }

        else if (elementsEqualIsh(property, handleProperty))
        {
            value = canonicalizeHandle(this.collection.getHandle());
        }
        else if (elementsEqualIsh(property, logoProperty))
        {
            Bitstream lbs = this.collection.getLogo();
            if (lbs != null)
            {
                Element le = DAVBitstream.makeXmlBitstream(lbs, this);
                if (le != null)
                {
                    Element p = new Element("logo", DAV.NS_DSPACE);
                    p.addContent(le);
                    return p;
                }
            }
        }

        else if (elementsEqualIsh(property, short_descriptionProperty))
        {
            value = getObjectMetadata("short_description");
        }
        else if (elementsEqualIsh(property, introductory_textProperty))
        {
            value = getObjectMetadata("introductory_text");
        }
        else if (elementsEqualIsh(property, side_bar_textProperty))
        {
            value = getObjectMetadata("side_bar_text");
        }
        else if (elementsEqualIsh(property, copyright_textProperty))
        {
            value = getObjectMetadata("copyright_text");
        }
        else if (elementsEqualIsh(property, default_licenseProperty))
        {
            value = this.collection.hasCustomLicense() ? this.collection.getLicense()
                    : null;
        }
        else if (elementsEqualIsh(property, provenance_descriptionProperty))
        {
            value = getObjectMetadata("provenance_description");
        }
        else
        {
            return super.propfindInternal(property);
        }

        // value was set up by "if" clause:
        if (value == null)
        {
            throw new DAVStatusException(HttpServletResponse.SC_NOT_FOUND,
                    "Not found.");
        }
        Element p = new Element(property.getName(), property.getNamespace());
        p.setText(filterForXML(value));
        return p;
    }

    // syntactic sugar around getting collection metadata values:
    /**
     * Gets the object metadata.
     *
     * @param mdname the mdname
     *
     * @return the object metadata
     */
    private String getObjectMetadata(String mdname)
    {
        try
        {
            return this.collection.getMetadata(mdname);
        }
        catch (IllegalArgumentException e)
        {
            return null;
        }
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#proppatchInternal(int, org.jdom.Element)
     */
    @Override
    protected int proppatchInternal(int action, Element prop)
            throws SQLException, AuthorizeException, IOException,
            DAVStatusException
    {
        String newValue = (action == DAV.PROPPATCH_REMOVE) ? null : prop
                .getText();

        // these are "metadata" values..
        if (elementsEqualIsh(prop, short_descriptionProperty)
                || elementsEqualIsh(prop, introductory_textProperty)
                || elementsEqualIsh(prop, side_bar_textProperty)
                || elementsEqualIsh(prop, copyright_textProperty)
                || elementsEqualIsh(prop, provenance_descriptionProperty))
        {
            this.collection.setMetadata(prop.getName(), newValue);
        }
        else if (elementsEqualIsh(prop, displaynameProperty))
        {
            this.collection.setMetadata("name", newValue);
        }
        else if (elementsEqualIsh(prop, default_licenseProperty))
        {
            this.collection.setLicense(newValue);
        }
        else if (elementsEqualIsh(prop, logoProperty))
        {
            if (action == DAV.PROPPATCH_REMOVE)
            {
                this.collection.setLogo(null);
            }
            else
            {
                Element bs = prop.getChild("bitstream", DAV.NS_DSPACE);
                if (bs != null)
                {
                    InputStream bis = DAVBitstream.getXmlBitstreamContent(
                            this.context, bs);
                    BitstreamFormat bsf = DAVBitstream.getXmlBitstreamFormat(
                            this.context, bs);
                    if (bis == null || bsf == null)
                    {
                        throw new DAVStatusException(DAV.SC_CONFLICT,
                                "Unacceptable value for logo property.");
                    }
                    Bitstream nbs = this.collection.setLogo(bis);
                    nbs.setFormat(bsf);
                    nbs.update();
                }
                else
                {
                    throw new DAVStatusException(DAV.SC_CONFLICT,
                            "No <bitstream> element value found for logo property.");
                }
            }
        }
        else
        {
            throw new DAVStatusException(DAV.SC_CONFLICT, "The "
                    + prop.getName() + " property cannot be changed.");
        }

        this.collection.update();
        return HttpServletResponse.SC_OK;
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#get()
     */
    @Override
    protected void get() throws SQLException, AuthorizeException,
            IOException, DAVStatusException
    {
        throw new DAVStatusException(HttpServletResponse.SC_NOT_IMPLEMENTED,
                "GET not implemented for Collection.");
    }

    /**
     * Wrapped input stream that hits end-of-file after reading a counted number
     * of bytes, even if its client stream appears to have more data. This fixes
     * a problem in the Servlet container's input stream which will try to read
     * past the end of the request body.
     */
    private static class CountedInputStream extends FilterInputStream
    {
       
        /** The count. */
        private long count = 0;

        /** The length. */
        private long length = -1;

        /**
         * Instantiates a new counted input stream.
         *
         * @param is the is
         * @param length the length
         */
        protected CountedInputStream(InputStream is, long length)
        {
            super(is);
            this.length = length;
        }

        /* (non-Javadoc)
         * @see java.io.FilterInputStream#read()
         */
        @Override
        public int read() throws IOException
        {
            if (++this.count > this.length)
            {
                return -1;
            }
            return super.read();
        }

        /* (non-Javadoc)
         * @see java.io.FilterInputStream#read(byte[])
         */
        @Override
        public int read(byte[] b) throws IOException
        {
            if (this.count >= this.length)
            {
                return -1;
            }
            int result = super.read(b);
            if (this.count > 0)
            {
                this.count += result;
            }
            return result;
        }

        /* (non-Javadoc)
         * @see java.io.FilterInputStream#read(byte[], int, int)
         */
        @Override
        public int read(byte[] b, int off, int len) throws IOException
        {
            if (this.count >= this.length)
            {
                return -1;
            }
            int result = super.read(b, off, len);
            if (this.count > 0)
            {
                this.count += result;
            }
            return result;
        }

        /* (non-Javadoc)
         * @see java.io.FilterInputStream#skip(long)
         */
        @Override
        public long skip(long n) throws IOException
        {
            long result = super.skip(n);
            this.count += result;
            return result;
        }
    }

    /**
     * PUT ingests a package as a new Item. Package type (must match pluggable
     * packager name) is in either (a) "package" query arg in URI (b)
     * content-type request header
     *
     * @throws SQLException the SQL exception
     * @throws AuthorizeException the authorize exception
     * @throws IOException Signals that an I/O exception has occurred.
     * @throws DAVStatusException the DAV status exception
     */
    @Override
    protected void put() throws SQLException, AuthorizeException,
            IOException, DAVStatusException
    {
        try
        {
            String packageType = this.request.getParameter("package");
            if (packageType == null)
            {
                packageType = this.request.getContentType();
            }
            if (packageType == null)
            {
                throw new DAVStatusException(
                        HttpServletResponse.SC_BAD_REQUEST,
                        "Cannot determine package type,  need content-type header or package param");
            }

            PackageIngester sip = (PackageIngester) PluginManager
                    .getNamedPlugin(PackageIngester.class, packageType);
            if (sip == null)
            {
                throw new DAVStatusException(
                        HttpServletResponse.SC_BAD_REQUEST,
                        "Cannot find importer for package type: " + packageType);
            }

            /*
             * Ugh. Servlet container doesn't get end-of-file right on input
             * stream so we have to count it, when possible.
             */
            int contentLength = this.request.getIntHeader("Content-Length");
            InputStream pis = this.request.getInputStream();
            if (contentLength >= 0)
            {
                pis = new CountedInputStream(pis, contentLength);
                log.debug("put: Using CountedInputStream, length="
                        + String.valueOf(contentLength));
            }

            // Write to temporary file (so that SIP ingester can process it)
            File tempDir = new File(tempDirectory);
            File tempFile = File.createTempFile("davUpload" + pis.hashCode(), null, tempDir);
            log.debug("Storing temporary file at " + tempFile.getCanonicalPath());

            FileOutputStream fos = new FileOutputStream(tempFile);
            Utils.copy(pis, fos);
            fos.close();
            pis.close();

            // Initialize parameters to packager
            PackageParameters params = PackageParameters.create(this.request);
            // Force package ingester to respect Collection workflows (i.e. start workflow automatically as needed)
            params.setWorkflowEnabled(true);

            //ingest from the temp file to create the new DSpaceObject
            DSpaceObject ingestedDso = sip.ingest(this.context, this.collection, tempFile,
                    params, null);
            Item item = (Item) ingestedDso;

            //schedule temp file for deletion
            tempFile.deleteOnExit();

            //get the new workflowitem (if it exists)
            WorkflowItem wfi = WorkflowItem.findByItem(context, item);

            //Get status of item
            //  if we found a WorkflowItem, then it is still in-process
            //  if we didn't find one, item is already in archive
            int state;
            if(wfi!=null)
            {
                state = wfi.getState();
            }
            else
            {
                state = WorkflowManager.WFSTATE_ARCHIVE;
            }

            // get new item's location: if workflow completed, then look
            // for handle (but be ready for disappointment); otherwise,
            // return the workflow item's resource.
            String location = null;
            if (state == WorkflowManager.WFSTATE_ARCHIVE)
            {
                // Item is already in the archive
                String handle = HandleManager.findHandle(this.context, item);
                String end = (handle != null) ? DAVDSpaceObject
                        .getPathElt(handle) : DAVItem.getPathElt(item);
                DAVItem newItem = new DAVItem(this.context, this.request, this.response,
                        makeChildPath(end), item);
                location = newItem.hrefURL();
            }
            else if (state == WorkflowManager.WFSTATE_SUBMIT
                    || state == WorkflowManager.WFSTATE_STEP1POOL)
            {
                // Item is still in-process in the workflow
                location = hrefPrefix() + DAVWorkflow.getPath(wfi);
            }
            else
            {
                throw new DAVStatusException(
                        HttpServletResponse.SC_INTERNAL_SERVER_ERROR,
                        "Workflow object in unexpected state, state="
                                + String.valueOf(state) + ", aborting PUT.");
            }

            this.context.commit();
            log.info("Created new Item, location=" + location);
            this.response.setHeader("Location", location);
            this.response.setStatus(HttpServletResponse.SC_CREATED);
        }
        catch (PackageException pe)
        {
            pe.log(log);
            throw new DAVStatusException(
                    HttpServletResponse.SC_INTERNAL_SERVER_ERROR, pe.toString(), pe);
        }
        catch (CrosswalkException ie)
        {
            String reason = "";
            if (ie.getCause() != null)
            {
                reason = ", Reason: " + ie.getCause().toString();
            }
            log.error(ie.toString() + reason);
            throw new DAVStatusException(
                    HttpServletResponse.SC_INTERNAL_SERVER_ERROR, ie.toString()
                            + reason, ie);
        }
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#copyInternal(org.dspace.app.dav.DAVResource, int, boolean, boolean)
     */
    @Override
    protected int copyInternal(DAVResource destination, int depth,
            boolean overwrite, boolean keepProperties)
            throws DAVStatusException, SQLException, AuthorizeException,
            IOException
    {
        throw new DAVStatusException(HttpServletResponse.SC_NOT_IMPLEMENTED,
                "COPY method not implemented for Collection.");
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#deleteInternal()
     */
    @Override
    protected int deleteInternal() throws DAVStatusException, SQLException,
            AuthorizeException, IOException
    {
        Community[] ca = this.collection.getCommunities();
        if (ca != null)
        {
            for (Community element : ca)
            {
                element.removeCollection(this.collection);
            }
        }
        // collection.delete();
        return HttpServletResponse.SC_OK; // HTTP OK
    }

    /* (non-Javadoc)
     * @see org.dspace.app.dav.DAVResource#mkcolInternal(java.lang.String)
     */
    @Override
    protected int mkcolInternal(String waste) throws DAVStatusException,
            SQLException, AuthorizeException, IOException
    {
        throw new DAVStatusException(HttpServletResponse.SC_METHOD_NOT_ALLOWED,
                "MKCOL method not allowed for Collection.");
    }
}
TOP

Related Classes of org.dspace.app.dav.DAVCollection$CountedInputStream

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.