youtube_explode/lib/src/reverse_engineering/youtube_http_client.dart

117 lines
3.6 KiB
Dart

import 'package:http/http.dart';
import '../exceptions/exceptions.dart';
import '../videos/streams/streams.dart';
class YoutubeHttpClient {
final Client _httpClient = Client();
final Map<String, String> _defaultHeaders = const {
'user-agent':
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36',
'accept-language': 'en-US,en;q=1.0',
'x-youtube-client-name': '1',
'x-youtube-client-version': '2.20200609.04.02',
};
/// Throws if something is wrong with the response.
void _validateResponse(BaseResponse response, int statusCode) {
var request = response.request;
if (request.url.host.endsWith('.google.com') &&
request.url.path.startsWith('/sorry/')) {
throw RequestLimitExceededException.httpRequest(response);
}
if (statusCode >= 500) {
throw TransientFailureException.httpRequest(response);
}
if (statusCode == 429) {
throw RequestLimitExceededException.httpRequest(response);
}
if (statusCode >= 400) {
throw FatalFailureException.httpRequest(response);
}
}
Future<Response> get(dynamic url, {Map<String, String> headers}) {
return _httpClient.get(url, headers: {...?headers, ..._defaultHeaders});
}
Future<Response> post(dynamic url, {Map<String, String> headers}) {
return _httpClient.post(url, headers: {...?headers, ..._defaultHeaders});
}
Future<Response> head(dynamic url, {Map<String, String> headers}) {
return _httpClient.head(url, headers: {...?headers, ..._defaultHeaders});
}
Future<String> getString(dynamic url,
{Map<String, String> headers, bool validate = true}) async {
var response =
await _httpClient.get(url, headers: {...?headers, ..._defaultHeaders});
if (validate) {
_validateResponse(response, response.statusCode);
}
return response.body;
}
Future<String> postString(dynamic url,
{Map<String, String> body,
Map<String, String> headers,
bool validate = true}) async {
var response = await _httpClient.post(url,
headers: {...?headers, ..._defaultHeaders}, body: body);
if (validate) {
_validateResponse(response, response.statusCode);
}
return response.body;
}
// TODO: Check why isRateLimited is not working.
Stream<List<int>> getStream(StreamInfo streamInfo,
{Map<String, String> headers, bool validate = true}) async* {
var url = streamInfo.url;
// if (streamInfo.isRateLimited()) {
// var request = Request('get', url);
// request.headers.addAll(_defaultHeaders);
// var response = await request.send();
// if (validate) {
// _validateResponse(response, response.statusCode);
// }
// yield* response.stream;
// } else {
for (var i = 0; i < streamInfo.size.totalBytes; i += 9898989) {
var request = Request('get', url);
request.headers['range'] = 'bytes=$i-${i + 9898989}';
request.headers.addAll(_defaultHeaders);
var response = await request.send();
if (validate) {
_validateResponse(response, response.statusCode);
}
yield* response.stream;
}
// }
}
Future<int> getContentLength(dynamic url,
{Map<String, String> headers, bool validate = true}) async {
var response = await head(url, headers: headers);
if (validate) {
_validateResponse(response, response.statusCode);
}
return int.tryParse(response.headers['content-length'] ?? '');
}
/// Closes the [Client] assigned to this [YoutubeHttpClient].
/// Should be called after this is not used anymore.
void close() => _httpClient.close();
}