com.microkubes.tools.security.spring.JwtOauth2Config Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tools Show documentation

Java integration library for Microkubes platform.

There is a newer version: 0.2.0.RELEASE

package com.microkubes.tools.security.spring;

import org.bouncycastle.util.io.pem.PemObject;
import org.bouncycastle.util.io.pem.PemReader;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.oauth2.provider.token.TokenStore;
import org.springframework.security.oauth2.provider.token.store.JwtAccessTokenConverter;
import org.springframework.security.oauth2.provider.token.store.JwtTokenStore;

import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.security.KeyFactory;
import java.security.KeyPair;
import java.security.NoSuchAlgorithmException;
import java.security.spec.InvalidKeySpecException;
import java.security.spec.KeySpec;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;


@Configuration
@ConditionalOnProperty(value = {"com.microkubes.security.oauth2_jwt"})
public class JwtOauth2Config {

    @Value("${com.microkubes.security.private_key.path}")
    private String privateKeyPath;

    @Value("${com.microkubes.security.public_key.path}")
    private String publicKeyPath;

    @Bean
    public TokenStore getTokenStore() throws IOException, NoSuchAlgorithmException, InvalidKeySpecException {
        JwtAccessTokenConverter jwtAccessTokenConverter = new JwtAccessTokenConverter();
        jwtAccessTokenConverter.setAccessTokenConverter(new CustomClaimsTokenConverter());
        jwtAccessTokenConverter.setKeyPair(loadKeyPair());
        return new JwtTokenStore(jwtAccessTokenConverter);
    }

    public KeyPair loadKeyPair() throws IOException, NoSuchAlgorithmException, InvalidKeySpecException {
        KeyFactory kf = KeyFactory.getInstance("RSA");
        return  new KeyPair(kf.generatePublic(readPublicKey()), kf.generatePrivate(readPrivateKey()));
    }


    private KeySpec readPrivateKey() throws IOException{
        PemReader reader = new PemReader(new FileReader(privateKeyPath));
        PKCS8EncodedKeySpec privateKey = new PKCS8EncodedKeySpec(reader.readPemObject().getContent());
        return  privateKey;
    }

    private KeySpec readPublicKey() throws IOException {
        PemReader reader = new PemReader(new FileReader(publicKeyPath));
        X509EncodedKeySpec pubKeySpec = new X509EncodedKeySpec(reader.readPemObject().getContent());
        return pubKeySpec;
    }

}