/* * mongo-edu * * Copyright (c) 2014 Przemyslaw Pluta * Licensed under the MIT license. * https://github.com/przemyslawpluta/mongo-edu/blob/master/LICENSE */ var request = require('request'), cheerio = require('cheerio'), ProgressBar = require('progress'), _ = require('lodash'), videoHandler = require('./videos'), coursewareHandler = require('./courseware'); var jar = request.jar(), host = 'https://university.mongodb.com'; function addCookies(cookies, url) { 'use strict'; _.each(cookies, function cookies(cookie) { jar.setCookie(cookie, url + '/login'); }); } module.exports = { init: function init(opt, argv, callback) { 'use strict'; var url = host, bar = new ProgressBar('[' + '>'.magenta + '] Searching [:bar] :percent', { complete: '=', incomplete: ' ', width: 20, total: 5 }), CSRFTokenCookie = '', login = function login(token) { request({ uri: url + '/login', method: 'POST', jar: jar, headers: { 'X-CSRFToken': token[0].split('=')[1] }, form: { 'email': opt.user, 'password': opt.password } }, function post(err, res, body) { if (err !== null) { return callback(err, null); } if (res.statusCode === 200) { var status = JSON.parse(body); if (status.success !== true) { return callback(new Error(status.value)); } bar.tick(); return checkCourses(bar); } callback(new Error(res.statusCode)); }); }, checkCourses = function checkCourses(bar) { bar.tick(); request({ url: url + '/dashboard', jar: jar }, function get(err, res, body) { if (err !== null) { return callback(err, null); } bar.tick(); if (res.statusCode === 200) { var list = [], $ = cheerio.load(body), current = $('article.my-course.panel').children(), link = $(current).children().filter('.span-7.offset-1').children().find('h3').children(); bar.tick(); $(link).each(function each(i, item) { list.push({ name: $(item).text(), value: url + $(item).attr('href').replace(/about|info|syllabus/g, '') + ((!argv.h) ? (argv.cw || argv.cwd) ? 'courseware' : 'course_wiki' : 'syllabus') }); }); bar.tick(); callback(null, list); } }); }; request(url, function (err, res) { if (err !== null) { return callback(err, null); } if (res.statusCode === 200) { addCookies(res.headers['set-cookie'], url); CSRFTokenCookie = res.headers['set-cookie'][0].split(';'); return login(CSRFTokenCookie); } callback(new Error(res.statusCode)); }); }, getList: function getList(opt, argv, callback) { 'use strict'; request({ url: opt.url, jar: jar }, function get(err, res, body) { if (err !== null) { return callback(err, null); } if (res.statusCode === 404) { return callback(null, [], true); } if (res.statusCode === 200) { var list = [], getCourses = [], $ = cheerio.load(body), options = (!argv.h)? 'Video|playlist' : 'Handout'; if (!argv.h) { if (argv.cw || argv.cwd) { return coursewareHandler.findDetails($, callback); } $('div.wiki-article p').children().filter('a').map(function map(i, item) { var current = $(item); if (current.text().match(options)) { list.push({ href: current.attr('href'), text: current.text() }); } }); getCourses = _.filter(list, function map(item) { if (item.href.match(/(wiki\/M101|wiki\/M102|wiki\/C100|wiki\/M202|wiki\/list-youtube-links|playlist\?list|view_play_list)/)) { return item; } }); return callback(null, _.map(getCourses, function map(item) { return { name: item.text, value: item.href }; })); } else { $('div.syllabus tbody tr').map(function map(i, item) { var current = $(item), text = current.children().first().text(), href = $(current.find('a')).attr('href'); if (href) { list.push({ name: text, value: host + $(current.find('a')).attr('href') }); } }); if (list.length) { list.unshift({name: 'All', value: 'all', checked: true}, {name: 'Cancel', value: 'cancel'}); } return callback(null, list, true); } } }); }, listVideos: function listVideos(opt, argv, callback) { 'use strict'; var videos = [], list, $, current; if (argv.cw || argv.cwd) { return coursewareHandler.filterVideos(opt, host, jar, argv, callback); } if (opt.url.match(/playlist\?list|view_play_list/)) { return videoHandler.listVideosFromPlaylist(opt, argv, callback); } request({ url: opt.url, jar: jar }, function get(err, res, body) { if (err !== null) { return callback(err, null); } if (res.statusCode === 200) { $ = cheerio.load(body); list = $('div.wiki-article'); current = list.find('code').text(); if (current.match(/http:\/\/|https:\/\//)) { videos = _.compact(current.split('\n')); } if (!videos.length) { list.children('p').children().each(function each(i, item) { var current = $(item), href = current.attr('href'); if (href && href.match(/http:\/\/|https:\/\//)) { videos.push(href); } }); } return callback(null, videos); } callback(new Error(res.statusCode)); }); } };