step.functions.packages.client.RemoteFunctionPackageClientImpl Maven / Gradle / Ivy

Go to download
package step.functions.packages.client;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import jakarta.ws.rs.client.Entity;
import jakarta.ws.rs.client.Invocation.Builder;
import jakarta.ws.rs.core.MediaType;

import step.attachments.FileResolver;
import step.client.AbstractRemoteClient;
import step.client.credentials.ControllerCredentials;
import step.client.resources.RemoteResourceManager;
import step.functions.packages.FunctionPackage;
import step.resources.Resource;
import step.resources.SimilarResourceExistingException;

public class RemoteFunctionPackageClientImpl extends AbstractRemoteClient implements FunctionPackageClient {

	private RemoteResourceManager remoteResourceManager;

	public RemoteFunctionPackageClientImpl() {
		super();
		remoteResourceManager = new RemoteResourceManager();
	}

	public RemoteFunctionPackageClientImpl(ControllerCredentials credentials) {
		super(credentials);
		remoteResourceManager = new RemoteResourceManager(credentials);
	}
	
	private FunctionPackage addOrUpdateKeywordPackage(FunctionPackage previousPackage, LibFileReference packageLibraryFile, File packageFile, Map packageAttributes, String trackingField) throws IOException {
		FunctionPackage functionPackage = null;
		if(previousPackage != null) {
			functionPackage = previousPackage;
		}else {
			functionPackage = new FunctionPackage();
		}
		if (packageLibraryFile != null) {
			Resource packageLibraryResource = null;
			if (packageLibraryFile.getFile() != null) {
				// upload new file as library (or reuse the existing resource with the same has sum)
				File file = packageLibraryFile.getFile();
				try {
					// TODO: here we want to protect against uploading duplicated resources (checkForDuplicates=true), but the remoteResourceManager now doesn't support this (should be fixed)
					remoteResourceManager.createResource("functions", new FileInputStream(file), file.getName(), true, null);
				} catch (SimilarResourceExistingException e) {
					// in case of existing resource with the same hash sum we want to use it
					if (e.getSimilarResources() != null && !e.getSimilarResources().isEmpty()) {
						packageLibraryResource = e.getSimilarResources().get(0);
					}
				}
			} else if (packageLibraryFile.getResourceId() != null && !packageLibraryFile.getResourceId().isEmpty()) {
				// reuse existing resource as package library
				packageLibraryResource = remoteResourceManager.getResource(packageLibraryFile.getResourceId());
				if (packageLibraryResource == null) {
					throw new IllegalArgumentException("Library resource not found by id: " + packageLibraryFile.getResourceId());
				}
			}

			if (packageLibraryResource != null) {
				functionPackage.setPackageLibrariesLocation(FileResolver.RESOURCE_PREFIX + packageLibraryResource.getId().toString());
			}
		}

		Resource packageResource = upload(packageFile);
		functionPackage.setPackageLocation(FileResolver.RESOURCE_PREFIX+packageResource.getId().toString());

		functionPackage.setPackageAttributes(packageAttributes);

		if (trackingField != null) {
			Map customFields = functionPackage.getCustomFields();
			if (customFields == null) {
				customFields = new HashMap<>();
				functionPackage.setCustomFields(customFields);
			}
			customFields.put(FunctionPackage.TRACKING_FIELD, trackingField);
		}

		functionPackage.setWatchForChange(false);

		Builder b = requestBuilder("/rest/functionpackages/");
		Entity entity = Entity.entity(functionPackage, MediaType.APPLICATION_JSON);
		
		return executeRequest(()->b.post(entity)).readEntity(FunctionPackage.class); 
	}

	@Override
	public FunctionPackage newKeywordPackage(File packageLibraryFile, File packageFile, Map packageAttributes, String trackingField) throws IOException {
		return addOrUpdateKeywordPackage(null, packageLibraryFile == null ? null : LibFileReference.file(packageFile), packageFile, packageAttributes, trackingField);
	}
	
	@Override
	public FunctionPackage updateKeywordPackageById(FunctionPackage previousPackage, File packageLibraryFile, File packageFile, Map packageAttributes, String trackingField) throws IOException {
		return addOrUpdateKeywordPackage(previousPackage, packageLibraryFile == null ? null : LibFileReference.file(packageFile), packageFile, packageAttributes, trackingField);
	}

	@Override
	public FunctionPackage newKeywordPackageWithLibReference(LibFileReference packageLib, File packageFile, Map packageAttributes, String trackingField) throws IOException {
		return addOrUpdateKeywordPackage(null, packageLib, packageFile, packageAttributes, trackingField);
	}

	@Override
	public FunctionPackage updateKeywordPackageWithLibReference(FunctionPackage previousPackage, LibFileReference packageLib, File packageFile, Map packageAttributes, String trackingField) throws IOException {
		return addOrUpdateKeywordPackage(previousPackage, packageLib, packageFile, packageAttributes, trackingField);
	}

	@Override
	public void deleteKeywordPackage(String packlageId) {
		Builder b = requestBuilder("/rest/functionpackages/"+packlageId);
		executeRequest(()->b.delete());
	}

	@Override
	public FunctionPackage updateResourceBasedKeywordPackage(File packageLibraryFile, File packageFile,
			Map packageAttributes) throws IOException {
		throw new RuntimeException(
				"This service has been removed. Lookup by resourceName isn't supported anymore. Use updateKeywordPackageById instead.");
	}

	@Override
	public FunctionPackage lookupPackageByResourceName(String resourceName) throws IOException {
		throw new RuntimeException(
				"This service has been removed. Lookup by resourceName isn't supported anymore. Use addOrUpdateKeywordPackage instead.");
	}

	protected Resource upload(File file) throws IOException {
		try {
			return remoteResourceManager.createResource("functions", new FileInputStream(file), file.getName(), false, null);
		} catch (IOException e) {
			throw e;
		} catch (SimilarResourceExistingException e) {
			throw new RuntimeException("Unexpected similar resource error. This should never occur.", e);
		}
	}

	@Override
	public void close() throws IOException {
		super.close();
		remoteResourceManager.close();
	}

}