/*
    * Copyright 2015 Data Archiving and Networked Services (an institute of
    * Koninklijke Nederlandse Akademie van Wetenschappen), King's College London,
    * Georg-August-Universitaet Goettingen Stiftung Oeffentlichen Rechts
    *
    * Licensed under the EUPL, Version 1.1 or – as soon they will be approved by
    * the European Commission - subsequent versions of the EUPL (the "Licence");
    * You may not use this work except in compliance with the Licence.
    * You may obtain a copy of the Licence at:
   *
   * https://joinup.ec.europa.eu/software/page/eupl
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the Licence is distributed on an "AS IS" basis,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the Licence for the specific language governing
   * permissions and limitations under the Licence.
   */
  
  package eu.ehri.extension.providers;
  
  import eu.ehri.extension.errors.WebDeserializationError;
  import eu.ehri.project.exceptions.DeserializationError;
  import eu.ehri.project.exceptions.SerializationError;
  import eu.ehri.project.persistence.Bundle;
  
  import javax.ws.rs.Consumes;
  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.MessageBodyReader;
  import javax.ws.rs.ext.MessageBodyWriter;
  import javax.ws.rs.ext.Provider;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.OutputStream;
  import java.lang.annotation.Annotation;
  import java.lang.reflect.Type;
  
  
  @Provider
  @Consumes(MediaType.APPLICATION_JSON)
  @Produces(MediaType.APPLICATION_JSON)
  public class BundleProvider implements MessageBodyReader<Bundle>, MessageBodyWriter<Bundle> {
      @Override
      public boolean isReadable(Class<?> aClass, Type type, Annotation[] annotations, MediaType mediaType) {
          return aClass == Bundle.class;
      }
  
      @Override
      public Bundle readFrom(Class<Bundle> bundleClass, Type type, Annotation[] annotations,
                             MediaType mediaType, MultivaluedMap<String,
              String> headers, InputStream stream) throws IOException, WebApplicationException {
          try {
              return Bundle.fromStream(stream);
          } catch (DeserializationError deserializationError) {
              throw new WebDeserializationError(deserializationError);
          }
      }
  
      @Override
      public boolean isWriteable(Class<?> aClass, Type type, Annotation[] annotations, MediaType mediaType) {
          return aClass == Bundle.class;
      }
  
      @Override
      public long getSize(Bundle bundle, Class<?> aClass, Type type, Annotation[] annotations, MediaType mediaType) {
          return -1;
      }
  
      @Override
      public void writeTo(Bundle bundle, Class<?> aClass, Type type, Annotation[] annotations, MediaType mediaType, MultivaluedMap<String, Object> multivaluedMap, OutputStream outputStream) throws IOException, WebApplicationException {
          try {
              Bundle.toStream(bundle, outputStream);
          } catch (SerializationError serializationError) {
              throw new WebApplicationException(serializationError);
          }
      }
  }