101 lines
3.2 KiB
TypeScript
101 lines
3.2 KiB
TypeScript
'use strict';
|
|
import request = require('request');
|
|
import cheerio = require('cheerio');
|
|
|
|
var isAuthenticated = false;
|
|
var isPremium = false;
|
|
|
|
var defaultHeaders: request.Headers = {
|
|
'User-Agent': 'Mozilla/5.0 (Windows NT 6.2; WOW64; rv:34.0) Gecko/20100101 Firefox/34.0',
|
|
'Connection': 'keep-alive'
|
|
};
|
|
|
|
/**
|
|
* Performs a GET request for the resource.
|
|
*/
|
|
export function get(config: IConfig, options: request.Options, done: (err: Error, result?: string) => void) {
|
|
authenticate(config, err => {
|
|
if (err) return done(err);
|
|
request.get(modify(options), (err: Error, response: any, body: any) => {
|
|
if (err) return done(err);
|
|
done(null, typeof body === 'string' ? body : String(body));
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Performs a POST request for the resource.
|
|
*/
|
|
export function post(config: IConfig, options: request.Options, done: (err: Error, result?: string) => void) {
|
|
authenticate(config, err => {
|
|
if (err) return done(err);
|
|
request.post(modify(options), (err: Error, response: any, body: any) => {
|
|
if (err) return done(err);
|
|
done(null, typeof body === 'string' ? body : String(body));
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Authenticates using the configured pass and user.
|
|
*/
|
|
function authenticate(config: IConfig, done: (err: Error) => void) {
|
|
if (isAuthenticated || !config.pass || !config.user) return done(null);
|
|
|
|
/* Bypass the login page and send a login request directly */
|
|
var options =
|
|
{
|
|
headers: defaultHeaders,
|
|
jar: true,
|
|
gzip: false,
|
|
url: 'https://www.crunchyroll.com/?a=formhandler&formname=RpcApiUser_Login&name=' + config.user + '&password=' + config.pass
|
|
};
|
|
|
|
request(options, (err: Error, rep: string, body: string) =>
|
|
{
|
|
if (err) return done(err);
|
|
/* The page return with a meta based redirection, as we wan't to check that everything is fine, reload
|
|
* the main page. A bit convoluted, but more sure.
|
|
*/
|
|
var options =
|
|
{
|
|
headers: defaultHeaders,
|
|
jar: true,
|
|
url: 'http://www.crunchyroll.com/'
|
|
};
|
|
request(options, (err: Error, rep: string, body: string) =>
|
|
{
|
|
if (err) return done(err);
|
|
var $ = cheerio.load(body);
|
|
/* Check if auth worked */
|
|
var regexps = /ga\(\'set\', \'dimension[5-8]\', \'([^']*)\'\);/g;
|
|
var dims = regexps.exec($('script').text());
|
|
for (var i = 1; i < 5; i++)
|
|
{
|
|
if ((dims[i] !== undefined) && (dims[i] !== '') && (dims[i] !== 'not-registered')) { isAuthenticated = true; }
|
|
if ((dims[i] === 'premium') || (dims[i] === 'premiumplus')) { isPremium = true; }
|
|
}
|
|
if (isAuthenticated === false)
|
|
{
|
|
var error = $('ul.message, li.error').text();
|
|
return done(new Error('Authentication failed: ' + error));
|
|
}
|
|
if (isPremium === false) { console.log('Do not use this app without a premium account.'); }
|
|
else { console.log('You have a premium account! Good!'); }
|
|
done(null);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Modifies the options to use the authenticated cookie jar.
|
|
*/
|
|
function modify(options: string|request.Options): request.Options {
|
|
if (typeof options !== 'string') {
|
|
options.jar = true;
|
|
options.headers = defaultHeaders;
|
|
return options;
|
|
}
|
|
return {jar: true, headers: defaultHeaders, url: options.toString()};
|
|
}
|