/******************************************************************************
    Copyright (C) 2025 by Taylor Giampaolo <warchamp7@obsproject.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/

#pragma once

#include "OBSBasic.hpp"

#include <components/MenuCheckBox.hpp>
#include <components/VolumeControl.hpp>

#include <QObject>
#include <QPointer>
#include <QTimer>

#include <unordered_set>

class QHBoxLayout;
class QMenu;
class QPushButton;
class QScrollArea;
class QStackedWidget;
class QToolBar;
class QVBoxLayout;

class AudioMixer : public QFrame {
	Q_OBJECT

	struct RankedVolume {
		VolumeControl *control;
		int sortingWeight;
	};

public:
	explicit AudioMixer(QWidget *parent = 0);
	~AudioMixer();

	void refreshVolumeColors();

	void setMixerLayoutVertical(bool vertical);
	void createMixerContextMenu();
	void showMixerContextMenu();

private:
	std::vector<OBSSignal> signalHandlers;
	std::vector<OBSSignal> previewSignals;
	static void onFrontendEvent(enum obs_frontend_event event, void *data);
	void handleFrontendEvent(enum obs_frontend_event event);
	void updatePreviewHandlers();

	std::unordered_map<QString, QPointer<VolumeControl>> volumeList;
	void addControlForUuid(QString uuid);
	void removeControlForUuid(QString uuid);

	std::unordered_set<QString> globalSources;
	std::unordered_set<QString> previewSources;
	bool mixerVertical{false};

	int hiddenCount{0};
	bool showHidden{false};
	bool showInactive{false};

	bool keepInactiveLast{false};
	bool keepHiddenLast{false};

	bool showToolbar{true};

	QVBoxLayout *mainLayout{nullptr};

	QStackedWidget *stackedMixerArea{nullptr};
	QToolBar *mixerToolbar{nullptr};
	QAction *layoutButton{nullptr};
	QAction *advAudio{nullptr};
	QPushButton *optionsButton{nullptr};
	QPushButton *toggleHiddenButton{nullptr};

	QPointer<QMenu> mixerMenu;
	QPointer<MenuCheckBox> showHiddenCheckBox;

	QScrollArea *hMixerScrollArea{nullptr};
	QWidget *hVolumeWidgets{nullptr};
	QVBoxLayout *hVolumeControlLayout{nullptr};

	QScrollArea *vMixerScrollArea{nullptr};
	QWidget *vVolumeWidgets{nullptr};
	QHBoxLayout *vVolumeControlLayout{nullptr};

	QBoxLayout *activeLayout() const;

	void reloadVolumeControls();
	bool getMixerVisibilityForControl(VolumeControl *widget);

	void clearPreviewSources();
	bool isSourcePreviewed(obs_source_t *source);
	bool isSourceGlobal(obs_source_t *source);
	void clearVolumeControls();
	void updateShowInactive();
	void updateKeepInactiveLast();
	void updateShowHidden();
	void updateKeepHiddenLast();
	void updateShowToolbar();

	QTimer updateTimer;
	void updateVolumeLayouts();

	// OBS Callbacks
	static void obsSourceActivated(void *data, calldata_t *params);
	static void obsSourceDeactivated(void *data, calldata_t *params);
	static void obsSourceAudioActivated(void *data, calldata_t *params);
	static void obsSourceAudioDeactivated(void *data, calldata_t *params);
	static void obsSourceCreate(void *data, calldata_t *params);
	static void obsSourceRemove(void *data, calldata_t *params);
	static void obsSourceRename(void *data, calldata_t *params);
	static void obsSceneItemVisibleChange(void *data, calldata_t *params);

private slots:
	void addSource(QString uuid);
	void removeSource(QString uuid);
	void updatePreviewSources();
	void updateGlobalSources();
	void unhideAllAudioControls();
	void queueLayoutUpdate();

	VolumeControl *createVolumeControl(obs_source_t *source);
	void updateControlVisibility(QString uuid);

	void updateLayout();
	void toggleShowInactive(bool checked);
	void toggleKeepInactiveLast(bool checked);
	void toggleShowHidden(bool checked);
	void toggleKeepHiddenLast(bool checked);

	void mixerContextMenuRequested();

signals:
	void advAudioPropertiesClicked();
};
