openpilot/frogpilot/ui/qt/offroad/data_settings.cc
2025-11-01 12:00:00 -07:00

598 lines
23 KiB
C++

#include <sys/xattr.h>
#include "frogpilot/ui/qt/offroad/data_settings.h"
FrogPilotDataPanel::FrogPilotDataPanel(FrogPilotSettingsWindow *parent) : FrogPilotListWidget(parent), parent(parent) {
QJsonObject shownDescriptions = QJsonDocument::fromJson(QString::fromStdString(params.get("ShownToggleDescriptions")).toUtf8()).object();
QString className = this->metaObject()->className();
bool forceOpenDescriptions = false;
if (!shownDescriptions.value(className).toBool(false)) {
forceOpenDescriptions = true;
shownDescriptions.insert(className, true);
params.put("ShownToggleDescriptions", QJsonDocument(shownDescriptions).toJson(QJsonDocument::Compact).toStdString());
}
QStackedLayout *dataLayout = new QStackedLayout();
addItem(dataLayout);
FrogPilotListWidget *dataMainList = new FrogPilotListWidget(this);
ScrollView *dataMainPanel = new ScrollView(dataMainList, this);
dataLayout->addWidget(dataMainPanel);
ButtonControl *deleteDrivingDataButton = new ButtonControl(tr("Delete Driving Data"), tr("DELETE"), tr("<b>Delete all stored driving footage and data</b> to free up space and clear private information."));
QObject::connect(deleteDrivingDataButton, &ButtonControl::clicked, [=]() {
QDir hdDataDir("/data/media/0/realdata_HD/");
QDir konikDataDir("/data/media/0/realdata_konik/");
QDir realDataDir("/data/media/0/realdata/");
if (ConfirmationDialog::confirm(tr("Delete all driving data and footage?"), tr("Delete"), this)) {
std::thread([=]() mutable {
parent->keepScreenOn = true;
deleteDrivingDataButton->setEnabled(false);
deleteDrivingDataButton->setValue(tr("Deleting..."));
QList<QDir> footageDirs = {hdDataDir, konikDataDir, realDataDir};
for (const QDir &footageDir : footageDirs) {
if (!footageDir.exists()) {
continue;
}
QFileInfoList entries = footageDir.entryInfoList(QDir::Dirs | QDir::NoDotAndDotDot);
for (const QFileInfo &entry : entries) {
char value[10] = {0};
if (!(getxattr(entry.absoluteFilePath().toUtf8().constData(), "user.preserve", value, sizeof(value)) > 0 && strcmp(value, "1") == 0)) {
QDir(entry.absoluteFilePath()).removeRecursively();
}
}
}
deleteDrivingDataButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
deleteDrivingDataButton->setEnabled(true);
deleteDrivingDataButton->setValue("");
parent->keepScreenOn = false;
}).detach();
}
});
if (forceOpenDescriptions) {
deleteDrivingDataButton->showDescription();
}
dataMainList->addItem(deleteDrivingDataButton);
ButtonControl *deleteErrorLogsButton = new ButtonControl(tr("Delete Error Logs"), tr("DELETE"), tr("<b>Delete collected error logs</b> to free up space and clear old crash records."));
QObject::connect(deleteErrorLogsButton, &ButtonControl::clicked, [=]() {
QDir errorLogsDir("/data/error_logs");
if (ConfirmationDialog::confirm(tr("Delete all error logs?"), tr("Delete"), this)) {
std::thread([=]() mutable {
parent->keepScreenOn = true;
deleteErrorLogsButton->setEnabled(false);
deleteErrorLogsButton->setValue(tr("Deleting..."));
errorLogsDir.removeRecursively();
errorLogsDir.mkpath(".");
deleteErrorLogsButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
deleteErrorLogsButton->setEnabled(true);
deleteErrorLogsButton->setValue("");
parent->keepScreenOn = false;
}).detach();
}
});
if (forceOpenDescriptions) {
deleteErrorLogsButton->showDescription();
}
dataMainList->addItem(deleteErrorLogsButton);
FrogPilotButtonsControl *screenRecordingsButton = new FrogPilotButtonsControl(tr("Screen Recordings"), tr("<b>Delete or rename screen recordings.</b>"), "", {tr("DELETE"), tr("DELETE ALL"), tr("RENAME")});
QObject::connect(screenRecordingsButton, &FrogPilotButtonsControl::buttonClicked, [=](int id) {
QDir recordingsDir("/data/media/screen_recordings");
QStringList recordingsNames = recordingsDir.entryList(QDir::Files | QDir::NoDotAndDotDot);
QStringList mp4Recordings;
for (const QString &name : recordingsNames) {
if (name.endsWith(".mp4", Qt::CaseInsensitive)) {
mp4Recordings << name;
}
}
if (id == 0) {
QString selection = MultiOptionDialog::getSelection(tr("Choose a screen recording to delete"), mp4Recordings, "", this);
if (!selection.isEmpty()) {
if (ConfirmationDialog::confirm(tr("Delete this screen recording?"), tr("Delete"), this)) {
std::thread([=]() {
parent->keepScreenOn = true;
screenRecordingsButton->setEnabled(false);
screenRecordingsButton->setValue(tr("Deleting..."));
screenRecordingsButton->setVisibleButton(1, false);
screenRecordingsButton->setVisibleButton(2, false);
QFile::remove(recordingsDir.absoluteFilePath(selection));
screenRecordingsButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
screenRecordingsButton->setEnabled(true);
screenRecordingsButton->setValue("");
screenRecordingsButton->setVisibleButton(1, true);
screenRecordingsButton->setVisibleButton(2, true);
parent->keepScreenOn = false;
}).detach();
}
}
} else if (id == 1) {
if (ConfirmationDialog::confirm(tr("Delete all screen recordings?"), tr("Delete All"), this)) {
std::thread([=]() mutable {
parent->keepScreenOn = true;
screenRecordingsButton->setEnabled(false);
screenRecordingsButton->setValue(tr("Deleting..."));
screenRecordingsButton->setVisibleButton(0, false);
screenRecordingsButton->setVisibleButton(2, false);
recordingsDir.removeRecursively();
recordingsDir.mkpath(".");
screenRecordingsButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
screenRecordingsButton->setEnabled(true);
screenRecordingsButton->setValue("");
screenRecordingsButton->setVisibleButton(0, true);
screenRecordingsButton->setVisibleButton(2, true);
parent->keepScreenOn = false;
}).detach();
}
} else if (id == 2) {
QString selection = MultiOptionDialog::getSelection(tr("Choose a screen recording to rename"), mp4Recordings, "", this);
if (!selection.isEmpty()) {
QString newBase = InputDialog::getText(tr("Enter a new name"), this, tr("Rename Screen Recording")).trimmed().replace(" ", "_");
if (!newBase.isEmpty()) {
QString newName = newBase + ".mp4";
if (recordingsNames.contains(newName)) {
ConfirmationDialog::alert(tr("Name already in use. Please choose a different name."), this);
return;
}
std::thread([=]() {
parent->keepScreenOn = true;
screenRecordingsButton->setEnabled(false);
screenRecordingsButton->setValue(tr("Renaming..."));
screenRecordingsButton->setVisibleButton(0, false);
screenRecordingsButton->setVisibleButton(1, false);
QString newPath = recordingsDir.absoluteFilePath(newName);
QString oldPath = recordingsDir.absoluteFilePath(selection);
QFile::rename(oldPath, newPath);
screenRecordingsButton->setValue(tr("Renamed!"));
util::sleep_for(2500);
screenRecordingsButton->setEnabled(true);
screenRecordingsButton->setValue("");
screenRecordingsButton->setVisibleButton(0, true);
screenRecordingsButton->setVisibleButton(1, true);
parent->keepScreenOn = false;
}).detach();
}
}
}
});
if (forceOpenDescriptions) {
screenRecordingsButton->showDescription();
}
dataMainList->addItem(screenRecordingsButton);
FrogPilotButtonsControl *frogpilotBackupButton = new FrogPilotButtonsControl(tr("FrogPilot Backups"), tr("<b>Create, delete, or restore FrogPilot backups.</b>"), "", {tr("BACKUP"), tr("DELETE"), tr("DELETE ALL"), tr("RESTORE")});
QObject::connect(frogpilotBackupButton, &FrogPilotButtonsControl::buttonClicked, [=](int id) {
QDir backupDir("/data/backups");
QStringList backupNames = backupDir.entryList(QDir::Dirs | QDir::Files | QDir::NoDotAndDotDot, QDir::Name).filter(QRegularExpression("^(?!.*_in_progress(?:\\..*)?$).*$"));
QRegularExpression autoRegex("^(.*)_(\\d{4}-\\d{2}-\\d{2})_auto(?:\\..*)?$");
QMap<QString, QString> backupFriendlyMap;
for (const QString &name : backupNames) {
QString friendly = name;
QRegularExpressionMatch match = autoRegex.match(name);
if (match.hasMatch()) {
friendly = match.captured(1) + ": " + match.captured(2);
}
backupFriendlyMap.insert(friendly, name);
}
if (id == 0) {
QString nameSelection = InputDialog::getText(tr("Enter a name for this backup"), this, "", false, 1).trimmed().replace(" ", "_");
if (!nameSelection.isEmpty()) {
if (backupNames.contains(nameSelection)) {
ConfirmationDialog::alert(tr("Name already in use. Please choose a different name."), this);
return;
}
bool compressed = FrogPilotConfirmationDialog::yesorno(tr("Compress this backup? This will save space and run in the background but take a bit longer."), this);
std::thread([=]() {
parent->keepScreenOn = true;
frogpilotBackupButton->setEnabled(false);
frogpilotBackupButton->setValue(tr("Backing up..."));
frogpilotBackupButton->setVisibleButton(1, false);
frogpilotBackupButton->setVisibleButton(2, false);
frogpilotBackupButton->setVisibleButton(3, false);
QString fullBackupPath = backupDir.filePath(nameSelection);
QString inProgressBackupPath = fullBackupPath + "_in_progress";
QDir().mkpath(inProgressBackupPath);
std::system(qPrintable("rsync -av /data/openpilot/ " + inProgressBackupPath + "/"));
if (compressed) {
frogpilotBackupButton->setValue(tr("Compressing..."));
std::system(qPrintable("tar -cf - -C " + inProgressBackupPath + " . | zstd -2 -T0 -o " + fullBackupPath + "_in_progress.tar.zst"));
QDir(inProgressBackupPath).removeRecursively();
QString oldTar = fullBackupPath + "_in_progress.tar.zst";
QString newTar = fullBackupPath + ".tar.zst";
QFile::rename(oldTar, newTar);
} else {
QDir().rename(inProgressBackupPath, fullBackupPath);
}
frogpilotBackupButton->setValue(tr("Backup created!"));
util::sleep_for(2500);
frogpilotBackupButton->setEnabled(true);
frogpilotBackupButton->setValue("");
frogpilotBackupButton->setVisibleButton(1, true);
frogpilotBackupButton->setVisibleButton(2, true);
frogpilotBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
} else if (id == 1) {
QString selectionFriendly = MultiOptionDialog::getSelection(tr("Choose a FrogPilot backup to delete"), backupFriendlyMap.keys(), "", this);
if (!selectionFriendly.isEmpty()) {
QString selection = backupFriendlyMap.value(selectionFriendly);
if (ConfirmationDialog::confirm(tr("Delete this backup?"), tr("Delete"), this)) {
std::thread([=]() {
parent->keepScreenOn = true;
frogpilotBackupButton->setEnabled(false);
frogpilotBackupButton->setValue(tr("Deleting..."));
frogpilotBackupButton->setVisibleButton(0, false);
frogpilotBackupButton->setVisibleButton(2, false);
frogpilotBackupButton->setVisibleButton(3, false);
if (selection.endsWith(".tar.gz") || selection.endsWith(".tar.zst")) {
QFile::remove(backupDir.filePath(selection));
} else {
QDir(backupDir.filePath(selection)).removeRecursively();
}
frogpilotBackupButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
frogpilotBackupButton->setEnabled(true);
frogpilotBackupButton->setValue("");
frogpilotBackupButton->setVisibleButton(0, true);
frogpilotBackupButton->setVisibleButton(2, true);
frogpilotBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
}
} else if (id == 2) {
if (ConfirmationDialog::confirm(tr("Delete all backups?"), tr("Delete All"), this)) {
std::thread([=]() mutable {
parent->keepScreenOn = true;
frogpilotBackupButton->setEnabled(false);
frogpilotBackupButton->setValue(tr("Deleting..."));
frogpilotBackupButton->setVisibleButton(0, false);
frogpilotBackupButton->setVisibleButton(1, false);
frogpilotBackupButton->setVisibleButton(3, false);
backupDir.removeRecursively();
backupDir.mkpath(".");
frogpilotBackupButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
frogpilotBackupButton->setEnabled(true);
frogpilotBackupButton->setValue("");
frogpilotBackupButton->setVisibleButton(0, true);
frogpilotBackupButton->setVisibleButton(1, true);
frogpilotBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
} else if (id == 3) {
QString selectionFriendly = MultiOptionDialog::getSelection(tr("Choose a backup to restore"), backupFriendlyMap.keys(), "", this);
if (!selectionFriendly.isEmpty()) {
QString selection = backupFriendlyMap.value(selectionFriendly);
if (ConfirmationDialog::confirm(tr("Restore this backup?"), tr("Restore"), this)) {
std::thread([=]() {
parent->keepScreenOn = true;
frogpilotBackupButton->setEnabled(false);
frogpilotBackupButton->setValue(tr("Restoring..."));
frogpilotBackupButton->setVisibleButton(0, false);
frogpilotBackupButton->setVisibleButton(1, false);
frogpilotBackupButton->setVisibleButton(2, false);
QString extractDirectory = "/data/restore_temp";
QString sourcePath = backupDir.filePath(selection);
QString targetPath = "/data/safe_staging/finalized";
QDir().mkpath(extractDirectory);
if (selection.endsWith(".tar.gz")) {
frogpilotBackupButton->setValue(tr("Extracting..."));
std::system(qPrintable("tar --strip-components=1 -xzf " + sourcePath + " -C " + extractDirectory));
} else if (selection.endsWith(".tar.zst")) {
frogpilotBackupButton->setValue(tr("Extracting..."));
std::system(qPrintable("zstd -d " + sourcePath + " -o " + extractDirectory + "/backup.tar"));
std::system(qPrintable("tar --strip-components=1 -xf " + extractDirectory + "/backup.tar -C " + extractDirectory));
QFile::remove(extractDirectory + "/backup.tar");
} else {
std::system(qPrintable("rsync -av " + sourcePath + "/ " + extractDirectory + "/"));
}
QDir().mkpath(targetPath);
std::system(qPrintable("rsync -av --delete -l " + extractDirectory + "/ " + targetPath + "/"));
QFile overlayFile(targetPath + "/.overlay_consistent");
overlayFile.open(QIODevice::WriteOnly);
overlayFile.close();
if (QFileInfo::exists(extractDirectory)) {
QDir(extractDirectory).removeRecursively();
}
QFile("/cache/on_backup").open(QIODevice::WriteOnly);
frogpilotBackupButton->setValue(tr("Restored!"));
util::sleep_for(2500);
frogpilotBackupButton->setValue(tr("Rebooting..."));
util::sleep_for(2500);
Hardware::reboot();
}).detach();
}
}
}
});
if (forceOpenDescriptions) {
frogpilotBackupButton->showDescription();
}
dataMainList->addItem(frogpilotBackupButton);
FrogPilotButtonsControl *toggleBackupButton = new FrogPilotButtonsControl(tr("Toggle Backups"), tr("<b>Create, delete, or restore toggle backups.</b>"), "", {tr("BACKUP"), tr("DELETE"), tr("DELETE ALL"), tr("RESTORE")});
QObject::connect(toggleBackupButton, &FrogPilotButtonsControl::buttonClicked, [=](int id) {
QDir backupDir("/data/toggle_backups");
QStringList backupNames = backupDir.entryList(QDir::Dirs | QDir::Files | QDir::NoDotAndDotDot, QDir::Name).filter(QRegularExpression("^(?!.*_in_progress$).*$"));
QRegularExpression autoRegex("^(\\d{4}-\\d{2}-\\d{2})_(\\d{2}-\\d{2}[APMapm]{2})_auto(?:\\..*)?$");
QMap<QString, QString> backupFriendlyMap;
for (const QString &name : backupNames) {
QRegularExpressionMatch match = autoRegex.match(name);
QString friendly = name;
if (match.hasMatch()) {
QString datePart = match.captured(1);
QString timePart = match.captured(2);
timePart.replace("-", ":");
if (timePart.endsWith("pm", Qt::CaseInsensitive)) {
timePart = timePart.left(timePart.size() - 2) + " PM";
} else if (timePart.endsWith("am", Qt::CaseInsensitive)) {
timePart = timePart.left(timePart.size() - 2) + " AM";
}
friendly = datePart + " - " + timePart;
}
backupFriendlyMap.insert(friendly, name);
}
if (id == 0) {
QString nameSelection = InputDialog::getText(tr("Enter a name for this backup"), this, "", false, 1).trimmed().replace(" ", "_");
if (!nameSelection.isEmpty()) {
if (backupNames.contains(nameSelection)) {
ConfirmationDialog::alert(tr("Name already in use. Please choose a different name."), this);
return;
}
std::thread([=]() {
parent->keepScreenOn = true;
toggleBackupButton->setEnabled(false);
toggleBackupButton->setValue(tr("Backing up..."));
toggleBackupButton->setVisibleButton(1, false);
toggleBackupButton->setVisibleButton(2, false);
toggleBackupButton->setVisibleButton(3, false);
QString fullBackupPath = backupDir.filePath(nameSelection);
QString inProgressBackupPath = fullBackupPath + "_in_progress";
QDir().mkpath(inProgressBackupPath);
std::system(qPrintable("rsync -av /data/params/d/ " + inProgressBackupPath + "/"));
QDir().rename(inProgressBackupPath, fullBackupPath);
toggleBackupButton->setValue(tr("Backup created!"));
util::sleep_for(2500);
toggleBackupButton->setEnabled(true);
toggleBackupButton->setValue("");
toggleBackupButton->setVisibleButton(1, true);
toggleBackupButton->setVisibleButton(2, true);
toggleBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
} else if (id == 1) {
QString selectionFriendly = MultiOptionDialog::getSelection(tr("Choose a backup to delete"), backupFriendlyMap.keys(), "", this);
if (!selectionFriendly.isEmpty()) {
QString selection = backupFriendlyMap.value(selectionFriendly);
if (ConfirmationDialog::confirm(tr("Delete this backup?"), tr("Delete"), this)) {
std::thread([=]() {
parent->keepScreenOn = true;
toggleBackupButton->setEnabled(false);
toggleBackupButton->setValue(tr("Deleting..."));
toggleBackupButton->setVisibleButton(0, false);
toggleBackupButton->setVisibleButton(2, false);
toggleBackupButton->setVisibleButton(3, false);
QDir dirToDelete(backupDir.filePath(selection));
dirToDelete.removeRecursively();
toggleBackupButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
toggleBackupButton->setEnabled(true);
toggleBackupButton->setValue("");
toggleBackupButton->setVisibleButton(0, true);
toggleBackupButton->setVisibleButton(2, true);
toggleBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
}
} else if (id == 2) {
if (ConfirmationDialog::confirm(tr("Delete all backups?"), tr("Delete All"), this)) {
std::thread([=]() mutable {
parent->keepScreenOn = true;
toggleBackupButton->setEnabled(false);
toggleBackupButton->setValue(tr("Deleting..."));
toggleBackupButton->setVisibleButton(0, false);
toggleBackupButton->setVisibleButton(1, false);
toggleBackupButton->setVisibleButton(3, false);
backupDir.removeRecursively();
backupDir.mkpath(".");
toggleBackupButton->setValue(tr("Deleted!"));
util::sleep_for(2500);
toggleBackupButton->setEnabled(true);
toggleBackupButton->setValue("");
toggleBackupButton->setVisibleButton(0, true);
toggleBackupButton->setVisibleButton(1, true);
toggleBackupButton->setVisibleButton(3, true);
parent->keepScreenOn = false;
}).detach();
}
} else if (id == 3) {
QString selectionFriendly = MultiOptionDialog::getSelection(tr("Choose a backup to restore"), backupFriendlyMap.keys(), "", this);
if (!selectionFriendly.isEmpty()) {
QString selection = backupFriendlyMap.value(selectionFriendly);
if (ConfirmationDialog::confirm(tr("Restore this backup?"), tr("Restore"), this)) {
std::thread([=]() {
parent->keepScreenOn = true;
toggleBackupButton->setEnabled(false);
toggleBackupButton->setValue(tr("Restoring..."));
toggleBackupButton->setVisibleButton(0, false);
toggleBackupButton->setVisibleButton(1, false);
toggleBackupButton->setVisibleButton(2, false);
QString sourcePath = backupDir.filePath(selection);
QString targetPath = "/data/params/d";
QDir().mkpath(targetPath);
std::system(qPrintable("rsync -av -l " + sourcePath + "/ " + targetPath + "/"));
updateFrogPilotToggles();
toggleBackupButton->setValue(tr("Restored!"));
util::sleep_for(2500);
toggleBackupButton->setEnabled(true);
toggleBackupButton->setValue("");
toggleBackupButton->setVisibleButton(0, true);
toggleBackupButton->setVisibleButton(1, true);
toggleBackupButton->setVisibleButton(2, true);
parent->keepScreenOn = false;
}).detach();
}
}
}
});
if (forceOpenDescriptions) {
toggleBackupButton->showDescription();
}
dataMainList->addItem(toggleBackupButton);
}