#pragma once

#include <cstddef>
#include <cstdint>
#include <memory>

#ifdef PS2
#include <audsrv.h>
#else
#include <Aulib/Stream.h>
#endif

#include "engine/sound_defs.hpp"
#include "utils/stdcompat/shared_ptr_array.hpp"

namespace devilution {

class SoundSample final {
public:
	SoundSample() = default;
	SoundSample(SoundSample &&) noexcept = default;
	SoundSample &operator=(SoundSample &&) noexcept = default;

	[[nodiscard]] bool IsLoaded() const
	{
		return stream_ != nullptr;
	}

	void Release();
	bool IsPlaying();

	// Returns 0 on success.
	int SetChunkStream(std::string filePath, bool isMp3, bool logErrors = true);

#ifndef PS2
	void SetFinishCallback(Aulib::Stream::Callback &&callback)
	{
		stream_->setFinishCallback(std::forward<Aulib::Stream::Callback>(callback));
	}
#endif

	/**
	 * @brief Sets the sample's WAV, FLAC, or Ogg/Vorbis data.
	 * @param fileData Buffer containing the data
	 * @param dwBytes Length of buffer
	 * @param isMp3 Whether the data is an MP3
	 * @return 0 on success, -1 otherwise
	 */
	int SetChunk(ArraySharedPtr<std::uint8_t> fileData, std::size_t dwBytes, bool isMp3);

	[[nodiscard]] bool IsStreaming() const
	{
#ifndef PS2
		return file_data_ == nullptr;
#else
		return sampleId_ == nullptr;
#endif
	}

	int DuplicateFrom(const SoundSample &other)
	{
#ifndef PS2
		if (other.IsStreaming())
			return SetChunkStream(other.file_path_, other.isMp3_);
		return SetChunk(other.file_data_, other.file_data_size_, other.isMp3_);
#else
		if (other.IsStreaming())
			return SetChunkStream(other.file_path_, false);
		sampleId_ = other.sampleId_;
		return 0;
#endif
	}

	/**
	 * @brief Start playing the sound for a given number of iterations (0 means loop).
	 */
	bool Play(int numIterations = 1);

	/**
	 * @brief Start playing the sound with the given sound and user volume, and a stereo position.
	 */
	bool PlayWithVolumeAndPan(int logSoundVolume, int logUserVolume, int logPan)
	{
		SetVolume(logSoundVolume + logUserVolume * (ATTENUATION_MIN / VOLUME_MIN), ATTENUATION_MIN, 0);
		SetStereoPosition(logPan);
		return Play();
	}

	/**
	 * @brief Stop playing the sound
	 */
	void Stop()
	{
#ifndef PS2
		stream_->stop();
#else
		/** Hack: Implement way to stop sounds in PS2SDK */
		if (channel_ != -1)
			audsrv_adpcm_set_volume_and_pan(channel_, 0, pan_);
#endif
	}

	void SetVolume(int logVolume, int logMin, int logMax);
	void SetStereoPosition(int logPan);

	void Mute()
	{
#ifndef PS2
		stream_->mute();
#else
		if (channel_ != -1)
			audsrv_adpcm_set_volume_and_pan(channel_, 0, pan_);
#endif
	}

	void Unmute()
	{
#ifndef PS2
		stream_->unmute();
#else
		if (channel_ != -1)
			audsrv_adpcm_set_volume_and_pan(channel_, volume_, pan_);
#endif
	}

	/**
	 * @return Audio duration in ms
	 */
	int GetLength() const;

private:
	// Set for streaming audio to allow for duplicating it:
	std::string file_path_;

#ifndef PS2
	// Non-streaming audio fields:
	ArraySharedPtr<std::uint8_t> file_data_;
	std::size_t file_data_size_;

	bool isMp3_;

	std::unique_ptr<Aulib::Stream> stream_;
#else
	int channel_ = -1;
	int pan_ = 0;
	int volume_ = 100;
	audsrv_adpcm_t *sampleId_ = nullptr;
	std::unique_ptr<audsrv_adpcm_t> stream_;
#endif
};

} // namespace devilution