es6 changes for npr one module

This commit is contained in:
Todd Treece 2016-05-27 13:15:09 -04:00
parent e6114d3907
commit 2a3d76914e

View file

@ -1,19 +1,18 @@
/************************ DEPENDENCIES *****************************/
var util = require('util'),
bunyan = require('bunyan'),
'use strict';
const bunyan = require('bunyan'),
helpers = require('./helpers'),
swagger = require('swagger-client');
Swagger = require('swagger-client');
var proto = One.prototype;
exports = module.exports = One;
class One {
/************************* CONSTRUCTOR ****************************/
function One(config) {
constructor(config) {
if (! (this instanceof One))
return new One(config);
this.token = false;
this.swagger_url = 'https://api.npr.org/documentation/beryllium/api-docs';
this.log = false;
util._extend(this, config || {});
Object.assign(this, config || {});
if(! this.log)
this.log = bunyan.createLogger({name: 'npr-one'});
@ -21,70 +20,63 @@ function One(config) {
// silence swagger log output
process.env.NODE_ENV = 'test';
}
}
/*************************** DEFAULTS *****************************/
proto.token = false;
proto.swagger_url = 'https://api.npr.org/documentation/beryllium/api-docs';
proto.log = false;
proto.init = function(token) {
var self = this;
init(token) {
if(token)
this.setAccessToken(token);
return new Promise(function(resolve, reject) {
return new Promise((resolve, reject) => {
var client = new swagger({
url: self.swagger_url,
success: function() {
self.client = client;
helpers.processSwagger(self);
resolve(self);
const client = new Swagger({
url: this.swagger_url,
success: () => {
this.client = client;
helpers.processSwagger(this);
resolve(this);
},
failure: function() {
failure: () => {
reject('swagger init failed');
}
});
});
};
}
proto.setAccessToken = function(token) {
var self = this;
setAccessToken(token) {
this.token = token;
return new Promise(function(resolve, reject) {
return new Promise((resolve, reject) => {
if(! self.token)
if(! this.token)
return reject('no token supplied');
var auth = new swagger.ApiKeyAuthorization('Authorization', 'Bearer ' + token, 'header');
const auth = new Swagger.ApiKeyAuthorization('Authorization', `Bearer ${token}`, 'header');
self.client.clientAuthorizations.add('oauth2', auth);
this.client.clientAuthorizations.add('oauth2', auth);
resolve();
});
};
}
proto.getAccessToken = function() {
getAccessToken() {
var self = this;
return new Promise((resolve, reject) => {
return new Promise(function(resolve, reject) {
if(! self.token)
if(! this.token)
return reject('no token supplied');
resolve(self.token);
resolve(this.token);
});
};
}
}
exports = module.exports = One;