123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562 |
- #include "moc_window-extra-browsers.cpp"
- #include "window-dock-browser.hpp"
- #include "window-basic-main.hpp"
- #include <qt-wrappers.hpp>
- #include <QLineEdit>
- #include <QHBoxLayout>
- #include <QUuid>
- #include <json11.hpp>
- #include "ui_OBSExtraBrowsers.h"
- using namespace json11;
- #define OBJ_NAME_SUFFIX "_extraBrowser"
- enum class Column : int {
- Title,
- Url,
- Delete,
- Count,
- };
- /* ------------------------------------------------------------------------- */
- void ExtraBrowsersModel::Reset()
- {
- items.clear();
- OBSBasic *main = OBSBasic::Get();
- for (int i = 0; i < main->extraBrowserDocks.size(); i++) {
- Item item;
- item.prevIdx = i;
- item.title = main->extraBrowserDockNames[i];
- item.url = main->extraBrowserDockTargets[i];
- items.push_back(item);
- }
- }
- int ExtraBrowsersModel::rowCount(const QModelIndex &) const
- {
- int count = items.size() + 1;
- return count;
- }
- int ExtraBrowsersModel::columnCount(const QModelIndex &) const
- {
- return (int)Column::Count;
- }
- QVariant ExtraBrowsersModel::data(const QModelIndex &index, int role) const
- {
- int column = index.column();
- int idx = index.row();
- int count = items.size();
- bool validRole = role == Qt::DisplayRole || role == Qt::AccessibleTextRole;
- if (!validRole)
- return QVariant();
- if (idx >= 0 && idx < count) {
- switch (column) {
- case (int)Column::Title:
- return items[idx].title;
- case (int)Column::Url:
- return items[idx].url;
- }
- } else if (idx == count) {
- switch (column) {
- case (int)Column::Title:
- return newTitle;
- case (int)Column::Url:
- return newURL;
- }
- }
- return QVariant();
- }
- QVariant ExtraBrowsersModel::headerData(int section, Qt::Orientation orientation, int role) const
- {
- bool validRole = role == Qt::DisplayRole || role == Qt::AccessibleTextRole;
- if (validRole && orientation == Qt::Orientation::Horizontal) {
- switch (section) {
- case (int)Column::Title:
- return QTStr("ExtraBrowsers.DockName");
- case (int)Column::Url:
- return QStringLiteral("URL");
- }
- }
- return QVariant();
- }
- Qt::ItemFlags ExtraBrowsersModel::flags(const QModelIndex &index) const
- {
- Qt::ItemFlags flags = QAbstractTableModel::flags(index);
- if (index.column() != (int)Column::Delete)
- flags |= Qt::ItemIsEditable;
- return flags;
- }
- class DelButton : public QPushButton {
- public:
- inline DelButton(QModelIndex index_) : QPushButton(), index(index_) {}
- QPersistentModelIndex index;
- };
- class EditWidget : public QLineEdit {
- public:
- inline EditWidget(QWidget *parent, QModelIndex index_) : QLineEdit(parent), index(index_) {}
- QPersistentModelIndex index;
- };
- void ExtraBrowsersModel::AddDeleteButton(int idx)
- {
- QTableView *widget = reinterpret_cast<QTableView *>(parent());
- QModelIndex index = createIndex(idx, (int)Column::Delete, nullptr);
- QPushButton *del = new DelButton(index);
- del->setProperty("class", "icon-trash");
- del->setObjectName("extraPanelDelete");
- del->setMinimumSize(QSize(20, 20));
- connect(del, &QPushButton::clicked, this, &ExtraBrowsersModel::DeleteItem);
- widget->setIndexWidget(index, del);
- widget->setRowHeight(idx, 20);
- widget->setColumnWidth(idx, 20);
- }
- void ExtraBrowsersModel::CheckToAdd()
- {
- if (newTitle.isEmpty() || newURL.isEmpty())
- return;
- int idx = items.size() + 1;
- beginInsertRows(QModelIndex(), idx, idx);
- Item item;
- item.prevIdx = -1;
- item.title = newTitle;
- item.url = newURL;
- items.push_back(item);
- newTitle = "";
- newURL = "";
- endInsertRows();
- AddDeleteButton(idx - 1);
- }
- void ExtraBrowsersModel::UpdateItem(Item &item)
- {
- int idx = item.prevIdx;
- OBSBasic *main = OBSBasic::Get();
- BrowserDock *dock = reinterpret_cast<BrowserDock *>(main->extraBrowserDocks[idx].get());
- dock->setWindowTitle(item.title);
- dock->setObjectName(item.title + OBJ_NAME_SUFFIX);
- if (main->extraBrowserDockNames[idx] != item.title) {
- main->extraBrowserDockNames[idx] = item.title;
- dock->toggleViewAction()->setText(item.title);
- dock->setTitle(item.title);
- }
- if (main->extraBrowserDockTargets[idx] != item.url) {
- dock->cefWidget->setURL(QT_TO_UTF8(item.url));
- main->extraBrowserDockTargets[idx] = item.url;
- }
- }
- void ExtraBrowsersModel::DeleteItem()
- {
- QTableView *widget = reinterpret_cast<QTableView *>(parent());
- DelButton *del = reinterpret_cast<DelButton *>(sender());
- int row = del->index.row();
- /* there's some sort of internal bug in Qt and deleting certain index
- * widgets or "editors" that can cause a crash inside Qt if the widget
- * is not manually removed, at least on 5.7 */
- widget->setIndexWidget(del->index, nullptr);
- del->deleteLater();
- /* --------- */
- beginRemoveRows(QModelIndex(), row, row);
- int prevIdx = items[row].prevIdx;
- items.removeAt(row);
- if (prevIdx != -1) {
- int i = 0;
- for (; i < deleted.size() && deleted[i] < prevIdx; i++)
- ;
- deleted.insert(i, prevIdx);
- }
- endRemoveRows();
- }
- void ExtraBrowsersModel::Apply()
- {
- OBSBasic *main = OBSBasic::Get();
- for (Item &item : items) {
- if (item.prevIdx != -1) {
- UpdateItem(item);
- } else {
- QString uuid = QUuid::createUuid().toString();
- uuid.replace(QRegularExpression("[{}-]"), "");
- main->AddExtraBrowserDock(item.title, item.url, uuid, true);
- }
- }
- for (int i = deleted.size() - 1; i >= 0; i--) {
- int idx = deleted[i];
- main->extraBrowserDockTargets.removeAt(idx);
- main->extraBrowserDockNames.removeAt(idx);
- main->extraBrowserDocks.removeAt(idx);
- }
- if (main->extraBrowserDocks.empty())
- main->extraBrowserMenuDocksSeparator.clear();
- deleted.clear();
- Reset();
- }
- void ExtraBrowsersModel::TabSelection(bool forward)
- {
- QListView *widget = reinterpret_cast<QListView *>(parent());
- QItemSelectionModel *selModel = widget->selectionModel();
- QModelIndex sel = selModel->currentIndex();
- int row = sel.row();
- int col = sel.column();
- switch (sel.column()) {
- case (int)Column::Title:
- if (!forward) {
- if (row == 0) {
- return;
- }
- row -= 1;
- }
- col += 1;
- break;
- case (int)Column::Url:
- if (forward) {
- if (row == items.size()) {
- return;
- }
- row += 1;
- }
- col -= 1;
- }
- sel = createIndex(row, col, nullptr);
- selModel->setCurrentIndex(sel, QItemSelectionModel::Clear);
- }
- void ExtraBrowsersModel::Init()
- {
- for (int i = 0; i < items.count(); i++)
- AddDeleteButton(i);
- }
- /* ------------------------------------------------------------------------- */
- QWidget *ExtraBrowsersDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &,
- const QModelIndex &index) const
- {
- QLineEdit *text = new EditWidget(parent, index);
- text->installEventFilter(const_cast<ExtraBrowsersDelegate *>(this));
- text->setSizePolicy(QSizePolicy(QSizePolicy::Policy::Expanding, QSizePolicy::Policy::Expanding,
- QSizePolicy::ControlType::LineEdit));
- return text;
- }
- void ExtraBrowsersDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
- {
- QLineEdit *text = reinterpret_cast<QLineEdit *>(editor);
- text->blockSignals(true);
- text->setText(index.data().toString());
- text->blockSignals(false);
- }
- bool ExtraBrowsersDelegate::eventFilter(QObject *object, QEvent *event)
- {
- QLineEdit *edit = qobject_cast<QLineEdit *>(object);
- if (!edit)
- return false;
- if (LineEditCanceled(event)) {
- RevertText(edit);
- }
- if (LineEditChanged(event)) {
- UpdateText(edit);
- if (event->type() == QEvent::KeyPress) {
- QKeyEvent *keyEvent = static_cast<QKeyEvent *>(event);
- if (keyEvent->key() == Qt::Key_Tab) {
- model->TabSelection(true);
- } else if (keyEvent->key() == Qt::Key_Backtab) {
- model->TabSelection(false);
- }
- }
- return true;
- }
- return false;
- }
- bool ExtraBrowsersDelegate::ValidName(const QString &name) const
- {
- for (auto &item : model->items) {
- if (name.compare(item.title, Qt::CaseInsensitive) == 0) {
- return false;
- }
- }
- return true;
- }
- void ExtraBrowsersDelegate::RevertText(QLineEdit *edit_)
- {
- EditWidget *edit = reinterpret_cast<EditWidget *>(edit_);
- int row = edit->index.row();
- int col = edit->index.column();
- bool newItem = (row == model->items.size());
- QString oldText;
- if (col == (int)Column::Title) {
- oldText = newItem ? model->newTitle : model->items[row].title;
- } else {
- oldText = newItem ? model->newURL : model->items[row].url;
- }
- edit->setText(oldText);
- }
- bool ExtraBrowsersDelegate::UpdateText(QLineEdit *edit_)
- {
- EditWidget *edit = reinterpret_cast<EditWidget *>(edit_);
- int row = edit->index.row();
- int col = edit->index.column();
- bool newItem = (row == model->items.size());
- QString text = edit->text().trimmed();
- if (!newItem && text.isEmpty()) {
- return false;
- }
- if (col == (int)Column::Title) {
- QString oldText = newItem ? model->newTitle : model->items[row].title;
- bool same = oldText.compare(text, Qt::CaseInsensitive) == 0;
- if (!same && !ValidName(text)) {
- edit->setText(oldText);
- return false;
- }
- }
- if (!newItem) {
- /* if edited existing item, update it*/
- switch (col) {
- case (int)Column::Title:
- model->items[row].title = text;
- break;
- case (int)Column::Url:
- model->items[row].url = text;
- break;
- }
- } else {
- /* if both new values filled out, create new one */
- switch (col) {
- case (int)Column::Title:
- model->newTitle = text;
- break;
- case (int)Column::Url:
- model->newURL = text;
- break;
- }
- model->CheckToAdd();
- }
- emit commitData(edit);
- return true;
- }
- /* ------------------------------------------------------------------------- */
- OBSExtraBrowsers::OBSExtraBrowsers(QWidget *parent) : QDialog(parent), ui(new Ui::OBSExtraBrowsers)
- {
- ui->setupUi(this);
- setAttribute(Qt::WA_DeleteOnClose, true);
- setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);
- model = new ExtraBrowsersModel(ui->table);
- ui->table->setModel(model);
- ui->table->setItemDelegateForColumn((int)Column::Title, new ExtraBrowsersDelegate(model));
- ui->table->setItemDelegateForColumn((int)Column::Url, new ExtraBrowsersDelegate(model));
- ui->table->horizontalHeader()->setSectionResizeMode(QHeaderView::ResizeMode::Stretch);
- ui->table->horizontalHeader()->setSectionResizeMode((int)Column::Delete, QHeaderView::ResizeMode::Fixed);
- ui->table->setEditTriggers(QAbstractItemView::EditTrigger::CurrentChanged);
- }
- OBSExtraBrowsers::~OBSExtraBrowsers() {}
- void OBSExtraBrowsers::closeEvent(QCloseEvent *event)
- {
- QDialog::closeEvent(event);
- model->Apply();
- }
- void OBSExtraBrowsers::on_apply_clicked()
- {
- model->Apply();
- }
- /* ------------------------------------------------------------------------- */
- void OBSBasic::ClearExtraBrowserDocks()
- {
- extraBrowserDockTargets.clear();
- extraBrowserDockNames.clear();
- extraBrowserDocks.clear();
- }
- void OBSBasic::LoadExtraBrowserDocks()
- {
- const char *jsonStr = config_get_string(App()->GetUserConfig(), "BasicWindow", "ExtraBrowserDocks");
- std::string err;
- Json json = Json::parse(jsonStr, err);
- if (!err.empty())
- return;
- Json::array array = json.array_items();
- if (!array.empty())
- extraBrowserMenuDocksSeparator = ui->menuDocks->addSeparator();
- for (Json &item : array) {
- std::string title = item["title"].string_value();
- std::string url = item["url"].string_value();
- std::string uuid = item["uuid"].string_value();
- AddExtraBrowserDock(title.c_str(), url.c_str(), uuid.c_str(), false);
- }
- }
- void OBSBasic::SaveExtraBrowserDocks()
- {
- Json::array array;
- for (int i = 0; i < extraBrowserDocks.size(); i++) {
- QDockWidget *dock = extraBrowserDocks[i].get();
- QString title = extraBrowserDockNames[i];
- QString url = extraBrowserDockTargets[i];
- QString uuid = dock->property("uuid").toString();
- Json::object obj{
- {"title", QT_TO_UTF8(title)},
- {"url", QT_TO_UTF8(url)},
- {"uuid", QT_TO_UTF8(uuid)},
- };
- array.push_back(obj);
- }
- std::string output = Json(array).dump();
- config_set_string(App()->GetUserConfig(), "BasicWindow", "ExtraBrowserDocks", output.c_str());
- }
- void OBSBasic::ManageExtraBrowserDocks()
- {
- if (!extraBrowsers.isNull()) {
- extraBrowsers->show();
- extraBrowsers->raise();
- return;
- }
- extraBrowsers = new OBSExtraBrowsers(this);
- extraBrowsers->show();
- }
- void OBSBasic::AddExtraBrowserDock(const QString &title, const QString &url, const QString &uuid, bool firstCreate)
- {
- static int panel_version = -1;
- if (panel_version == -1) {
- panel_version = obs_browser_qcef_version();
- }
- BrowserDock *dock = new BrowserDock(title);
- QString bId(uuid.isEmpty() ? QUuid::createUuid().toString() : uuid);
- bId.replace(QRegularExpression("[{}-]"), "");
- dock->setProperty("uuid", bId);
- dock->setObjectName(title + OBJ_NAME_SUFFIX);
- dock->resize(460, 600);
- dock->setMinimumSize(80, 80);
- dock->setWindowTitle(title);
- dock->setAllowedAreas(Qt::AllDockWidgetAreas);
- QCefWidget *browser = cef->create_widget(dock, QT_TO_UTF8(url), nullptr);
- if (browser && panel_version >= 1)
- browser->allowAllPopups(true);
- dock->SetWidget(browser);
- /* Add support for Twitch Dashboard panels */
- if (url.contains("twitch.tv/popout") && url.contains("dashboard/live")) {
- QRegularExpression re("twitch.tv\\/popout\\/([^/]+)\\/");
- QRegularExpressionMatch match = re.match(url);
- QString username = match.captured(1);
- if (username.length() > 0) {
- std::string script;
- script = "Object.defineProperty(document, 'referrer', { get: () => '";
- script += "https://twitch.tv/";
- script += QT_TO_UTF8(username);
- script += "/dashboard/live";
- script += "'});";
- browser->setStartupScript(script);
- }
- }
- AddDockWidget(dock, Qt::RightDockWidgetArea, true);
- extraBrowserDocks.push_back(std::shared_ptr<QDockWidget>(dock));
- extraBrowserDockNames.push_back(title);
- extraBrowserDockTargets.push_back(url);
- if (firstCreate) {
- dock->setFloating(true);
- QPoint curPos = pos();
- QSize wSizeD2 = size() / 2;
- QSize dSizeD2 = dock->size() / 2;
- curPos.setX(curPos.x() + wSizeD2.width() - dSizeD2.width());
- curPos.setY(curPos.y() + wSizeD2.height() - dSizeD2.height());
- dock->move(curPos);
- dock->setVisible(true);
- }
- }
|