com.marklogic.appdeployer.command.security.DeployPrivilegeRolesCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ml-app-deployer Show documentation

Java client for the MarkLogic REST Management API and for deploying applications to MarkLogic

There is a newer version: 5.0.0

package com.marklogic.appdeployer.command.security;

import com.marklogic.appdeployer.command.AbstractResourceCommand;
import com.marklogic.appdeployer.command.CommandContext;
import com.marklogic.appdeployer.command.ResourceReference;
import com.marklogic.appdeployer.command.SortOrderConstants;
import com.marklogic.mgmt.api.API;
import com.marklogic.mgmt.api.security.Privilege;
import com.marklogic.mgmt.mapper.DefaultResourceMapper;
import com.marklogic.mgmt.mapper.ResourceMapper;
import com.marklogic.mgmt.resource.ResourceManager;
import com.marklogic.mgmt.resource.security.PrivilegeManager;

import java.io.File;
import java.util.function.BiPredicate;

/**
 * Intended to run after roles and privileges have been deployed so that any roles associated with privileges can be
 * safely deployed.
 */
public class DeployPrivilegeRolesCommand extends AbstractResourceCommand {

	private ResourceMapper resourceMapper;

	public DeployPrivilegeRolesCommand() {
		setExecuteSortOrder(SortOrderConstants.DEPLOY_PRIVILEGE_ROLES);
		setUndoSortOrder(SortOrderConstants.DELETE_PRIVILEGES);

		setSupportsResourceMerging(true);
		setResourceClassType(Privilege.class);
	}

	@Override
	public void undo(CommandContext context) {
		logger.info("Nothing to do, as DeployPrivilegesCommand is expected to delete privileges");
	}

	@Override
	protected File[] getResourceDirs(CommandContext context) {
		return findResourceDirs(context, configDir -> configDir.getPrivilegesDir());
	}

	@Override
	protected ResourceManager getResourceManager(CommandContext context) {
		return new PrivilegeManager(context.getManageClient());
	}

	@Override
	protected String adjustPayloadBeforeSavingResource(CommandContext context, File f, String payload) {
		payload = super.adjustPayloadBeforeSavingResource(context, f, payload);
		if (payload != null) {
			if (resourceMapper == null) {
				resourceMapper = new DefaultResourceMapper(new API(context.getManageClient()));
			}
			Privilege p = resourceMapper.readResource(payload, Privilege.class);
			if (p.getRole() == null || p.getRole().isEmpty()) {
				return null;
			}
		}
		return payload;
	}

	@Override
	protected BiPredicate getBiPredicateForMergingResources() {
		return new PrivilegeBiPredicate();
	}
}