no.nav.common.utils.NaisUtils Maven / Gradle / Ivy

Go to download
package no.nav.common.utils;

import lombok.SneakyThrows;

import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import static java.lang.String.format;

public class NaisUtils {

    public static String SECRETS_BASE_PATH_PROPERTY_NAME = "SECRETS_BASE_PATH";
    private static String DEFAULT_SECRETS_BASE_PATH = "/var/run/secrets/nais.io";
    private static String DEFAULT_CREDENTIALS_USERNAME_FILE = "username";
    private static String DEFAULT_CREDENTIALS_PASSWORD_FILE = "password";

    public static String CONFIG_MAPS_BASE_PATH_PROPERTY_NAME = "CONFIG_BASE_MAPS_PATH";
    private static String DEFAULT_CONFIG_MAPS_PATH = "/var/run/configmaps";

    public static Credentials getCredentials(String secretName,
                                             String usernameFileName,
                                             String passwordFileName) {
        String credentialsBasePath = EnvironmentUtils.getOptionalProperty(SECRETS_BASE_PATH_PROPERTY_NAME).orElse(DEFAULT_SECRETS_BASE_PATH);
        Path path = Paths.get(credentialsBasePath, secretName);
        String username = getFileContent(path.resolve(usernameFileName));
        String password = getFileContent(path.resolve(passwordFileName));

        return new Credentials(username, password);
    }

    public static Credentials getCredentials(String secretName) {
        return getCredentials(secretName, DEFAULT_CREDENTIALS_USERNAME_FILE, DEFAULT_CREDENTIALS_PASSWORD_FILE);
    }

    public static String getFileContent(String path) {
        return getFileContent(Paths.get(path));
    }


    public static String getFileContent(Path path) {
        return Optional.of(path)
                .filter(Files::isRegularFile)
                .map(NaisUtils::readAllLines)
                .flatMap(lines ->
                        lines.stream().reduce((a, b) -> a + "\n" + b))
                .orElseThrow(() -> new IllegalStateException(format("Fant ikke fil %s", path.toString())));
    }

    @SneakyThrows
    private static List readAllLines(Path path) {
        return Files.readAllLines(path);
    }

    public static void addConfigMapToEnv(String configMap) {
        addMapToEnv(readConfigMap(configMap));
    }

    public static void addConfigMapToEnv(String configMap, String... keys) {
        addMapToEnv(readConfigMap(configMap, keys));
    }

    private static void addMapToEnv(Map map) {
        map.forEach((key, value) -> EnvironmentUtils.setProperty(key, value, EnvironmentUtils.Type.PUBLIC));
    }

    @SneakyThrows
    public static Map readConfigMap(String configMap) {
        String configMapsPath = EnvironmentUtils.getOptionalProperty(CONFIG_MAPS_BASE_PATH_PROPERTY_NAME).orElse(DEFAULT_CONFIG_MAPS_PATH);
        Path path = Paths.get(configMapsPath, configMap);
        Stream files = Files.walk(path, 1).filter(Files::isRegularFile);

        return files.collect(Collectors.toMap(file -> file.getFileName().toString(), NaisUtils::getFileContent));
    }

    public static Map readConfigMap(String configMap, String... keys) {
        Map configMapMap = readConfigMap(configMap);

        return Arrays.stream(keys).map(key ->
                Optional.ofNullable(configMapMap.get(key))
                        .map(value -> Pair.of(key, value))
                        .orElseThrow(() ->
                                new IllegalStateException(String.format("Fant ikke key %s i config map %s", key, configMap))))
                .collect(Collectors.toMap(Pair::getFirst, Pair::getSecond));
    }
}