/*
obs-websocket
Copyright (C) 2016-2017	Stéphane Lepin <stephane.lepin@gmail.com>
Copyright (C) 2017	Mikhail Swift <https://github.com/mikhailswift>

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 <https://www.gnu.org/licenses/>
*/

#ifndef WSREQUESTHANDLER_H
#define WSREQUESTHANDLER_H

#include <obs-frontend-api.h>
#include <QtWebSockets/QWebSocket>
#include <QtWebSockets/QWebSocketServer>

class WSRequestHandler : public QObject
{
	Q_OBJECT

	public:
		explicit WSRequestHandler(QWebSocket *client);
		~WSRequestHandler();
		void processIncomingMessage(QString textMessage);
		bool hasField(const char* name);

	private:
		QWebSocket *_client;
		const char *_messageId;
		const char *_requestType;
		obs_data_t *data;

		QMap<QString, void(*)(WSRequestHandler*)> messageMap;
		QSet<QString> authNotRequired;

		void SendOKResponse(obs_data_t *additionalFields = NULL);
		void SendErrorResponse(const char *errorMessage);
		
		static void HandleGetVersion(WSRequestHandler *req);
		static void HandleGetAuthRequired(WSRequestHandler *req);
		static void HandleAuthenticate(WSRequestHandler *req);

		static void HandleSetCurrentScene(WSRequestHandler *req);
		static void HandleGetCurrentScene(WSRequestHandler *req);
		static void HandleGetSceneList(WSRequestHandler *req);
		
		static void HandleSetSceneItemRender(WSRequestHandler *req);
		static void HandleSetSceneItemPosition(WSRequestHandler *req);
		static void HandleSetSceneItemTransform(WSRequestHandler *req);
		static void HandleSetSceneItemCrop(WSRequestHandler *req);

		static void HandleGetStreamingStatus(WSRequestHandler *req);
		static void HandleStartStopStreaming(WSRequestHandler *req);
		static void HandleStartStopRecording(WSRequestHandler *req);
		static void HandleStartStreaming(WSRequestHandler *req);
		static void HandleStopStreaming(WSRequestHandler *req);
		static void HandleStartRecording(WSRequestHandler *req);
		static void HandleStopRecording(WSRequestHandler *req);

		static void HandleGetTransitionList(WSRequestHandler *req);
		static void HandleGetCurrentTransition(WSRequestHandler *req);
		static void HandleSetCurrentTransition(WSRequestHandler *req);

		static void HandleSetVolume(WSRequestHandler *req);
		static void HandleGetVolume(WSRequestHandler *req);
		static void HandleToggleMute(WSRequestHandler *req);
		static void HandleSetMute(WSRequestHandler *req);
		static void HandleGetMute(WSRequestHandler *req);

		static void HandleSetCurrentSceneCollection(WSRequestHandler *req);
		static void HandleGetCurrentSceneCollection(WSRequestHandler *req);
		static void HandleListSceneCollections(WSRequestHandler *req);

		static void HandleSetCurrentProfile(WSRequestHandler *req);
		static void HandleGetCurrentProfile(WSRequestHandler *req);
		static void HandleListProfiles(WSRequestHandler *req);

		static void HandleSetTransitionDuration(WSRequestHandler *req);
		static void HandleGetTransitionDuration(WSRequestHandler *req);

		static void HandleGetStudioModeStatus(WSRequestHandler *req);
		static void HandleGetPreviewScene(WSRequestHandler *req);
		static void HandleSetPreviewScene(WSRequestHandler *req);
		static void HandleTransitionToProgram(WSRequestHandler *req);
		static void HandleEnableStudioMode(WSRequestHandler *req);
		static void HandleDisableStudioMode(WSRequestHandler *req);
		static void HandleToggleStudioMode(WSRequestHandler *req);
};

#endif // WSPROTOCOL_H