index.js

#! /usr/bin/env node

'use strict';

require('app-module-path').addPath(__dirname);

const request = require('request');
const cheerio = require('cheerio');
const https = require('https');
const fs = require('fs');
const download = require('download');
const path = require('path');
const progress = require('request-progress');
const colors = require('colors');
const readline = require('readline');

const validateParams = require('src/validate/validateParams');
const getFlagIndex = require('src/validate/getFlagIndex');
const createFolder = require('src/create/createFolder');
const createLogger = require('src/create/createLogger');
const downloadVideos = require('src/download/downloadVideos');
const downloadMaterials = require('src/download/downloadMaterials');
const getVideos = require('src/download/getVideos.js');
const versionCheck = require('github-version-checker');
const pkg = require('./package.json');
const getToken = require('./src/download/getToken');

function printHeader() {
  console.log('#############################'.green);
  console.log(`###  CH DOWNLOADER ${pkg.version}  ###`.green);
  console.log('### this is alpha version ###'.toUpperCase().green);
  console.log('#############################\n'.green);
}

function checkNewVersion(startDownloading) {
  const options = {
    repo: 'alekseylovchikov/ch-download',
    currentVersion: pkg.version,
    includePreReleases: true
  };

  versionCheck(options, function (update, error) {
    if (error) throw error;
    if (update) {
      console.log('An update is available!');
      console.log('New version: ' + update.tag_name);
      console.log('Details here: ' + update.html_url);
    }

    console.log('Starting app...\n');
    startDownloading();
  });
}

function startDownloading() {
  // get email password
  const e = process.argv.indexOf('-e');
  const p = process.argv.indexOf('-p');
  if (e > -1 && p > -1) {
    // with email and password
    const email = process.argv[e + 1];
    const password = process.argv[p + 1];
    getToken(email, password)
      .then(token => runGetVideos(token))
      .catch(err => console.log('Check your email or password'.red));
  } else {
    // without email and password
    runGetVideos();
  }
}

function runGetVideos(token) {
  const courseUrl = process.argv[indexUrlFlag + 1];
  const downloadFolder = (indexDirFlag == -1) ? getLastSegment(courseUrl) : process.argv[indexDirFlag + 1];

  createFolder(downloadFolder);

  const logger = createLogger(downloadFolder);

  const lessonNumbers = (indexLessonsFlag === -1) ? null : getLessonNumbers(process.argv[indexLessonsFlag + 1]);
  const videos = [];

  getVideos(courseUrl, token)
    .then(data => {
      data.result.map((url, i) => {
        const name = data.names[i].toString().replace(/[^A-Za-zА-Яа-я\d\s]/gmi, ''); // alelov
        videos.push({ url, name });
      });
      if (data.urlMaterials !== undefined) {
        downloadMaterials(data.urlMaterials, downloadFolder);
      }
      console.log('Start download videos, please wait...');
      downloadVideos(logger, videos, downloadFolder, lessonNumbers);
    })
    .catch(err => console.log(`${err}`.red));
}

function getLastSegment(url) {
  let parts = url.split('/');
  return parts.pop() || parts.pop(); // handle potential trailing slash
}

function getLessonNumbers(lessonsString) {
  let lessonNumbers = [];
  const regExpComma = /\s*,\s*/,
        regExpDash = /\s*-\s*/,
        lessonList = lessonsString.split(regExpComma);
  for (let item of lessonList) {
    const dashCounter = (item.match(/-/g) || []).length;
    if (dashCounter === 1) {
      const periodList = item.split(regExpDash);
      let firstNumber = Number(periodList[0]),
          lastNumber = Number(periodList[1]),
          buf = 0;
      if (firstNumber > lastNumber) {
        buf = firstNumber;
        firstNumber = lastNumber;
        lastNumber = buf;
      }
      for (let i = firstNumber; i <= lastNumber; i++) {
        lessonNumbers.push(i);
      }
    } else if (dashCounter === 0) {
      lessonNumbers.push(Number(item));
    }
  }
  lessonNumbers = lessonNumbers.sort(function (a, b) {
      return a - b;
    })
    .filter(function (value, index, self) {
      return self.indexOf(value) === index;
    });
  return lessonNumbers;
}

const url = 'url';
const dirName = 'dirName';
let flags = { url: ['--url', '-u'], dirName: ['--dir', '-d'], lessons: ['--lessons', '-l'] };
const indexUrlFlag = getFlagIndex(flags.url);
const indexDirFlag = getFlagIndex(flags.dirName);
const indexLessonsFlag = getFlagIndex(flags.lessons);

validateParams(flags, indexUrlFlag, indexDirFlag);
printHeader();
checkNewVersion(startDownloading);