/*

 * Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package jdk.internal.net.http;

import java.io.IOException;

import java.net.InetSocketAddress;

import java.net.http.HttpResponse.BodyHandler;

import java.net.http.HttpResponse.BodySubscriber;

import java.nio.ByteBuffer;

import java.util.Objects;

import java.util.concurrent.CompletableFuture;

import java.util.LinkedList;

import java.util.List;

import java.util.concurrent.ConcurrentLinkedDeque;

import java.util.concurrent.Executor;

import java.util.concurrent.Flow;

import jdk.internal.net.http.common.Demand;

import jdk.internal.net.http.common.Log;

import jdk.internal.net.http.common.FlowTube;

import jdk.internal.net.http.common.Logger;

import jdk.internal.net.http.common.SequentialScheduler;

import jdk.internal.net.http.common.MinimalFuture;

import jdk.internal.net.http.common.Utils;

import static java.net.http.HttpClient.Version.HTTP_1_1;

/**

 * Encapsulates one HTTP/1.1 request/response exchange.

 */

class Http1Exchange<T> extends ExchangeImpl<T> {

    final Logger debug = Utils.getDebugLogger(this::dbgString, Utils.DEBUG);

    final HttpRequestImpl request; // main request

    final Http1Request requestAction;

    private volatile Http1Response<T> response;

    final HttpConnection connection;

    final HttpClientImpl client;

    final Executor executor;

    private final Http1AsyncReceiver asyncReceiver;

    /** Records a possible cancellation raised before any operation

     * has been initiated, or an error received while sending the request. */

    private Throwable failed;

    private final List<CompletableFuture<?>> operations; // used for cancel

    /** Must be held when operating on any internal state or data. */

    private final Object lock = new Object();

    /** Holds the outgoing data, either the headers or a request body part. Or

     * an error from the request body publisher. At most there can be ~2 pieces

     * of outgoing data ( onComplete|onError can be invoked without demand ).*/

    final ConcurrentLinkedDeque<DataPair> outgoing = new ConcurrentLinkedDeque<>();

    /** The write publisher, responsible for writing the complete request ( both

     * headers and body ( if any ). */

    private final Http1Publisher writePublisher = new Http1Publisher();

    /** Completed when the header have been published, or there is an error */

    private final CompletableFuture<ExchangeImpl<T>> headersSentCF  = new MinimalFuture<>();

     /** Completed when the body has been published, or there is an error */

    private final CompletableFuture<ExchangeImpl<T>> bodySentCF = new MinimalFuture<>();

    /** The subscriber to the request's body published. Maybe null. */

    private volatile Http1BodySubscriber bodySubscriber;

    enum State { INITIAL,

                 HEADERS,

                 BODY,

                 ERROR,          // terminal state

                 COMPLETING,

                 COMPLETED }     // terminal state

    private State state = State.INITIAL;

    /** A carrier for either data or an error. Used to carry data, and communicate

     * errors from the request ( both headers and body ) to the exchange. */

    static class DataPair {

        Throwable throwable;

        List<ByteBuffer> data;

        DataPair(List<ByteBuffer> data, Throwable throwable){

            this.data = data;

            this.throwable = throwable;

        }

        @Override

        public String toString() {

            return "DataPair [data=" + data + ", throwable=" + throwable + "]";

        }

    }

    /** An abstract supertype for HTTP/1.1 body subscribers. There are two

     * concrete implementations: {@link Http1Request.StreamSubscriber}, and

     * {@link Http1Request.FixedContentSubscriber}, for receiving chunked and

     * fixed length bodies, respectively. */

    static abstract class Http1BodySubscriber implements Flow.Subscriber<ByteBuffer> {

        final MinimalFuture<Flow.Subscription> whenSubscribed = new MinimalFuture<>();

        private volatile Flow.Subscription subscription;

        volatile boolean complete;

        private final Logger debug;

        Http1BodySubscriber(Logger debug) {

            assert debug != null;

            this.debug = debug;

        }

        /** Final sentinel in the stream of request body. */

        static final List<ByteBuffer> COMPLETED = List.of(ByteBuffer.allocate(0));

        final void request(long n) {

            if (debug.on())

                debug.log("Http1BodySubscriber requesting %d, from %s",

                          n, subscription);

            subscription.request(n);

        }

        final boolean isSubscribed() {

            return subscription != null;

        }

        final void setSubscription(Flow.Subscription subscription) {

            this.subscription = subscription;

            whenSubscribed.complete(subscription);

        }

        final void cancelSubscription() {

            try {

                subscription.cancel();

            } catch(Throwable t) {

                String msg = "Ignoring exception raised when canceling BodyPublisher subscription";

                if (debug.on()) debug.log("%s: %s", msg, t);

                Log.logError("{0}: {1}", msg, (Object)t);

            }

        }

        static Http1BodySubscriber completeSubscriber(Logger debug) {

            return new Http1BodySubscriber(debug) {

                @Override public void onSubscribe(Flow.Subscription subscription) { error(); }

                @Override public void onNext(ByteBuffer item) { error(); }

                @Override public void onError(Throwable throwable) { error(); }

                @Override public void onComplete() { error(); }

                private void error() {

                    throw new InternalError("should not reach here");

                }

            };

        }

    }

    @Override

    public String toString() {

        return "HTTP/1.1 " + request.toString();

    }

    HttpRequestImpl request() {

        return request;

    }

    Http1Exchange(Exchange<T> exchange, HttpConnection connection)

        throws IOException

    {

        super(exchange);

        this.request = exchange.request();

        this.client = exchange.client();

        this.executor = exchange.executor();

        this.operations = new LinkedList<>();

        operations.add(headersSentCF);

        operations.add(bodySentCF);

        if (connection != null) {

            this.connection = connection;

        } else {

            InetSocketAddress addr = request.getAddress();

            this.connection = HttpConnection.getConnection(addr, client, request, HTTP_1_1);

        }

        this.requestAction = new Http1Request(request, this);

        this.asyncReceiver = new Http1AsyncReceiver(executor, this);

    }

    @Override

    HttpConnection connection() {

        return connection;

    }

    private void connectFlows(HttpConnection connection) {

        FlowTube tube =  connection.getConnectionFlow();

        if (debug.on()) debug.log("%s connecting flows", tube);

        // Connect the flow to our Http1TubeSubscriber:

        //   asyncReceiver.subscriber().

        tube.connectFlows(writePublisher,

                          asyncReceiver.subscriber());

    }

    @Override

    CompletableFuture<ExchangeImpl<T>> sendHeadersAsync() {

        // create the response before sending the request headers, so that

        // the response can set the appropriate receivers.

        if (debug.on()) debug.log("Sending headers only");

        if (response == null) {

            response = new Http1Response<>(connection, this, asyncReceiver);

        }

        if (debug.on()) debug.log("response created in advance");

        CompletableFuture<Void> connectCF;

        if (!connection.connected()) {

            if (debug.on()) debug.log("initiating connect async");

            connectCF = connection.connectAsync();

            Throwable cancelled;

            synchronized (lock) {

                if ((cancelled = failed) == null) {

                    operations.add(connectCF);

                }

            }

            if (cancelled != null) {

                if (client.isSelectorThread()) {

                    executor.execute(() ->

                        connectCF.completeExceptionally(cancelled));

                } else {

                    connectCF.completeExceptionally(cancelled);

                }

            }

        } else {

            connectCF = new MinimalFuture<>();

            connectCF.complete(null);

        }

        return connectCF

                .thenCompose(unused -> {

                    CompletableFuture<Void> cf = new MinimalFuture<>();

                    try {

                        connectFlows(connection);

                        if (debug.on()) debug.log("requestAction.headers");

                        List<ByteBuffer> data = requestAction.headers();

                        synchronized (lock) {

                            state = State.HEADERS;

                        }

                        if (debug.on()) debug.log("setting outgoing with headers");

                        assert outgoing.isEmpty() : "Unexpected outgoing:" + outgoing;

                        appendToOutgoing(data);

                        cf.complete(null);

                        return cf;

                    } catch (Throwable t) {

                        if (debug.on()) debug.log("Failed to send headers: %s", t);

                        connection.close();

                        cf.completeExceptionally(t);

                        return cf;

                    } })

                .thenCompose(unused -> headersSentCF);

    }

    @Override

    CompletableFuture<ExchangeImpl<T>> sendBodyAsync() {

        assert headersSentCF.isDone();

        try {

            bodySubscriber = requestAction.continueRequest();

            if (bodySubscriber == null) {

                bodySubscriber = Http1BodySubscriber.completeSubscriber(debug);

                appendToOutgoing(Http1BodySubscriber.COMPLETED);

            } else {

                // start

                bodySubscriber.whenSubscribed

                        .thenAccept((s) -> requestMoreBody());

            }

        } catch (Throwable t) {

            cancelImpl(t);

            bodySentCF.completeExceptionally(t);

        }

    }

    @Override

    CompletableFuture<Response> getResponseAsync(Executor executor) {

        CompletableFuture<Response> cf = response.readHeadersAsync(executor);

        Throwable cause;

        synchronized (lock) {

            operations.add(cf);

            cause = failed;

            failed = null;

        }

        if (cause != null) {

            Log.logTrace("Http1Exchange: request [{0}/timeout={1}ms]"

                            + "\n\tCompleting exceptionally with {2}\n",

                         request.uri(),

                         request.timeout().isPresent() ?

                            // calling duration.toMillis() can throw an exception.

                            // this is just debugging, we don't care if it overflows.

                            (request.timeout().get().getSeconds() * 1000

                             + request.timeout().get().getNano() / 1000000) : -1,

                         cause);

            boolean acknowledged = cf.completeExceptionally(cause);

            if (debug.on())

                debug.log(acknowledged ? ("completed response with " + cause)

                          : ("response already completed, ignoring " + cause));

        }

    }

    @Override

    CompletableFuture<T> readBodyAsync(BodyHandler<T> handler,

                                       boolean returnConnectionToPool,

                                       Executor executor)

    {

        BodySubscriber<T> bs = handler.apply(new ResponseInfoImpl(response.responseCode(),

                                                                  response.responseHeaders(),

                                                                  HTTP_1_1));

        CompletableFuture<T> bodyCF = response.readBody(bs,

                                                        returnConnectionToPool,

                                                        executor);

        return bodyCF;

    }

    @Override

    CompletableFuture<Void> ignoreBody() {

        return response.ignoreBody(executor);

    }

    ByteBuffer drainLeftOverBytes() {

        synchronized (lock) {

            asyncReceiver.stop();

            return asyncReceiver.drain(Utils.EMPTY_BYTEBUFFER);

        }

    }

    void released() {

        Http1Response<T> resp = this.response;

        if (resp != null) resp.completed();

        asyncReceiver.clear();

    }

    void completed() {

        Http1Response<T> resp = this.response;

        if (resp != null) resp.completed();

    }

    /**

     * Cancel checks to see if request and responseAsync finished already.

     * If not it closes the connection and completes all pending operations

     */

    @Override

    void cancel() {

        cancelImpl(new IOException("Request cancelled"));

    }

    /**

     * Cancel checks to see if request and responseAsync finished already.

     * If not it closes the connection and completes all pending operations

     */

    @Override

    void cancel(IOException cause) {

        cancelImpl(cause);

    }

    private void cancelImpl(Throwable cause) {

        LinkedList<CompletableFuture<?>> toComplete = null;

        int count = 0;

        Throwable error;

        synchronized (lock) {

            if ((error = failed) == null) {

                failed = error = cause;

            }

            if (debug.on()) {

                debug.log(request.uri() + ": " + error);

            }

            if (requestAction != null && requestAction.finished()

                    && response != null && response.finished()) {

                return;

            }

            writePublisher.writeScheduler.stop();

            if (operations.isEmpty()) {

                Log.logTrace("Http1Exchange: request [{0}/timeout={1}ms] no pending operation."

                                + "\n\tCan''t cancel yet with {2}",

                             request.uri(),

                             request.timeout().isPresent() ?

                                // calling duration.toMillis() can throw an exception.

                                // this is just debugging, we don't care if it overflows.

                                (request.timeout().get().getSeconds() * 1000

                                 + request.timeout().get().getNano() / 1000000) : -1,

                             cause);

            } else {

                for (CompletableFuture<?> cf : operations) {

                    if (!cf.isDone()) {

                        if (toComplete == null) toComplete = new LinkedList<>();

                        toComplete.add(cf);

                        count++;

                    }

                }

                operations.clear();

            }

        }

            }

        }

    }

    private void runInline(Runnable run) {

        assert !client.isSelectorThread();

        run.run();

    }

    /** Returns true if this exchange was canceled. */

    boolean isCanceled() {

        synchronized (lock) {

            return failed != null;

        }

    }

    /** Returns the cause for which this exchange was canceled, if available. */

    Throwable getCancelCause() {

        synchronized (lock) {

            return failed;

        }

    }

    /** Convenience for {@link #appendToOutgoing(DataPair)}, with just a Throwable. */