com.vmware.pscoe.maven.plugins.SshPushMojo Maven / Gradle / Ivy

Go to download
/*
 * #%L
 * ssh-maven-plugin
 * %%
 * Copyright (C) 2023 VMware
 * %%
 * Build Tools for VMware Aria
 * Copyright 2023 VMware, Inc.
 * 
 * This product is licensed to you under the BSD-2 license (the "License"). You may not use this product except in compliance with the BSD-2 License.  
 * 
 * This product may include a number of subcomponents with separate copyright notices and license terms. Your use of these subcomponents is subject to the terms and conditions of the subcomponent's license, as noted in the LICENSE file.
 * #L%
 */
package com.vmware.pscoe.maven.plugins;

import java.util.Collection;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

import org.apache.maven.artifact.Artifact;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.LifecyclePhase;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.plugins.annotations.ResolutionScope;
import org.apache.maven.project.MavenProject;

import com.vmware.pscoe.iac.artifact.PackageStore;
import com.vmware.pscoe.iac.artifact.PackageStoreFactory;
import com.vmware.pscoe.iac.artifact.configuration.ConfigurationException;
import com.vmware.pscoe.iac.artifact.model.Package;
import com.vmware.pscoe.iac.artifact.model.PackageFactory;
import com.vmware.pscoe.iac.artifact.model.PackageType;

@Mojo(name = "push", defaultPhase = LifecyclePhase.PRE_INTEGRATION_TEST, requiresDependencyResolution = ResolutionScope.RUNTIME_PLUS_SYSTEM)
public class SshPushMojo extends AbstractIacMojo {

	@Parameter(defaultValue = "${project}")
	private MavenProject project;

	@Parameter(required = false, property = "dryrun", defaultValue = "false")
	private boolean dryrun;

	@Parameter(required = true, property = "includeDependencies", defaultValue = "true")
	private boolean includeDependencies;

	@Parameter(required = false, property = "files", defaultValue = "")
	private List filesChanged;

	private static Package packageFromArtifact(Artifact artifact) {
		return PackageFactory.getInstance(PackageType.fromExtension(artifact.getType()), artifact.getFile(),
				new MavenArtifactPackageInfoProvider(artifact).getPackageName());
	}

	private void importArtifacts(Collection allArtifacts) throws MojoExecutionException {
		List bscArtifacts = allArtifacts.stream()
				.filter(a -> a.getType().equals(PackageType.BASIC.getPackageExtention())).collect(Collectors.toList());

		try {
			List packages = bscArtifacts.stream().map(SshPushMojo::packageFromArtifact)
					.collect(Collectors.toList());
			PackageStore store = PackageStoreFactory.getInstance(getConfigurationForSsh());

			boolean mergePackages = filesChanged.size() != 0; // it means that only a few files was selected to create the package
			store.importAllPackages(packages, dryrun, mergePackages);
		} catch (ConfigurationException e) {
			getLog().error(e);
			String message = String.format("Error processing configuration : %s", e.getMessage());
			throw new MojoExecutionException(e, message, message);
		}
	}

	@Override
	public void execute() throws MojoExecutionException, MojoFailureException {
		super.execute();

		final String artifactType = project.getArtifact().getType();
		final PackageType packageType = PackageType.fromExtension(artifactType);
		if (packageType == null || !packageType.equals(PackageType.BASIC)) {
			getLog().warn(String.format("Skipping push because of unsupported artifact type '%s'", artifactType));
			return;
		}
		if (project.getArtifact().getFile() == null) {
			throw new MojoExecutionException("You need to have the package goal as well when pushing SSH projects.");
		}

		LinkedList artifacts = new LinkedList<>();
		if (includeDependencies) {
			for (Object artifact : project.getArtifacts()) {
				artifacts.addLast((Artifact) artifact);
			}
		}
		artifacts.addLast(project.getArtifact());

		importArtifacts(artifacts);
	}
}