Examples with Policy com.google.api.services.iam.v1.model.Policy used on opensource projects

Example 6 with Policy

use of com.google.api.services.iam.v1.model.Policy in project java-docs-samples by GoogleCloudPlatform.

the class Snippets method removeMemberFromKeyRingPolicy.

// [END kms_remove_member_from_cryptokey_policy]
// [START kms_remove_member_from_keyring_policy]
/**
 * Removes the given member from the given policy.
 */
public static Policy removeMemberFromKeyRingPolicy(String projectId, String locationId, String keyRingId, String member, String role) throws IOException {
    // Create the Cloud KMS client.
    CloudKMS kms = createAuthorizedClient();
    // The resource name of the cryptoKey
    String cryptoKey = String.format("projects/%s/locations/%s/keyRings/%s", projectId, locationId, keyRingId);
    // Get the current IAM policy and add the new account to it.
    Policy iamPolicy = getKeyRingPolicy(projectId, locationId, keyRingId);
    // Filter out the given member
    for (Binding b : iamPolicy.getBindings()) {
        if (role.equals(b.getRole()) && b.getMembers().contains(member)) {
            b.getMembers().remove(member);
            break;
        }
    }
    // Set the new IAM Policy.
    Policy newIamPolicy = kms.projects().locations().keyRings().setIamPolicy(cryptoKey, new SetIamPolicyRequest().setPolicy(iamPolicy)).execute();
    System.out.println("Response: " + newIamPolicy);
    return newIamPolicy;
}

Example 7 with Policy

use of com.google.api.services.iam.v1.model.Policy in project platinum by hartwigmedical.

the class PipelineIamPolicyTest method addsServiceAccountRolesToExistingPolicy.

@Test
public void addsServiceAccountRolesToExistingPolicy() throws Exception {
    CloudResourceManager resourceManager = mock(CloudResourceManager.class);
    CloudResourceManager.Projects projects = mock(CloudResourceManager.Projects.class);
    CloudResourceManager.Projects.GetIamPolicy getIamPolicy = mock(CloudResourceManager.Projects.GetIamPolicy.class);
    CloudResourceManager.Projects.SetIamPolicy setIamPolicy = mock(CloudResourceManager.Projects.SetIamPolicy.class);
    Policy policy = new Policy().setBindings(new ArrayList<>());
    when(resourceManager.projects()).thenReturn(projects);
    when(projects.getIamPolicy("projects/test", new GetIamPolicyRequest())).thenReturn(getIamPolicy);
    when(getIamPolicy.execute()).thenReturn(policy);
    ArgumentCaptor<String> projectArgumentCaptor = ArgumentCaptor.forClass(String.class);
    ArgumentCaptor<SetIamPolicyRequest> policyArgumentCaptor = ArgumentCaptor.forClass(SetIamPolicyRequest.class);
    when(projects.setIamPolicy(projectArgumentCaptor.capture(), policyArgumentCaptor.capture())).thenReturn(setIamPolicy);
    PipelineIamPolicy victim = new PipelineIamPolicy(resourceManager);
    victim.apply(new ServiceAccount().setEmail(EMAIL).setProjectId(PROJECT_ID));
    assertThat(projectArgumentCaptor.getValue()).isEqualTo(PROJECT_ID);
    assertThat(policyArgumentCaptor.getValue().getPolicy().getBindings()).contains(new Binding().setMembers(List.of(IDENTITY)).setRole(PipelineIamPolicy.COMPUTE_ADMIN), new Binding().setMembers(List.of(IDENTITY)).setRole(PipelineIamPolicy.STORAGE_ADMIN));
}

Example 8 with Policy

use of com.google.api.services.iam.v1.model.Policy in project terra-workspace-manager by DataBiosphere.

the class GrantPetUsagePermissionStep method doStep.

@Override
public StepResult doStep(FlightContext context) throws InterruptedException, RetryException {
    String userEmail = SamRethrow.onInterrupted(() -> samService.getUserEmailFromSam(userRequest), "enablePet");
    Policy modifiedPolicy;
    try {
        modifiedPolicy = petSaService.enablePetServiceAccountImpersonation(workspaceUuid, userEmail, userRequest);
    } catch (ConflictException e) {
        // There was a conflict enabling the service account. Request retry.
        return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
    }
    // Store the eTag value of the modified policy in case this step needs to be undone.
    FlightMap workingMap = context.getWorkingMap();
    workingMap.put(PetSaKeys.MODIFIED_PET_SA_POLICY_ETAG, modifiedPolicy.getEtag());
    return StepResult.getStepResultSuccess();
}

Example 9 with Policy

use of com.google.api.services.iam.v1.model.Policy in project terra-workspace-manager by DataBiosphere.

the class NotebookCloudSyncStep method doStep.

@Override
public StepResult doStep(FlightContext flightContext) throws InterruptedException, RetryException {
    FlightMap workingMap = flightContext.getWorkingMap();
    FlightUtils.validateRequiredEntries(workingMap, ControlledResourceKeys.GCP_CLOUD_CONTEXT);
    GcpCloudContext cloudContext = workingMap.get(ControlledResourceKeys.GCP_CLOUD_CONTEXT, GcpCloudContext.class);
    List<Binding> newBindings = createBindings(cloudContext, flightContext.getWorkingMap());
    AIPlatformNotebooksCow notebooks = crlService.getAIPlatformNotebooksCow();
    InstanceName instanceName = resource.toInstanceName(cloudContext.getGcpProjectId());
    try {
        Policy policy = notebooks.instances().getIamPolicy(instanceName).execute();
        // Duplicating bindings is harmless (e.g. on retry). GCP de-duplicates.
        Optional.ofNullable(policy.getBindings()).ifPresent(newBindings::addAll);
        policy.setBindings(newBindings);
        notebooks.instances().setIamPolicy(instanceName, new SetIamPolicyRequest().setPolicy(policy)).execute();
    } catch (IOException e) {
        return new StepResult(StepStatus.STEP_RESULT_FAILURE_RETRY, e);
    }
    return StepResult.getStepResultSuccess();
}

Example 10 with Policy

use of com.google.api.services.iam.v1.model.Policy in project terra-workspace-manager by DataBiosphere.

the class CreateGcpContextFlightV2Test method assertPolicyGroupsSynced.

/**
 * Asserts that Sam groups are granted their appropriate IAM roles on a GCP project.
 */
private void assertPolicyGroupsSynced(UUID workspaceUuid, Project project) throws Exception {
    Map<WsmIamRole, String> roleToSamGroup = Arrays.stream(WsmIamRole.values()).collect(Collectors.toMap(Function.identity(), role -> "group:" + SamRethrow.onInterrupted(() -> mockSamService.syncWorkspacePolicy(workspaceUuid, role, userAccessUtils.defaultUserAuthRequest()), "syncWorkspacePolicy")));
    Policy currentPolicy = crl.getCloudResourceManagerCow().projects().getIamPolicy(project.getProjectId(), new GetIamPolicyRequest()).execute();
    for (WsmIamRole role : WsmIamRole.values()) {
        // Don't check MANAGER role, which isn't synced to GCP.
        if (role.equals(WsmIamRole.MANAGER)) {
            continue;
        }
        assertRoleBindingInPolicy(role, roleToSamGroup.get(role), currentPolicy, project.getProjectId());
    }
}