com.adobe.pdfservices.operation.pdfjobs.jobs.SplitPDFJob Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pdfservices-sdk Show documentation
Adobe PDF Services SDK allows you to access RESTful APIs to create, convert, and manipulate PDFs within your applications. Older versions can be found under groupId: com.adobe.documentservices, artifactId: pdftools-sdk
There is a newer version: 4.2.0
Show newest version
/*
 * Copyright 2024 Adobe
 * All Rights Reserved.
 *
 * NOTICE: Adobe permits you to use, modify, and distribute this file in
 * accordance with the terms of the Adobe license agreement accompanying
 * it. If you have received this file from a source other than Adobe,
 * then your use, modification, or distribution of it requires the prior
 * written permission of Adobe.
 */

package com.adobe.pdfservices.operation.pdfjobs.jobs;

import com.adobe.pdfservices.operation.PDFServicesJob;
import com.adobe.pdfservices.operation.config.notifier.NotifierConfig;
import com.adobe.pdfservices.operation.exception.ServiceApiException;
import com.adobe.pdfservices.operation.internal.ExecutionContext;
import com.adobe.pdfservices.operation.internal.PDFServicesHelper;
import com.adobe.pdfservices.operation.internal.constants.CustomErrorMessages;
import com.adobe.pdfservices.operation.internal.constants.OperationHeaderInfoEndpointMap;
import com.adobe.pdfservices.operation.internal.dto.request.PlatformApiRequest;
import com.adobe.pdfservices.operation.internal.dto.request.splitpdf.SplitPDFExternalAssetRequest;
import com.adobe.pdfservices.operation.internal.dto.request.splitpdf.SplitPDFInternalAssetRequest;
import com.adobe.pdfservices.operation.internal.http.DefaultRequestHeaders;
import com.adobe.pdfservices.operation.internal.http.HttpResponse;
import com.adobe.pdfservices.operation.internal.util.ObjectUtil;
import com.adobe.pdfservices.operation.io.Asset;
import com.adobe.pdfservices.operation.io.CloudAsset;
import com.adobe.pdfservices.operation.io.ExternalAsset;
import com.adobe.pdfservices.operation.pdfjobs.params.splitpdf.SplitPDFParams;

import java.util.List;
import java.util.UUID;

/**
 * A job that splits PDF document into multiple smaller documents by simply specifying either the number of files,
 * pages per file, or page ranges.
 *
 * 
 * Sample Usage:
 * 
{@code
 *             InputStream inputStream = new FileInputStream(new File("SOURCE_PATH"));
 *
 *             Credentials credentials = new ServicePrincipalCredentials(
 *                     System.getenv("PDF_SERVICES_CLIENT_ID"),
 *                     System.getenv("PDF_SERVICES_CLIENT_SECRET"));
 *
 *             PDFServices pdfServices = new PDFServices(credentials);
 *
 *             Asset asset = pdfServices.upload(inputStream, PDFServicesMediaType.PDF.getMediaType());
 *
 *             SplitPDFParams splitPDFParams = new SplitPDFParams().setPageCount(2);
 *             SplitPDFJob splitPDFJob = new SplitPDFJob(asset, splitPDFParams);
 *
 *             String location = pdfServices.submit(splitPDFJob);
 *             PDFServicesResponse pdfServicesResponse = pdfServices.getJobResult(location, SplitPDFResult.class);
 *
 *             List resultAssets = pdfServicesResponse.getResult().getAssets();
 *
 *             int index = 0;
 *             for (Asset resultAsset : resultAssets) {
 *                 StreamAsset streamAsset = pdfServices.getContent(resultAsset);
 *                 index++;
 *             }
 * }
 */
public class SplitPDFJob extends PDFServicesJob {
    private Asset inputAsset;

    private Asset outputAsset;

    private SplitPDFParams splitPDFParams;

    /**
     * Constructs a new {@code SplitPDFJob} instance.
     *
     * @param asset          {@link Asset} object containing the input file; can not be null.
     * @param splitPDFParams {@link SplitPDFParams} object containing the parameters for splitting the PDF; can not
     *                       be null.
     */
    public SplitPDFJob(Asset asset, SplitPDFParams splitPDFParams) {
        ObjectUtil.requireNonNull(asset, String.format(CustomErrorMessages.GENERIC_CAN_NOT_BE_NULL, "Input Asset"));
        ObjectUtil.requireNonNull(splitPDFParams, String.format(CustomErrorMessages.GENERIC_CAN_NOT_BE_NULL,
                "Split " + "PDF parameters"));
        this.inputAsset = asset;
        this.splitPDFParams = splitPDFParams;
    }

    @Override
    protected String process(ExecutionContext executionContext) throws ServiceApiException {
        return this.process(executionContext, null);
    }

    @Override
    protected String process(ExecutionContext executionContext, List notifyConfigList) throws ServiceApiException {
        this.validate(executionContext);
        PlatformApiRequest splitPDFRequest = generatePlatformApiRequest(notifyConfigList);
        String xRequestId = UUID.randomUUID().toString();
        HttpResponse response = PDFServicesHelper.submitJob(executionContext, splitPDFRequest, xRequestId,
                OperationHeaderInfoEndpointMap.SPLIT_PDF);
        return response.getHeaders().get(DefaultRequestHeaders.LOCATION_HEADER_NAME);
    }

    /**
     * Sets the output asset for the job.
     * {@code @note} External assets can be set as output only when input is external asset as well
     *
     * @param asset {@link Asset} object representing the output asset; can not be null.
     * @return {@code SplitPDFJob} instance
     */
    public SplitPDFJob setOutput(Asset asset) {
        ObjectUtil.requireNonNull(asset, String.format(CustomErrorMessages.GENERIC_CAN_NOT_BE_NULL, "Output asset"));
        if (this.inputAsset instanceof CloudAsset) {
            throw new IllegalArgumentException(CustomErrorMessages.SET_OUTPUT_VALIDATE);
        }

        this.outputAsset = asset;
        return this;
    }

    private PlatformApiRequest generatePlatformApiRequest(List notifyConfigList) {
        PlatformApiRequest splitPDFRequest;
        if (this.inputAsset instanceof CloudAsset) {
            splitPDFRequest = new SplitPDFInternalAssetRequest(( (CloudAsset) this.inputAsset ).getAssetId(),
                    this.splitPDFParams, notifyConfigList);
        } else {
            splitPDFRequest = new SplitPDFExternalAssetRequest((ExternalAsset) this.inputAsset, this.splitPDFParams,
                    notifyConfigList).setOutput((ExternalAsset) outputAsset);
        }
        return splitPDFRequest;
    }
}