下载专栏音频文件

Author: jjeejj

columnArticleList.js

@@ -4,7 +4,11 @@ const superagent = require('superagent');
 const utils = require('./utils');
 const path = require('path');
 const generaterPdf = require('./generaterPdf.js');
+const downloadAudio = require('./downloadAudio.js');
 
+/**
+ * 执行方法
+ */
 (async function getColumnArticleList (firstArticalId){
     await utils.createDir('geektime_'+config.columnName);
     console.log('专栏文章链接开始获取');
@@ -46,6 +50,14 @@ const generaterPdf = require('./generaterPdf.js');
                 columnArticle.article_title + '.pdf',
                 path.resolve(__dirname, 'geektime_' + config.columnName)
             );
+            // 是否下载音频
+            if (config.isdownloadVideo) {
+                await downloadAudio(
+                    columnArticle.audio_download_url,
+                    columnArticle.article_title + '.mp3',
+                    path.resolve(__dirname, 'geektime_' + config.columnName)
+                );
+            };
             // 判断是否还有下一篇文章
             let neighborRight = columnArticle.neighbors.right;
             if (neighborRight && neighborRight.id){

downloadAudio.js

@@ -0,0 +1,27 @@
+const path = require('path');
+const superagent = require('superagent');
+const fs = require('fs');
+/**
+ * 下载对应的音频文件
+ * @param {String} url 音频文件的地址
+ * @param {String} fileName 文件名称
+ * @param {String} fileDir 保存文件夹地址
+ */
+const downloadAudio = async (url, fileName, fileDir = __dirname) => {
+   console.log('开始下载 音频文件: ', fileName);
+   if (!url) throw '请传入一个音频地址';
+   if (path.extname(fileName) !== '.mp3'){ // 判断传的文件后缀是否是 mp3
+      fileName = fileName + '.mp3';
+   };
+   let filePath = path.resolve(fileDir, fileName);
+   let writeStream = fs.createWriteStream(filePath);
+   superagent.get(url).pipe(writeStream);
+   console.log('结束下载 音频文件: ', fileName);
+};
+
+// downloadAudio(
+//    'https://static001.geekbang.org/resource/audio/e7/b1/e7ffca8ca5b09224969b3237723a0bb1.mp3',
+//    '305 | 学会几个系统调用：咱们公司能接哪些类型的项目？.mp3',
+//    '/Users/jiang/Project/geektime2pdf/geektime_趣谈Linux操作系统')
+
+module.exports = downloadAudio;