com.auth0.jwt.impl.ClaimsSerializer Maven / Gradle / Ivy

Go to download
package com.auth0.jwt.impl;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;

import java.io.IOException;
import java.time.Instant;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * Custom serializer used to write the resulting JWT.
 *
 * @param  the type this serializer operates on.
 */
public class ClaimsSerializer extends StdSerializer {

    public ClaimsSerializer(Class t) {
        super(t);
    }

    @Override
    public void serialize(T holder, JsonGenerator gen, SerializerProvider provider) throws IOException {
        gen.writeStartObject();
        for (Map.Entry entry : holder.getClaims().entrySet()) {
            writeClaim(entry, gen);
        }
        gen.writeEndObject();
    }

    /**
     * Writes the given entry to the JSON representation. Custom claim serialization handling can override this method
     * to provide use-case specific serialization. Implementors who override this method must write
     * the field name and the field value.
     *
     * @param entry The entry that corresponds to the JSON field to write
     * @param gen The {@code JsonGenerator} to use
     * @throws IOException if there is either an underlying I/O problem or encoding issue at format layer
     */
    protected void writeClaim(Map.Entry entry, JsonGenerator gen) throws IOException {
        gen.writeFieldName(entry.getKey());
        handleSerialization(entry.getValue(), gen);
    }

    private static void handleSerialization(Object value, JsonGenerator gen) throws IOException {
        if (value instanceof Date) {
            gen.writeNumber(dateToSeconds((Date) value));
        } else if (value instanceof Instant) { // EXPIRES_AT, ISSUED_AT, NOT_BEFORE, custom Instant claims
            gen.writeNumber(instantToSeconds((Instant) value));
        } else if (value instanceof Map) {
            serializeMap((Map) value, gen);
        } else if (value instanceof List) {
            serializeList((List) value, gen);
        } else {
            gen.writeObject(value);
        }
    }

    private static void serializeMap(Map map, JsonGenerator gen) throws IOException {
        gen.writeStartObject();
        for (Map.Entry entry : map.entrySet()) {
            gen.writeFieldName((String) entry.getKey());
            Object value = entry.getValue();
            handleSerialization(value, gen);
        }
        gen.writeEndObject();
    }

    private static void serializeList(List list, JsonGenerator gen) throws IOException {
        gen.writeStartArray();
        for (Object entry : list) {
            handleSerialization(entry, gen);
        }
        gen.writeEndArray();
    }

    private static long instantToSeconds(Instant instant) {
        return instant.getEpochSecond();
    }

    private static long dateToSeconds(Date date) {
        return date.getTime() / 1000;
    }
}