io.bdeploy.bhive.op.VerifyOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Public API including dependencies, ready to be used for integrations and plugins.
There is a newer version: 7.3.6
package io.bdeploy.bhive.op;

import static io.bdeploy.bhive.op.VerifyOperation.VerifyOpStatus.MISSING;
import static io.bdeploy.bhive.op.VerifyOperation.VerifyOpStatus.MODIFIED;
import static io.bdeploy.bhive.op.VerifyOperation.VerifyOpStatus.UNMODIFIED;

import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;

import io.bdeploy.bhive.BHive;
import io.bdeploy.bhive.ReadOnlyOperation;
import io.bdeploy.bhive.model.Manifest;
import io.bdeploy.bhive.model.ObjectId;
import io.bdeploy.bhive.objects.view.BlobView;
import io.bdeploy.bhive.objects.view.TreeView;
import io.bdeploy.bhive.objects.view.scanner.TreeVisitor;
import io.bdeploy.bhive.op.VerifyOperation.VerifiedBlobView;

/**
 * Scan a given target folder and origin {@link Manifest} and identify which files were deleted or modified.
 */
@ReadOnlyOperation
public class VerifyOperation extends BHive.Operation> {

    private Path targetPath;
    private Manifest.Key manifest;

    @Override
    public List call() throws Exception {
        List result = new ArrayList<>();
        TreeView state = execute(new ScanOperation().setManifest(manifest));
        state.visit(new TreeVisitor.Builder().onBlob(blob -> this.visit(blob, result)).build());
        return result;
    }

    private void visit(BlobView blob, List result) {
        Path path = targetPath.resolve(blob.getPathString());
        if (!path.toFile().exists()) {
            result.add(new VerifiedBlobView(MISSING, blob.getPathString()));
            return;
        }
        try (InputStream is = new BufferedInputStream(new FileInputStream(path.toFile()))) {
            ObjectId objectId = ObjectId.createFromStreamNoCopy(is);
            if (objectId.equals(blob.getElementId())) {
                result.add(new VerifiedBlobView(UNMODIFIED, blob.getPathString()));
            } else {
                result.add(new VerifiedBlobView(MODIFIED, blob.getPathString()));
            }
        } catch (Exception e) {
            result.add(new VerifiedBlobView(MODIFIED, blob.getPathString()));
        }
    }

    /**
     * Set the target path to check files for. The given directory will be compared against scanned {@link Manifest}
     */
    public VerifyOperation setTargetPath(Path targetPath) {
        this.targetPath = targetPath;
        return this;
    }

    /**
     * Set the {@link Manifest} to scan. Scans the {@link Manifest}s root tree.
     */
    public VerifyOperation setManifest(Manifest.Key manifest) {
        this.manifest = manifest;
        return this;
    }

    public static class VerifiedBlobView {

        public final VerifyOpStatus status;
        public final String path;

        private VerifiedBlobView(VerifyOpStatus status, String path) {
            this.status = status;
            this.path = path;
        }
    }

    public enum VerifyOpStatus {
        MISSING,
        MODIFIED,
        UNMODIFIED
    }
}