package svanimpe.reminders.json; import java.io.IOException; import java.io.OutputStream; import java.lang.annotation.Annotation; import java.lang.reflect.Type; import javax.json.Json; import javax.json.JsonArrayBuilder; import javax.json.JsonObjectBuilder; import javax.json.JsonWriter; import javax.ws.rs.Produces; import javax.ws.rs.WebApplicationException; import javax.ws.rs.core.MediaType; import javax.ws.rs.core.MultivaluedMap; import javax.ws.rs.ext.MessageBodyWriter; import javax.ws.rs.ext.Provider; import svanimpe.reminders.domain.Role; import svanimpe.reminders.domain.User; @Provider @Produces(MediaType.APPLICATION_JSON) public class UserWriter implements MessageBodyWriter<User> { @Override public boolean isWriteable(Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType) { return User.class.isAssignableFrom(type); } @Override public long getSize(User user, Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType) { return -1; } @Override public void writeTo(User user, Class<?> type, Type genericType, Annotation[] annotations, MediaType mediaType, MultivaluedMap<String, Object> httpHeaders, OutputStream entityStream) throws IOException, WebApplicationException { JsonObjectBuilder jsonUser = Json.createObjectBuilder(); jsonUser.add("username", user.getUsername()); if (user.getFullName() != null) { jsonUser.add("fullName", user.getFullName()); } JsonArrayBuilder roles = Json.createArrayBuilder(); for (Role role : user.getRoles()) { roles.add(role.name()); } jsonUser.add("roles", roles); try (JsonWriter writer = Json.createWriter(entityStream)) { writer.writeObject(jsonUser.build()); } } }