org.apereo.cas.config.CasAmazonS3BucketsCloudConfigBootstrapAutoConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cas-server-support-configuration-cloud-aws-s3 Show documentation
cas-server-support-configuration-cloud-aws-s3
There is a newer version: 7.2.0-RC2
Show newest version
package org.apereo.cas.config;

import org.apereo.cas.aws.AmazonEnvironmentAwareClientBuilder;
import org.apereo.cas.configuration.CasCoreConfigurationUtils;
import org.apereo.cas.configuration.features.CasFeatureModule;
import org.apereo.cas.util.LoggingUtils;
import org.apereo.cas.util.spring.boot.ConditionalOnFeatureEnabled;

import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import lombok.val;
import org.springframework.boot.autoconfigure.AutoConfiguration;
import org.springframework.cloud.bootstrap.config.PropertySourceLocator;
import org.springframework.core.env.Environment;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.InputStreamResource;
import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.S3Configuration;
import software.amazon.awssdk.services.s3.model.GetObjectRequest;
import software.amazon.awssdk.services.s3.model.ListObjectsV2Request;

import java.util.LinkedHashMap;
import java.util.Properties;

/**
 * This is {@link CasAmazonS3BucketsCloudConfigBootstrapAutoConfiguration}.
 *
 * @author Misagh Moayyed
 * @since 5.3.0
 */
@Slf4j
@Getter
@ConditionalOnFeatureEnabled(feature = CasFeatureModule.FeatureCatalog.CasConfiguration, module = "aws-s3")
@AutoConfiguration
public class CasAmazonS3BucketsCloudConfigBootstrapAutoConfiguration implements PropertySourceLocator {
    /**
     * Amazon S3 CAS configuration key prefix.
     */
    public static final String CAS_CONFIGURATION_PREFIX = "cas.spring.cloud.aws.s3";

    @Override
    public PropertySource locate(final Environment environment) {
        val properties = new LinkedHashMap();
        val builder = new AmazonEnvironmentAwareClientBuilder(CAS_CONFIGURATION_PREFIX, environment);
        val clientBuilder = S3Client.builder().serviceConfiguration(S3Configuration.Builder::pathStyleAccessEnabled).forcePathStyle(true);
        try (val s3Client = builder.build(clientBuilder, S3Client.class)) {
            val bucketName = builder.getSetting("bucket-name", "cas-properties");
            LOGGER.debug("Locating S3 object(s) from bucket [{}]...", bucketName);
            val result = s3Client.listObjectsV2(ListObjectsV2Request.builder().bucket(bucketName).build());
            val objects = result.contents();
            LOGGER.debug("Located [{}] S3 object(s) from bucket [{}]", objects.size(), bucketName);

            objects.forEach(obj -> {
                val objectKey = obj.key();
                LOGGER.debug("Fetching object [{}] from bucket [{}]", objectKey, bucketName);
                try (val is = s3Client.getObject(GetObjectRequest.builder().bucket(bucketName).key(objectKey).build())) {
                    if (objectKey.endsWith("properties")) {
                        val props = new Properties();
                        props.load(is);
                        props.forEach((key, value) -> properties.put(key.toString(), value));
                    } else if (objectKey.endsWith("yml") || objectKey.endsWith("yaml")) {
                        val yamlProps = CasCoreConfigurationUtils.loadYamlProperties(new InputStreamResource(is));
                        properties.putAll(yamlProps);
                    }
                } catch (final Exception e) {
                    LoggingUtils.error(LOGGER, e);
                }
            });
        } catch (final Exception e) {
            LoggingUtils.error(LOGGER, e);
        }
        return new MapPropertySource(getClass().getSimpleName(), properties);
    }
}