#!/usr/bin/env python3
"""
Contains helper functions for accessing the database.
"""
import os
from urllib import parse

import config


async def get_random_track(request):
	"""Selects a random track."""
	async with request.app.state.db_pool.acquire() as conn:
		track = await conn.fetchrow(
			"SELECT * FROM track ORDER BY random() LIMIT 1"
		)
	track = dict(track)
	fpath = track.pop('filepath')
	art256 = os.path.join(os.path.dirname(fpath), 'folder-256x256.jpg')
	art64 = os.path.join(os.path.dirname(fpath), 'folder-64x64.jpg')
	track['source'] = convert_filepath(fpath)
	track['artwork'] = [{
		'src': convert_filepath(art256),
		'sizes': '256x256',
		'type': 'image/jpeg'
		}
	]
	return track


async def get_artists(request):
	"""Returns all artists in the database."""
	async with request.app.state.db_pool.acquire() as conn:
		artists = await conn.fetch(
			"SELECT DISTINCT albumartist FROM track ORDER BY albumartist"
		)
	artists = [a['albumartist'] for a in artists]
	return artists


async def get_albums(request, albumartist):
	"""Return all albums associated with a particular albumartist."""
	async with request.app.state.db_pool.acquire() as conn:
		albums = await conn.fetch(
			"SELECT DISTINCT ON (album, date) filepath, album, date "
			"FROM track "
			"WHERE albumartist = $1 ORDER BY date",
			albumartist
		)
	albums = [dict(a) for a in albums]
	for album in albums:
		fpath = album.pop('filepath')
		cpath = os.path.join(os.path.dirname(fpath), 'folder-64x64.jpg')
		album['cover_art_path'] = convert_filepath(cpath)
	return albums


async def get_tracks(request, albumartist, album, date):
	"""Select an album and return all associated tracks."""
	async with request.app.state.db_pool.acquire() as conn:
		tracks = await conn.fetch(
			"SELECT * FROM track "
			"WHERE albumartist = $1 AND "
			"album = $2 AND date = $3 "
			"ORDER BY discnumber ASC, tracknumber ASC",
			albumartist, album, date
		)
	tracks = [dict(t) for t in tracks]
	for track in tracks:
		track['filepath'] = convert_filepath(track['filepath'])
		track.pop('last_modified')
	return tracks


def convert_filepath(path):
	"""Convert a filepath to a URL."""
	path = os.path.relpath(path, config.music_dir)
	path = config.server_homepage + '/library/' + path
	#path = parse.quote(path)
	return path