import { Track } from "@db_models"
import Library from "@classes/library"

async function fullfillData(list, { user_id = null }) {
	if (!Array.isArray(list)) {
		list = [list]
	}

	// if user_id is provided, fetch likes
	if (user_id) {
		const trackIds = list.map((track) => {
			return track._id
		})

		const tracksLikes = await Library.isFavorite(
			user_id,
			trackIds,
			"tracks",
		)

		list = list.map(async (track) => {
			const trackLike = tracksLikes.find((trackLike) => {
				return trackLike.item_id.toString() === track._id.toString()
			})

			if (trackLike) {
				track.liked_at = trackLike.created_at
				track.liked = trackLike.liked
			}

			return track
		})

		list = await Promise.all(list)
	} else {
		list = list.map((track) => {
			delete track.source
			delete track.publisher

			return track
		})
	}

	return list
}

export default async (track_id, { user_id = null, onlyList = false } = {}) => {
	if (!track_id) {
		throw new OperationError(400, "Missing track_id")
	}

	const isMultiple = Array.isArray(track_id) || track_id.includes(",")

	if (isMultiple) {
		const track_ids = Array.isArray(track_id)
			? track_id
			: track_id.split(",")

		let tracks = await Track.find({
			_id: { $in: track_ids },
		}).lean()

		tracks = await fullfillData(tracks, {
			user_id,
		})

		if (onlyList) {
			return tracks
		}

		return {
			total_count: await Track.countDocuments({
				_id: { $in: track_ids },
			}),
			list: tracks,
		}
	}

	let track = await Track.findOne({
		_id: track_id,
	}).lean()

	if (!track) {
		throw new OperationError(404, "Track not found")
	}

	track = await fullfillData(track, {
		user_id,
	})

	return track[0]
}