SceneSwitcher/lib/macro/macro-action-variable-json.cpp

#include "macro-action-variable-json.hpp"
#include "json-helpers.hpp"

namespace advss {

using Action = MacroActionVariable::Action;

static bool registered = RegisterVariableActionGroup(
	{20,
	 {
		 {Action::EXTRACT_JSON,
		  "AdvSceneSwitcher.action.variable.type.extractJsonField",
		  "AdvSceneSwitcher.action.variable.layout.strValue", true},
		 {Action::QUERY_JSON,
		  "AdvSceneSwitcher.action.variable.type.queryJson",
		  "AdvSceneSwitcher.action.variable.layout.queryJson", true},
		 {Action::ARRAY_JSON,
		  "AdvSceneSwitcher.action.variable.type.accessJsonArray",
		  "AdvSceneSwitcher.action.variable.layout.accessJsonArray"},
		 {Action::JSON_ARRAY_LENGTH,
		  "AdvSceneSwitcher.action.variable.type.jsonArrayLength",
		  "AdvSceneSwitcher.action.variable.layout.jsonArrayLength"},
		 {Action::SET_JSON_FIELD,
		  "AdvSceneSwitcher.action.variable.type.setJsonField",
		  "AdvSceneSwitcher.action.variable.layout.setJsonField", true},
	 },
	 []() { return std::make_unique<VariableActionJson>(); },
	 VariableActionJsonEdit::Create});

std::unique_ptr<VariableActionGroup> VariableActionJson::Copy() const
{
	return std::make_unique<VariableActionJson>(*this);
}

void VariableActionJson::Save(obs_data_t *obj,
			      const MacroActionVariable &) const
{
	_jsonQuery.Save(obj, "jsonQuery");
	_jsonIndex.Save(obj, "jsonIndex");
	obs_data_set_bool(obj, "jsonQueryExtractSingle",
			  _jsonQueryExtractSingle);
	_jsonValue.Save(obj, "jsonValue");
}

void VariableActionJson::Load(obs_data_t *obj, MacroActionVariable &)
{
	_jsonQuery.Load(obj, "jsonQuery");
	_jsonIndex.Load(obj, "jsonIndex");
	_jsonQueryExtractSingle =
		obs_data_get_bool(obj, "jsonQueryExtractSingle");
	if (obs_data_has_user_value(obj, "jsonValue")) {
		_jsonValue.Load(obj, "jsonValue");
	}
}

void VariableActionJson::ResolveVariables()
{
	_jsonQuery.ResolveVariables();
	_jsonIndex.ResolveVariables();
	_jsonValue.ResolveVariables();
}

bool VariableActionJson::Perform(MacroActionVariable &action,
				 MacroActionVariable::Action type,
				 Variable &var)
{
	switch (type) {
	case Action::EXTRACT_JSON: {
		auto value = GetJsonField(var.Value(), action._strValue);
		if (!value.has_value()) {
			return true;
		}
		var.SetValue(*value);
		return true;
	}
	case Action::QUERY_JSON: {
		auto value = QueryJson(var.Value(), _jsonQuery);
		if (!value.has_value()) {
			return true;
		}

		if (!_jsonQueryExtractSingle) {
			var.SetValue(*value);
			return true;
		}

		auto extracted = ExtractSingleJsonArrayElement(*value);
		if (extracted.has_value()) {
			var.SetValue(*extracted);
		} else {
			var.SetValue(*value);
		}
		return true;
	}
	case Action::ARRAY_JSON: {
		auto value = AccessJsonArrayIndex(var.Value(), _jsonIndex);
		if (!value.has_value()) {
			return true;
		}
		var.SetValue(*value);
		return true;
	}
	case Action::JSON_ARRAY_LENGTH: {
		auto input = action._variable2.lock();
		if (!input) {
			return true;
		}
		auto length = GetJsonArrayLength(input->Value());
		if (!length.has_value()) {
			return true;
		}
		var.SetValue(*length);
		return true;
	}
	case Action::SET_JSON_FIELD: {
		auto value =
			SetJsonField(var.Value(), action._strValue, _jsonValue);
		if (!value.has_value()) {
			return true;
		}
		var.SetValue(*value);
		return true;
	}
	default:
		return true;
	}
}

VariableActionJsonEdit::VariableActionJsonEdit(
	QWidget *parent, std::shared_ptr<MacroActionVariable> entryData)
	: VariableActionGroupEdit(parent, entryData),
	  _data(entryData ? entryData->GetGroup<VariableActionJson>()
			  : nullptr),
	  _jsonQuery(new VariableLineEdit(this)),
	  _jsonQueryHelp(new HelpIcon(
		  obs_module_text(
			  "AdvSceneSwitcher.action.variable.type.queryJson.info"),
		  this)),
	  _jsonIndex(new VariableSpinBox(this)),
	  _jsonExtractSingle(new QCheckBox(
		  obs_module_text(
			  "AdvSceneSwitcher.action.variable.type.queryJson.extractSingle"),
		  this)),
	  _jsonExtractSingleHelp(new HelpIcon(
		  obs_module_text(
			  "AdvSceneSwitcher.action.variable.type.queryJson.extractSingle.info"),
		  this)),
	  _jsonValue(new VariableTextEdit(this, 5, 1, 1))
{
	_jsonIndex->setMaximum(999);

	QWidget::connect(_jsonQuery, SIGNAL(editingFinished()), this,
			 SLOT(JsonQueryChanged()));
	QWidget::connect(
		_jsonIndex,
		SIGNAL(NumberVariableChanged(const NumberVariable<int> &)),
		this, SLOT(JsonIndexChanged(const NumberVariable<int> &)));
	QWidget::connect(_jsonExtractSingle, SIGNAL(stateChanged(int)), this,
			 SLOT(JsonQueryExtractSingleChanged(int)));
	QWidget::connect(_jsonValue, SIGNAL(textChanged()), this,
			 SLOT(JsonValueChanged()));

	auto layout = new QHBoxLayout();
	layout->setContentsMargins(0, 0, 0, 0);
	layout->addWidget(_jsonExtractSingle);
	layout->addWidget(_jsonExtractSingleHelp);
	layout->addStretch();
	setLayout(layout);
}

VariableActionGroupEdit *
VariableActionJsonEdit::Create(QWidget *parent,
			       std::shared_ptr<MacroActionVariable> entryData)
{
	return new VariableActionJsonEdit(parent, entryData);
}

std::unordered_map<std::string, QWidget *>
VariableActionJsonEdit::GetInlineWidgets()
{
	return {
		{"{{jsonQuery}}", _jsonQuery},
		{"{{jsonQueryHelp}}", _jsonQueryHelp},
		{"{{jsonIndex}}", _jsonIndex},
		{"{{jsonValue}}", _jsonValue},
	};
}

void VariableActionJsonEdit::UpdateEntryData()
{
	if (!_data) {
		return;
	}

	_jsonQuery->setText(_data->_jsonQuery);
	_jsonIndex->SetValue(_data->_jsonIndex);
	_jsonExtractSingle->setChecked(_data->_jsonQueryExtractSingle);
	_jsonValue->setPlainText(_data->_jsonValue);
	_loading = false;
}

void VariableActionJsonEdit::SetWidgetVisibility(
	MacroActionVariable::Action action)
{
	setVisible(action == Action::QUERY_JSON);
}

void VariableActionJsonEdit::JsonQueryChanged()
{
	GUARD_LOADING_AND_LOCK();
	_data->_jsonQuery = _jsonQuery->text().toStdString();
}

void VariableActionJsonEdit::JsonIndexChanged(const NumberVariable<int> &value)
{
	GUARD_LOADING_AND_LOCK();
	_data->_jsonIndex = value;
}

void VariableActionJsonEdit::JsonQueryExtractSingleChanged(int value)
{
	GUARD_LOADING_AND_LOCK();
	_data->_jsonQueryExtractSingle = value;
}

void VariableActionJsonEdit::JsonValueChanged()
{
	GUARD_LOADING_AND_LOCK();
	_data->_jsonValue = _jsonValue->toPlainText().toStdString();
	UpdateSize();
}

} // namespace advss