/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.eclipse.aether.transport.jetty;

import java.io.EOFException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.file.Files;
import java.nio.file.StandardOpenOption;

import org.eclipse.aether.spi.connector.transport.PutTask;
import org.eclipse.jetty.client.util.AbstractRequestContent;
import org.eclipse.jetty.io.ByteBufferPool;
import org.eclipse.jetty.util.BufferUtil;
import org.eclipse.jetty.util.Callback;
import org.eclipse.jetty.util.IO;

class PutTaskRequestContent extends AbstractRequestContent {
    private final PutTask putTask;
    private final int bufferSize;
    private ByteBufferPool bufferPool;
    private boolean useDirectByteBuffers = true;

    PutTaskRequestContent(PutTask putTask) {
        this(putTask, 4096);
    }

    PutTaskRequestContent(PutTask putTask, int bufferSize) {
        super("application/octet-stream");
        this.putTask = putTask;
        this.bufferSize = bufferSize;
    }

    @Override
    public long getLength() {
        return putTask.getDataLength();
    }

    @Override
    public boolean isReproducible() {
        return true;
    }

    public ByteBufferPool getByteBufferPool() {
        return bufferPool;
    }

    public void setByteBufferPool(ByteBufferPool byteBufferPool) {
        this.bufferPool = byteBufferPool;
    }

    public boolean isUseDirectByteBuffers() {
        return useDirectByteBuffers;
    }

    public void setUseDirectByteBuffers(boolean useDirectByteBuffers) {
        this.useDirectByteBuffers = useDirectByteBuffers;
    }

    @Override
    protected Subscription newSubscription(Consumer consumer, boolean emitInitialContent) {
        return new SubscriptionImpl(consumer, emitInitialContent);
    }

    private class SubscriptionImpl extends AbstractSubscription {
        private ReadableByteChannel channel;
        private long readTotal;

        private SubscriptionImpl(Consumer consumer, boolean emitInitialContent) {
            super(consumer, emitInitialContent);
        }

        @Override
        protected boolean produceContent(Producer producer) throws IOException {
            ByteBuffer buffer;
            boolean last;
            if (channel == null) {
                if (putTask.getDataPath() != null) {
                    channel = Files.newByteChannel(putTask.getDataPath(), StandardOpenOption.READ);
                } else {
                    channel = Channels.newChannel(putTask.newInputStream());
                }
            }

            buffer = bufferPool == null
                    ? BufferUtil.allocate(bufferSize, isUseDirectByteBuffers())
                    : bufferPool.acquire(bufferSize, isUseDirectByteBuffers());

            BufferUtil.clearToFill(buffer);
            int read = channel.read(buffer);
            BufferUtil.flipToFlush(buffer, 0);
            if (!channel.isOpen() && read < 0) {
                throw new EOFException("EOF reached for " + putTask);
            }
            if (read > 0) {
                readTotal += read;
            }
            last = readTotal == getLength();
            if (last) {
                IO.close(channel);
            }
            return producer.produce(buffer, last, Callback.from(() -> release(buffer)));
        }

        private void release(ByteBuffer buffer) {
            if (bufferPool != null) {
                bufferPool.release(buffer);
            }
        }

        @Override
        public void fail(Throwable failure) {
            super.fail(failure);
            IO.close(channel);
        }
    }
}
