« get me outta code hell

process-metadata.js « src - http-music - Command-line music player + utils (not a server!)
about summary refs log tree commit diff
path: root/src/process-metadata.js
blob: 903a73c85f3d2ffc7f84d583c8fee03e9764ace9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
const fs = require('fs')
const processArgv = require('./process-argv')
const promisifyProcess = require('./promisify-process')
const { spawn } = require('child_process')
const { promisify } = require('util')
const { showTrackProcessStatus } = require('./general-util')
const { updatePlaylistFormat, flattenGrouplike } = require('./playlist-utils')
const { processSmartPlaylist } = require('./smart-playlist')

const readFile = promisify(fs.readFile)
const writeFile = promisify(fs.writeFile)

async function probe(filePath) {
  const ffprobe = spawn('ffprobe', [
    '-print_format', 'json',
    '-show_entries', 'stream=codec_name:format',
    '-select_streams', 'a:0',
    '-v', 'quiet',
    filePath
  ])

  let probeDataString = ''

  ffprobe.stdout.on('data', data => {
    probeDataString += data
  })

  await promisifyProcess(ffprobe, false)

  return JSON.parse(probeDataString)
}

async function main(args) {
  if (args.length < 2) {
    console.error('Usage: http-music process-metadata <in> <out> (..args..)')
    console.error('See \x1b[1mman http-music-process-metadata\x1b[0m!')
    return false
  }

  const inFile = args[0]
  const outFile = args[1]

  // Whether or not to save actual audio tag data. (This includes things like
  // genre, track #, and album, as well as any non-standard data set on the
  // file.)
  let saveTags = false

  // Whether or not to skip tracks which have already been processed.
  let skipCompleted = true

  await processArgv(args.slice(1), {
    '-save-tags': function() {
      saveTags = true
    },

    '-tags': util => util.alias('-save-tags'),
    't': util => util.alias('-save-tags'),

    '-skip-completed': function() {
      skipCompleted = true
    },

    '-skip-done': util => util.alias('-skip-completed'),
    '-faster': util => util.alias('-skip-completed'),

    '-no-skip-completed': function() {
      skipCompleted = false
    },

    '-no-skip-done': util => util.alias('-no-skip-completed'),
    '-slower': util => util.alias('-no-skip-completed')
  })

  let doneCount = 0

  const playlist = await processSmartPlaylist(JSON.parse(await readFile(args[0])))

  const flattened = flattenGrouplike(playlist)
  for (const item of flattened.items) {
    if (!(skipCompleted && 'metadata' in item)) {
      const probeData = await probe(item.downloaderArg)

      item.metadata = Object.assign(item.metadata || {}, {
        duration: parseInt(probeData.format.duration),
        size: parseInt(probeData.format.size),
        bitrate: parseInt(probeData.format.bit_rate)
      })

      if (saveTags) {
        item.metadata.tags = probeData.tags
      }
    }

    doneCount++
    showTrackProcessStatus(flattened.items.length, doneCount, true)
    process.stdout.write('   \r')
  }

  await writeFile(outFile, JSON.stringify(playlist, null, 2))

  console.log(`\nDone! Processed ${flattened.items.length} tracks.`)
}

module.exports = main

if (require.main === module) {
  main(process.argv.slice(2))
    .catch(err => console.error(err))
}