#pragma once

#include <solanaceae/object_store/fwd.hpp>
#include <solanaceae/contact/contact_model3.hpp>
#include <solanaceae/tox_contacts/tox_contact_model2.hpp>
#include "./frame_streams/voip_model.hpp"
#include "./tox_av.hpp"

#include <unordered_map>
#include <variant>
#include <deque>
#include <mutex>

// fwd
struct ToxAVCallAudioSink;
struct ToxAVCallVideoSink;

class ToxAVVoIPModel : protected ToxAVEventI, public VoIPModelI {
	ObjectStore2& _os;
	ToxAVI& _av;
	Contact3Registry& _cr;
	ToxContactModel2& _tcm;

	uint64_t _pad0;
	// these events need to be worked on the main thread instead
	// TODO: replac ewith lockless queue
	std::deque<
	std::variant<
		Events::FriendCall,
		Events::FriendCallState
		// bitrates
	>> _e_queue;
	std::mutex _e_queue_mutex;
	uint64_t _pad1;

	std::vector<ToxAVCallAudioSink*> _audio_sinks;
	std::mutex _audio_sinks_mutex;
	uint64_t _pad2;

	std::vector<ToxAVCallVideoSink*> _video_sinks;
	std::mutex _video_sinks_mutex;
	uint64_t _pad3;

	// for faster lookup
	std::unordered_map<uint32_t, ObjectHandle> _audio_sources;
	std::unordered_map<uint32_t, ObjectHandle> _video_sources;

	// TODO: virtual? strategy? protected?
	virtual void addAudioSource(ObjectHandle session, uint32_t friend_number);
	virtual void addAudioSink(ObjectHandle session, uint32_t friend_number);
	virtual void addVideoSource(ObjectHandle session, uint32_t friend_number);
	virtual void addVideoSink(ObjectHandle session, uint32_t friend_number);

	void destroySession(ObjectHandle session);

	// TODO: this needs to move to the toxav thread
	// we could use "events" as pre/post audio/video iterate...
	void audio_thread_tick(void);
	void video_thread_tick(void);

	void handleEvent(const Events::FriendCall&);
	void handleEvent(const Events::FriendCallState&);

	public:
		ToxAVVoIPModel(ObjectStore2& os, ToxAVI& av, Contact3Registry& cr, ToxContactModel2& tcm);
		~ToxAVVoIPModel(void);

		// handle events coming from toxav thread(s)
		void tick(void);

	public: // voip model
		ObjectHandle enter(const Contact3 c, const Components::VoIP::DefaultConfig& defaults) override;
		bool accept(ObjectHandle session, const Components::VoIP::DefaultConfig& defaults) override;
		bool leave(ObjectHandle session) override;

	protected: // toxav events
		bool onEvent(const Events::FriendCall&) override;
		bool onEvent(const Events::FriendCallState&) override;
		bool onEvent(const Events::FriendAudioBitrate&) override;
		bool onEvent(const Events::FriendVideoBitrate&) override;
		bool onEvent(const Events::FriendAudioFrame&) override;
		bool onEvent(const Events::FriendVideoFrame&) override;
		bool onEvent(const Events::IterateAudio&) override;
		bool onEvent(const Events::IterateVideo&) override;
};