obs-websocket/src/requesthandler/RequestHandler_Scenes.cpp

#include "RequestHandler.h"

#include "../plugin-macros.generated.h"

RequestResult RequestHandler::GetSceneList(const Request& request)
{
	json responseData;

	OBSSourceAutoRelease currentProgramScene = obs_frontend_get_current_scene();
	responseData["currentProgramSceneName"] = obs_source_get_name(currentProgramScene);


	OBSSourceAutoRelease currentPreviewScene = obs_frontend_get_current_preview_scene();
	if (currentPreviewScene)
		responseData["currentPreviewSceneName"] = obs_source_get_name(currentPreviewScene);
	else
		responseData["currentPreviewSceneName"] = nullptr;

	responseData["scenes"] = Utils::Obs::ListHelper::GetSceneList();

	return RequestResult::Success(responseData);
}

RequestResult RequestHandler::GetCurrentProgramScene(const Request& request)
{
	json responseData;

	OBSSourceAutoRelease currentProgramScene = obs_frontend_get_current_scene();
	responseData["currentProgramSceneName"] = obs_source_get_name(currentProgramScene);

	return RequestResult::Success(responseData);
}

RequestResult RequestHandler::SetCurrentProgramScene(const Request& request)
{
	RequestStatus::RequestStatus statusCode;
	std::string comment;
	if (!request.ValidateString("sceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	std::string sceneName = request.RequestData["sceneName"];

	OBSSourceAutoRelease scene = obs_get_source_by_name(sceneName.c_str());
	if (!scene)
		return RequestResult::Error(RequestStatus::SceneNotFound);

	if (obs_source_get_type(scene) != OBS_SOURCE_TYPE_SCENE)
		return RequestResult::Error(RequestStatus::InvalidSourceType, "The specified source is not a scene.");

	obs_frontend_set_current_scene(scene);

	return RequestResult::Success();
}

RequestResult RequestHandler::GetCurrentPreviewScene(const Request& request)
{
	json responseData;

	OBSSourceAutoRelease currentPreviewScene = obs_frontend_get_current_preview_scene();
	if (currentPreviewScene) {
		responseData["currentPreviewSceneName"] = obs_source_get_name(currentPreviewScene);
	} else if (request.IgnoreNonFatalRequestChecks) {
		responseData["currentPreviewSceneName"] = nullptr;
	} else {
		return RequestResult::Error(RequestStatus::StudioModeNotActive);
	}

	return RequestResult::Success(responseData);
}

RequestResult RequestHandler::SetCurrentPreviewScene(const Request& request)
{
	if (!obs_frontend_preview_program_mode_active())
		return RequestResult::Error(RequestStatus::StudioModeNotActive);

	RequestStatus::RequestStatus statusCode;
	std::string comment;
	if (!request.ValidateString("sceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	std::string sceneName = request.RequestData["sceneName"];

	OBSSourceAutoRelease scene = obs_get_source_by_name(sceneName.c_str());
	if (!scene)
		return RequestResult::Error(RequestStatus::SceneNotFound);

	if (obs_source_get_type(scene) != OBS_SOURCE_TYPE_SCENE)
		return RequestResult::Error(RequestStatus::InvalidSourceType, "The specified source is not a scene.");

	obs_frontend_set_current_preview_scene(scene);

	return RequestResult::Success();
}

RequestResult RequestHandler::SetSceneName(const Request& request)
{
	RequestStatus::RequestStatus statusCode;
	std::string comment;
	if (!request.ValidateString("sceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	if (!request.ValidateString("newSceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	std::string sceneName = request.RequestData["sceneName"];
	std::string newSceneName = request.RequestData["newSceneName"];

	OBSSourceAutoRelease scene = obs_get_source_by_name(sceneName.c_str());
	if (!scene)
		return RequestResult::Error(RequestStatus::SceneNotFound);

	if (obs_source_get_type(scene) != OBS_SOURCE_TYPE_SCENE)
		return RequestResult::Error(RequestStatus::InvalidSourceType, "The specified source is not a scene.");

	OBSSourceAutoRelease existingSource = obs_get_source_by_name(newSceneName.c_str());
	if (existingSource)
		return RequestResult::Error(RequestStatus::SourceAlreadyExists, "A source already exists by that new scene name.");

	obs_source_set_name(scene, newSceneName.c_str());

	return RequestResult::Success();
}

RequestResult RequestHandler::CreateScene(const Request& request)
{
	RequestStatus::RequestStatus statusCode;
	std::string comment;
	if (!request.ValidateString("sceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	std::string sceneName = request.RequestData["sceneName"];

	OBSSourceAutoRelease scene = obs_get_source_by_name(sceneName.c_str());
	if (scene)
		return RequestResult::Error(RequestStatus::SourceAlreadyExists, "A source already exists by that scene name.");

	obs_scene_t *createdScene = obs_scene_create(sceneName.c_str());
	obs_scene_release(createdScene);

	return RequestResult::Success();
}

RequestResult RequestHandler::RemoveScene(const Request& request)
{
	RequestStatus::RequestStatus statusCode;
	std::string comment;
	if (!request.ValidateString("sceneName", statusCode, comment)) {
		return RequestResult::Error(statusCode, comment);
	}

	std::string sceneName = request.RequestData["sceneName"];

	OBSSourceAutoRelease scene = obs_get_source_by_name(sceneName.c_str());
	if (!scene)
		return RequestResult::Error(RequestStatus::SceneNotFound);

	if (obs_source_get_type(scene) != OBS_SOURCE_TYPE_SCENE)
		return RequestResult::Error(RequestStatus::InvalidSourceType, "The specified source is not a scene.");

	obs_source_remove(scene);

	return RequestResult::Success();
}