Add Get Viewers

README.md

@@ -88,6 +88,13 @@ For get your live stream information you have to call `liveStream`
 youtube.liveStream();
 ```
 
+### Get Viewers:
+For get live viewers you have to call `getViewers`
+
+```js
+youtube.getViewers();
+```
+
 ### Get Live Chat:
 For get live chat message you have to call `liveChat`
 
@@ -125,4 +132,5 @@ First change the `clientId` and `clientSecret` in `tests/integration.js` with yo
 - `http://localhost:8080/getChannel` return information of your [channel](#get-channel)
 - `http://localhost:8080/credentials` [get credentials](#get-credentials)
 - `http://localhost:8080/reconnect` [refresh access token](#refresh-access-token)
+- `http://localhost:8080/getViewers` [get viewers](#get-viewers)
 

app.js

@@ -1,6 +1,7 @@
 let axios = require('axios');
 let OAuth2 = require('oauth20');
 let getYoutube = Symbol('getYoutube');
+let getViewers = Symbol('getViewers');
 let credentialsYoutube = Symbol('credentialsYoutube');
 let urlsYoutube = Symbol('urlsYoutube');
 
@@ -35,7 +36,7 @@ class Youtube extends OAuth2 {
 	}
 
 	getChannel() {
-		let url = `${this[urlsYoutube].channels}`;
+		let url = this[urlsYoutube].channels;
 		let params = {
 			part: 'snippet,contentDetails,brandingSettings,invideoPromotion,statistics',
 			mine: true,
@@ -46,7 +47,7 @@ class Youtube extends OAuth2 {
 	}
 
 	liveStream() {
-		let url = `${this[urlsYoutube].videos}`;
+		let url = this[urlsYoutube].videos;
 		let params = {
 			part: 'statistics',
 			id: this[credentialsYoutube].liveId,
@@ -57,8 +58,12 @@ class Youtube extends OAuth2 {
 		return this[getYoutube](url, params);	
 	}
 
+	getViewers() {
+		return this[getViewers]();
+	}
+
 	liveChat() {
-		let url = `${this[urlsYoutube].chats}`;
+		let url = this[urlsYoutube].chats;
 		let params = {
 			part: 'id,snippet,authorDetails',
 			key: this[credentialsYoutube].key,
@@ -69,7 +74,7 @@ class Youtube extends OAuth2 {
 	}
 
 	liveBroadcast() {
-		let url = `${this[urlsYoutube].broadcasts}`;
+		let url = this[urlsYoutube].broadcasts;
 		let params = {
 			part: 'snippet',
 			broadcastType: 'persistent',
@@ -91,6 +96,11 @@ class Youtube extends OAuth2 {
 		    headers: {Authorization: `Bearer ${this.getCredentials().accessToken}`}
 		});
 	}
+
+	[getViewers]() {
+		let url = `https://www.youtube.com/live_stats?v=${this[credentialsYoutube].liveId}`;
+		return require('axios').get(url);
+	}
 }
 
 module.exports = Youtube;

tests/app.js

@@ -34,7 +34,8 @@ describe('youtube', function() {
 			channels: 'channels',
 			chats: 'liveChat/messages',
 			broadcasts: 'liveBroadcasts',
-			videos: 'videos'
+			videos: 'videos',
+			viewers: 'https://www.youtube.com/live_stats?v='
 		};
 	  });
 
@@ -153,6 +154,20 @@ describe('youtube', function() {
 		youtube.liveBroadcast().catch((err) => expect(500).to.equal(err.response.status));
 	});
 
+	it('getViewers() should get a count of viewers', () => {	
+		var viewers = 1240;
+
+		mock.onGet(urls.viewers).replyOnce(200, {response: viewers});
+
+		youtube.getViewers().then(() => expect(JSON.stringify(response)).to.equal(JSON.stringify(response)));
+	});
+
+	it('getViewers() should throw error with a message', () => {	
+		mock.onGet(urls.broadcasts).replyOnce(500, { error: 'error' });
+
+		youtube.getViewers().catch((err) => expect(500).to.equal(err.response.status));
+	});
+
 	it('getCredentials() should get credentials', () => {
 		var credentials = {
 			accessToken: 'token',

tests/integration.js

@@ -24,7 +24,11 @@ app.get('/getChannel', (req, res) => {
 
 app.get('/credentials', (req, res) => {
 	res.json(youtube.getCredentials());
-})
+});
+
+app.get('/getViewers', (req, res) => {
+	youtube.getViewers().then((result) => res.json(result.data)).catch((err) => res.json(err.response.data));
+});
 
 app.get('/', (req, res) => {
 	res.json(youtube.authorizationUrl());