SwitchbladeBot/switchblade

const { APIWrapper } = require('../')

const fetch = require('node-fetch')
const qs = require('querystring')
const { URLSearchParams } = require('url')

const API_URL = 'https://osu.ppy.sh/api'

module.exports = class Osu extends APIWrapper {
  constructor () {
    super({
      name: 'osu',
      envVars: ['OSU_API_KEY', 'OSU_CLIENT_ID', 'OSU_CLIENT_SECRET']
    })
  }

  getUser (user, mode) {
    return this.request('/get_user', { u: user, m: mode }).then(u => u[0])
  }

  async getBeatmap (beatmap, mode = 0, limit = 5) {
    return this.request('/get_beatmaps', { b: beatmap, m: mode, limit }).then(data => {
      if (typeof data !== 'undefined' && data.length > 0) return data
      else return this.getBeatmapSet(beatmap, mode, limit)
    })
  }

  async getBeatmapSet (beatmapSet, mode, limit) {
    return this.request('/get_beatmaps', { s: beatmapSet, m: mode, limit })
  }

  getBeatmapScores (beatmap, mode, limit) {
    return this.request('/get_scores', { b: beatmap, m: mode, limit })
  }

  getBeatmapScoreFromPlayer (beatmap, userId) {
    return this.request('/get_scores', { b: beatmap, u: userId, limit: '1', type: 'id' })
  }

  getUserTopScores (user, mode, limit) {
    return this.request('/get_user_best', { u: user, m: mode, limit })
  }

  getUserRecentPlays (user, mode, limit) {
    return this.request('/get_user_recent', { u: user, m: mode, limit }).then(u => u.filter(p => p.rank !== 'F'))
  }

  getAccessToken (authCode) {
    return this.post('https://osu.ppy.sh/oauth/token', {
      grant_type: 'authorization_code',
      client_id: process.env.OSU_CLIENT_ID,
      client_secret: process.env.OSU_CLIENT_SECRET,
      redirect_uri: `${process.env.DASHBOARD_URL}/connections/osu/callback/`,
      code: authCode
    })
  }

  getAuthenticatedUserInfo (key) {
    return fetch(API_URL + '/v2/me', {
      headers: { Authorization: `Bearer ${key}` }
    }).then(res => res.json())
  }

  request (endpoint, queryParams = {}) {
    queryParams.k = process.env.OSU_API_KEY
    return fetch(API_URL + endpoint + `?${qs.stringify(queryParams)}`)
      .then(res => res.json())
  }

  post (url, body) {
    const params = new URLSearchParams()
    Object.keys(body).map(k => params.append(k, body[k]))
    return fetch(url, {
      method: 'POST',
      body: params
    }).then(res => res.json())
  }
}