io.automatiko.addon.files.s3.AzureBlobStoreFileVariableAugmentor Maven / Gradle / Ivy

Go to download
package io.automatiko.addon.files.s3;

import java.util.ArrayList;
import java.util.Collection;

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;

import io.automatiko.engine.api.workflow.Variable;
import io.automatiko.engine.api.workflow.VariableAugmentor;
import io.automatiko.engine.api.workflow.files.HasFiles;
import io.automatiko.engine.workflow.file.ByteArrayFile;

@ApplicationScoped
public class AzureBlobStoreFileVariableAugmentor implements VariableAugmentor {

    private String serviceUrl;

    private AzureBlobStoreStore store;

    @Inject
    public AzureBlobStoreFileVariableAugmentor(AzureBlobStoreStore store) {
        this.store = store;
        this.serviceUrl = store.urlPrefix();
    }

    @Override
    public boolean accept(Variable variable, Object value) {
        if (value == null) {
            // check variable definition
            if (variable.getType().getClassType() != null) {
                return ByteArrayFile.class.isAssignableFrom(variable.getType().getClassType());
            }
            return false;
        }

        if (value instanceof HasFiles) {
            value = ((HasFiles) value).files();
        }

        if (value instanceof ByteArrayFile) {
            return true;
        }

        if (value instanceof Collection) {
            return ((Collection) value).stream().anyMatch(item -> item instanceof ByteArrayFile);
        }
        return false;
    }

    @Override
    public Object augmentOnCreate(String processId, String processVersion, String processInstanceId, Variable variable,
            Object value) {
        Object originalValue = value;
        value = retrieveValue(value);
        if (value == null) {
            return value;
        }
        StringBuilder url = new StringBuilder(serviceUrl);
        url.append(processId).append("/");
        if (processVersion != null && !processVersion.isEmpty()) {
            url.append(processVersion).append("/");
        }
        url.append(processInstanceId).append("/").append(variable.getName());

        if (value instanceof ByteArrayFile) {
            ByteArrayFile file = (ByteArrayFile) value;
            if (file.content() != null) {
                AzureBlobStoreFile fsFile = new AzureBlobStoreFile(file.name(), null, file.attributes());
                fsFile.url(url.toString() + "/" + file.name());

                // store file on file system
                store.save(file, processId, processVersion, processInstanceId, variable.getName(),
                        file.name());
                value = updateValue(originalValue, fsFile);
            }
        } else if (value instanceof Collection) {
            Collection fsFiles = new ArrayList<>();
            for (Object potentialFile : (Collection) value) {
                if (potentialFile instanceof ByteArrayFile) {
                    ByteArrayFile file = (ByteArrayFile) potentialFile;
                    if (file.content() != null) {
                        AzureBlobStoreFile fsFile = new AzureBlobStoreFile(file.name(), null, file.attributes());
                        fsFile.url(url.toString() + "/" + file.name());

                        // store file on file system
                        store.save(file, processId, processVersion, processInstanceId, variable.getName(),
                                file.name());
                        fsFiles.add(fsFile);
                    } else {
                        fsFiles.add(file);
                    }
                }
            }
            return updateValue(originalValue, fsFiles);
        }

        return value;
    }

    @Override
    public Object augmentOnUpdate(String processId, String processVersion, String processInstanceId, Variable variable,
            Object value) {
        Object originalValue = value;
        value = retrieveValue(value);
        if (value == null) {
            return value;
        }
        StringBuilder url = new StringBuilder(serviceUrl);
        url.append(processId).append("/");
        if (processVersion != null && !processVersion.isEmpty()) {
            url.append(processVersion).append("/");
        }
        url.append(processInstanceId).append("/").append(variable.getName());

        if (value instanceof ByteArrayFile) {
            ByteArrayFile file = (ByteArrayFile) value;
            if (file.content() != null) {
                AzureBlobStoreFile fsFile = new AzureBlobStoreFile(file.name(), null, file.attributes());
                fsFile.url(url.toString() + "/" + file.name());

                // replace file on file system
                store.replace(file, processId, processVersion, processInstanceId, variable.getName(),
                        file.name());
                value = updateValue(originalValue, fsFile);
            }
        } else if (value instanceof Collection) {
            Collection fsFiles = new ArrayList<>();
            for (Object potentialFile : (Collection) value) {
                if (potentialFile instanceof ByteArrayFile) {
                    ByteArrayFile file = (ByteArrayFile) potentialFile;
                    if (file.content() != null) {
                        AzureBlobStoreFile fsFile = new AzureBlobStoreFile(file.name(), null, file.attributes());
                        fsFile.url(url.toString() + "/" + file.name());

                        // replace file on file system
                        store.replace(file, processId, processVersion, processInstanceId, variable.getName(),
                                file.name());
                        fsFiles.add(fsFile);
                    } else {
                        fsFiles.add(file);
                    }
                }
            }
            return updateValue(originalValue, fsFiles);
        }

        return value;
    }

    @Override
    public void augmentOnDelete(String processId, String processVersion, String processInstanceId, Variable variable,
            Object value) {
        value = retrieveValue(value);
        if (value instanceof ByteArrayFile) {
            ByteArrayFile file = (ByteArrayFile) value;
            store.remove(processId, processVersion, processInstanceId, variable.getName(), file.name());
        } else if (value instanceof Collection) {
            for (Object potentialFile : (Collection) value) {
                if (potentialFile instanceof ByteArrayFile) {
                    ByteArrayFile file = (ByteArrayFile) potentialFile;
                    store.remove(processId, processVersion, processInstanceId, variable.getName(), file.name());
                }
            }

        }

    }

    protected Object retrieveValue(Object value) {
        if (value instanceof HasFiles) {
            return ((HasFiles) value).files();
        }

        return value;
    }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    protected Object updateValue(Object variable, Object value) {
        if (variable instanceof HasFiles) {
            ((HasFiles) variable).augmentFiles(value);

            return variable;
        }

        return value;
    }

}