com.sendsafely.handlers.AddFileHandler Maven / Gradle / Ivy

Go to download
package com.sendsafely.handlers;

import java.io.IOException;

import com.sendsafely.File;
import com.sendsafely.Package;
import com.sendsafely.ProgressInterface;
import com.sendsafely.dto.request.CreateFileIdRequest;
import com.sendsafely.dto.response.CreateFileIdResponse;
import com.sendsafely.enums.APIResponse;
import com.sendsafely.enums.Endpoint;
import com.sendsafely.exceptions.FileOperationFailedException;
import com.sendsafely.exceptions.LimitExceededException;
import com.sendsafely.exceptions.PackageInformationFailedException;
import com.sendsafely.exceptions.SendFailedException;
import com.sendsafely.exceptions.UploadFileException;
import com.sendsafely.file.FileManager;
import com.sendsafely.progress.DefaultProgress;
import com.sendsafely.upload.UploadManager;
import com.sendsafely.utils.FileUploadUtility;

public class AddFileHandler extends BaseHandler 
{	
	
	private CreateFileIdRequest request;
	
	public AddFileHandler(UploadManager uploadManager) {
		super(uploadManager);

        request = new CreateFileIdRequest(uploadManager.getJsonManager());
	}

	public File makeRequest(String packageId, String keyCode, FileManager file) throws LimitExceededException, UploadFileException 
	{	
		return makeRequest(packageId, null, keyCode, file, new DefaultProgress());
	}
	
	public File makeRequest(String packageId, String directoryId, String keyCode, FileManager file) throws LimitExceededException, UploadFileException 
	{
		return makeRequest(packageId, directoryId,  keyCode, file, new DefaultProgress());
	}

	public File makeRequest(String packageId, String directoryId, String keyCode, FileManager file, ProgressInterface progress) throws LimitExceededException, UploadFileException {
			
			Package packageInfo = getPackageInfo(packageId);
			return makeRequest(packageId, directoryId, keyCode, packageInfo.getServerSecret(), file, progress);
	}
	
	public File makeRequest(String packageId, String keyCode, FileManager file, ProgressInterface progress) throws LimitExceededException, UploadFileException 
	{	
		Package packageInfo = getPackageInfo(packageId);
		return makeRequest(packageId, (String)null, keyCode, packageInfo.getServerSecret(), file, progress);
	}
	
	public File makeRequest(String packageId, String directoryId, String keyCode, String serverSecret,  FileManager file, ProgressInterface progress) throws LimitExceededException, UploadFileException 
	{	
		FileUploadUtility uploadUtility = new FileUploadUtility(super.uploadManager);

		String fileId = createFileId(file, packageId, directoryId, uploadUtility.calculateParts(file));
		progress.gotFileId(fileId);
		
		String encryptionKey = createEncryptionKey(serverSecret, keyCode);
		
		UploadFileHandler handler = (UploadFileHandler)HandlerFactory.getInstance(super.uploadManager, Endpoint.UPLOAD_FILE);
		com.sendsafely.File metaFile = makeRequest(handler, packageId, fileId, directoryId, encryptionKey, file, progress);
		
		return metaFile;
	}
	
	protected File makeRequest(UploadFileHandler handler, String packageId, String fileId, String encryptionKey, FileManager file, ProgressInterface progress) throws UploadFileException, LimitExceededException
	{
		try {
			return handler.makeRequest(packageId, fileId, (String)null, encryptionKey, file, progress);
		} catch (SendFailedException e) {
			throw new UploadFileException(e);
		} catch (IOException e) {
			throw new UploadFileException(e);
		}
	}
	
	protected File makeRequest(UploadFileHandler handler, String packageId, String fileId, String directoryId, String encryptionKey, FileManager file, ProgressInterface progress) throws UploadFileException, LimitExceededException
	{
		try {
			return handler.makeRequest(packageId, fileId, directoryId, encryptionKey, file, progress);
		} catch (SendFailedException e) {
			throw new UploadFileException(e);
		} catch (IOException e) {
			throw new UploadFileException(e);
		}
	}
	
	protected Package getPackageInfo(String packageId) throws UploadFileException
	{
		Package info;
		try {
			info = ((PackageInformationHandler)(HandlerFactory
					.getInstance(super.uploadManager, Endpoint.PACKAGE_INFORMATION))).makeRequest(packageId);
		} catch (PackageInformationFailedException e) {
			throw new UploadFileException(e);
		}
		
		return info;
	}
	
	protected String createEncryptionKey(String serverSecret, String keyCode)
    {
        return serverSecret + keyCode;
    }
		
	protected String createFileId(FileManager file, String packageId, String directoryId, int calculateParts) throws LimitExceededException, UploadFileException {
		request.setDirectoryId(directoryId);
		return createFileId(file, packageId, calculateParts);
	}
	
	protected String createFileId(FileManager file, String packageId, int parts) throws LimitExceededException, UploadFileException
	{
        try {
            request.setFilename(file.getName());
            request.setFilesize(file.length());
            request.setParts(parts);
            request.setPackageId(packageId);
        } catch (IOException e) {
            throw new UploadFileException(e);
        }
		
		CreateFileIdResponse response = send();
		
		if(response.getResponse() == APIResponse.LIMIT_EXCEEDED)
		{
			throw new LimitExceededException(response.getMessage());
		}
		else if (response.getResponse() != APIResponse.SUCCESS)
		{
			throw new UploadFileException(response.getMessage());
		}
		
		return response.getMessage();
	}
	
	protected CreateFileIdResponse send() throws UploadFileException
	{
		try {
			return send(request, new CreateFileIdResponse());
		} catch (SendFailedException e) {
			throw new UploadFileException(e);
		} catch (IOException e) {
			throw new UploadFileException(e);
		}
	}
}