« get me outta code hell

crawl-itunes.js « src - http-music - Command-line music player + utils (not a server!)
about summary refs log tree commit diff
path: root/src/crawl-itunes.js
blob: ec6c3ecb91ef9cf4121216ccc39521b10e47eb92 (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

const fs = require('fs')
const path = require('path')
const xmldoc = require('xmldoc')

const { promisify } = require('util')
const readFile = promisify(fs.readFile)

function getDictValue(dict, key) {
  if (dict.name !== 'dict') {
    throw new Error("Not a dict: " + dict.name)
  }

  for (let i = 0; i < dict.children.length; i++) {
    const child = dict.children[i]
    if (child.name === 'key') {
      if (child.val === key) {
        return dict.children.slice(i + 1).find(item => !item.text)
      }
    }
  }

  return null
}

async function crawl(libraryXML) {
  const document = new xmldoc.XmlDocument(libraryXML)

  const libraryDict = document.children.find(child => child.name === 'dict')

  const tracksDict = getDictValue(libraryDict, 'Tracks')

  const trackDicts = tracksDict.children.filter(child => child.name === 'dict')

  const result = []

  for (let trackDict of trackDicts) {
    let kind = getDictValue(trackDict, 'Kind')
    kind = kind && kind.val
    kind = kind || ''

    if (!kind.includes('audio file')) {
      continue
    }

    let location = getDictValue(trackDict, 'Location')
    location = location && location.val
    location = location || ''

    if (!location) {
      continue
    }

    let name = getDictValue(trackDict, 'Name')
    name = name && name.val
    name = name || 'Unknown Name'

    let album = getDictValue(trackDict, 'Album')
    album = album && album.val
    album = album || 'Unknown Album'

    let artist = getDictValue(trackDict, 'Artist')
    artist = artist && artist.val
    artist = artist || 'Unknown Artist'

    // console.log(`${artist} - ${name} (${album})`)

    const group = (arr, title) => arr.find(g => g[0] === title)

    let artistGroup = group(result, artist)

    if (!artistGroup) {
      artistGroup = [artist, []]
      result.push(artistGroup)
    }

    let albumGroup = group(artistGroup[1], album)

    if (!albumGroup) {
      albumGroup = [album, []]
      artistGroup[1].push(albumGroup)
    }

    albumGroup[1].push([name, location])
  }

  return result
}

async function main() {
  const libraryPath = process.argv[2] || (
    `${process.env.HOME}/Music/iTunes/iTunes Music Library.xml`
  )

  const library = await readFile(libraryPath)

  const playlist = await crawl(library)

  console.log(JSON.stringify(playlist, null, 2))
}

main()
  .catch(err => console.error(err))