package org.jboss.resteasy.core.interception.jaxrs;

import java.io.OutputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.Collection;
import java.util.Map;

import jakarta.ws.rs.ProcessingException;
import jakarta.ws.rs.core.MediaType;
import jakarta.ws.rs.core.MultivaluedMap;
import jakarta.ws.rs.ext.MessageBodyWriter;
import jakarta.ws.rs.ext.WriterInterceptor;

import org.jboss.resteasy.resteasy_jaxrs.i18n.Messages;
import org.jboss.resteasy.spi.ResteasyProviderFactory;
import org.jboss.resteasy.tracing.RESTEasyTracingLogger;

/**
 * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
 * @version $Revision: 1 $
 */
public class ClientWriterInterceptorContext extends AbstractWriterInterceptorContext {
    protected Map<String, Object> properties;

    // use the tracingLogger version instead.
    @Deprecated
    public ClientWriterInterceptorContext(final WriterInterceptor[] interceptors, final ResteasyProviderFactory providerFactory,
            final Object entity, final Class type, final Type genericType, final Annotation[] annotations,
            final MediaType mediaType, final MultivaluedMap<String, Object> headers,
            final OutputStream outputStream, final Map<String, Object> properties) {
        super(interceptors, annotations, entity, genericType, mediaType, type, outputStream, providerFactory, headers);
        this.properties = properties;
    }

    public ClientWriterInterceptorContext(final WriterInterceptor[] interceptors, final ResteasyProviderFactory providerFactory,
            final Object entity, final Class type, final Type genericType, final Annotation[] annotations,
            final MediaType mediaType, final MultivaluedMap<String, Object> headers,
            final OutputStream outputStream, final Map<String, Object> properties, final RESTEasyTracingLogger tracingLogger) {
        //      super(interceptors, annotations, entity, genericType, mediaType, type, outputStream, providerFactory, headers);
        super(interceptors, annotations, entity, genericType, mediaType, type, outputStream, providerFactory, headers,
                tracingLogger);
        this.properties = properties;
    }

    @Override
    void throwWriterNotFoundException() {
        throw new ProcessingException(Messages.MESSAGES.couldNotFindWriterForContentType(mediaType, type.getName()));
    }

    @Override
    protected MessageBodyWriter resolveWriter() {
        @SuppressWarnings(value = "unchecked")
        MessageBodyWriter writer = providerFactory.getClientMessageBodyWriter(
                type, genericType, annotations, mediaType);
        //logger.info("********* WRITER: " + writer.getClass().getName());
        return writer;
    }

    @Override
    public Object getProperty(String name) {
        return properties.get(name);
    }

    @Override
    public Collection<String> getPropertyNames() {
        return properties.keySet();
    }

    @Override
    public void setProperty(String name, Object object) {
        if (object == null) {
            properties.remove(name);
        } else {
            properties.put(name, object);
        }
    }

    @Override
    public void removeProperty(String name) {
        properties.remove(name);
    }
}
