de.adorsys.sts.token.JwtClaimSetHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sts-token Show documentation

The newest version!

package de.adorsys.sts.token;

import com.nimbusds.jwt.JWTClaimsSet;
import de.adorsys.sts.common.ObjectMapperSPI;
import de.adorsys.sts.resourceserver.model.ResourceServerAndSecret;
import org.apache.commons.lang3.StringUtils;

import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class JwtClaimSetHelper {

    public static JWTClaimsSet.Builder handleResources(
            JWTClaimsSet.Builder claimSetBuilder,
            List processedResources,
            ObjectMapperSPI mapper
    ) {

        if (processedResources.isEmpty()) return claimSetBuilder;
        Map map = new HashMap<>();
        for (ResourceServerAndSecret rs : processedResources) {
            String userSecretClaimName = rs.getResourceServer().getUserSecretClaimName();
            if (StringUtils.isBlank(userSecretClaimName)) userSecretClaimName = rs.getResourceServer().getAudience();
            if (StringUtils.isNotBlank(userSecretClaimName))
                map.put(userSecretClaimName, rs.getEncryptedSecret());
        }

        try {
            return claimSetBuilder.claim("user-secret", mapper.writeValueAsString(map));
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}