org.eclipse.skalli.core/src/main/java/org/eclipse/skalli/core/rest/admin/StatisticsBackupResource.java

/*******************************************************************************
 * Copyright (c) 2010-2014 SAP AG and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     SAP AG - initial API and implementation
 *******************************************************************************/
package org.eclipse.skalli.core.rest.admin;

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;

import org.apache.commons.io.IOUtils;
import org.eclipse.skalli.commons.Statistics;
import org.eclipse.skalli.commons.Statistics.BrowserInfo;
import org.eclipse.skalli.commons.Statistics.RefererInfo;
import org.eclipse.skalli.commons.Statistics.ResponseTimeInfo;
import org.eclipse.skalli.commons.Statistics.SearchInfo;
import org.eclipse.skalli.commons.Statistics.UsageInfo;
import org.eclipse.skalli.commons.Statistics.UserInfo;
import org.eclipse.skalli.services.extension.rest.ResourceBase;
import org.eclipse.skalli.services.extension.rest.ResourceRepresentation;
import org.eclipse.skalli.services.permit.Permits;
import org.restlet.data.Disposition;
import org.restlet.data.MediaType;
import org.restlet.data.Status;
import org.restlet.representation.OutputRepresentation;
import org.restlet.representation.Representation;
import org.restlet.resource.Get;
import org.restlet.resource.Put;

import com.thoughtworks.xstream.XStreamException;

public class StatisticsBackupResource extends ResourceBase {

    private static final String FILE_NAME = "statistics.zip"; //$NON-NLS-1$
    private static final String ENTRY_NAME = "statistics.xml"; //$NON-NLS-1$

    private static final String ID_PREFIX = "rest:api/admin/statistics/backup:"; //$NON-NLS-1$
    private static final String ERROR_ID_IO_ERROR = ID_PREFIX + "00"; //$NON-NLS-1$
    private static final String ERROR_ID_INVALID_BACKUP = ID_PREFIX + ":30"; //$NON-NLS-1$
    private static final String ERROR_ID_PARSING_FAILED = ID_PREFIX + ":40"; //$NON-NLS-1$
    private static final String ERROR_ID_CLASS_MISMATCH = ID_PREFIX + ":50"; //$NON-NLS-1$


    @Get
    public Representation backup() {
        if (!Permits.isAllowed(getAction(), getPath())) {
            return createUnauthorizedRepresentation();
        }
        StatisticsQuery query = new StatisticsQuery(getQueryAttributes());
        Statistics statistics = new Statistics(Statistics.getDefault(), query.getFrom(), query.getTo());
        ZipOutputRepresentation zipRepresentation = new ZipOutputRepresentation(statistics);
        Disposition disposition = new Disposition(Disposition.TYPE_ATTACHMENT);
        disposition.setFilename(FILE_NAME);
        zipRepresentation.setDisposition(disposition);
        return zipRepresentation;
    }

    @Put
    public Representation restore(Representation entity) {
        if (!Permits.isAllowed(getAction(), getPath())) {
            return createUnauthorizedRepresentation();
        }

        ZipInputStream zipStream = null;
        try {
            zipStream = new ZipInputStream(entity.getStream());
            ZipEntry entry = zipStream.getNextEntry();
            try {
                String entryName = entry.getName();
                if (!entryName.equals(ENTRY_NAME)) {
                    return createErrorRepresentation(Status.CLIENT_ERROR_BAD_REQUEST, ERROR_ID_INVALID_BACKUP,
                            "Invalid backup file: Expected entry ''statistics.xml'', but found ''{0}''", entryName); //$NON-NLS-1$
                }
                ResourceRepresentation<Statistics> representation = new ResourceRepresentation<Statistics>();
                addAliases(representation);
                Statistics.getDefault().restore(representation.read(zipStream, Statistics.class));
            } finally {
                zipStream.closeEntry();
            }
        } catch (IOException e) {
            return createIOErrorRepresentation(ERROR_ID_IO_ERROR, e);
        } catch (XStreamException e) {
            return createErrorRepresentation(Status.CLIENT_ERROR_BAD_REQUEST, ERROR_ID_PARSING_FAILED,
                    "Invalid backup file: {0}", e.getMessage()); //$NON-NLS-1$
        } catch (ClassCastException e) {
            return createErrorRepresentation(Status.CLIENT_ERROR_BAD_REQUEST, ERROR_ID_CLASS_MISMATCH,
                    "Invalid backup file: {0}", e.getMessage()); //$NON-NLS-1$
        } finally {
            IOUtils.closeQuietly(zipStream);
        }
        setStatus(Status.SUCCESS_NO_CONTENT);
        return null;
    }

    private static class ZipOutputRepresentation extends OutputRepresentation {

        private Statistics statistics;

        public ZipOutputRepresentation(Statistics statistics) {
            super(MediaType.APPLICATION_ZIP);
            this.statistics = statistics;
        }

        @Override
        public void write(OutputStream out) throws IOException {
            ZipOutputStream zipStream = null;
            try {
                zipStream = new ZipOutputStream(new BufferedOutputStream(out));
                ZipEntry entry = new ZipEntry(ENTRY_NAME);
                zipStream.putNextEntry(entry);
                ResourceRepresentation<Statistics> representation = new ResourceRepresentation<Statistics>(statistics);
                addAliases(representation);
                representation.write(zipStream);
            } finally {
                IOUtils.closeQuietly(zipStream);
            }
        }
    }

    @SuppressWarnings("nls")
    private static void addAliases(ResourceRepresentation<Statistics> representation) {
        representation.addAlias("statistics", Statistics.class);
        representation.addAlias("user", UserInfo.class);
        representation.addAlias("usage", UsageInfo.class);
        representation.addAlias("referer", RefererInfo.class);
        representation.addAlias("browser", BrowserInfo.class);
        representation.addAlias("search", SearchInfo.class);
        representation.addAlias("responseTime", ResponseTimeInfo.class);
    }
}