obs-aitum-multistream/output-dialog.cpp

#include "output-dialog.hpp"

#include <QDialog>
#include <QVBoxLayout>
#include <QHBoxLayout>
#include <QLabel>
#include <QStackedWidget>
#include <QPushButton>
#include <QAbstractButton>
#include <QToolButton>
#include <QFormLayout>
#include <QComboBox>
#include <QLineEdit>
#include <QSizePolicy>
#include "obs-module.h"
#include "util/platform.h"

// Reset output values, e.g. when user hits the back button
void OutputDialog::resetOutputs() {
	outputName = QString("");
	outputServer = QString("");
	outputKey = QString("");
}

QToolButton *OutputDialog::selectionButton(std::string title, QIcon icon, int selectionStep) {
	auto button = new QToolButton;
	
	button->setText(QString::fromUtf8(title));
	button->setIcon(icon);
	button->setIconSize(QSize(32, 32));
	button->setToolButtonStyle(Qt::ToolButtonTextUnderIcon);
	button->setStyleSheet("min-width: 110px; max-width: 110px; min-height: 90px; max-height: 90px; padding-top: 16px; font-weight: bold;");
	
	connect(button, &QPushButton::clicked, [this, selectionStep] {
		stackedWidget->setCurrentIndex(selectionStep);
	});
	
	return button;
}

// Gets a service from the service array matched by name
obs_data_t *OutputDialog::getService(std::string serviceName) {
	auto total = obs_data_array_count(servicesData);

	for (size_t idx = 0; idx < total; idx++) {
		auto item = obs_data_array_item(servicesData, idx);
//		blog(LOG_WARNING, "[Aitum Multistream] %s", obs_data_get_string(item, "name"));

		if (serviceName == obs_data_get_string(item, "name")) {
			return item;
		}
	}
	
	return nullptr;
}

OutputDialog::OutputDialog(QDialog *parent) : QDialog(parent) {
	// Load the services from rtmp-services plugin
	auto servicesPath = obs_module_get_config_path(obs_get_module("rtmp-services"), "services.json");
	auto absolutePath = os_get_abs_path_ptr(servicesPath);
	auto allData = obs_data_create_from_json_file(absolutePath);
	
	servicesData = obs_data_get_array(allData, "services");
	
	bfree(allData);
	bfree(servicesPath);
	bfree(absolutePath);
	
	//
	setModal(true);
	setContentsMargins(0, 0, 0, 0);
	setFixedSize(650, 400);
	
	stackedWidget = new QStackedWidget;
	
	stackedWidget->setStyleSheet("padding: 0px; margin: 0px;");
	
	// Service selection page
	stackedWidget->addWidget(WizardServicePage());
	
	// Pages for each service
	stackedWidget->addWidget(WizardInfoKick());
	stackedWidget->addWidget(WizardInfoYouTube());
	stackedWidget->addWidget(WizardInfoTwitter());
	stackedWidget->addWidget(WizardInfoUnknown());
	stackedWidget->addWidget(WizardInfoTwitch());
	stackedWidget->addWidget(WizardInfoTrovo());
	stackedWidget->addWidget(WizardInfoTikTok());
	stackedWidget->addWidget(WizardInfoFacebook());

	stackedWidget->setCurrentIndex(0);
	
	stackedWidget->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
	setWindowTitle(obs_module_text("NewOutputWindowTitle"));
	
	auto stackedLayout = new QVBoxLayout;
	stackedLayout->setContentsMargins(0, 0, 0, 0);
	stackedLayout->addWidget(stackedWidget, 1);

	setLayout(stackedLayout);
	show();
}

QWidget *OutputDialog::WizardServicePage() {
	auto page = new QWidget(this);
	auto pageLayout = new QVBoxLayout;
	pageLayout->setAlignment(Qt::AlignTop);
	
	auto description = new QLabel(QString::fromUtf8(obs_module_text("NewOutputSelectService")));
	description->setStyleSheet("margin-bottom: 20px;");
	pageLayout->addWidget(description);
	
	// layout for service selection
	auto selectionLayout = new QVBoxLayout;
	selectionLayout->setAlignment(Qt::AlignCenter);
	
	auto spacerTest = new QSpacerItem(20, 45, QSizePolicy::Fixed, QSizePolicy::Fixed);
	
	pageLayout->addSpacerItem(spacerTest);
	
	// row 1
	auto rowOne = new QHBoxLayout;
	
	rowOne->addWidget(selectionButton("Twitch", platformIconTwitch, 5));
	rowOne->addWidget(selectionButton("YouTube", platformIconYouTube, 2));
	rowOne->addWidget(selectionButton("TikTok", platformIconTikTok, 7));
	rowOne->addWidget(selectionButton("Facebook", platformIconFacebook, 8));
	
	selectionLayout->addLayout(rowOne);
	
	// row 2
	auto rowTwo = new QHBoxLayout;
	
	rowTwo->addWidget(selectionButton("Trovo", platformIconTrovo, 6));
	rowTwo->addWidget(selectionButton("X (Twitter)", platformIconTwitter, 3));
	rowTwo->addWidget(selectionButton("Kick", platformIconKick, 1));
	rowTwo->addWidget(selectionButton(obs_module_text("OtherService"), platformIconUnknown, 4));
	
	selectionLayout->addLayout(rowTwo);
	
	//
	pageLayout->addLayout(selectionLayout);
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoKick() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Kick Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoYouTube() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("YouTube Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoTwitter() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Twitter Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoUnknown() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Unknown Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoTwitch() {
	auto page = new QWidget(this);

	page->setStyleSheet("padding: 0px; margin: 0px;");
	
	auto pageLayout = new QVBoxLayout;
	pageLayout->setSpacing(16);
	
	
	// Heading
	auto title = new QLabel(QString::fromUtf8(obs_module_text("TwitchServiceInfo")));
	pageLayout->addWidget(title);
	
	// Content
	auto contentLayout = new QVBoxLayout;
//	auto filler = new QLabel(QString("filler text"));
//	contentLayout->addWidget(filler);
	
	// Form
	auto formLayout = new QFormLayout;
	formLayout->setFieldGrowthPolicy(QFormLayout::AllNonFixedFieldsGrow);
	formLayout->setLabelAlignment(Qt::AlignRight | Qt::AlignTrailing | Qt::AlignVCenter);
	formLayout->setSpacing(12);
	
	// Output name
	auto outputNameTitle = new QLabel(QString::fromUtf8(obs_module_text("OutputName")));
	outputNameTitle->setStyleSheet("font-weight: bold;");
	
	auto outputName = new QLineEdit;
	outputName->setStyleSheet("padding: 4px 8px;");
	formLayout->addRow(outputNameTitle, outputName);
	
	// Server selection
	
	auto serverSelectionTitle = new QLabel(QString::fromUtf8(obs_module_text("TwitchServer")));
	serverSelectionTitle->setStyleSheet("font-weight: bold;");
	auto serverSelection = new QComboBox;
	serverSelection->setMinimumHeight(30);
	serverSelection->setStyleSheet("padding: 4px 8px;");
	auto rawOptions = getService("Twitch");
	
	// turn raw options into actual selectable options
	if (rawOptions != nullptr) {
		auto servers = obs_data_get_array(rawOptions, "servers");
		auto totalServers = obs_data_array_count(servers);
		
		for (size_t idx = 0; idx < totalServers; idx++) {
			auto item = obs_data_array_item(servers, idx);
			serverSelection->addItem(obs_data_get_string(item, "name"), obs_data_get_string(item, "url"));
		}
	}
	
	
	formLayout->addRow(serverSelectionTitle, serverSelection);
	
	// Server info
	auto serverInfo = new QLabel;
	serverInfo->setTextFormat(Qt::RichText);
	serverInfo->setText(QString::fromUtf8(obs_module_text("TwitchServerInfo")));
	serverInfo->setOpenExternalLinks(true);
	serverInfo->setWordWrap(true);
	formLayout->addWidget(serverInfo);
	
	// Server key
	auto outputKeyTitle = new QLabel(QString::fromUtf8(obs_module_text("TwitchStreamKey")));
	outputNameTitle->setStyleSheet("font-weight: bold;");
	
	auto outputKey = new QLineEdit;
	outputName->setStyleSheet("padding: 4px 8px;");
	formLayout->addRow(outputKeyTitle, outputKey);

	// Server key info
	auto keyInfo = new QLabel;
	keyInfo->setTextFormat(Qt::RichText);
	keyInfo->setText(QString::fromUtf8(obs_module_text("TwitchStreamKeyInfo")));
	keyInfo->setOpenExternalLinks(true);
	keyInfo->setWordWrap(true);
	keyInfo->setAlignment(Qt::AlignTop);
	formLayout->addWidget(keyInfo);
	
	
	contentLayout->addLayout(formLayout);
	
	// spacing
	auto spacer = new QSpacerItem(1, 20, QSizePolicy::Minimum, QSizePolicy::MinimumExpanding);
	contentLayout->addSpacerItem(spacer);
	
	pageLayout->addLayout(contentLayout);
	
	// Controls
	auto controlsLayout = new QHBoxLayout;
	controlsLayout->setSpacing(12);
	
		
	// back button
	auto serviceButton = new QPushButton;
	serviceButton->setText(QString("< Back"));
	serviceButton->setStyleSheet("padding: 4px 12px;");
	serviceButton->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
	
	connect(serviceButton, &QPushButton::clicked, [this] {
		stackedWidget->setCurrentIndex(0);
		resetOutputs();
	});
	
	controlsLayout->addWidget(serviceButton, 0);
	
	controlsLayout->addStretch(1);
	// confirm button
	auto confirmButton = new QPushButton;
	confirmButton->setEnabled(false);
	confirmButton->setText(QString("Create Output"));
	confirmButton->setStyleSheet("padding: 4px 12px;");
	confirmButton->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
	
	controlsLayout->addWidget(confirmButton, 0);


	// Hook it all together
	pageLayout->addLayout(controlsLayout);
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoTrovo() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Trovo Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoTikTok() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Tiktok Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}

QWidget *OutputDialog::WizardInfoFacebook() {
	auto page = new QWidget(this);
		
	auto pageLayout = new QVBoxLayout;
	
	auto title = new QLabel(QString("Facebook Service page"));
	pageLayout->addWidget(title);
	
	page->setLayout(pageLayout);
	
	return page;
}