com.amazonaws.services.ecr.AmazonECR Java Examples

@Override
protected List<Map<String, String>> run() throws Exception {
	AmazonECR ecr = AWSClientFactory.create(AmazonECRClientBuilder.standard(), this.getContext());

	ListImagesRequest request = new ListImagesRequest()
			.withRegistryId(this.step.getRegistryId())
			.withRepositoryName(this.step.getRepositoryName())
			.withFilter(this.step.getFilter());
	List<ImageIdentifier> images = new LinkedList<>();
	ListImagesResult result;
	do {
		result = ecr.listImages(request);
		images.addAll(result.getImageIds());
		request.setNextToken(result.getNextToken());
	} while (result.getNextToken() != null);
	return images.stream().map(image -> new HashMap<String, String>() {
		{
			put("imageTag", image.getImageTag());
			put("imageDigest", image.getImageDigest());
		}
	}).collect(Collectors.toList());
}
 

@Override
protected List<ImageIdentifier> run() throws Exception {
	AmazonECR ecr = AWSClientFactory.create(AmazonECRClientBuilder.standard(), this.getContext());

	BatchDeleteImageResult result = ecr.batchDeleteImage(new BatchDeleteImageRequest()
			.withImageIds(new ArrayList<>(this.step.getImageIds()))
			.withRegistryId(this.step.getRegistryId())
			.withRepositoryName(this.step.getRepositoryName())
	);
	if (!result.getFailures().isEmpty()) {
		TaskListener listener = this.getContext().get(TaskListener.class);
		listener.error("Unable to delete images:");
		for (ImageFailure failure : result.getFailures()) {
			listener.error("%s %s %s", failure.getFailureCode(), failure.getFailureReason(), failure.getImageId());
		}
	}

	return result.getImageIds();
}
 

@Override
protected String run() throws Exception {
	AmazonECR ecr = AWSClientFactory.create(AmazonECRClientBuilder.standard(), this.getContext());

	GetAuthorizationTokenRequest request = new GetAuthorizationTokenRequest();
	List<String> registryIds;
	if((registryIds = this.step.getRegistryIds()) != null) {
		request.setRegistryIds(registryIds);
	}
	GetAuthorizationTokenResult token = ecr.getAuthorizationToken(request);

	if (token.getAuthorizationData().size() != 1) {
		throw new RuntimeException("Did not get authorizationData from AWS");
	}

	AuthorizationData authorizationData = token.getAuthorizationData().get(0);
	byte[] bytes = org.apache.commons.codec.binary.Base64.decodeBase64(authorizationData.getAuthorizationToken());
	String data = new String(bytes, Charsets.UTF_8);
	String[] parts = data.split(":");
	if (parts.length != 2) {
		throw new RuntimeException("Got invalid authorizationData from AWS");
	}

	String emailString = this.step.getEmail() ? "-e none" : "";
	return String.format("docker login -u %s -p %s %s %s", parts[0], parts[1], emailString, authorizationData.getProxyEndpoint());
}
 

@Override
public AwsToken load(AwsCredentials awsCredentials) throws Exception {
    AmazonECR amazonECR = new AmazonECRClient(new AWSCredentialsProvider() {
        @Override
        public AWSCredentials getCredentials() {
            return awsCredentials;
        }

        @Override
        public void refresh() {
        }
    });
    amazonECR.setRegion(RegionUtils.getRegion(awsCredentials.getRegion()));
    GetAuthorizationTokenResult authorizationToken = amazonECR.getAuthorizationToken(new GetAuthorizationTokenRequest());
    List<AuthorizationData> authorizationData = authorizationToken.getAuthorizationData();
    Assert.isTrue(!CollectionUtils.isEmpty(authorizationData), "authorizationData is null or empty for token " + authorizationToken);
    AuthorizationData data = authorizationData.get(0);
    byte[] decode = Base64.getDecoder().decode(data.getAuthorizationToken());
    String token = new String(decode);
    String[] split = token.split(":");
    log.info("about to connect to AWS endpoint: {}", data.getProxyEndpoint());
    return AwsToken.builder().username(split[0]).password(split[1])
            .expiresAt(data.getExpiresAt()).proxyEndpoint(data.getProxyEndpoint()).build();
}
 

@Override
protected SetRepositoryPolicyResult run() throws Exception {
	AmazonECR ecr = AWSClientFactory.create(AmazonECRClientBuilder.standard(), this.getContext());

	SetRepositoryPolicyRequest request = new SetRepositoryPolicyRequest()
			.withRegistryId(this.step.getRegistryId())
			.withRepositoryName(this.step.getRepositoryName())
			.withPolicyText(this.step.getPolicyText());
	// https://docs.aws.amazon.com/AWSJavaSDK/latest/javadoc/com/amazonaws/services/ecr/model/SetRepositoryPolicyResult.html
	SetRepositoryPolicyResult result = ecr.setRepositoryPolicy(request);
	return result;
}
 

@Before
public void setupSdk() throws Exception {
	this.ecr = Mockito.mock(AmazonECR.class);
	PowerMockito.mockStatic(AWSClientFactory.class);
	PowerMockito.when(AWSClientFactory.create(Mockito.any(AwsSyncClientBuilder.class), Mockito.any(StepContext.class)))
			.thenReturn(ecr);
}
 

@Before
public void setupSdk() throws Exception {
	this.ecr = Mockito.mock(AmazonECR.class);
	PowerMockito.mockStatic(AWSClientFactory.class);
	PowerMockito.when(AWSClientFactory.create(Mockito.any(AwsSyncClientBuilder.class), Mockito.any(StepContext.class)))
			.thenReturn(ecr);
}
 

@Before
public void setupSdk() throws Exception {
	this.ecr = Mockito.mock(AmazonECR.class);
	PowerMockito.mockStatic(AWSClientFactory.class);
	PowerMockito.when(AWSClientFactory.create(Mockito.any(AwsSyncClientBuilder.class), Mockito.any(StepContext.class)))
			.thenReturn(ecr);
}
 

@Override
public HttpHeaders getAuthorizationHeaders(ContainerImage containerImage,
		RegistryConfiguration registryConfiguration) {

	Assert.isTrue(registryConfiguration.getAuthorizationType() == this.getType(),
			"Incorrect type: " + registryConfiguration.getAuthorizationType());

	AmazonECRClientBuilder ecrBuilder = AmazonECRClientBuilder.standard();
	if (registryConfiguration.getExtra().containsKey(AWS_REGION)) {
		ecrBuilder.withRegion(registryConfiguration.getExtra().get(AWS_REGION));
	}
	if (StringUtils.hasText(registryConfiguration.getUser()) && StringUtils.hasText(registryConfiguration.getSecret())) {
		// Expects that the 'user' == 'Access Key ID' and 'secret' == 'Secret Access Key'
		ecrBuilder.withCredentials(new AWSStaticCredentialsProvider(
				new BasicAWSCredentials(registryConfiguration.getUser(), registryConfiguration.getSecret())));
	}

	AmazonECR client = ecrBuilder.build();

	GetAuthorizationTokenRequest request = new GetAuthorizationTokenRequest();
	if (registryConfiguration.getExtra().containsKey(REGISTRY_IDS)) {
		request.withRegistryIds(registryConfiguration.getExtra().get(REGISTRY_IDS).split(","));
	}
	GetAuthorizationTokenResult response = client.getAuthorizationToken(request);
	String token = response.getAuthorizationData().iterator().next().getAuthorizationToken();
	final HttpHeaders headers = new HttpHeaders();
	headers.setBasicAuth(token);
	return headers;
}
 

AWSTokenRequestGenerator(AwsSyncClientBuilder<AmazonECRClientBuilder, AmazonECR> builder) {
    this.builder = builder;
}
 

AwsSyncClientBuilder<AmazonECRClientBuilder, AmazonECR> getBuilder() {
    return builder;
}
 

@Override
protected AmazonECR build(AwsSyncClientParams clientParams) {
    return mockAmazonEcrClient;
}