yakworks.security.rest.okta.OktaClient.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rally-security Show documentation

integrates spring security with shiro for rest

There is a newer version: 7.3.27

/*
* Copyright 2022 Yak.Works - Licensed under the Apache License, Version 2.0 (the "License")
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*/
package yakworks.security.rest.okta

import groovy.transform.CompileStatic

import org.pac4j.core.profile.creator.ProfileCreator
import org.pac4j.core.util.CommonHelper
import org.pac4j.oidc.client.OidcClient
import org.pac4j.oidc.config.OidcConfiguration
import org.pac4j.oidc.credentials.OidcCredentials
import org.pac4j.oidc.profile.OidcProfileDefinition
import org.pac4j.oidc.profile.creator.OidcProfileCreator

/**
 * This is the client for Oauth/OpenId with Okta.
 * Spring security rest expects client class to have methods setClientId/setSecret etc so custom client class implementation is required.
*/
@CompileStatic
class OktaClient extends OidcClient{

    OktaClient() {
        super(new OidcConfiguration())
    }

    OktaClient(final OidcConfiguration configuration) {
        super(configuration);
    }

    void setClientId(final String id) {
        configuration.setClientId(id)
    }

    void setSecret(final String sc) {
        configuration.setSecret(sc)
    }

    void setDiscoveryUrl(final String u) {
        configuration.setDiscoveryURI(u)
    }

    // public void setCallbackUrl(final String url) {
    //     super.setCallbackUrl(url)
    // }

    @Override
    protected void clientInit() {
        CommonHelper.assertNotNull("configuration", getConfiguration())
        final OidcProfileCreator profileCreator = new OidcProfileCreator<>(getConfiguration())
        profileCreator.setProfileDefinition(new OidcProfileDefinition<>({ def x -> new OktaProfile()}))
        defaultProfileCreator(profileCreator as ProfileCreator)
        super.clientInit()
    }
}