gcube-cms-suite/geoportal-common/src/main/java/org/gcube/application/geoportal/common/utils/FileSets.java

package org.gcube.application.geoportal.common.utils;

import org.bson.Document;
import org.gcube.application.geoportal.common.model.legacy.InputStreamDescriptor;
import org.gcube.application.geoportal.common.model.rest.RegisterFileSetRequest;
import org.gcube.application.geoportal.common.model.rest.TempFile;
import org.gcube.contentmanagement.blobstorage.transport.backend.RemoteBackendException;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.ArrayList;

public class FileSets {

    public static class RequestBuilder {
        RegisterFileSetRequest theRequest=new RegisterFileSetRequest();

        public RequestBuilder add(TempFile... f){
            if(theRequest.getStreams()==null)
                theRequest.setStreams(new ArrayList<TempFile>());
            for(TempFile temp: f )
                theRequest.getStreams().add(temp);
            return this;
        }

        public RequestBuilder add(TempFile f){
            if(theRequest.getStreams()==null)
                theRequest.setStreams(new ArrayList<TempFile>());
            theRequest.getStreams().add(f);
            return this;
        }

        public RequestBuilder setFieldPath(String path){
            theRequest.setDestinationPath(path);
            return this;
        }

        public RequestBuilder setFieldDescriptionPath(String path){
            theRequest.setFieldPath(path);
            return this;
        }

        public RequestBuilder setClashPolicy(RegisterFileSetRequest.ClashOptions path){
            theRequest.setClashOption(path);
            return this;
        }

        public RequestBuilder setAttributes(Document attributes){
            theRequest.setAttributes(attributes);
            return this;
        }

        public RegisterFileSetRequest getTheRequest(){return theRequest;}
    }


    public static RequestBuilder build(String path) {
        return new RequestBuilder().setFieldPath(path);
    }

    public static RequestBuilder build(String path, TempFile...files) {
        return new RequestBuilder().setFieldPath(path).add(files);
    }

    public static TempFile asTemp(StorageUtils storage,InputStreamDescriptor descriptor) throws RemoteBackendException, FileNotFoundException {
        return storage.putOntoStorage(descriptor.getStream(), descriptor.getFilename());
    }

    public static TempFile[] asTemp(StorageUtils storage,InputStreamDescriptor... descriptors) throws RemoteBackendException, FileNotFoundException {
        ArrayList<TempFile> toReturn=new ArrayList<TempFile>();
        for(InputStreamDescriptor desc:descriptors)
            toReturn.add(storage.putOntoStorage(desc.getStream(), desc.getFilename()));
        return toReturn.toArray(new TempFile[toReturn.size()]);
    }

    public static RegisterFileSetRequest prepareRequestFromFolder(StorageUtils storage, String path, File directory) throws FileNotFoundException {

        File[] children =directory.listFiles();
        InputStreamDescriptor[] iss=new InputStreamDescriptor[children.length];
        return prepareRequest(storage,path,children);

    }

    public static RegisterFileSetRequest prepareRequest(StorageUtils storage, String path, File... toUpload) throws FileNotFoundException {

        FileSets.RequestBuilder builder = FileSets.build(path);
        for (File f : toUpload) {
            if(!f.isDirectory())
            builder.add(FileSets.asTemp(storage, new InputStreamDescriptor(new FileInputStream(f), f.getName())));
        }
        return builder.getTheRequest();
    }
}