com.centurylink.mdw.hub.servlet.asset.VersionedAssetServer Maven / Gradle / Ivy

Go to download
package com.centurylink.mdw.hub.servlet.asset;

import com.centurylink.mdw.cache.impl.AssetRefCache;
import com.centurylink.mdw.common.service.ServiceException;
import com.centurylink.mdw.dataaccess.AssetRef;
import com.centurylink.mdw.model.asset.Asset;
import com.centurylink.mdw.model.asset.AssetVersionSpec;
import com.centurylink.mdw.util.AssetRefConverter;
import com.centurylink.mdw.util.log.LoggerUtil;
import com.centurylink.mdw.util.log.StandardLogger;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;

/**
 * Serves up previous asset versions.
 */
public class VersionedAssetServer {
    private static final StandardLogger logger = LoggerUtil.getStandardLogger();

    private HttpServletRequest servletRequest;
    private HttpServletResponse servletResponse;

    public VersionedAssetServer(HttpServletRequest servletRequest, HttpServletResponse servletResponse) {
        this.servletRequest = servletRequest;
        this.servletResponse = servletResponse;
    }

    public void serveAsset(String path, String version) throws IOException, ServiceException {
        // if older version check in AssetRefCache
        AssetRef assetRef = AssetRefCache.getRef(path, version);
        if (assetRef == null)
            throw new ServiceException(ServiceException.NOT_FOUND, "Asset ref not found: " + path + " v" + version);
        if ("true".equalsIgnoreCase(servletRequest.getParameter("download"))) {
            String filename = path.substring(path.indexOf("/") + 1);
            servletResponse.setHeader("Content-Disposition", "attachment;filename=\"" + filename + "\"");
            servletResponse.setContentType("application/octet-stream");
        }
        try {
            Asset asset = AssetRefConverter.getAsset(assetRef);
            OutputStream out = servletResponse.getOutputStream();
            out.write(asset.getContent());
        }
        catch (Exception ex) {
            logger.severeException(ex.getMessage(), ex);
        }
    }
}