pixiv-illust
Version:
A third-party Node.js library for Pixiv.
310 lines (250 loc) • 12.4 kB
JavaScript
'use strict';
Object.defineProperty(exports, "__esModule", {
value: true
});
var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
var _desc, _value, _class;
require('babel-polyfill');
var _login = require('./login');
var _login2 = _interopRequireDefault(_login);
var _utility = require('./utility');
var _Bluebird = require('Bluebird');
var _Bluebird2 = _interopRequireDefault(_Bluebird);
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new _Bluebird2.default(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return _Bluebird2.default.resolve(value).then(function (value) { step("next", value); }, function (err) { step("throw", err); }); } } return step("next"); }); }; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _applyDecoratedDescriptor(target, property, decorators, descriptor, context) {
var desc = {};
Object['ke' + 'ys'](descriptor).forEach(function (key) {
desc[key] = descriptor[key];
});
desc.enumerable = !!desc.enumerable;
desc.configurable = !!desc.configurable;
if ('value' in desc || desc.initializer) {
desc.writable = true;
}
desc = decorators.slice().reverse().reduce(function (desc, decorator) {
return decorator(target, property, desc) || desc;
}, desc);
if (context && desc.initializer !== void 0) {
desc.value = desc.initializer ? desc.initializer.call(context) : void 0;
desc.initializer = undefined;
}
if (desc.initializer === void 0) {
Object['define' + 'Property'](target, property, desc);
desc = null;
}
return desc;
}
var request = _Bluebird2.default.promisifyAll(require('request'));
var MODES = ['daily', 'weekly', 'monthly', 'rookie', 'male', 'female'];
var RESTRICT_MODES = ['daily_r18', 'weekly_r18', 'r18g', 'male_r18', 'female_r18'];
var PAGE = 'http://www.pixiv.net/ranking.php?format=json';
var Rank = (_class = function () {
function Rank() {
var mode = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 'daily';
var date = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : null;
_classCallCheck(this, Rank);
if (!MODES.includes(mode) && !RESTRICT_MODES.includes(mode)) {
throw Error('Unvalid mode.');
}
this.mode = mode;
// default 30 hours before.
this.date = (0, _utility.getDate)(date || new Date() - 30 * 3600 * 1000);
}
_createClass(Rank, [{
key: 'getPage',
value: function () {
var _ref = _asyncToGenerator(regeneratorRuntime.mark(function _callee() {
var page = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 1;
return regeneratorRuntime.wrap(function _callee$(_context) {
while (1) {
switch (_context.prev = _context.next) {
case 0:
if (!MODES.includes(this.mode)) {
_context.next = 6;
break;
}
_context.next = 3;
return this._getPage(page);
case 3:
return _context.abrupt('return', _context.sent);
case 6:
if (!RESTRICT_MODES.includes(this.mode)) {
_context.next = 10;
break;
}
_context.next = 9;
return this._getRestrictPage(page);
case 9:
return _context.abrupt('return', _context.sent);
case 10:
case 'end':
return _context.stop();
}
}
}, _callee, this);
}));
function getPage(_x3) {
return _ref.apply(this, arguments);
}
return getPage;
}()
}, {
key: '_getRestrictPage',
value: function () {
var _ref2 = _asyncToGenerator(regeneratorRuntime.mark(function _callee2() {
var page = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 1;
return regeneratorRuntime.wrap(function _callee2$(_context2) {
while (1) {
switch (_context2.prev = _context2.next) {
case 0:
return _context2.abrupt('return', this._getPage(page));
case 1:
case 'end':
return _context2.stop();
}
}
}, _callee2, this);
}));
function _getRestrictPage(_x5) {
return _ref2.apply(this, arguments);
}
return _getRestrictPage;
}()
}, {
key: '_getPage',
value: function () {
var _ref3 = _asyncToGenerator(regeneratorRuntime.mark(function _callee3(page) {
var resp;
return regeneratorRuntime.wrap(function _callee3$(_context3) {
while (1) {
switch (_context3.prev = _context3.next) {
case 0:
_context3.next = 2;
return request.getAsync({
url: this.url + '&p=' + page,
jar: _login2.default.cookieJar
});
case 2:
resp = _context3.sent;
if (!(resp.statusCode !== 200)) {
_context3.next = 5;
break;
}
throw resp;
case 5:
return _context3.abrupt('return', JSON.parse(resp.body));
case 6:
case 'end':
return _context3.stop();
}
}
}, _callee3, this);
}));
function _getPage(_x7) {
return _ref3.apply(this, arguments);
}
return _getPage;
}()
}, {
key: 'get',
value: function () {
var _ref4 = _asyncToGenerator(regeneratorRuntime.mark(function _callee4(page) {
return regeneratorRuntime.wrap(function _callee4$(_context4) {
while (1) {
switch (_context4.prev = _context4.next) {
case 0:
_context4.next = 2;
return this.getPage(page);
case 2:
return _context4.abrupt('return', _context4.sent);
case 3:
case 'end':
return _context4.stop();
}
}
}, _callee4, this);
}));
function get(_x8) {
return _ref4.apply(this, arguments);
}
return get;
}()
}, {
key: 'getRank',
value: function () {
var _ref5 = _asyncToGenerator(regeneratorRuntime.mark(function _callee5() {
var rank = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 500;
var contents, page, data, len;
return regeneratorRuntime.wrap(function _callee5$(_context5) {
while (1) {
switch (_context5.prev = _context5.next) {
case 0:
contents = [];
page = 1;
case 2:
if (!(page && contents.length < rank)) {
_context5.next = 13;
break;
}
_context5.next = 5;
return this.getPage(page);
case 5:
data = _context5.sent;
len = data && data.contents && data.contents.length;
if (len) {
_context5.next = 9;
break;
}
return _context5.abrupt('break', 13);
case 9:
page = data.next;
contents = contents.concat(data.contents);
_context5.next = 2;
break;
case 13:
return _context5.abrupt('return', contents.slice(0, rank));
case 14:
case 'end':
return _context5.stop();
}
}
}, _callee5, this);
}));
function getRank(_x9) {
return _ref5.apply(this, arguments);
}
return getRank;
}()
}, {
key: 'getAll',
value: function () {
var _ref6 = _asyncToGenerator(regeneratorRuntime.mark(function _callee6() {
return regeneratorRuntime.wrap(function _callee6$(_context6) {
while (1) {
switch (_context6.prev = _context6.next) {
case 0:
return _context6.abrupt('return', this.getRank(Infinity));
case 1:
case 'end':
return _context6.stop();
}
}
}, _callee6, this);
}));
function getAll() {
return _ref6.apply(this, arguments);
}
return getAll;
}()
}, {
key: 'url',
get: function get() {
return PAGE + '&mode=' + this.mode + '&date=' + this.date;
}
}]);
return Rank;
}(), (_applyDecoratedDescriptor(_class.prototype, '_getRestrictPage', [_login.loginRequired], Object.getOwnPropertyDescriptor(_class.prototype, '_getRestrictPage'), _class.prototype), _applyDecoratedDescriptor(_class.prototype, 'get', [_utility.cachedProperty], Object.getOwnPropertyDescriptor(_class.prototype, 'get'), _class.prototype), _applyDecoratedDescriptor(_class.prototype, 'getRank', [_utility.cachedProperty], Object.getOwnPropertyDescriptor(_class.prototype, 'getRank'), _class.prototype), _applyDecoratedDescriptor(_class.prototype, 'getAll', [_utility.cachedProperty], Object.getOwnPropertyDescriptor(_class.prototype, 'getAll'), _class.prototype)), _class);
exports.default = Rank;
module.exports = exports['default'];