io.trino.server.security.oauth2.TokenPairSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of trino-main Show documentation

Trino - Engine

There is a newer version: 465

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.trino.server.security.oauth2;

import io.trino.server.security.oauth2.OAuth2Client.Response;
import jakarta.annotation.Nullable;

import java.util.Date;
import java.util.Optional;

import static java.lang.Long.MAX_VALUE;
import static java.util.Objects.requireNonNull;

public interface TokenPairSerializer
{
    TokenPairSerializer ACCESS_TOKEN_ONLY_SERIALIZER = new TokenPairSerializer()
    {
        @Override
        public TokenPair deserialize(String token)
        {
            return TokenPair.withAccessToken(token);
        }

        @Override
        public String serialize(TokenPair tokenPair)
        {
            return tokenPair.accessToken();
        }
    };

    TokenPair deserialize(String token);

    String serialize(TokenPair tokenPair);

    record TokenPair(String accessToken, Date expiration, Optional refreshToken)
    {
        public TokenPair
        {
            requireNonNull(accessToken, "accessToken is nul");
            requireNonNull(expiration, "expiration is null");
            requireNonNull(refreshToken, "refreshToken is null");
        }

        public static TokenPair withAccessToken(String accessToken)
        {
            return new TokenPair(accessToken, new Date(MAX_VALUE), Optional.empty());
        }

        public static TokenPair fromOAuth2Response(Response tokens)
        {
            requireNonNull(tokens, "tokens is null");
            return new TokenPair(tokens.getAccessToken(), Date.from(tokens.getExpiration()), tokens.getRefreshToken());
        }

        public static TokenPair withAccessAndRefreshTokens(String accessToken, Date expiration, @Nullable String refreshToken)
        {
            return new TokenPair(accessToken, expiration, Optional.ofNullable(refreshToken));
        }
    }
}