| OLD | NEW |
| 1 // Copyright (c) 2016, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2016, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 import "dart:async" show Future, Stream; | 5 import "dart:async" show Future, Stream; |
| 6 import "dart:convert" show Encoding, LATIN1, UTF8; | 6 import "dart:convert" show Encoding, LATIN1, UTF8; |
| 7 import "dart:io" show | 7 import "dart:io" show File, |
| 8 File, HttpClient, HttpClientResponse, HttpClientRequest, HttpHeaders; | 8 HttpStatus, |
| 9 HttpClient, |
| 10 HttpClientResponse, |
| 11 HttpClientRequest, |
| 12 HttpHeaders; |
| 9 | 13 |
| 10 import "package:typed_data/typed_buffers.dart" show Uint8Buffer; | 14 import "package:typed_data/typed_buffers.dart" show Uint8Buffer; |
| 11 | 15 |
| 12 /// Read the bytes of a URI as a stream of bytes. | 16 /// Read the bytes of a URI as a stream of bytes. |
| 13 Stream<List<int>> readAsStream(Uri uri) async* { | 17 Stream<List<int>> readAsStream(Uri uri) async* { |
| 14 if (uri.scheme == "file") { | 18 if (uri.scheme == "file") { |
| 15 yield* new File.fromUri(uri).openRead(); | 19 yield* new File.fromUri(uri).openRead(); |
| 16 return; | 20 return; |
| 17 } | 21 } |
| 18 if (uri.scheme == "http" || uri.scheme == "https") { | 22 if (uri.scheme == "http" || uri.scheme == "https") { |
| 19 HttpClientResponse response = await _httpGetBytes(uri); | 23 HttpClientResponse response = await _httpGetBytes(uri); |
| 24 _throwIfFailed(response, uri); |
| 20 yield* response; | 25 yield* response; |
| 21 return; | 26 return; |
| 22 } | 27 } |
| 23 if (uri.scheme == "data") { | 28 if (uri.scheme == "data") { |
| 24 yield uri.data.contentAsBytes(); | 29 yield uri.data.contentAsBytes(); |
| 25 return; | 30 return; |
| 26 } | 31 } |
| 27 throw new UnsupportedError("Unsupported scheme: $uri"); | 32 throw new UnsupportedError("Unsupported scheme: $uri"); |
| 28 } | 33 } |
| 29 | 34 |
| 30 /// Read the bytes of a URI as a list of bytes. | 35 /// Read the bytes of a URI as a list of bytes. |
| 31 Future<List<int>> readAsBytes(Uri uri) async { | 36 Future<List<int>> readAsBytes(Uri uri) async { |
| 32 if (uri.scheme == "file") { | 37 if (uri.scheme == "file") { |
| 33 return new File.fromUri(uri).readAsBytes(); | 38 return new File.fromUri(uri).readAsBytes(); |
| 34 } | 39 } |
| 35 if (uri.scheme == "http" || uri.scheme == "https") { | 40 if (uri.scheme == "http" || uri.scheme == "https") { |
| 36 HttpClientResponse response = await _httpGetBytes(uri); | 41 HttpClientResponse response = await _httpGetBytes(uri); |
| 42 _throwIfFailed(response, uri); |
| 37 int length = response.contentLength; | 43 int length = response.contentLength; |
| 38 if (length < 0) length = 0; | 44 if (length < 0) length = 0; |
| 39 var buffer = new Uint8Buffer(length); | 45 var buffer = new Uint8Buffer(length); |
| 40 await for (var bytes in response) { | 46 await for (var bytes in response) { |
| 41 buffer.addAll(bytes); | 47 buffer.addAll(bytes); |
| 42 } | 48 } |
| 43 return buffer.toList(); | 49 return buffer.toList(); |
| 44 } | 50 } |
| 45 if (uri.scheme == "data") { | 51 if (uri.scheme == "data") { |
| 46 return uri.data.contentAsBytes(); | 52 return uri.data.contentAsBytes(); |
| 47 } | 53 } |
| 48 throw new UnsupportedError("Unsupported scheme: $uri"); | 54 throw new UnsupportedError("Unsupported scheme: $uri"); |
| 49 } | 55 } |
| 50 | 56 |
| 51 /// Read the bytes of a URI as a string. | 57 /// Read the bytes of a URI as a string. |
| 52 Future<String> readAsString(Uri uri, Encoding encoding) async { | 58 Future<String> readAsString(Uri uri, Encoding encoding) async { |
| 53 if (uri.scheme == "file") { | 59 if (uri.scheme == "file") { |
| 54 if (encoding == null) encoding = UTF8; | 60 if (encoding == null) encoding = UTF8; |
| 55 return new File.fromUri(uri).readAsString(encoding: encoding); | 61 return new File.fromUri(uri).readAsString(encoding: encoding); |
| 56 } | 62 } |
| 57 if (uri.scheme == "http" || uri.scheme == "https") { | 63 if (uri.scheme == "http" || uri.scheme == "https") { |
| 58 HttpClientRequest request = await new HttpClient().getUrl(uri); | 64 HttpClientRequest request = await new HttpClient().getUrl(uri); |
| 59 // Prefer text/plain, text/* if possible, otherwise take whatever is there. | 65 // Prefer text/plain, text/* if possible, otherwise take whatever is there. |
| 60 request.headers.set(HttpHeaders.ACCEPT, "text/plain, text/*, */*"); | 66 request.headers.set(HttpHeaders.ACCEPT, "text/plain, text/*, */*"); |
| 61 if (encoding != null) { | 67 if (encoding != null) { |
| 62 request.headers.set(HttpHeaders.ACCEPT_CHARSET, encoding.name); | 68 request.headers.set(HttpHeaders.ACCEPT_CHARSET, encoding.name); |
| 63 } | 69 } |
| 64 HttpClientResponse response = await request.close(); | 70 HttpClientResponse response = await request.close(); |
| 71 _throwIfFailed(response, uri); |
| 65 encoding ??= Encoding.getByName(response.headers.contentType?.charset); | 72 encoding ??= Encoding.getByName(response.headers.contentType?.charset); |
| 66 if (encoding == null || encoding == LATIN1) { | 73 if (encoding == null || encoding == LATIN1) { |
| 67 // Default to LATIN-1 if no encoding found. | 74 // Default to LATIN-1 if no encoding found. |
| 68 // Special case LATIN-1 since it is common and doesn't need decoding. | 75 // Special case LATIN-1 since it is common and doesn't need decoding. |
| 69 int length = response.contentLength; | 76 int length = response.contentLength; |
| 70 if (length < 0) length = 0; | 77 if (length < 0) length = 0; |
| 71 var buffer = new Uint8Buffer(length); | 78 var buffer = new Uint8Buffer(length); |
| 72 await for (var bytes in response) { | 79 await for (var bytes in response) { |
| 73 buffer.addAll(bytes); | 80 buffer.addAll(bytes); |
| 74 } | 81 } |
| 75 var byteList = buffer.buffer.asUint8List(0, buffer.length); | 82 var byteList = buffer.buffer.asUint8List(0, buffer.length); |
| 76 return new String.fromCharCodes(byteList); | 83 return new String.fromCharCodes(byteList); |
| 77 } | 84 } |
| 78 return response.transform(encoding.decoder).join(); | 85 return response.transform(encoding.decoder).join(); |
| 79 } | 86 } |
| 80 if (uri.scheme == "data") { | 87 if (uri.scheme == "data") { |
| 81 return uri.data.contentAsString(encoding: encoding); | 88 return uri.data.contentAsString(encoding: encoding); |
| 82 } | 89 } |
| 83 throw new UnsupportedError("Unsupported scheme: $uri"); | 90 throw new UnsupportedError("Unsupported scheme: $uri"); |
| 84 } | 91 } |
| 85 | 92 |
| 86 Future<HttpClientResponse> _httpGetBytes(Uri uri) async { | 93 Future<HttpClientResponse> _httpGetBytes(Uri uri) async { |
| 87 HttpClientRequest request = await new HttpClient().getUrl(uri); | 94 HttpClientRequest request = await new HttpClient().getUrl(uri); |
| 88 request.headers.set(HttpHeaders.ACCEPT, "application/octet-stream, */*"); | 95 request.headers.set(HttpHeaders.ACCEPT, "application/octet-stream, */*"); |
| 89 return request.close(); | 96 return request.close(); |
| 90 } | 97 } |
| 98 |
| 99 void _throwIfFailed(HttpClientResponse response, Uri uri) { |
| 100 var statusCode = response.statusCode; |
| 101 if (statusCode < HttpStatus.OK || statusCode > HttpStatus.NO_CONTENT) { |
| 102 throw new HttpException(response.reasonPhrase, uri: uri); |
| 103 } |
| 104 } |
| OLD | NEW |