io.freefair.gradle.plugins.compress.tasks.SevenZip Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of compress-plugin Show documentation
Collection of Apache Commons Compress related Gradle plugins
There is a newer version: 8.11
package io.freefair.gradle.plugins.compress.tasks;

import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.SneakyThrows;
import org.apache.commons.compress.archivers.sevenz.SevenZArchiveEntry;
import org.apache.commons.compress.archivers.sevenz.SevenZMethod;
import org.apache.commons.compress.archivers.sevenz.SevenZOutputFile;
import org.gradle.api.internal.file.CopyActionProcessingStreamAction;
import org.gradle.api.internal.file.copy.CopyAction;
import org.gradle.api.internal.file.copy.CopyActionProcessingStream;
import org.gradle.api.internal.file.copy.FileCopyDetailsInternal;
import org.gradle.api.provider.Property;
import org.gradle.api.tasks.Input;
import org.gradle.api.tasks.WorkResult;
import org.gradle.api.tasks.WorkResults;
import org.gradle.api.tasks.bundling.AbstractArchiveTask;

import java.io.IOException;
import java.io.OutputStream;
import java.util.Date;

/**
 * @author Lars Grefer
 */
@Getter
@Setter
public abstract class SevenZip extends AbstractArchiveTask {

    @Input
    public abstract Property getContentCompression();

    public SevenZip() {
        getArchiveExtension().convention("7z");
        getContentCompression().set(SevenZMethod.LZMA2);
    }

    @Override
    protected CopyAction createCopyAction() {
        return this::execute;
    }

    @SneakyThrows
    private WorkResult execute(CopyActionProcessingStream stream) {
        try (SevenZOutputFile outputFile = new SevenZOutputFile(getArchiveFile().get().getAsFile())) {

            if (getContentCompression().isPresent()) {
                outputFile.setContentCompression(getContentCompression().get());
            }

            stream.process(new StreamAction(outputFile));

            outputFile.finish();

            return WorkResults.didWork(true);
        }
    }

    @RequiredArgsConstructor
    private static class StreamAction implements CopyActionProcessingStreamAction {
        private final SevenZOutputFile outputFile;

        @Override
        @SneakyThrows
        public void processFile(FileCopyDetailsInternal details) {
            SevenZArchiveEntry archiveEntry = new SevenZArchiveEntry();
            archiveEntry.setLastModifiedDate(new Date(details.getLastModified()));
            archiveEntry.setName(details.getPath());

            if (details.isDirectory()) {
                archiveEntry.setDirectory(true);

                outputFile.putArchiveEntry(archiveEntry);

            } else {
                archiveEntry.setDirectory(false);

                outputFile.putArchiveEntry(archiveEntry);

                details.copyTo(new OutputStream() {
                    @Override
                    public void write(int b) throws IOException {
                        outputFile.write(b);
                    }

                    @Override
                    public void write(byte[] b) throws IOException {
                        outputFile.write(b);
                    }

                    @Override
                    public void write(byte[] b, int off, int len) throws IOException {
                        outputFile.write(b, off, len);
                    }
                });
            }
            outputFile.closeArchiveEntry();
        }
    }
}