org.jboss.resteasy.jwt.JsonSerialization Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jose-jwt Show documentation
The newest version!
package org.jboss.resteasy.jwt;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;

import jakarta.ws.rs.core.MediaType;
import jakarta.ws.rs.core.MultivaluedHashMap;
import jakarta.ws.rs.ext.MessageBodyReader;
import jakarta.ws.rs.ext.MessageBodyWriter;
import jakarta.ws.rs.ext.Providers;

import org.jboss.resteasy.core.ResteasyContext;
import org.jboss.resteasy.core.providerfactory.ResteasyProviderFactoryImpl;
import org.jboss.resteasy.jose.i18n.Messages;
import org.jboss.resteasy.plugins.providers.jackson.ResteasyJackson2Provider;
import org.jboss.resteasy.spi.ResteasyProviderFactory;

/**
 * @author Bill Burke
 * @version $Revision: 1 $
 */
public class JsonSerialization {
    public static byte[] toByteArray(Object token, boolean indent) throws Exception {
        ResteasyProviderFactory factory = new ResteasyProviderFactoryImpl();
        factory.register(new JWTContextResolver(indent));
        factory.register(ResteasyJackson2Provider.class);

        return toByteArray(token, factory);

    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public static byte[] toByteArray(Object token, ResteasyProviderFactory factory) throws IOException {
        MessageBodyWriter writer = factory.getMessageBodyWriter(token.getClass(), null, null, MediaType.APPLICATION_JSON_TYPE);
        if (writer == null)
            throw new NullPointerException(Messages.MESSAGES.couldNotFindMessageBodyWriterForJSON());
        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        Providers old = ResteasyContext.getContextData(Providers.class);
        ResteasyContext.pushContext(Providers.class, factory);
        try {
            writer.writeTo(token, token.getClass(), null, null, MediaType.APPLICATION_JSON_TYPE,
                    new MultivaluedHashMap(), baos);

            return baos.toByteArray();
        } finally {
            ResteasyContext.popContextData(Providers.class);
            if (old != null)
                ResteasyContext.pushContext(Providers.class, old);
        }
    }

    public static String toString(Object token, ResteasyProviderFactory factory) throws Exception {
        byte[] bytes = toByteArray(token, factory);
        return new String(bytes);
    }

    public static String toString(Object token, boolean indent) throws Exception {
        byte[] bytes = toByteArray(token, indent);
        return new String(bytes);
    }

    public static  T fromString(Class type, String json) throws Exception {
        byte[] bytes = json.getBytes(StandardCharsets.UTF_8);
        return fromBytes(type, bytes);

    }

    public static  T fromString(Class type, String json, ResteasyProviderFactory factory) throws Exception {
        byte[] bytes = json.getBytes(StandardCharsets.UTF_8);
        return fromBytes(type, bytes, factory);

    }

    public static  T fromBytes(Class type, byte[] bytes) throws IOException {
        ResteasyProviderFactory factory = new ResteasyProviderFactoryImpl();
        factory.register(ResteasyJackson2Provider.class);
        factory.register(JWTContextResolver.class);

        return fromBytes(type, bytes, factory);
    }

    public static  T fromBytes(Class type, byte[] bytes, ResteasyProviderFactory factory) throws IOException {
        MessageBodyReader reader = factory.getMessageBodyReader(type, type, null, MediaType.APPLICATION_JSON_TYPE);
        if (reader == null)
            throw new NullPointerException(Messages.MESSAGES.couldNotFindMessageBodyReaderForJSON());
        ByteArrayInputStream bais = new ByteArrayInputStream(bytes);

        Providers old = ResteasyContext.getContextData(Providers.class);
        ResteasyContext.pushContext(Providers.class, factory);
        try {
            return reader.readFrom(type, type, null, MediaType.APPLICATION_JSON_TYPE, new MultivaluedHashMap(),
                    bais);
        } finally {
            ResteasyContext.popContextData(Providers.class);
            if (old != null)
                ResteasyContext.pushContext(Providers.class, old);
        }
    }
}