Added option for custom callback token request handler (#50)

Author: bflorian

index.js

@@ -23,5 +23,7 @@ module.exports = {
  DiscoveryResponse: require('./lib/discovery/DiscoveryResponse'),
  SchemaConnector: require('./lib/SchemaConnector'),
  StateRefreshResponse: require('./lib/state/StateRefreshResponse'),
- StateUpdateRequest: require('./lib/callbacks/StateUpdateRequest')
+ StateUpdateRequest: require('./lib/callbacks/StateUpdateRequest'),
+ STBase: require('./lib/STBase'),
+ checkFetchStatus: require('./lib/util/checkFetchStatus')
 };

lib/SchemaConnector.js

@@ -12,6 +12,7 @@ const clientSecret = Symbol('private)');
 const discoveryHandler = Symbol('private)');
 const stateRefreshHandler = Symbol('private)');
 const commandHandler = Symbol('private)');
+const callbackTokenRequestHandler = Symbol('private)');
 const callbackAccessHandler = Symbol('private)');
 const integrationDeletedHandler = Symbol('private)');
 const interactionResultHandler = Symbol('private)');
@@ -39,6 +40,19 @@ module.exports = class SchemaConnector {
 
  this[callbackAccessHandler] = null;
 
+ this[callbackTokenRequestHandler] = (async (clientId, clientSecret, body) => {
+ const tokenRequest = new AccessTokenRequest(
+ clientId,
+ clientSecret,
+ body.headers.requestId
+ );
+
+ return await tokenRequest.getCallbackToken(
+ body.callbackUrls.oauthToken,
+ body.callbackAuthentication.code
+ );
+ });
+
  this[integrationDeletedHandler] = ((accessToken, data) => {
  console.log('integrationDeletedHandler not defined')
  });
@@ -100,13 +114,22 @@ module.exports = class SchemaConnector {
  }
 
  /**
- * Sets the grant callback access handler
+ * Sets the handlers called after callback access is granted
  */
  callbackAccessHandler(callback) {
  this[callbackAccessHandler] = callback;
  return this
  }
 
+ /**
+ * Overrides the built in handler that requests callback access tokens.
+ * That handler is expected to return the response of the callback token request.
+ */
+ callbackTokenRequestHandler(callback) {
+ this[callbackTokenRequestHandler] = callback
+ return this
+ }
+
  /**
  * Sets integration deleted handler
  */
@@ -239,16 +262,7 @@ module.exports = class SchemaConnector {
  if (this[callbackAccessHandler]) {
  if (body.callbackAuthentication.clientId === this[clientId] ) {
 
- const tokenRequest = new AccessTokenRequest(
- this[clientId],
- this[clientSecret],
- body.headers.requestId
- );
-
- const tokenResponse = await tokenRequest.getCallbackToken(
- body.callbackUrls.oauthToken,
- body.callbackAuthentication.code
- );
+ const tokenResponse = await this[callbackTokenRequestHandler](this[clientId], this[clientSecret], body)
 
  await this[callbackAccessHandler](body.authentication.token, tokenResponse.callbackAuthentication, body.callbackUrls, body)
  }