package org.gcube.common.keycloak;

import java.io.File;

import org.gcube.com.fasterxml.jackson.databind.ObjectMapper;
import org.gcube.common.keycloak.model.AccessToken;
import org.gcube.common.keycloak.model.ModelUtils;
import org.gcube.common.keycloak.model.RefreshToken;
import org.gcube.common.keycloak.model.TokenIntrospectionResponse;
import org.gcube.common.keycloak.model.TokenResponse;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.FixMethodOrder;
import org.junit.Test;
import org.junit.runners.MethodSorters;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@FixMethodOrder(MethodSorters.NAME_ASCENDING)
public class TestModels {

    protected static final Logger logger = LoggerFactory.getLogger(TestModels.class);

    @Before
    public void setUp() throws Exception {
    }

    @After
    public void tearDown() throws Exception {
    }

    @Test
    public void testTokenResponseForOIDC() throws Exception {
        logger.info("Start testing OIDC token response object binding...");
        TokenResponse tr = new ObjectMapper().readValue(new File("src/test/resources/oidc-token-response.json"),
                TokenResponse.class);

        logger.debug("OIDC token response:\n{}", ModelUtils.toJSONString(tr, true));
        checkTokenResponse(tr);
        
    }

    @Test
    public void testTokenResponseForUMA() throws Exception {
        logger.info("Start testing UMA token response object binding...");
        TokenResponse tr = new ObjectMapper().readValue(new File("src/test/resources/uma-token-response.json"),
                TokenResponse.class);

        logger.debug("UMA token response:\n{}", ModelUtils.toJSONString(tr, true));
        checkTokenResponse(tr);
    }

    @Test
    public void testUMAAccessToken() throws Exception {
        logger.info("Start testing access token object binding...");
        AccessToken at = new ObjectMapper().readValue(new File("src/test/resources/uma-access-token.json"),
                AccessToken.class);

        checkAccessToken(at, null, true);
    }

    @Test
    public void testRemoveBearerPrefixInHeader() throws Exception {
        TokenResponse tr = new ObjectMapper().readValue(new File("src/test/resources/oidc-token-response.json"),
                TokenResponse.class);

        AccessToken at1 = ModelUtils.getAccessTokenFrom(tr.getAccessToken());
        AccessToken at2 = ModelUtils.getAccessTokenFrom("Bearer " + tr.getAccessToken());
        AccessToken at3 = ModelUtils.getAccessTokenFrom("bearer " + tr.getAccessToken());

        checkAccessToken(at1, null, true);
        checkAccessToken(at2, null, true);
        checkAccessToken(at3, null, true);
        Assert.assertEquals(ModelUtils.toJSONString(at1), ModelUtils.toJSONString(at2));
        Assert.assertEquals(ModelUtils.toJSONString(at2), ModelUtils.toJSONString(at3));
    }

    @Test
    public void testUMARefreshToken() throws Exception {
        logger.info("Start testing refresh token object binding...");
        RefreshToken rt = new ObjectMapper().readValue(new File("src/test/resources/uma-refresh-token.json"),
                RefreshToken.class);

        checkRefreshToken(rt);
    }

    public static void checkTokenResponse(TokenResponse tr) throws Exception {
        Assert.assertNotNull(tr);
        Assert.assertEquals("bearer", tr.getTokenType().toLowerCase());
        Assert.assertNotNull(tr.getAccessToken());
        Assert.assertNotNull(tr.getRefreshToken());
    }

    public static void checkAccessToken(AccessToken at, String preferredUsername, boolean checkAudience) {
        logger.debug("Access token:\n{}", ModelUtils.toJSONString(at, true));
        Assert.assertNotNull(at.getPreferredUsername());
        if (preferredUsername != null) {
            Assert.assertEquals(preferredUsername, at.getPreferredUsername());
        }
        if (checkAudience) {
            Assert.assertNotNull(at.getAudience());
        }
    }

    public static void checkRefreshToken(RefreshToken rt) {
        logger.debug("Refresh token:\n{}", ModelUtils.toJSONString(rt, true));
        Assert.assertNotNull(rt.getOtherClaims());
        Assert.assertNotNull(rt.getAudience());
    }

    public static void checkTokenIntrospectionResponse(TokenIntrospectionResponse tir) {
        logger.debug("Token introspection response :\n{}", ModelUtils.toJSONString(tir, true));
        Assert.assertTrue(tir.isActive());
    }

}