UNPKG

gogo-dl

Version:

<h1 align="center">gogo-dl</h1> <p align="center"> <b> Powerful, lightweight, user-friendly tool for downloading anime.</b> </p>

231 lines (230 loc) 13.4 kB
"use strict"; var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } return new (P || (P = Promise))(function (resolve, reject) { function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } step((generator = generator.apply(thisArg, _arguments || [])).next()); }); }; var __generator = (this && this.__generator) || function (thisArg, body) { var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g; return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g; function verb(n) { return function (v) { return step([n, v]); }; } function step(op) { if (f) throw new TypeError("Generator is already executing."); while (_) try { if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t; if (y = 0, t) op = [op[0] & 2, t.value]; switch (op[0]) { case 0: case 1: t = op; break; case 4: _.label++; return { value: op[1], done: false }; case 5: _.label++; y = op[1]; op = [0]; continue; case 7: op = _.ops.pop(); _.trys.pop(); continue; default: if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; } if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; } if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; } if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; } if (t[2]) _.ops.pop(); _.trys.pop(); continue; } op = body.call(thisArg, _); } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; } if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true }; } }; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); exports.dl = void 0; var Aria2 = require("aria2"); var aria2 = new Aria2(); var spawn = require('child_process').spawn; var path = require('path'); var chalk_1 = __importDefault(require("chalk")); var driver_1 = require("../driver/driver"); var utils_1 = require("../utils/utils"); var ConfigFile_1 = __importDefault(require("../utils/ConfigFile")); var cliProgress = require('cli-progress'); var commandExists = require('command-exists'); exports.dl = { download: function (anime, lower, upper, type) { return __awaiter(this, void 0, void 0, function () { function delay(ms) { return new Promise(function (resolve) { return setTimeout(resolve, ms); }); } function update_bar(results, i, bars) { return __awaiter(this, void 0, void 0, function () { var speed; return __generator(this, function (_a) { switch (_a.label) { case 0: if (!(bars[i] !== null)) return [3 /*break*/, 2]; if (parseInt(results[i][0].totalLength) != 0) { speed = chalk_1.default.yellowBright('waiting'); if (parseInt(results[i][0].downloadSpeed) !== 0) speed = parseInt(((results[i][0].totalLength - results[i][0].completedLength) / parseInt(results[i][0].downloadSpeed)).toFixed(0)) + 's'; bars[i].update(parseFloat(((results[i][0].completedLength / results[i][0].totalLength) * 100).toFixed(2)), { et: speed }); } if (!(results[i][0].status === 'complete')) return [3 /*break*/, 2]; return [4 /*yield*/, bars[i].update(100, { et: chalk_1.default.greenBright('completed') })]; case 1: _a.sent(); bars[i] = null; _a.label = 2; case 2: return [2 /*return*/]; } }); }); } var multibar, aria_bin, e_1, cli, multi, bars, wait_for, nconf, _a, i, url, name_1, temp_id, gap, tbar, gap, tbar; return __generator(this, function (_b) { switch (_b.label) { case 0: multibar = new cliProgress.MultiBar({ clearOnComplete: false, hideCursor: false, format: 'ep: {ep}{gap}|' + chalk_1.default.cyanBright('[{bar}]') + ' {percentage}% | ETA: {et} | {value}/{total}' }); aria_bin = ''; _b.label = 1; case 1: _b.trys.push([1, 6, , 7]); if (!process.platform.startsWith('win')) return [3 /*break*/, 3]; return [4 /*yield*/, commandExists('aria2c.exe')]; case 2: _b.sent(); aria_bin = 'aria2c.exe'; return [3 /*break*/, 5]; case 3: return [4 /*yield*/, commandExists('aria2c')]; case 4: _b.sent(); aria_bin = 'aria2c'; _b.label = 5; case 5: return [3 /*break*/, 7]; case 6: e_1 = _b.sent(); console.log(chalk_1.default.redBright('aria2 is required to download videos download at https://github.com/aria2/aria2/releases')); process.exit(34); return [3 /*break*/, 7]; case 7: cli = spawn(aria_bin, ['--enable-rpc', '--rpc-listen-all=true', '--rpc-allow-origin-all' /*,'--rpc-secret='+secret*/]); cli.stdout.on('data', function (data) { //console.log(data.toString()); }); process.on('SIGINT', function () { return __awaiter(this, void 0, void 0, function () { return __generator(this, function (_a) { switch (_a.label) { case 0: return [4 /*yield*/, cli.kill('SIGINT')]; case 1: _a.sent(); process.exit(0); return [2 /*return*/]; } }); }); }); multi = []; bars = []; wait_for = new Promise(function (resolve) { setTimeout(function () { return __awaiter(this, void 0, void 0, function () { var downloading, results, i; return __generator(this, function (_a) { switch (_a.label) { case 0: downloading = true; _a.label = 1; case 1: if (!downloading) return [3 /*break*/, 8]; return [4 /*yield*/, delay(100)]; case 2: _a.sent(); return [4 /*yield*/, aria2.multicall(multi)]; case 3: results = _a.sent(); i = 0; _a.label = 4; case 4: if (!(i < results.length)) return [3 /*break*/, 7]; return [4 /*yield*/, update_bar(results, i, bars)]; case 5: _a.sent(); if (bars.length >= (1 + upper - lower) && bars.every(function (element) { return element === null; })) { downloading = false; return [3 /*break*/, 7]; } _a.label = 6; case 6: i++; return [3 /*break*/, 4]; case 7: return [3 /*break*/, 1]; case 8: resolve(downloading); return [2 /*return*/]; } }); }); }, 0); }); return [4 /*yield*/, delay(2000)]; case 8: _b.sent(); _a = ConfigFile_1.default.bind; return [4 /*yield*/, utils_1.utils.getConfigPath()]; case 9: nconf = new (_a.apply(ConfigFile_1.default, [void 0, _b.sent()]))(); i = lower; _b.label = 10; case 10: if (!(i <= upper)) return [3 /*break*/, 15]; return [4 /*yield*/, driver_1.driver.getOptimizedPlayer(anime, i, type) //let url = 'https://file-examples-com.github.io/uploads/2017/04/file_example_MP4_480_1_5MG.mp4' ]; case 11: url = _b.sent(); if (!(url !== null && url !== undefined)) return [3 /*break*/, 14]; name_1 = anime.name.trim().split(':').join(' ') /*.split('/').join('-').split(' ').join('-')*/; if (!(url !== '' && !url.endsWith('.m3u8'))) return [3 /*break*/, 13]; return [4 /*yield*/, aria2.call('addUri', [url], { dir: path.join(process.cwd(), name_1), out: name_1 + ' episode ' + i + '.mp4' })]; case 12: temp_id = _b.sent(); if (nconf.get('token') !== undefined) { utils_1.utils.getMal().update_list(anime.id, { num_watched_episodes: i }); } multi.push(["tellStatus", temp_id]); gap = ' '.repeat(3 - ('' + i).length); tbar = multibar.create(100, 0, { ep: i, gap: gap, et: chalk_1.default.yellowBright('waiting') }); if (tbar !== undefined) bars.push(tbar); return [3 /*break*/, 14]; case 13: gap = ' '.repeat(3 - ('' + i).length); tbar = (multibar.create(100, 0, { ep: i, gap: gap, et: chalk_1.default.red('failed') })); if (tbar !== undefined) bars.push(tbar); _b.label = 14; case 14: i++; return [3 /*break*/, 10]; case 15: return [4 /*yield*/, wait_for]; case 16: _b.sent(); return [4 /*yield*/, delay(1000)]; case 17: _b.sent(); return [4 /*yield*/, cli.kill('SIGINT')]; case 18: _b.sent(); console.log(chalk_1.default.greenBright('\ndownloads complete!')); process.exit(0); return [2 /*return*/]; } }); }); } };