mirror of
https://github.com/obsproject/obs-studio.git
synced 2024-09-20 13:08:50 +02:00
12236d73f4
When running the program for the first time, no scene collections will show up in the scene collection menu. This changes it to forcibly save the first scene collection on the first run of the program, and then re-enumerates the list to ensure it's listed.
470 lines
12 KiB
C++
470 lines
12 KiB
C++
/******************************************************************************
|
|
Copyright (C) 2015 by Hugh Bailey <obs.jim@gmail.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/>.
|
|
******************************************************************************/
|
|
|
|
#include <obs.hpp>
|
|
#include <util/util.hpp>
|
|
#include <QMessageBox>
|
|
#include <QVariant>
|
|
#include <QFileDialog>
|
|
#include <QStandardPaths>
|
|
#include "item-widget-helpers.hpp"
|
|
#include "window-basic-main.hpp"
|
|
#include "window-namedialog.hpp"
|
|
#include "qt-wrappers.hpp"
|
|
|
|
using namespace std;
|
|
|
|
void EnumSceneCollections(std::function<bool (const char *, const char *)> &&cb)
|
|
{
|
|
char path[512];
|
|
os_glob_t *glob;
|
|
|
|
int ret = GetConfigPath(path, sizeof(path),
|
|
"obs-studio/basic/scenes/*.json");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get config path for scene "
|
|
"collections");
|
|
return;
|
|
}
|
|
|
|
if (os_glob(path, 0, &glob) != 0) {
|
|
blog(LOG_WARNING, "Failed to glob scene collections");
|
|
return;
|
|
}
|
|
|
|
for (size_t i = 0; i < glob->gl_pathc; i++) {
|
|
const char *filePath = glob->gl_pathv[i].path;
|
|
|
|
if (glob->gl_pathv[i].directory)
|
|
continue;
|
|
|
|
obs_data_t *data = obs_data_create_from_json_file_safe(filePath,
|
|
"bak");
|
|
std::string name = obs_data_get_string(data, "name");
|
|
|
|
/* if no name found, use the file name as the name
|
|
* (this only happens when switching to the new version) */
|
|
if (name.empty()) {
|
|
name = strrchr(filePath, '/') + 1;
|
|
name.resize(name.size() - 5);
|
|
}
|
|
|
|
obs_data_release(data);
|
|
|
|
if (!cb(name.c_str(), filePath))
|
|
break;
|
|
}
|
|
|
|
os_globfree(glob);
|
|
}
|
|
|
|
static bool SceneCollectionExists(const char *findName)
|
|
{
|
|
bool found = false;
|
|
auto func = [&](const char *name, const char*)
|
|
{
|
|
if (strcmp(name, findName) == 0) {
|
|
found = true;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
};
|
|
|
|
EnumSceneCollections(func);
|
|
return found;
|
|
}
|
|
|
|
static bool GetSceneCollectionName(QWidget *parent, std::string &name,
|
|
std::string &file, const char *oldName = nullptr)
|
|
{
|
|
bool rename = oldName != nullptr;
|
|
const char *title;
|
|
const char *text;
|
|
char path[512];
|
|
size_t len;
|
|
int ret;
|
|
|
|
if (rename) {
|
|
title = Str("Basic.Main.RenameSceneCollection.Title");
|
|
text = Str("Basic.Main.AddSceneCollection.Text");
|
|
} else {
|
|
title = Str("Basic.Main.AddSceneCollection.Title");
|
|
text = Str("Basic.Main.AddSceneCollection.Text");
|
|
}
|
|
|
|
for (;;) {
|
|
bool success = NameDialog::AskForName(parent, title, text,
|
|
name, QT_UTF8(oldName));
|
|
if (!success) {
|
|
return false;
|
|
}
|
|
if (name.empty()) {
|
|
OBSMessageBox::information(parent,
|
|
QTStr("NoNameEntered.Title"),
|
|
QTStr("NoNameEntered.Text"));
|
|
continue;
|
|
}
|
|
if (SceneCollectionExists(name.c_str())) {
|
|
OBSMessageBox::information(parent,
|
|
QTStr("NameExists.Title"),
|
|
QTStr("NameExists.Text"));
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (!GetFileSafeName(name.c_str(), file)) {
|
|
blog(LOG_WARNING, "Failed to create safe file name for '%s'",
|
|
name.c_str());
|
|
return false;
|
|
}
|
|
|
|
ret = GetConfigPath(path, sizeof(path), "obs-studio/basic/scenes/");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get scene collection config path");
|
|
return false;
|
|
}
|
|
|
|
len = file.size();
|
|
file.insert(0, path);
|
|
|
|
if (!GetClosestUnusedFileName(file, "json")) {
|
|
blog(LOG_WARNING, "Failed to get closest file name for %s",
|
|
file.c_str());
|
|
return false;
|
|
}
|
|
|
|
file.erase(file.size() - 5, 5);
|
|
file.erase(0, file.size() - len);
|
|
return true;
|
|
}
|
|
|
|
void OBSBasic::AddSceneCollection(bool create_new)
|
|
{
|
|
std::string name;
|
|
std::string file;
|
|
|
|
if (!GetSceneCollectionName(this, name, file))
|
|
return;
|
|
|
|
SaveProjectNow();
|
|
|
|
config_set_string(App()->GlobalConfig(), "Basic", "SceneCollection",
|
|
name.c_str());
|
|
config_set_string(App()->GlobalConfig(), "Basic", "SceneCollectionFile",
|
|
file.c_str());
|
|
if (create_new) {
|
|
CreateDefaultScene(false);
|
|
}
|
|
SaveProjectNow();
|
|
RefreshSceneCollections();
|
|
|
|
blog(LOG_INFO, "Added scene collection '%s' (%s, %s.json)",
|
|
name.c_str(), create_new ? "clean" : "duplicate",
|
|
file.c_str());
|
|
blog(LOG_INFO, "------------------------------------------------");
|
|
|
|
UpdateTitleBar();
|
|
|
|
if (api) {
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_LIST_CHANGED);
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CHANGED);
|
|
}
|
|
}
|
|
|
|
void OBSBasic::RefreshSceneCollections()
|
|
{
|
|
QList<QAction*> menuActions = ui->sceneCollectionMenu->actions();
|
|
int count = 0;
|
|
|
|
for (int i = 0; i < menuActions.count(); i++) {
|
|
QVariant v = menuActions[i]->property("file_name");
|
|
if (v.typeName() != nullptr)
|
|
delete menuActions[i];
|
|
}
|
|
|
|
const char *cur_name = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollection");
|
|
|
|
auto addCollection = [&](const char *name, const char *path)
|
|
{
|
|
std::string file = strrchr(path, '/') + 1;
|
|
file.erase(file.size() - 5, 5);
|
|
|
|
QAction *action = new QAction(QT_UTF8(name), this);
|
|
action->setProperty("file_name", QT_UTF8(path));
|
|
connect(action, &QAction::triggered,
|
|
this, &OBSBasic::ChangeSceneCollection);
|
|
action->setCheckable(true);
|
|
|
|
action->setChecked(strcmp(name, cur_name) == 0);
|
|
|
|
ui->sceneCollectionMenu->addAction(action);
|
|
count++;
|
|
return true;
|
|
};
|
|
|
|
EnumSceneCollections(addCollection);
|
|
|
|
/* force saving of first scene collection on first run, otherwise
|
|
* no scene collections will show up */
|
|
if (!count) {
|
|
long prevDisableVal = disableSaving;
|
|
|
|
disableSaving = 0;
|
|
SaveProjectNow();
|
|
disableSaving = prevDisableVal;
|
|
|
|
EnumSceneCollections(addCollection);
|
|
}
|
|
|
|
ui->actionRemoveSceneCollection->setEnabled(count > 1);
|
|
|
|
OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
|
|
|
|
main->OpenSavedProjectors();
|
|
main->ui->actionPasteFilters->setEnabled(false);
|
|
main->ui->actionPasteRef->setEnabled(false);
|
|
main->ui->actionPasteDup->setEnabled(false);
|
|
}
|
|
|
|
void OBSBasic::on_actionNewSceneCollection_triggered()
|
|
{
|
|
AddSceneCollection(true);
|
|
}
|
|
|
|
void OBSBasic::on_actionDupSceneCollection_triggered()
|
|
{
|
|
AddSceneCollection(false);
|
|
}
|
|
|
|
void OBSBasic::on_actionRenameSceneCollection_triggered()
|
|
{
|
|
std::string name;
|
|
std::string file;
|
|
|
|
std::string oldFile = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollectionFile");
|
|
const char *oldName = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollection");
|
|
|
|
bool success = GetSceneCollectionName(this, name, file, oldName);
|
|
if (!success)
|
|
return;
|
|
|
|
config_set_string(App()->GlobalConfig(), "Basic", "SceneCollection",
|
|
name.c_str());
|
|
config_set_string(App()->GlobalConfig(), "Basic", "SceneCollectionFile",
|
|
file.c_str());
|
|
SaveProjectNow();
|
|
|
|
char path[512];
|
|
int ret = GetConfigPath(path, 512, "obs-studio/basic/scenes/");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get scene collection config path");
|
|
return;
|
|
}
|
|
|
|
oldFile.insert(0, path);
|
|
oldFile += ".json";
|
|
os_unlink(oldFile.c_str());
|
|
oldFile += ".bak";
|
|
os_unlink(oldFile.c_str());
|
|
|
|
blog(LOG_INFO, "------------------------------------------------");
|
|
blog(LOG_INFO, "Renamed scene collection to '%s' (%s.json)",
|
|
name.c_str(), file.c_str());
|
|
blog(LOG_INFO, "------------------------------------------------");
|
|
|
|
UpdateTitleBar();
|
|
RefreshSceneCollections();
|
|
|
|
if (api) {
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_LIST_CHANGED);
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CHANGED);
|
|
}
|
|
}
|
|
|
|
void OBSBasic::on_actionRemoveSceneCollection_triggered()
|
|
{
|
|
std::string newName;
|
|
std::string newPath;
|
|
|
|
std::string oldFile = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollectionFile");
|
|
std::string oldName = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollection");
|
|
|
|
auto cb = [&](const char *name, const char *filePath)
|
|
{
|
|
if (strcmp(oldName.c_str(), name) != 0) {
|
|
newName = name;
|
|
newPath = filePath;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
};
|
|
|
|
EnumSceneCollections(cb);
|
|
|
|
/* this should never be true due to menu item being grayed out */
|
|
if (newPath.empty())
|
|
return;
|
|
|
|
QString text = QTStr("ConfirmRemove.Text");
|
|
text.replace("$1", QT_UTF8(oldName.c_str()));
|
|
|
|
QMessageBox::StandardButton button = OBSMessageBox::question(this,
|
|
QTStr("ConfirmRemove.Title"), text);
|
|
if (button == QMessageBox::No)
|
|
return;
|
|
|
|
char path[512];
|
|
int ret = GetConfigPath(path, 512, "obs-studio/basic/scenes/");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get scene collection config path");
|
|
return;
|
|
}
|
|
|
|
oldFile.insert(0, path);
|
|
oldFile += ".json";
|
|
os_unlink(oldFile.c_str());
|
|
oldFile += ".bak";
|
|
os_unlink(oldFile.c_str());
|
|
|
|
Load(newPath.c_str());
|
|
RefreshSceneCollections();
|
|
|
|
const char *newFile = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollectionFile");
|
|
|
|
blog(LOG_INFO, "Removed scene collection '%s' (%s.json), "
|
|
"switched to '%s' (%s.json)",
|
|
oldName.c_str(), oldFile.c_str(),
|
|
newName.c_str(), newFile);
|
|
blog(LOG_INFO, "------------------------------------------------");
|
|
|
|
UpdateTitleBar();
|
|
|
|
if (api) {
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_LIST_CHANGED);
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CHANGED);
|
|
}
|
|
}
|
|
|
|
void OBSBasic::on_actionImportSceneCollection_triggered()
|
|
{
|
|
char path[512];
|
|
|
|
QString home = QDir::homePath();
|
|
|
|
int ret = GetConfigPath(path, 512, "obs-studio/basic/scenes/");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get scene collection config path");
|
|
return;
|
|
}
|
|
|
|
QString file = QFileDialog::getOpenFileName(
|
|
this,
|
|
QTStr("Basic.MainMenu.SceneCollection.Import"),
|
|
home,
|
|
"JSON Files (*.json)");
|
|
|
|
QFileInfo finfo(file);
|
|
QString filename = finfo.fileName();
|
|
QFileInfo destinfo(path + filename);
|
|
|
|
if (!file.isEmpty() && !file.isNull()) {
|
|
if (!destinfo.exists()) {
|
|
QFile::copy(file, path + filename);
|
|
RefreshSceneCollections();
|
|
} else {
|
|
OBSMessageBox::information(this,
|
|
QTStr("Basic.MainMenu.SceneCollection.Import"),
|
|
QTStr("Basic.MainMenu.SceneCollection.Exists"));
|
|
}
|
|
}
|
|
}
|
|
|
|
void OBSBasic::on_actionExportSceneCollection_triggered()
|
|
{
|
|
char path[512];
|
|
|
|
QString home = QDir::homePath();
|
|
|
|
QString currentFile = QT_UTF8(config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollectionFile"));
|
|
|
|
int ret = GetConfigPath(path, 512, "obs-studio/basic/scenes/");
|
|
if (ret <= 0) {
|
|
blog(LOG_WARNING, "Failed to get scene collection config path");
|
|
return;
|
|
}
|
|
|
|
QString exportFile = QFileDialog::getSaveFileName(
|
|
this,
|
|
QTStr("Basic.MainMenu.SceneCollection.Export"),
|
|
home + "/" + currentFile,
|
|
"JSON Files (*.json)");
|
|
|
|
string file = QT_TO_UTF8(exportFile);
|
|
|
|
if (!exportFile.isEmpty() && !exportFile.isNull())
|
|
QFile::copy(path + currentFile + ".json", exportFile);
|
|
}
|
|
|
|
void OBSBasic::ChangeSceneCollection()
|
|
{
|
|
QAction *action = reinterpret_cast<QAction*>(sender());
|
|
std::string fileName;
|
|
|
|
if (!action)
|
|
return;
|
|
|
|
fileName = QT_TO_UTF8(action->property("file_name").value<QString>());
|
|
if (fileName.empty())
|
|
return;
|
|
|
|
const char *oldName = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollection");
|
|
if (action->text().compare(QT_UTF8(oldName)) == 0) {
|
|
action->setChecked(true);
|
|
return;
|
|
}
|
|
|
|
SaveProjectNow();
|
|
|
|
Load(fileName.c_str());
|
|
RefreshSceneCollections();
|
|
|
|
const char *newName = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollection");
|
|
const char *newFile = config_get_string(App()->GlobalConfig(),
|
|
"Basic", "SceneCollectionFile");
|
|
|
|
blog(LOG_INFO, "Switched to scene collection '%s' (%s.json)",
|
|
newName, newFile);
|
|
blog(LOG_INFO, "------------------------------------------------");
|
|
|
|
UpdateTitleBar();
|
|
|
|
if (api)
|
|
api->on_event(OBS_FRONTEND_EVENT_SCENE_COLLECTION_CHANGED);
|
|
}
|