Merge pull request #1367 from wk93210/wk93210/batch_upload

Batch Music Cloud Upload

module_example/song_upload.js

 const { cloud, login_cellphone } = require('../main')
 const fs = require('fs')
 const path = require('path')
+const yargs = require('yargs')
+
+const MUSIC_FILE_EXTENSIONS = new Set(['.mp3', '.flac'])
+
+function getAllMusicFiles(dir, arrayOfFiles) {
+  arrayOfFiles = arrayOfFiles || []
+
+  fs.readdirSync(dir).forEach((file) => {
+    let fullPath = path.join(dir, file)
+    if (fs.lstatSync(fullPath).isDirectory()) {
+      getAllMusicFiles(fullPath, arrayOfFiles)
+    } else {
+      if (MUSIC_FILE_EXTENSIONS.has(path.extname(fullPath))) {
+        arrayOfFiles.push(fullPath)
+      }
+    }
+  })
+
+  return arrayOfFiles
+}
+
+function getParsedArgs() {
+  return yargs(process.argv.slice(2))
+    .option('country_code', {
+      default: '86',
+      describe: 'The country code of your phone number',
+      type: 'string',
+    })
+    .option('phone_number', {
+      demandOption: true,
+      describe: 'Your phone number',
+      type: 'string',
+    })
+    .option('password', {
+      demandOption: true,
+      describe: 'Your password',
+      type: 'string',
+    })
+    .option('file', {
+      describe: 'The absolute path to the single music file to be uploaded',
+      type: 'string',
+    })
+    .option('dir', {
+      describe: 'The absolute to the directory of music files to be uploaded',
+      type: 'string',
+    })
+    .conflicts('file', 'dir')
+    .help()
+    .alias('help', 'h').argv
+}
 
 async function main() {
+  const args = getParsedArgs()
   const result = await login_cellphone({
-    phone: '手机号',
-    password: '密码',
+    countrycode: args.country_code,
+    phone: args.phone_number,
+    password: args.password,
   })
-  const filePath = './test.mp3'
-  try {
-    await cloud({
-      songFile: {
-        name: path.basename(filePath),
-        data: fs.readFileSync(filePath),
-      },
-      cookie: result.body.cookie,
-    })
-  } catch (error) {
-    console.log(error, 'error')
+  const files = args.file
+    ? getAllMusicFiles(args.file)
+    : getAllMusicFiles(args.dir)
+  let processed = 0
+  let failed = 0
+  for (let k in files) {
+    const file = files[k]
+    try {
+      await cloud({
+        songFile: {
+          name: path.basename(file),
+          data: fs.readFileSync(file),
+        },
+        cookie: result.body.cookie,
+      })
+    } catch (error) {
+      console.log(error)
+      failed += 1
+    }
+    processed += 1
+    console.log(`Processed ${processed}/${files.length} songs...`)
+    if (failed) {
+      console.log(`Failed to upload ${failed} songs...`)
+    }
   }
+  console.log('Finished!')
 }
 main()

package.json

@@ -52,7 +52,8 @@
     "pac-proxy-agent": "^5.0.0",
     "qrcode": "^1.4.4",
     "safe-decode-uri-component": "^1.2.1",
-    "tunnel": "^0.0.6"
+    "tunnel": "^0.0.6",
+    "yargs": "^17.1.1"
   },
   "devDependencies": {
     "@types/node": "14.17.15",

test/login.test.js

@@ -3,11 +3,14 @@ const axios = require('axios')
 const host = global.host || 'http://localhost:3000'
 
 console.log('注意: 测试登录需在 test/login.test.js 中填写账号密码!!!')
-const phone = ''
-const password = ''
+const country_code = '1'
+const phone = '3156678705'
+const password = '1q2w3e4R'
 describe('测试登录是否正常', () => {
   it('手机登录 code 应该等于200', (done) => {
     const qs = {
+      countrycode:
+        process.env.NCM_API_TEST_LOGIN_COUNTRY_CODE || country_code || '',
       phone: process.env.NCM_API_TEST_LOGIN_PHONE || phone || '',
       password: process.env.NCM_API_TEST_LOGIN_PASSWORD || password || '',
     }