// Copyright (c) 2017 The Authors of 'JWTS for Java'

//

// Permission is hereby granted, free of charge, to any person obtaining a copy of

// this software and associated documentation files (the "Software"), to deal in

// the Software without restriction, including without limitation the rights to

// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of

// the Software, and to permit persons to whom the Software is furnished to do so,

// subject to the following conditions:

//

// The above copyright notice and this permission notice shall be included in all

// copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS

// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR

// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER

// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN

// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package com.auth0.jwt.impl;

import com.auth0.jwt.exceptions.JWTDecodeException;

import com.auth0.jwt.interfaces.Header;

import com.auth0.jwt.interfaces.JWTPartsParser;

import com.auth0.jwt.interfaces.Payload;

import com.fasterxml.jackson.annotation.JsonInclude;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.fasterxml.jackson.databind.SerializationFeature;

import com.fasterxml.jackson.databind.module.SimpleModule;

import java.io.IOException;

public class JWTParser implements JWTPartsParser {

private ObjectMapper mapper;

public JWTParser() {

this(getDefaultObjectMapper());

}

JWTParser(ObjectMapper mapper) {

addDeserializers(mapper);

this.mapper = mapper;

}

@Override

public Payload parsePayload(String json) throws JWTDecodeException {

return convertFromJSON(json, Payload.class);

}

@Override

public Header parseHeader(String json) throws JWTDecodeException {

return convertFromJSON(json, Header.class);

}

private void addDeserializers(ObjectMapper mapper) {

SimpleModule module = new SimpleModule();

module.addDeserializer(Payload.class, new PayloadDeserializer());

module.addDeserializer(Header.class, new HeaderDeserializer());

mapper.registerModule(module);

}

static ObjectMapper getDefaultObjectMapper() {

ObjectMapper mapper = new ObjectMapper();

mapper.disable(SerializationFeature.FAIL_ON_EMPTY_BEANS);

mapper.setSerializationInclusion(JsonInclude.Include.NON_EMPTY);

return mapper;

}

@SuppressWarnings("WeakerAccess")

<T> T convertFromJSON(String json, Class<T> tClazz) throws JWTDecodeException {

if (json == null) {

throw exceptionForInvalidJson(null);

}

try {

return mapper.readValue(json, tClazz);

} catch (IOException e) {

throw exceptionForInvalidJson(json);

}

}

private JWTDecodeException exceptionForInvalidJson(String json) {

return new JWTDecodeException(String.format("The string '%s' doesn't have a valid JSON format.", json));

}

}