1 /*

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

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

     4  *

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

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

     7  * published by the Free Software Foundation.  Oracle designates this

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

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

    10  *

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

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

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

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

    15  * accompanied this code).

    16  *

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

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

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

    20  *

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

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

    23  * questions.

    24  */

    25 

    26 package jdk.internal.net.http;

    27 

    28 import java.net.URI;

    29 import java.time.Duration;

    30 import java.util.Optional;

    31 import java.net.http.HttpClient;

    32 import java.net.http.HttpRequest;

    33 import java.net.http.HttpRequest.BodyPublisher;

    34 import jdk.internal.net.http.common.HttpHeadersImpl;

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

    36 import static java.lang.String.format;

    37 import static java.util.Objects.requireNonNull;

    38 import static jdk.internal.net.http.common.Utils.isValidName;

    39 import static jdk.internal.net.http.common.Utils.isValidValue;

    40 

    41 public class HttpRequestBuilderImpl extends HttpRequest.Builder {

    42 

    43     private HttpHeadersImpl userHeaders;

    44     private URI uri;

    45     private String method;

    46     private boolean expectContinue;

    47     private BodyPublisher bodyPublisher;

    48     private volatile Optional<HttpClient.Version> version;

    49     private Duration duration;

    50 

    51     public HttpRequestBuilderImpl(URI uri) {

    52         requireNonNull(uri, "uri must be non-null");

    53         checkURI(uri);

    54         this.uri = uri;

    55         this.userHeaders = new HttpHeadersImpl();

    56         this.method = "GET"; // default, as per spec

    57         this.version = Optional.empty();

    58     }

    59 

    60     public HttpRequestBuilderImpl() {

    61         this.userHeaders = new HttpHeadersImpl();

    62         this.method = "GET"; // default, as per spec

    63         this.version = Optional.empty();

    64     }

    65 

    66     @Override

    67     public HttpRequestBuilderImpl uri(URI uri) {

    68         requireNonNull(uri, "uri must be non-null");

    69         checkURI(uri);

    70         this.uri = uri;

    71         return this;

    72     }

    73 

    74     private static IllegalArgumentException newIAE(String message, Object... args) {

    75         return new IllegalArgumentException(format(message, args));

    76     }

    77 

    78     private static void checkURI(URI uri) {

    79         String scheme = uri.getScheme();

    80         if (scheme == null)

    81             throw newIAE("URI with undefined scheme");

    82         scheme = scheme.toLowerCase();

    83         if (!(scheme.equals("https") || scheme.equals("http"))) {

    84             throw newIAE("invalid URI scheme %s", scheme);

    85         }

    86         if (uri.getHost() == null) {

    87             throw newIAE("unsupported URI %s", uri);

    88         }

    89     }

    90 

    91     @Override

    92     public HttpRequestBuilderImpl copy() {

    93         HttpRequestBuilderImpl b = new HttpRequestBuilderImpl(this.uri);

    94         b.userHeaders = this.userHeaders.deepCopy();

    95         b.method = this.method;

    96         b.expectContinue = this.expectContinue;

    97         b.bodyPublisher = bodyPublisher;

    98         b.uri = uri;

    99         b.duration = duration;

   100         b.version = version;

   101         return b;

   102     }

   103 

   104     private void checkNameAndValue(String name, String value) {

   105         requireNonNull(name, "name");

   106         requireNonNull(value, "value");

   107         if (!isValidName(name)) {

   108             throw newIAE("invalid header name: \"%s\"", name);

   109         }

   110         if (!Utils.ALLOWED_HEADERS.test(name)) {

   111             throw newIAE("restricted header name: \"%s\"", name);

   112         }

   113         if (!isValidValue(value)) {

   114             throw newIAE("invalid header value: \"%s\"", value);

   115         }

   116     }

   117 

   118     @Override

   119     public HttpRequestBuilderImpl setHeader(String name, String value) {

   120         checkNameAndValue(name, value);

   121         userHeaders.setHeader(name, value);

   122         return this;

   123     }

   124 

   125     @Override

   126     public HttpRequestBuilderImpl header(String name, String value) {

   127         checkNameAndValue(name, value);

   128         userHeaders.addHeader(name, value);

   129         return this;

   130     }

   131 

   132     @Override

   133     public HttpRequestBuilderImpl headers(String... params) {

   134         requireNonNull(params);

   135         if (params.length == 0 || params.length % 2 != 0) {

   136             throw newIAE("wrong number, %d, of parameters", params.length);

   137         }

   138         for (int i = 0; i < params.length; i += 2) {

   139             String name  = params[i];

   140             String value = params[i + 1];

   141             header(name, value);

   142         }

   143         return this;

   144     }

   145 

   146     @Override

   147     public HttpRequestBuilderImpl expectContinue(boolean enable) {

   148         expectContinue = enable;

   149         return this;

   150     }

   151 

   152     @Override

   153     public HttpRequestBuilderImpl version(HttpClient.Version version) {

   154         requireNonNull(version);

   155         this.version = Optional.of(version);

   156         return this;

   157     }

   158 

   159     HttpHeadersImpl headers() {  return userHeaders; }

   160 

   161     URI uri() { return uri; }

   162 

   163     String method() { return method; }

   164 

   165     boolean expectContinue() { return expectContinue; }

   166 

   167     BodyPublisher bodyPublisher() { return bodyPublisher; }

   168 

   169     Optional<HttpClient.Version> version() { return version; }

   170 

   171     @Override

   172     public HttpRequest.Builder GET() {

   173         return method0("GET", null);

   174     }

   175 

   176     @Override

   177     public HttpRequest.Builder POST(BodyPublisher body) {

   178         return method0("POST", requireNonNull(body));

   179     }

   180 

   181     @Override

   182     public HttpRequest.Builder DELETE(BodyPublisher body) {

   183         return method0("DELETE", requireNonNull(body));

   184     }

   185 

   186     @Override

   187     public HttpRequest.Builder PUT(BodyPublisher body) {

   188         return method0("PUT", requireNonNull(body));

   189     }

   190 

   191     @Override

   192     public HttpRequest.Builder method(String method, BodyPublisher body) {

   193         requireNonNull(method);

   194         if (method.equals(""))

   195             throw newIAE("illegal method <empty string>");

   196         if (method.equals("CONNECT"))

   197             throw newIAE("method CONNECT is not supported");

   198         return method0(method, requireNonNull(body));

   199     }

   200 

   201     private HttpRequest.Builder method0(String method, BodyPublisher body) {

   202         assert method != null;

   203         assert !method.equals("GET") ? body != null : true;

   204         assert !method.equals("");

   205         this.method = method;

   206         this.bodyPublisher = body;

   207         return this;

   208     }

   209 

   210     @Override

   211     public HttpRequest build() {

   212         if (uri == null)

   213             throw new IllegalStateException("uri is null");

   214         assert method != null;

   215         return new HttpRequestImpl(this);

   216     }

   217 

   218     @Override

   219     public HttpRequest.Builder timeout(Duration duration) {

   220         requireNonNull(duration);

   221         if (duration.isNegative() || Duration.ZERO.equals(duration))

   222             throw new IllegalArgumentException("Invalid duration: " + duration);

   223         this.duration = duration;

   224         return this;

   225     }

   226 

   227     Duration timeout() { return duration; }

   228 

   229 }