ch.cyberduck.core.sds.SDSDelegatingCopyFeature.java Source code

Introduction

Here is the source code for ch.cyberduck.core.sds.SDSDelegatingCopyFeature.java

Source

package ch.cyberduck.core.sds;

/*
 * Copyright (c) 2002-2017 iterate GmbH. All rights reserved.
 * https://cyberduck.io/
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

import ch.cyberduck.core.ConnectionCallback;
import ch.cyberduck.core.DefaultIOExceptionMappingService;
import ch.cyberduck.core.Path;
import ch.cyberduck.core.PathContainerService;
import ch.cyberduck.core.Session;
import ch.cyberduck.core.exception.BackgroundException;
import ch.cyberduck.core.features.Copy;
import ch.cyberduck.core.sds.io.swagger.client.model.FileKey;
import ch.cyberduck.core.sds.triplecrypt.TripleCryptConverter;
import ch.cyberduck.core.shared.DefaultCopyFeature;
import ch.cyberduck.core.transfer.TransferStatus;

import org.apache.commons.lang3.StringUtils;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;

import com.fasterxml.jackson.databind.ObjectWriter;
import eu.ssp_europe.sds.crypto.Crypto;

public class SDSDelegatingCopyFeature implements Copy {

    private final SDSSession session;
    private final SDSCopyFeature proxy;

    private final PathContainerService containerService = new PathContainerService();

    public SDSDelegatingCopyFeature(final SDSSession session, final SDSCopyFeature proxy) {
        this.session = session;
        this.proxy = proxy;
    }

    @Override
    public void copy(final Path source, final Path target, final TransferStatus status,
            final ConnectionCallback callback) throws BackgroundException {
        if (containerService.getContainer(target).getType().contains(Path.Type.vault)) {
            final FileKey fileKey = TripleCryptConverter.toSwaggerFileKey(Crypto.generateFileKey());
            final ObjectWriter writer = session.getClient().getJSON().getContext(null).writerFor(FileKey.class);
            final ByteArrayOutputStream out = new ByteArrayOutputStream();
            try {
                writer.writeValue(out, fileKey);
            } catch (IOException e) {
                throw new DefaultIOExceptionMappingService().map(e);
            }
            status.setFilekey(ByteBuffer.wrap(out.toByteArray()));
        }
        if (containerService.getContainer(source).getType().contains(Path.Type.vault)
                || containerService.getContainer(target).getType().contains(Path.Type.vault)) {
            new DefaultCopyFeature(session).copy(source, target, status, callback);
        } else {
            if (StringUtils.equals(source.getName(), target.getName())) {
                proxy.copy(source, target, status, callback);
            } else {
                new DefaultCopyFeature(session).copy(source, target, status, callback);
            }
        }
    }

    @Override
    public boolean isRecursive(final Path source, final Path target) {
        if (containerService.getContainer(source).getType().contains(Path.Type.vault)
                || containerService.getContainer(target).getType().contains(Path.Type.vault)) {
            return new DefaultCopyFeature(session).isRecursive(source, target);
        }
        if (StringUtils.equals(source.getName(), target.getName())) {
            return proxy.isRecursive(source, target);
        }
        return new DefaultCopyFeature(session).isRecursive(source, target);
    }

    @Override
    public boolean isSupported(final Path source, final Path target) {
        if (containerService.getContainer(source).getType().contains(Path.Type.vault)
                || containerService.getContainer(target).getType().contains(Path.Type.vault)) {
            return new DefaultCopyFeature(session).isSupported(source, target);
        }
        if (StringUtils.equals(source.getName(), target.getName())) {
            return proxy.isSupported(source, target);
        }
        return new DefaultCopyFeature(session).isSupported(source, target);
    }

    @Override
    public Copy withTarget(final Session<?> session) {
        proxy.withTarget(session);
        return this;
    }
}