/*
This file is part of Telegram Desktop,
the official desktop application for the Telegram messaging service.

For license and copyright information please follow this link:
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
*/
#include "media/player/media_player_instance.h"

#include "data/data_document.h"
#include "data/data_session.h"
#include "data/data_changes.h"
#include "data/data_streaming.h"
#include "data/data_file_click_handler.h"
#include "base/options.h"
#include "base/random.h"
#include "base/power_save_blocker.h"
#include "media/audio/media_audio.h"
#include "media/audio/media_audio_capture.h"
#include "media/streaming/media_streaming_instance.h"
#include "media/streaming/media_streaming_player.h"
#include "media/view/media_view_playback_progress.h"
#include "calls/calls_instance.h"
#include "history/history.h"
#include "history/history_item.h"
#include "data/data_media_types.h"
#include "data/data_file_origin.h"
#include "core/shortcuts.h"
#include "core/application.h"
#include "core/core_settings.h"
#include "window/window_controller.h"
#include "mainwindow.h"
#include "main/main_domain.h" // Domain::activeSessionValue.
#include "main/main_session.h"
#include "main/main_account.h" // session->account().sessionChanges().
#include "main/main_session_settings.h"

namespace Media {
namespace Player {
namespace {

Instance *SingleInstance = nullptr;

// Preload X message ids before and after current.
constexpr auto kIdsLimit = 32;

// Preload next messages if we went further from current than that.
constexpr auto kIdsPreloadAfter = 28;

constexpr auto kShufflePlaylistLimit = 10'000;
constexpr auto kRememberShuffledOrderItems = 16;

constexpr auto kMinLengthForSavePosition = 20 * TimeId(60); // 20 minutes.

base::options::toggle OptionDisableAutoplayNext({
	.id = kOptionDisableAutoplayNext,
	.name = "Disable auto-play of the next track",
	.description = "Disable auto-play of the next "
		"Audio file / Voice Message / Video message.",
});

} // namespace

const char kOptionDisableAutoplayNext[] = "disable-autoplay-next";

struct Instance::Streamed {
	Streamed(
		AudioMsgId id,
		std::shared_ptr<Streaming::Document> document);

	AudioMsgId id;
	Streaming::Instance instance;
	View::PlaybackProgress progress;
	bool clearing = false;
	rpl::lifetime lifetime;
};

struct Instance::ShuffleData {
	using UniversalMsgId = MsgId;

	std::vector<UniversalMsgId> playlist;
	std::vector<UniversalMsgId> nonPlayedIds;
	std::vector<UniversalMsgId> playedIds;
	History *history = nullptr;
	MsgId topicRootId = 0;
	History *migrated = nullptr;
	bool scheduled = false;
	int indexInPlayedIds = 0;
	bool allLoaded = false;
	rpl::lifetime nextSliceLifetime;
	rpl::lifetime lifetime;
};

void start(not_null<Audio::Instance*> instance) {
	Audio::Start(instance);
	Capture::Start();

	SingleInstance = new Instance();
}

void finish(not_null<Audio::Instance*> instance) {
	delete base::take(SingleInstance);

	Capture::Finish();
	Audio::Finish(instance);
}

void SaveLastPlaybackPosition(
		not_null<DocumentData*> document,
		const TrackState &state) {
	const auto time = (state.position == kTimeUnknown
		|| state.length == kTimeUnknown
		|| state.state == State::PausedAtEnd
		|| IsStopped(state.state))
		? TimeId(0)
		: (state.length >= kMinLengthForSavePosition * state.frequency)
		? (state.position / state.frequency) * crl::time(1000)
		: TimeId(0);
	auto &session = document->session();
	if (session.settings().mediaLastPlaybackPosition(document->id) != time) {
		session.settings().setMediaLastPlaybackPosition(document->id, time);
		session.saveSettingsDelayed();
	}
}

Instance::Streamed::Streamed(
	AudioMsgId id,
	std::shared_ptr<Streaming::Document> document)
: id(id)
, instance(std::move(document), nullptr) {
}

Instance::Data::Data(AudioMsgId::Type type, SharedMediaType overview)
: type(type)
, overview(overview) {
}

Instance::Data::Data(Data &&other) = default;
Instance::Data &Instance::Data::operator=(Data &&other) = default;
Instance::Data::~Data() = default;

Instance::Instance()
: _songData(AudioMsgId::Type::Song, SharedMediaType::MusicFile)
, _voiceData(AudioMsgId::Type::Voice, SharedMediaType::RoundVoiceFile) {
	Media::Player::Updated(
	) | rpl::start_with_next([=](const AudioMsgId &audioId) {
		handleSongUpdate(audioId);
	}, _lifetime);

	repeatChanges(
		&_songData
	) | rpl::start_with_next([=](RepeatMode mode) {
		if (mode == RepeatMode::All) {
			refreshPlaylist(&_songData);
		}
	}, _lifetime);

	orderChanges(
		&_songData
	) | rpl::start_with_next([=](OrderMode mode) {
		if (mode == OrderMode::Shuffle) {
			validateShuffleData(&_songData);
		} else {
			_songData.shuffleData = nullptr;
		}
	}, _lifetime);

	using namespace rpl::mappers;
	rpl::combine(
		Core::App().calls().currentCallValue(),
		Core::App().calls().currentGroupCallValue(),
		_1 || _2
	) | rpl::start_with_next([=](bool call) {
		if (call) {
			pauseOnCall(AudioMsgId::Type::Voice);
			pauseOnCall(AudioMsgId::Type::Song);
		} else {
			resumeOnCall(AudioMsgId::Type::Voice);
			resumeOnCall(AudioMsgId::Type::Song);
		}
	}, _lifetime);

	setupShortcuts();
}

Instance::~Instance() = default;

AudioMsgId::Type Instance::getActiveType() const {
	if (const auto data = getData(AudioMsgId::Type::Voice)) {
		if (data->current) {
			const auto state = getState(data->type);
			if (!IsStoppedOrStopping(state.state)) {
				return data->type;
			}
		}
	}
	return AudioMsgId::Type::Song;
}

void Instance::handleSongUpdate(const AudioMsgId &audioId) {
	emitUpdate(audioId.type(), [&](const AudioMsgId &playing) {
		return (audioId == playing);
	});
}

void Instance::setCurrent(const AudioMsgId &audioId) {
	if (const auto data = getData(audioId.type())) {
		if (data->current == audioId) {
			return;
		}
		const auto changed = [&](const AudioMsgId & check) {
			return (check.audio() != audioId.audio())
				|| (check.contextId() != audioId.contextId());
		};
		if (changed(data->current)
			&& data->streamed
			&& changed(data->streamed->id)) {
			clearStreamed(data);
		}
		data->current = audioId;
		data->isPlaying = false;

		const auto item = (audioId.audio() && audioId.contextId())
			? audioId.audio()->owner().message(audioId.contextId())
			: nullptr;
		if (item) {
			setHistory(data, item->history());
		} else {
			setHistory(
				data,
				nullptr,
				audioId.audio() ? &audioId.audio()->session() : nullptr);
		}
		_trackChanged.fire_copy(data->type);
		refreshPlaylist(data);
	}
}

void Instance::setHistory(
		not_null<Data*> data,
		History *history,
		Main::Session *sessionFallback) {
	if (history) {
		data->history = history->migrateToOrMe();
		data->topicRootId = 0;
		data->migrated = data->history->migrateFrom();
		setSession(data, &history->session());
	} else {
		data->history = data->migrated = nullptr;
		setSession(data, sessionFallback);
	}
}

void Instance::setSession(not_null<Data*> data, Main::Session *session) {
	if (data->session == session) {
		return;
	}
	data->playlistLifetime.destroy();
	data->playlistOtherLifetime.destroy();
	data->sessionLifetime.destroy();
	data->session = session;
	if (session) {
		session->account().sessionChanges(
		) | rpl::start_with_next([=] {
			setSession(data, nullptr);
		}, data->sessionLifetime);

		session->data().documentLoadProgress(
		) | rpl::filter([=](not_null<DocumentData*> document) {
			// Before refactoring it was called only for audio files.
			return document->isAudioFile();
		}) | rpl::start_with_next([=](not_null<DocumentData*> document) {
			const auto type = AudioMsgId::Type::Song;
			emitUpdate(type, [&](const AudioMsgId &audioId) {
				return (audioId.audio() == document);
			});
		}, data->sessionLifetime);

		session->data().itemRemoved(
		) | rpl::filter([=](not_null<const HistoryItem*> item) {
			return (data->current.contextId() == item->fullId());
		}) | rpl::start_with_next([=] {
			stopAndClear(data);
		}, data->sessionLifetime);
	} else {
		stopAndClear(data);
	}
}

void Instance::clearStreamed(not_null<Data*> data, bool savePosition) {
	if (!data->streamed || data->streamed->clearing) {
		return;
	}
	data->streamed->clearing = true;
	if (savePosition) {
		SaveLastPlaybackPosition(
			data->current.audio(),
			data->streamed->instance.player().prepareLegacyState());
	}
	data->streamed->instance.stop();
	data->isPlaying = false;
	requestRoundVideoResize();
	emitUpdate(data->type);
	data->streamed = nullptr;

	_roundPlaying = false;
	Core::App().floatPlayerToggleGifsPaused(false);
}

void Instance::refreshPlaylist(not_null<Data*> data) {
	if (!validPlaylist(data)) {
		validatePlaylist(data);
	} else {
		refreshOtherPlaylist(data);
	}
}

void Instance::refreshOtherPlaylist(not_null<Data*> data) {
	if (!validOtherPlaylist(data)) {
		validateOtherPlaylist(data);
	}
	playlistUpdated(data);
}

void Instance::playlistUpdated(not_null<Data*> data) {
	if (data->playlistSlice) {
		const auto fullId = data->current.contextId();
		data->playlistIndex = data->playlistSlice->indexOf(fullId);
		if (order(data) == OrderMode::Shuffle) {
			validateShuffleData(data);
		}
	} else {
		data->playlistIndex = std::nullopt;
		data->shuffleData = nullptr;
	}
	data->playlistChanges.fire({});
}

bool Instance::validPlaylist(not_null<const Data*> data) const {
	if (const auto key = playlistKey(data)) {
		if (!data->playlistSlice) {
			return false;
		}
		using Key = SliceKey;
		const auto inSameDomain = [](const Key &a, const Key &b) {
			return (a.peerId == b.peerId)
				&& (a.topicRootId == b.topicRootId)
				&& (a.migratedPeerId == b.migratedPeerId);
		};
		const auto countDistanceInData = [&](const Key &a, const Key &b) {
			return [&](const SparseIdsMergedSlice &data) {
				return inSameDomain(a, b)
					? data.distance(a, b)
					: std::optional<int>();
			};
		};

		if (key == data->playlistRequestedKey) {
			return true;
		} else if (!data->playlistSliceKey
			|| !data->playlistRequestedKey
			|| *data->playlistRequestedKey != *data->playlistSliceKey) {
			return false;
		}
		auto distance = data->playlistSlice
			| countDistanceInData(*key, *data->playlistRequestedKey)
			| func::abs;
		if (distance) {
			return (*distance < kIdsPreloadAfter);
		}
	}
	return !data->playlistSlice;
}

void Instance::validatePlaylist(not_null<Data*> data) {
	data->playlistLifetime.destroy();
	if (const auto key = playlistKey(data)) {
		data->playlistRequestedKey = key;

		const auto sharedMediaViewer = (key->topicRootId
			== SparseIdsMergedSlice::kScheduledTopicId)
			? SharedScheduledMediaViewer
			: SharedMediaMergedViewer;
		sharedMediaViewer(
			&data->history->session(),
			SharedMediaMergedKey(*key, data->overview),
			kIdsLimit,
			kIdsLimit
		) | rpl::start_with_next([=](SparseIdsMergedSlice &&update) {
			data->playlistSlice = std::move(update);
			data->playlistSliceKey = key;
			refreshOtherPlaylist(data);
		}, data->playlistLifetime);
	} else {
		data->playlistSlice = std::nullopt;
		data->playlistSliceKey = data->playlistRequestedKey = std::nullopt;
		refreshOtherPlaylist(data);
	}
}

auto Instance::playlistKey(not_null<const Data*> data) const
-> std::optional<SliceKey> {
	const auto contextId = data->current.contextId();
	const auto history = data->history;
	if (!contextId || !history) {
		return {};
	}
	const auto item = data->history->owner().message(contextId);
	if (!item || (!item->isRegular() && !item->isScheduled())) {
		return {};
	}

	const auto universalId = (contextId.peer == history->peer->id)
		? contextId.msg
		: (contextId.msg - ServerMaxMsgId);
	return SliceKey(
		data->history->peer->id,
		(item->isScheduled()
			? SparseIdsMergedSlice::kScheduledTopicId
			: data->topicRootId),
		data->migrated ? data->migrated->peer->id : 0,
		universalId);
}

bool Instance::validOtherPlaylist(not_null<const Data*> data) const {
	if (const auto key = playlistOtherKey(data)) {
		return data->playlistOtherSlice
			&& (key == data->playlistOtherRequestedKey);
	}
	return !data->playlistOtherSlice;
}

void Instance::validateOtherPlaylist(not_null<Data*> data) {
	data->playlistOtherLifetime.destroy();
	if (const auto key = playlistOtherKey(data)) {
		data->playlistOtherRequestedKey = key;

		SharedMediaMergedViewer(
			&data->history->session(),
			SharedMediaMergedKey(*key, data->overview),
			kIdsLimit,
			kIdsLimit
		) | rpl::start_with_next([=](SparseIdsMergedSlice &&update) {
			data->playlistOtherSlice = std::move(update);
			playlistUpdated(data);
		}, data->playlistOtherLifetime);
	} else {
		data->playlistOtherSlice = std::nullopt;
		data->playlistOtherRequestedKey = std::nullopt;
		playlistUpdated(data);
	}
}

auto Instance::playlistOtherKey(not_null<const Data*> data) const
-> std::optional<SliceKey> {
	if (repeat(data) != RepeatMode::All
		|| order(data) == OrderMode::Shuffle
		|| !data->playlistSlice
		|| (data->playlistSlice->skippedBefore() != 0
			&& data->playlistSlice->skippedAfter() != 0)
		|| (data->playlistSlice->skippedBefore() == 0
			&& data->playlistSlice->skippedAfter() == 0)) {
		return {};
	}
	const auto contextId = data->current.contextId();
	const auto history = data->history;
	if (!contextId || !history) {
		return {};
	}
	const auto item = data->history->owner().message(contextId);
	if (!item || !item->isRegular()) {
		return {};
	}

	return SliceKey(
		data->history->peer->id,
		data->topicRootId,
		data->migrated ? data->migrated->peer->id : 0,
		(data->playlistSlice->skippedBefore() == 0
			? ServerMaxMsgId - 1
			: data->migrated
			? (1 - ServerMaxMsgId)
			: 1));
}

HistoryItem *Instance::itemByIndex(not_null<Data*> data, int index) {
	if (!data->playlistSlice
		|| index < 0
		|| index >= data->playlistSlice->size()) {
		return nullptr;
	}
	Assert(data->history != nullptr);
	const auto fullId = (*data->playlistSlice)[index];
	return data->history->owner().message(fullId);
}

bool Instance::moveInPlaylist(
		not_null<Data*> data,
		int delta,
		bool autonext) {
	if (!data->playlistIndex) {
		return false;
	}
	const auto jumpByItem = [&](not_null<HistoryItem*> item) {
		if (const auto media = item->media()) {
			if (media->ttlSeconds()) {
				return false;
			}
			if (const auto document = media->document()) {
				if (autonext) {
					_switchToNext.fire({
						data->current,
						item->fullId()
					});
				}
				if (document->isAudioFile()
					|| document->isVoiceMessage()
					|| document->isVideoMessage()) {
					play(AudioMsgId(document, item->fullId()));
				}
				return true;
			}
		}
		return false;
	};
	const auto jumpById = [&](FullMsgId id) {
		return jumpByItem(data->history->owner().message(id));
	};
	const auto repeatAll = (repeat(data) == RepeatMode::All);

	if (order(data) == OrderMode::Shuffle) {
		const auto raw = data->shuffleData.get();
		if (!raw || !raw->history) {
			return false;
		}
		const auto universal = computeCurrentUniversalId(data);
		const auto byUniversal = [&](ShuffleData::UniversalMsgId id) {
			return (id < 0 && raw->migrated)
				? jumpById({ raw->migrated->peer->id, id + ServerMaxMsgId })
				: jumpById({ raw->history->peer->id, id });
		};
		if (universal && raw->indexInPlayedIds == raw->playedIds.size()) {
			raw->playedIds.push_back(universal);
			const auto i = ranges::find(raw->nonPlayedIds, universal);
			if (i != end(raw->nonPlayedIds)) {
				raw->nonPlayedIds.erase(i);
			}
		}
		if (repeatAll) {
			ensureShuffleMove(data, delta);
		}
		if (raw->nonPlayedIds.empty()
			&& raw->indexInPlayedIds + 1 == raw->playedIds.size()) {
			raw->nonPlayedIds.push_back(raw->playedIds.back());
			raw->playedIds.pop_back();
		}
		const auto shuffleCompleted = raw->nonPlayedIds.empty()
			|| (raw->nonPlayedIds.size() == 1
				&& raw->nonPlayedIds.front() == universal);
		if (delta < 0) {
			return (raw->indexInPlayedIds > 0)
				&& byUniversal(raw->playedIds[--raw->indexInPlayedIds]);
		} else if (raw->indexInPlayedIds + 1 < raw->playedIds.size()) {
			return byUniversal(raw->playedIds[++raw->indexInPlayedIds]);
		}
		if (shuffleCompleted) {
			return false;
		} else if (raw->indexInPlayedIds < raw->playedIds.size()) {
			++raw->indexInPlayedIds;
		}
		const auto index = base::RandomIndex(raw->nonPlayedIds.size());
		return byUniversal(raw->nonPlayedIds[index]);
	}

	const auto newIndex = *data->playlistIndex
		+ (order(data) == OrderMode::Reverse ? -delta : delta);
	const auto useIndex = (!repeatAll
		|| !data->playlistSlice
		|| data->playlistSlice->skippedAfter() != 0
		|| data->playlistSlice->skippedBefore() != 0
		|| !data->playlistSlice->size())
		? newIndex
		: ((newIndex + int(data->playlistSlice->size()))
			% int(data->playlistSlice->size()));
	if (const auto item = itemByIndex(data, useIndex)) {
		return jumpByItem(item);
	} else if (repeatAll
		&& data->playlistOtherSlice
		&& data->playlistOtherSlice->size() > 0) {
		const auto &other = *data->playlistOtherSlice;
		if (newIndex < 0 && other.skippedAfter() == 0) {
			return jumpById(other[other.size() - 1]);
		} else if (newIndex > 0 && other.skippedBefore() == 0) {
			return jumpById(other[0]);
		}
	}
	return false;
}

void Instance::updatePowerSaveBlocker(
		not_null<Data*> data,
		const TrackState &state) {
	const auto block = !IsPausedOrPausing(state.state)
		&& !IsStoppedOrStopping(state.state);
	const auto blockVideo = block
		&& data->current.audio()
		&& data->current.audio()->isVideoMessage();
	const auto windowResolver = [] {
		const auto window = Core::App().activeWindow();
		return window ? window->widget()->windowHandle() : nullptr;
	};
	base::UpdatePowerSaveBlocker(
		data->powerSaveBlocker,
		block,
		base::PowerSaveBlockType::PreventAppSuspension,
		[] { return u"Audio playback is active"_q; },
		windowResolver);
	base::UpdatePowerSaveBlocker(
		data->powerSaveBlockerVideo,
		blockVideo,
		base::PowerSaveBlockType::PreventDisplaySleep,
		[] { return u"Video playback is active"_q; },
		windowResolver);
}

void Instance::ensureShuffleMove(not_null<Data*> data, int delta) {
	const auto raw = data->shuffleData.get();
	if (delta < 0) {
		if (raw->indexInPlayedIds > 0) {
			return;
		} else if (raw->nonPlayedIds.size() < 2) {
			const auto freeUp = std::max(
				int(raw->playedIds.size() / 2),
				int(raw->playlist.size()) - kRememberShuffledOrderItems);
			const auto till = end(raw->playedIds);
			const auto from = end(raw->playedIds) - freeUp;
			raw->nonPlayedIds.insert(end(raw->nonPlayedIds), from, till);
			raw->playedIds.erase(from, till);
		}
		if (raw->nonPlayedIds.empty()) {
			return;
		}
		const auto index = base::RandomIndex(raw->nonPlayedIds.size());
		raw->playedIds.insert(
			begin(raw->playedIds),
			raw->nonPlayedIds[index]);
		raw->nonPlayedIds.erase(begin(raw->nonPlayedIds) + index);
		++raw->indexInPlayedIds;
		if (raw->nonPlayedIds.empty() && raw->playedIds.size() > 1) {
			raw->nonPlayedIds.push_back(raw->playedIds.back());
			raw->playedIds.pop_back();
		}
		return;
	} else if (raw->indexInPlayedIds + 1 < raw->playedIds.size()) {
		return;
	} else if (raw->nonPlayedIds.size() < 2) {
		const auto freeUp = std::max(
			int(raw->playedIds.size() / 2),
			int(raw->playlist.size()) - kRememberShuffledOrderItems);
		const auto from = begin(raw->playedIds);
		const auto till = begin(raw->playedIds) + freeUp;
		raw->nonPlayedIds.insert(end(raw->nonPlayedIds), from, till);
		raw->playedIds.erase(from, till);
		raw->indexInPlayedIds -= freeUp;
	}
}

MsgId Instance::computeCurrentUniversalId(not_null<const Data*> data) const {
	const auto raw = data->shuffleData.get();
	if (!raw) {
		return MsgId(0);
	}
	const auto current = data->current.contextId();
	const auto item = raw->history->owner().message(current);
	return !item
		? MsgId(0)
		: (item->history() == raw->history)
		? item->id
		: (item->history() == raw->migrated)
		? (item->id - ServerMaxMsgId)
		: MsgId(0);
}

bool Instance::previousAvailable(AudioMsgId::Type type) const {
	const auto data = getData(type);
	Assert(data != nullptr);

	if (!data->playlistIndex || !data->playlistSlice) {
		return false;
	} else if (repeat(data) == RepeatMode::All) {
		return true;
	} else if (order(data) == OrderMode::Shuffle) {
		const auto raw = data->shuffleData.get();
		return raw && (raw->indexInPlayedIds > 0);
	}
	return (order(data) == OrderMode::Reverse)
		? (*data->playlistIndex + 1 < data->playlistSlice->size())
		: (*data->playlistIndex > 0);
}

bool Instance::nextAvailable(AudioMsgId::Type type) const {
	const auto data = getData(type);
	Assert(data != nullptr);

	if (!data->playlistIndex || !data->playlistSlice) {
		return false;
	} else if (repeat(data) == RepeatMode::All) {
		return true;
	} else if (order(data) == OrderMode::Shuffle) {
		const auto raw = data->shuffleData.get();
		const auto universal = computeCurrentUniversalId(data);
		return raw
			&& ((raw->indexInPlayedIds + 1 < raw->playedIds.size())
				|| (raw->nonPlayedIds.size() > 1)
				|| (!raw->nonPlayedIds.empty()
					&& raw->nonPlayedIds.front() != universal));
	}
	return (order(data) == OrderMode::Reverse)
		? (*data->playlistIndex > 0)
		: (*data->playlistIndex + 1 < data->playlistSlice->size());
}

rpl::producer<> Media::Player::Instance::playlistChanges(
		AudioMsgId::Type type) const {
	const auto data = getData(type);
	Assert(data != nullptr);

	return rpl::merge(
		data->playlistChanges.events(),
		orderChanges(data) | rpl::to_empty,
		repeatChanges(data) | rpl::to_empty);
}

rpl::producer<> Media::Player::Instance::stops(AudioMsgId::Type type) const {
	return _playerStopped.events(
	) | rpl::filter([=](auto t) {
		return t == type;
	}) | rpl::to_empty;
}

rpl::producer<> Media::Player::Instance::startsPlay(
		AudioMsgId::Type type) const {
	return _playerStartedPlay.events(
	) | rpl::filter([=](auto t) {
		return t == type;
	}) | rpl::to_empty;
}

auto Media::Player::Instance::seekingChanges(AudioMsgId::Type type) const
-> rpl::producer<Media::Player::Instance::Seeking> {
	return _seekingChanges.events(
	) | rpl::filter([=](SeekingChanges data) {
		return data.type == type;
	}) | rpl::map([](SeekingChanges data) {
		return data.seeking;
	});
}

not_null<Instance*> instance() {
	Expects(SingleInstance != nullptr);

	return SingleInstance;
}

void Instance::play(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		if (!data->streamed || IsStopped(getState(type).state)) {
			play(data->current);
		} else {
			if (data->streamed->instance.active()) {
				data->streamed->instance.resume();
			}
			emitUpdate(type);
		}
		data->resumeOnCallEnd = false;
	}
}

void Instance::play(const AudioMsgId &audioId) {
	const auto document = audioId.audio();
	if (!document) {
		return;
	}
	if (document->isAudioFile()
		|| document->isVoiceMessage()
		|| document->isVideoMessage()) {
		auto shared = document->owner().streaming().sharedDocument(
			document,
			audioId.contextId());
		if (!shared) {
			return;
		}
		playStreamed(audioId, std::move(shared));
	}
	if (document->isVoiceMessage() || document->isVideoMessage()) {
		document->owner().markMediaRead(document);
	}
	_playerStartedPlay.fire_copy({audioId.type()});
}

void Instance::playPause(const AudioMsgId &audioId) {
	const auto now = current(audioId.type());
	if (now.audio() == audioId.audio()
		&& now.contextId() == audioId.contextId()) {
		playPause(audioId.type());
	} else {
		play(audioId);
	}
}

void Instance::playStreamed(
		const AudioMsgId &audioId,
		std::shared_ptr<Streaming::Document> shared) {
	Expects(audioId.audio() != nullptr);

	const auto data = getData(audioId.type());
	Assert(data != nullptr);

	clearStreamed(data, data->current.audio() != audioId.audio());
	data->streamed = std::make_unique<Streamed>(
		audioId,
		std::move(shared));
	data->streamed->instance.lockPlayer();

	data->streamed->instance.player().updates(
	) | rpl::start_with_next_error([=](Streaming::Update &&update) {
		handleStreamingUpdate(data, std::move(update));
	}, [=](Streaming::Error &&error) {
		handleStreamingError(data, std::move(error));
	}, data->streamed->lifetime);

	data->streamed->instance.play(streamingOptions(audioId));

	emitUpdate(audioId.type());
}

Streaming::PlaybackOptions Instance::streamingOptions(
		const AudioMsgId &audioId,
		crl::time position) {
	const auto document = audioId.audio();
	auto result = Streaming::PlaybackOptions();
	result.mode = (document && document->isVideoMessage())
		? Streaming::Mode::Both
		: Streaming::Mode::Audio;
	result.speed = audioId.changeablePlaybackSpeed()
		? Core::App().settings().voicePlaybackSpeed()
		: 1.;
	result.audioId = audioId;
	if (position >= 0) {
		result.position = position;
	} else if (document) {
		auto &settings = document->session().settings();
		result.position = settings.mediaLastPlaybackPosition(document->id);
		settings.setMediaLastPlaybackPosition(document->id, 0);
	} else {
		result.position = 0;
	}
	return result;
}

void Instance::pause(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		if (data->streamed) {
			if (data->streamed->instance.active()) {
				data->streamed->instance.pause();
			}
			emitUpdate(type);
		}
	}
}

void Instance::stop(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		if (data->streamed) {
			clearStreamed(data);
		}
		data->resumeOnCallEnd = false;
		_playerStopped.fire_copy({type});
	}
}

void Instance::stopAndClear(not_null<Data*> data) {
	stop(data->type);
	*data = Data(data->type, data->overview);
	_tracksFinished.fire_copy(data->type);
}

void Instance::validateShuffleData(not_null<Data*> data) {
	if (!data->history) {
		data->shuffleData = nullptr;
		return;
	} else if (!data->shuffleData) {
		setupShuffleData(data);
	}
	const auto raw = data->shuffleData.get();
	const auto key = playlistKey(data);
	const auto scheduled = key
		&& (key->topicRootId == SparseIdsMergedSlice::kScheduledTopicId);
	if (raw->history != data->history
		|| raw->topicRootId != data->topicRootId
		|| raw->migrated != data->migrated
		|| raw->scheduled != scheduled) {
		raw->history = data->history;
		raw->migrated = data->migrated;
		raw->scheduled = scheduled;
		raw->nextSliceLifetime.destroy();
		raw->allLoaded = false;
		raw->playlist.clear();
		raw->nonPlayedIds.clear();
		raw->playedIds.clear();
		raw->indexInPlayedIds = 0;
	} else if (raw->nextSliceLifetime) {
		return;
	} else if (raw->allLoaded) {
		const auto universal = computeCurrentUniversalId(data);
		if (!universal
			|| (raw->indexInPlayedIds < raw->playedIds.size()
				? (raw->playedIds[raw->indexInPlayedIds] == universal)
				: ranges::contains(raw->nonPlayedIds, universal))) {
			return;
		}
		// We started playing some track not from the tracks that are left.
		// Start the whole playlist thing once again.
		raw->playedIds.clear();
		raw->indexInPlayedIds = 0;
		if (ranges::contains(raw->playlist, universal)) {
			raw->nonPlayedIds = raw->playlist;
		} else {
			raw->allLoaded = false;
			raw->playlist.clear();
			raw->nonPlayedIds.clear();
		}
	}
	if (raw->scheduled) {
		const auto count = data->playlistSlice
			? int(data->playlistSlice->size())
			: 0;
		if (raw->playlist.empty() && count > 0) {
			raw->playlist.reserve(count);
			for (auto i = 0; i != count; ++i) {
				raw->playlist.push_back((*data->playlistSlice)[i].msg);
			}
			raw->nonPlayedIds = raw->playlist;
			raw->allLoaded = true;
			data->playlistChanges.fire({});
		}
		return;
	}
	const auto last = raw->playlist.empty()
		? MsgId(ServerMaxMsgId - 1)
		: raw->playlist.back();
	SharedMediaMergedViewer(
		&raw->history->session(),
		SharedMediaMergedKey(
			SliceKey(
				raw->history->peer->id,
				raw->topicRootId,
				raw->migrated ? raw->migrated->peer->id : 0,
				last),
			data->overview),
		kIdsLimit,
		kIdsLimit
	) | rpl::start_with_next([=](SparseIdsMergedSlice &&update) {
		raw->nextSliceLifetime.destroy();

		const auto size = update.size();
		const auto peer = raw->history->peer->id;
		raw->playlist.reserve(raw->playlist.size() + size);
		raw->nonPlayedIds.reserve(raw->nonPlayedIds.size() + size);
		for (auto i = size; i != 0;) {
			const auto fullId = update[--i];
			const auto universal = (fullId.peer == peer)
				? fullId.msg
				: (fullId.msg - ServerMaxMsgId);
			if (raw->playlist.empty() || raw->playlist.back() > universal) {
				raw->playlist.push_back(universal);
				raw->nonPlayedIds.push_back(universal);
			}
		}
		if (update.skippedBefore() == 0
			|| raw->playlist.size() >= kShufflePlaylistLimit) {
			raw->allLoaded = true;
		}
		data->playlistChanges.fire({});
	}, raw->nextSliceLifetime);
}

void Instance::setupShuffleData(not_null<Data*> data) {
	data->shuffleData = std::make_unique<ShuffleData>();
	const auto raw = data->shuffleData.get();
	data->history->session().changes().messageUpdates(
		::Data::MessageUpdate::Flag::Destroyed
	) | rpl::map([=](const ::Data::MessageUpdate &update) {
		const auto item = update.item;
		const auto history = item->history().get();
		return (history == raw->history)
			? item->id
			: (history == raw->migrated)
			? (item->id - ServerMaxMsgId)
			: MsgId(0);
	}) | rpl::filter(
		rpl::mappers::_1 != MsgId(0)
	) | rpl::start_with_next([=](MsgId id) {
		const auto i = ranges::find(raw->playlist, id);
		if (i != end(raw->playlist)) {
			raw->playlist.erase(i);
		}
		const auto j = ranges::find(raw->nonPlayedIds, id);
		if (j != end(raw->nonPlayedIds)) {
			raw->nonPlayedIds.erase(j);
		}
		const auto k = ranges::find(raw->playedIds, id);
		if (k != end(raw->playedIds)) {
			const auto index = (k - begin(raw->playedIds));
			raw->playedIds.erase(k);
			if (raw->indexInPlayedIds > index) {
				--raw->indexInPlayedIds;
			}
		}
	}, data->shuffleData->lifetime);
}

void Instance::playPause(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		if (!data->streamed) {
			play(data->current);
		} else {
			auto &streamed = data->streamed->instance;
			if (!streamed.active()) {
				streamed.play(streamingOptions(data->streamed->id));
			} else if (streamed.paused()) {
				streamed.resume();
			} else {
				streamed.pause();
			}
			emitUpdate(type);
		}
		data->resumeOnCallEnd = false;
	}
}

void Instance::pauseOnCall(AudioMsgId::Type type) {
	const auto state = getState(type);
	if (!state.id
		|| IsStopped(state.state)
		|| IsPaused(state.state)
		|| state.state == State::Pausing) {
		return;
	}
	pause(type);
	if (const auto data = getData(type)) {
		data->resumeOnCallEnd = true;
	}
}

void Instance::resumeOnCall(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		if (data->resumeOnCallEnd) {
			data->resumeOnCallEnd = false;
			play(type);
		}
	}
}

bool Instance::next(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		return moveInPlaylist(data, 1, false);
	}
	return false;
}

bool Instance::previous(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		return moveInPlaylist(data, -1, false);
	}
	return false;
}

void Instance::playPauseCancelClicked(AudioMsgId::Type type) {
	if (isSeeking(type)) {
		return;
	}

	const auto data = getData(type);
	if (!data) {
		return;
	}
	const auto state = getState(type);
	const auto showPause = ShowPauseIcon(state.state);
	const auto audio = state.id.audio();
	if (audio && audio->loading() && !data->streamed) {
		audio->cancel();
	} else if (showPause) {
		pause(type);
	} else {
		play(type);
	}
}

void Instance::startSeeking(AudioMsgId::Type type) {
	if (auto data = getData(type)) {
		data->seeking = data->current;
	}
	pause(type);
	emitUpdate(type);
	_seekingChanges.fire({ .seeking = Seeking::Start, .type = type });
}

void Instance::finishSeeking(AudioMsgId::Type type, float64 progress) {
	if (const auto data = getData(type)) {
		if (const auto streamed = data->streamed.get()) {
			const auto &info = streamed->instance.info();
			const auto duration = info.audio.state.duration;
			if (duration != kTimeUnknown) {
				const auto position = crl::time(base::SafeRound(
					std::clamp(progress, 0., 1.) * duration));
				streamed->instance.play(streamingOptions(
					streamed->id,
					position));
				emitUpdate(type);
			}
		}
	}
	cancelSeeking(type);
	_seekingChanges.fire({ .seeking = Seeking::Finish, .type = type });
}

void Instance::cancelSeeking(AudioMsgId::Type type) {
	if (const auto data = getData(type)) {
		data->seeking = AudioMsgId();
	}
	emitUpdate(type);
	_seekingChanges.fire({ .seeking = Seeking::Cancel, .type = type });
}

void Instance::updateVoicePlaybackSpeed() {
	if (const auto data = getData(getActiveType())) {
		if (!data->current.changeablePlaybackSpeed()) {
			return;
		}
		if (const auto streamed = data->streamed.get()) {
			streamed->instance.setSpeed(
				Core::App().settings().voicePlaybackSpeed());
		}
	}
}

void Instance::emitUpdate(AudioMsgId::Type type) {
	emitUpdate(type, [](const AudioMsgId &playing) { return true; });
}

RepeatMode Instance::repeat(not_null<const Data*> data) const {
	return (data->type == AudioMsgId::Type::Song)
		? Core::App().settings().playerRepeatMode()
		: RepeatMode::None;
}

rpl::producer<RepeatMode> Instance::repeatChanges(
		not_null<const Data*> data) const {
	return (data->type == AudioMsgId::Type::Song)
		? Core::App().settings().playerRepeatModeChanges()
		: rpl::never<RepeatMode>();
}

OrderMode Instance::order(not_null<const Data*> data) const {
	return (data->type == AudioMsgId::Type::Song)
		? Core::App().settings().playerOrderMode()
		: OrderMode::Default;
}

rpl::producer<OrderMode> Instance::orderChanges(
		not_null<const Data*> data) const {
	return (data->type == AudioMsgId::Type::Song)
		? Core::App().settings().playerOrderModeChanges()
		: rpl::never<OrderMode>();
}

TrackState Instance::getState(AudioMsgId::Type type) const {
	if (const auto data = getData(type)) {
		if (data->streamed) {
			return data->streamed->instance.player().prepareLegacyState();
		}
	}
	return TrackState();
}

Streaming::Instance *Instance::roundVideoStreamed(HistoryItem *item) const {
	if (!item) {
		return nullptr;
	} else if (const auto data = getData(AudioMsgId::Type::Voice)) {
		if (const auto streamed = data->streamed.get()) {
			if (streamed->id.contextId() == item->fullId()) {
				const auto player = &streamed->instance.player();
				if (player->ready() && !player->videoSize().isEmpty()) {
					return &streamed->instance;
				}
			}
		}
	}
	return nullptr;
}

View::PlaybackProgress *Instance::roundVideoPlayback(
		HistoryItem *item) const {
	return roundVideoStreamed(item)
		? &getData(AudioMsgId::Type::Voice)->streamed->progress
		: nullptr;
}

template <typename CheckCallback>
void Instance::emitUpdate(AudioMsgId::Type type, CheckCallback check) {
	if (const auto data = getData(type)) {
		const auto state = getState(type);
		if (!state.id || !check(state.id)) {
			return;
		}
		setCurrent(state.id);
		if (const auto streamed = data->streamed.get()) {
			if (!streamed->instance.info().video.size.isEmpty()) {
				streamed->progress.updateState(state);
			}
		}
		updatePowerSaveBlocker(data, state);

		auto finished = false;
		_updatedNotifier.fire_copy({state});
		if (data->isPlaying && state.state == State::StoppedAtEnd) {
			if (repeat(data) == RepeatMode::One) {
				play(data->current);
			} else if (OptionDisableAutoplayNext.value()) {
				finished = true;
			} else if (!moveInPlaylist(data, 1, true)) {
				finished = true;
			}
		}
		data->isPlaying = !IsStopped(state.state);
		if (finished) {
			_tracksFinished.fire_copy(type);
		}
	}
}

void Instance::setupShortcuts() {
	Shortcuts::Requests(
	) | rpl::start_with_next([=](not_null<Shortcuts::Request*> request) {
		using Command = Shortcuts::Command;
		request->check(Command::MediaPlay) && request->handle([=] {
			playPause();
			return true;
		});
		request->check(Command::MediaPause) && request->handle([=] {
			pause();
			return true;
		});
		request->check(Command::MediaPlayPause) && request->handle([=] {
			playPause();
			return true;
		});
		request->check(Command::MediaStop) && request->handle([=] {
			stop();
			return true;
		});
		request->check(Command::MediaPrevious) && request->handle([=] {
			previous();
			return true;
		});
		request->check(Command::MediaNext) && request->handle([=] {
			next();
			return true;
		});
	}, _lifetime);
}

void Instance::stopAndClose() {
	_closePlayerRequests.fire({});

	stop(AudioMsgId::Type::Voice);
	stop(AudioMsgId::Type::Song);

	Shortcuts::ToggleMediaShortcuts(false);
}

void Instance::handleStreamingUpdate(
		not_null<Data*> data,
		Streaming::Update &&update) {
	using namespace Streaming;

	v::match(update.data, [&](Information &update) {
		if (!update.video.size.isEmpty()) {
			data->streamed->progress.setValueChangedCallback([=](
					float64,
					float64) {
				requestRoundVideoRepaint();
			});
			_roundPlaying = true;
			Core::App().floatPlayerToggleGifsPaused(true);
			requestRoundVideoResize();
		}
		emitUpdate(data->type);
	}, [&](PreloadedVideo &update) {
		//emitUpdate(data->type, [](AudioMsgId) { return true; });
	}, [&](UpdateVideo &update) {
		emitUpdate(data->type);
	}, [&](PreloadedAudio &update) {
		//emitUpdate(data->type, [](AudioMsgId) { return true; });
	}, [&](UpdateAudio &update) {
		emitUpdate(data->type);
	}, [&](WaitingForData) {
	}, [&](MutedByOther) {
	}, [&](Finished) {
		emitUpdate(data->type);
		if (data->streamed && data->streamed->instance.player().finished()) {
			clearStreamed(data);
		}
	});
}

HistoryItem *Instance::roundVideoItem() const {
	const auto data = getData(AudioMsgId::Type::Voice);
	return (data->streamed
		&& !data->streamed->instance.info().video.size.isEmpty()
		&& data->history)
		? data->history->owner().message(data->streamed->id.contextId())
		: nullptr;
}

void Instance::requestRoundVideoResize() const {
	if (const auto item = roundVideoItem()) {
		item->history()->owner().requestItemResize(item);
	}
}

void Instance::requestRoundVideoRepaint() const {
	if (const auto item = roundVideoItem()) {
		item->history()->owner().requestItemRepaint(item);
	}
}

void Instance::handleStreamingError(
		not_null<Data*> data,
		Streaming::Error &&error) {
	Expects(data->streamed != nullptr);

	const auto document = data->streamed->id.audio();
	const auto contextId = data->streamed->id.contextId();
	if (error == Streaming::Error::NotStreamable) {
		DocumentSaveClickHandler::SaveAndTrack(
			contextId,
			document);
	} else if (error == Streaming::Error::OpenFailed) {
		DocumentSaveClickHandler::SaveAndTrack(
			contextId,
			document,
			DocumentSaveClickHandler::Mode::ToFile);
	}
	emitUpdate(data->type);
	if (data->streamed && data->streamed->instance.player().failed()) {
		clearStreamed(data);
	}
}

} // namespace Player
} // namespace Media