Play-/Source/ui_qt/S3FileBrowser.cpp
Jean-Philip Desjardins a15ce14193 Update S3 client.
2021-04-23 17:39:19 -04:00

108 lines
3.1 KiB
C++

#include "S3FileBrowser.h"
#include "ui_s3filebrowser.h"
#include "amazon/AmazonS3Client.h"
#include "../s3stream/S3ObjectStream.h"
#include "string_format.h"
#include "AppConfig.h"
#include "ui_shared/AmazonS3Utils.h"
#define PREF_S3FILEBROWSER_BUCKETNAME "s3.filebrowser.bucketname"
S3FileBrowser::S3FileBrowser(QWidget* parent)
: QDialog(parent)
, ui(new Ui::S3FileBrowser())
{
CAppConfig::GetInstance().RegisterPreferenceString(PREF_S3FILEBROWSER_BUCKETNAME, "");
ui->setupUi(this);
ui->bucketNameEdit->setText(CAppConfig::GetInstance().GetPreferenceString(PREF_S3FILEBROWSER_BUCKETNAME));
m_continuationChecker = new CContinuationChecker(this);
m_filterTimer = new QTimer(this);
m_filterTimer->setSingleShot(true);
connect(m_filterTimer, SIGNAL(timeout()), this, SLOT(updateFilter()));
launchUpdate();
}
S3FileBrowser::~S3FileBrowser()
{
delete ui;
}
bool S3FileBrowser::IsAvailable()
{
return CS3ObjectStream::CConfig::GetInstance().GetCredentials().IsValid();
}
fs::path S3FileBrowser::GetSelectedPath() const
{
return m_selectedPath;
}
void S3FileBrowser::refreshButton_clicked()
{
launchUpdate();
}
void S3FileBrowser::objectList_itemSelectionChanged()
{
updateOkButtonState();
}
void S3FileBrowser::searchFilterEdit_textChanged(QString)
{
m_filterTimer->start(100);
}
void S3FileBrowser::updateFilter()
{
ui->objectList->clear();
auto filter = ui->searchFilterEdit->text().toStdString();
for(const auto& item : m_bucketItems.objects)
{
if(!filter.empty() && (item.key.find(filter, 0) == std::string::npos)) continue;
ui->objectList->addItem(QString::fromStdString(item.key));
}
}
void S3FileBrowser::accept()
{
auto bucketName = m_lastUpdateBucketName.toStdString();
auto selectedItems = ui->objectList->selectedItems();
if(selectedItems.size() != 0)
{
auto selectedItem = selectedItems[0];
auto objectName = selectedItem->text().toStdString();
m_selectedPath = string_format("//s3/%s/%s", bucketName.c_str(), objectName.c_str());
}
CAppConfig::GetInstance().SetPreferenceString(PREF_S3FILEBROWSER_BUCKETNAME, bucketName.c_str());
CAppConfig::GetInstance().Save();
QDialog::accept();
}
void S3FileBrowser::updateOkButtonState()
{
bool hasSelection = ui->objectList->selectedItems().size() != 0;
ui->buttonBox->button(QDialogButtonBox::Ok)->setEnabled(hasSelection);
}
void S3FileBrowser::launchUpdate()
{
setEnabled(false);
m_lastUpdateBucketName = ui->bucketNameEdit->text();
ui->objectList->clear();
updateOkButtonState();
auto getListFuture = std::async([bucketName = m_lastUpdateBucketName.toStdString()]() {
auto credentials = CS3ObjectStream::CConfig::GetInstance().GetCredentials();
return AmazonS3Utils::GetListObjects(credentials, bucketName);
});
m_continuationChecker->GetContinuationManager().Register(std::move(getListFuture),
[this](auto& result) {
m_bucketItems = result;
this->updateFilter();
this->setEnabled(true);
});
}