Sandboxie/SandboxiePlus/SandMan/Windows/OptionsGrouping.cpp

289 lines
7.7 KiB
C++

#include "stdafx.h"
#include "OptionsWindow.h"
#include "SandMan.h"
#include "SettingsWindow.h"
#include "../MiscHelpers/Common/Settings.h"
#include "../MiscHelpers/Common/Common.h"
#include "../MiscHelpers/Common/ComboInputDialog.h"
#include "../MiscHelpers/Common/SettingsWidgets.h"
#include "Helpers/WinAdmin.h"
void COptionsWindow::LoadGroups()
{
ui.treeGroups->clear();
QStringList ProcessGroups = m_pBox->GetTextList("ProcessGroup", m_Template);
foreach(const QString& Group, ProcessGroups)
{
QStringList Entries = Group.split(",");
QString GroupName = Entries.takeFirst();
QTreeWidgetItem* pItem = new QTreeWidgetItem();
pItem->setData(0, Qt::UserRole, GroupName);
if (GroupName.length() > 2)
GroupName = GroupName.mid(1, GroupName.length() - 2);
pItem->setText(0, GroupName);
for (int i = 0; i < Entries.count(); i++)
{
if (Entries[i].isEmpty())
continue;
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
SetProgramItem(Entries[i], pSubItem, 0);
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
pItem->addChild(pSubItem);
}
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
ui.treeGroups->addTopLevelItem(pItem);
}
LoadGroupsTmpl();
ui.treeGroups->expandAll();
m_GroupsChanged = false;
}
void COptionsWindow::LoadGroupsTmpl(bool bUpdate)
{
if (ui.chkShowGroupTmpl->isChecked())
{
foreach(const QString& Template, m_pBox->GetTemplates())
{
foreach(const QString& Group, m_pBox->GetTextListTmpl("ProcessGroup", Template))
{
QStringList Entries = Group.split(",");
QString GroupName = Entries.takeFirst();
QTreeWidgetItem* pItem = new QTreeWidgetItem();
if (GroupName.length() > 2)
GroupName = GroupName.mid(1, GroupName.length() - 2);
pItem->setText(0, GroupName + " (" + Template + ")");
for (int i = 0; i < Entries.count(); i++)
{
if (Entries[i].isEmpty())
continue;
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
SetProgramItem(Entries[i], pSubItem, 0);
pItem->addChild(pSubItem);
}
ui.treeGroups->addTopLevelItem(pItem);
}
}
if (bUpdate)
ui.treeGroups->expandAll();
}
else if (bUpdate)
{
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); )
{
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
QString GroupName = pItem->data(0, Qt::UserRole).toString();
if (GroupName.isEmpty()) {
delete pItem;
continue; // entry from template
}
i++;
}
}
}
QStringList COptionsWindow::GetCurrentGroups()
{
QStringList Groups;
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
{
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
QString GroupName = pItem->data(0, Qt::UserRole).toString();
Groups.append(GroupName);
}
foreach(const QString& Template, m_pBox->GetTemplates())
{
foreach(const QString& Group, m_pBox->GetTextListTmpl("ProcessGroup", Template))
{
QStringList Entries = Group.split(",");
QString GroupName = Entries.takeFirst();
if (!Groups.contains(GroupName))
Groups.append(GroupName);
}
}
return Groups;
}
void COptionsWindow::SaveGroups()
{
QStringList ProcessGroups;
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
{
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
QString GroupName = pItem->data(0, Qt::UserRole).toString();
if (GroupName.isEmpty())
continue; // this is a template entry, dont save
QStringList Programs;
for (int j = 0; j < pItem->childCount(); j++)
Programs.append(pItem->child(j)->data(0, Qt::UserRole).toString());
QString Group = GroupName + "," + Programs.join(",");
ProcessGroups.append(Group);
}
WriteTextList("ProcessGroup", ProcessGroups);
m_GroupsChanged = false;
}
void COptionsWindow::OnAddGroup()
{
QString Value = QInputDialog::getText(this, "Sandboxie-Plus", tr("Please enter a name for the new group"), QLineEdit::Normal, "NewGroup");
if (Value.isEmpty())
return;
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++) {
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
if (pItem->text(0).compare(Value, Qt::CaseInsensitive) == 0)
return;
}
QTreeWidgetItem* pItem = new QTreeWidgetItem();
pItem->setText(0, Value);
pItem->setData(0, Qt::UserRole, "<" + Value + ">");
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
ui.treeGroups->addTopLevelItem(pItem);
m_GroupsChanged = true;
OnOptChanged();
}
void COptionsWindow::AddProgramToGroup(const QString& Program, const QString& Group)
{
QTreeWidgetItem* pItem = FindGroupByName(Group, true);
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
SetProgramItem(Program, pSubItem, 0);
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
pItem->addChild(pSubItem);
m_GroupsChanged = true;
OnOptChanged();
}
bool COptionsWindow::DelProgramFromGroup(const QString& Program, const QString& Group)
{
QTreeWidgetItem* pItem = FindGroupByName(Group, false);
if (!pItem) return false;
bool bFound = false;
for (int j = 0; j < pItem->childCount(); j++){
QTreeWidgetItem* pProgItem = pItem->child(j);
if (pProgItem->data(0, Qt::UserRole).toString().compare(Program, Qt::CaseInsensitive) == 0) {
delete pProgItem;
bFound = true;
if (pItem->childCount() == 0)
delete pItem;
break;
}
}
m_GroupsChanged = true;
OnOptChanged();
return bFound;
}
QTreeWidgetItem* COptionsWindow::FindGroupByName(const QString& Group, bool bAdd)
{
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++) {
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
if (pItem->data(0, Qt::UserRole).toString().compare(Group, Qt::CaseInsensitive) == 0)
return pItem;
}
if (bAdd)
{
QString GroupName = Group;
QTreeWidgetItem* pItem = new QTreeWidgetItem();
pItem->setData(0, Qt::UserRole, GroupName);
if (GroupName.length() > 2)
GroupName = GroupName.mid(1, GroupName.length() - 2);
pItem->setText(0, GroupName);
pItem->setFlags(pItem->flags() | Qt::ItemIsEditable);
ui.treeGroups->addTopLevelItem(pItem);
return pItem;
}
return NULL;
}
void COptionsWindow::OnAddProg()
{
QTreeWidgetItem* pItem = ui.treeGroups->currentItem();
while (pItem && pItem->parent())
pItem = pItem->parent();
if (!pItem)
{
QMessageBox::warning(this, "SandboxiePlus", tr("Please select group first."));
return;
}
if (pItem->data(0, Qt::UserRole).toString().isEmpty()) {
QMessageBox::warning(this, "SandboxiePlus", tr("Template values can not be edited."));
return;
}
QString Value = SelectProgram();
if (Value.isEmpty())
return;
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
SetProgramItem(Value, pSubItem, 0);
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
pItem->addChild(pSubItem);
m_GroupsChanged = true;
OnOptChanged();
}
void COptionsWindow::OnDelProg()
{
QTreeWidgetItem* pItem = ui.treeGroups->currentItem();
if (!pItem)
return;
if ((pItem->parent() ? pItem->parent() : pItem)->data(0, Qt::UserRole).toString().isEmpty()) {
QMessageBox::warning(this, "SandboxiePlus", tr("Template values can not be edited."));
return;
}
delete pItem;
m_GroupsChanged = true;
OnOptChanged();
}
void COptionsWindow::CopyGroupToList(const QString& Groupe, QTreeWidget* pTree, bool disabled)
{
for (int i = 0; i < ui.treeGroups->topLevelItemCount(); i++)
{
QTreeWidgetItem* pItem = ui.treeGroups->topLevelItem(i);
if (pItem->data(0, Qt::UserRole).toString().compare(Groupe, Qt::CaseInsensitive) == 0)
{
for (int j = 0; j < pItem->childCount(); j++)
{
QString Value = pItem->child(j)->data(0, Qt::UserRole).toString();
QTreeWidgetItem* pSubItem = new QTreeWidgetItem();
pSubItem->setCheckState(0, disabled ? Qt::Unchecked : Qt::Checked);
SetProgramItem(Value, pSubItem, 0);
pSubItem->setFlags(pSubItem->flags() | Qt::ItemIsEditable);
pTree->addTopLevelItem(pSubItem);
}
break;
}
}
}