platform/packages/auth_twitter/lib/angel_auth_twitter.dart
2021-06-26 21:13:43 +08:00

147 lines
4.9 KiB
Dart

import 'dart:async';
import 'dart:convert';
import 'package:angel_auth/angel_auth.dart';
import 'package:angel_framework/angel_framework.dart';
import 'package:http/http.dart' as http;
import 'package:oauth/oauth.dart' as oauth;
import 'package:path/path.dart' as p;
import 'package:twitter/twitter.dart';
/// Authenticates users by connecting to Twitter's API.
class TwitterStrategy<User> extends AuthStrategy<User> {
/// The options defining how to connect to the third-party.
final ExternalAuthOptions options;
/// A callback that uses Twitter to authenticate a [User].
///
/// As always, return `null` if authentication fails.
final FutureOr<User> Function(Twitter, RequestContext, ResponseContext)
verifier;
/// A callback that is triggered when an OAuth2 error occurs (i.e. the user declines to login);
final FutureOr<dynamic> Function(
TwitterAuthorizationException, RequestContext, ResponseContext) onError;
/// The root of Twitter's API. Defaults to `'https://api.twitter.com'`.
final Uri baseUrl;
oauth.Client _client;
/// The underlying [oauth.Client] used to query Twitter.
oauth.Client get client => _client;
TwitterStrategy(this.options, this.verifier, this.onError,
{http.BaseClient client, Uri baseUrl})
: baseUrl = baseUrl ?? Uri.parse('https://api.twitter.com') {
var tokens = oauth.Tokens(
consumerId: options.clientId, consumerKey: options.clientSecret);
_client = oauth.Client(tokens, client: client);
}
/// Handle a response from Twitter.
Future<Map<String, String>> handleUrlEncodedResponse(http.Response rs) async {
var body = rs.body;
if (rs.statusCode != 200) {
var err = json.decode(rs.body) as Map;
var errors = err['errors'] as List;
if (errors.isNotEmpty) {
throw TwitterAuthorizationException(
errors[0]['message'] as String, false);
} else {
throw StateError(
'Twitter returned an error response without an error message: ${rs.body}');
}
}
return Uri.splitQueryString(body);
}
/// Get an access token.
Future<Map<String, String>> getAccessToken(String token, String verifier) {
return _client.post(
baseUrl.replace(path: p.join(baseUrl.path, 'oauth/access_token')),
headers: {
'accept': 'application/json'
},
body: {
'oauth_token': token,
'oauth_verifier': verifier
}).then(handleUrlEncodedResponse);
// var request = await createRequest("oauth/access_token",
// method: "POST", data: {"verifier": verifier}, accessToken: token);
}
/// Get a request token.
Future<Map<String, String>> getRequestToken() {
return _client.post(
baseUrl.replace(path: p.join(baseUrl.path, 'oauth/request_token')),
headers: {
'accept': 'application/json'
},
body: {
'oauth_callback': options.redirectUri.toString()
}).then(handleUrlEncodedResponse);
}
@override
Future<User> authenticate(RequestContext req, ResponseContext res,
[AngelAuthOptions options]) async {
try {
if (options != null) {
var result = await authenticateCallback(req, res, options);
if (result is User) {
return result;
} else {
return null;
}
} else {
var result = await getRequestToken();
var token = result['oauth_token'];
var url = baseUrl.replace(
path: p.join(baseUrl.path, 'oauth/authorize'),
queryParameters: {'oauth_token': token});
await res.redirect(url);
return null;
}
} on TwitterAuthorizationException catch (e) {
var result = await onError(e, req, res);
await req.app.executeHandler(result, req, res);
await res.close();
return null;
}
}
Future authenticateCallback(
RequestContext req, ResponseContext res, AngelAuthOptions options) async {
try {
if (req.queryParameters.containsKey('denied')) {
throw TwitterAuthorizationException(
'The user denied the Twitter authorization attempt.', true);
}
var token = req.queryParameters['oauth_token'] as String;
var verifier = req.queryParameters['oauth_verifier'] as String;
var loginData = await getAccessToken(token, verifier);
var twitter = Twitter(this.options.clientId, this.options.clientSecret,
loginData['oauth_token'], loginData['oauth_token_secret']);
return await this.verifier(twitter, req, res);
} on TwitterAuthorizationException catch (e) {
return await onError(e, req, res);
}
}
}
class TwitterAuthorizationException implements Exception {
/// The message associated with this exception.
final String message;
/// Whether the user denied the authorization attempt.
final bool isDenial;
TwitterAuthorizationException(this.message, this.isDenial);
@override
String toString() => 'TwitterAuthorizationException: $message';
}