graphbrainz/src/extensions/the-audio-db/client.js

import Client from '../../api/client'

export default class TheAudioDBClient extends Client {
  constructor ({
    apiKey = process.env.THEAUDIODB_API_KEY,
    baseURL = process.env.THEAUDIODB_BASE_URL || 'http://www.theaudiodb.com/api/v1/json/',
    limit = 10,
    period = 1000,
    ...options
  } = {}) {
    super({ baseURL, limit, period, ...options })
    this.apiKey = apiKey
  }

  get (path, options = {}) {
    const ClientError = this.errorClass
    if (!this.apiKey) {
      return Promise.reject(new ClientError(
        'No API key was configured for TheAudioDB client.'
      ))
    }
    return super.get(`${this.apiKey}/${path}`, { json: true, ...options })
  }

  entity (entityType, mbid) {
    const ClientError = this.errorClass
    switch (entityType) {
      case 'artist':
        return this.artist(mbid)
      case 'release-group':
        return this.album(mbid)
      case 'recording':
        return this.track(mbid)
      default:
        return Promise.reject(new ClientError(
          `Entity type unsupported: ${entityType}`
        ))
    }
  }

  artist (mbid) {
    return this.get('artist-mb.php', { qs: { i: mbid } })
      .then(body => {
        if (body.artists && body.artists.length === 1) {
          return body.artists[0]
        }
        return null
      })
  }

  album (mbid) {
    return this.get('album-mb.php', { qs: { i: mbid } })
      .then(body => {
        if (body.album && body.album.length === 1) {
          return body.album[0]
        }
        return null
      })
  }

  track (mbid) {
    return this.get('track-mb.php', { qs: { i: mbid } })
      .then(body => {
        if (body.track && body.track.length === 1) {
          return body.track[0]
        }
        return null
      })
  }
}