FileModel.cpp

Go to the documentation of this file.

// DUNE DAQ modification notice:
// This file has been modified from the original ATLAS dbe source for the DUNE DAQ project.
// Fork baseline commit: dbe-02-12-17 (2022-05-12).
// Renamed since fork: no.

#include "dbe/StyleUtility.hpp"
#include "dbe/confaccessor.hpp"
#include "dbe/ui_constants.hpp"
 
#include <QDir>
#include <QFileInfo>
#include "dbe/FileModel.hpp"
#include "dbe/MainWindow.hpp"
#include "dbe/config_api_get.hpp"
 
dbe::FileModel::FileModel(QObject* parent)
  : QAbstractTableModel(parent)
  , Headers{ tr("Name"), tr("Folder"), tr("Access"), tr("Status") }
{
  initpaths();
  initconnections();
  initmodel();
}
 
dbe::FileModel::FileModel(QList<QStringList> const& FileList, QObject* parent)
  : QAbstractTableModel(parent)
  , IncludedFiles(FileList)
  , Headers{ tr("Name"), tr("Folder"), tr("Access"), tr("Status") }
{
  initconnections();
}
 
dbe::FileModel::~FileModel() {}
 
int
dbe::FileModel::rowCount(const QModelIndex& parent) const
{
  if (!parent.isValid()) {
    return IncludedFiles.size();
  }
 
  return 0;
}
 
int
dbe::FileModel::columnCount(const QModelIndex& parent) const
{
  Q_UNUSED(parent)
  return Headers.size();
}
 
QVariant
dbe::FileModel::data(const QModelIndex& index, int role) const
{
  if (index.isValid()) {
    if (role == Qt::DisplayRole) {
      return QVariant(IncludedFiles.at(index.row()).at(index.column()));
    }
    if (role == Qt::ForegroundRole) {
      if (IncludedFiles.at(index.row()).at(static_cast<int>(tablepositions::filepermission)) == "RO") {
        return QBrush(StyleUtility::FileReadOnlyForeground);
      }
    }
    if (role == Qt::BackgroundRole) {
      if (IncludedFiles.at(index.row()).at(static_cast<int>(tablepositions::filepermission)) == "RO") {
        return QBrush(StyleUtility::FileReadOnlyBackground);
      }
    }
  }
 
  return QVariant();
}
 
QVariant
dbe::FileModel::headerData(int section, Qt::Orientation orientation, int role) const
{
  if (role == Qt::DisplayRole) {
    if (orientation == Qt::Horizontal) {
      return Headers.at(section);
    }
 
    if (orientation == Qt::Vertical) {
      return section + 1;
    }
  }
 
  return QVariant();
}
 
Qt::ItemFlags
dbe::FileModel::flags(const QModelIndex& /*index*/) const
{
  // if ( IncludedFiles.at ( index.row() ).at ( static_cast<int>
  //                                            ( tablepositions::filepermission ) ) == "RW" )
  // {
  return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
  // }
 
  // return Qt::ItemIsSelectable;
}
 
void
dbe::FileModel::initpaths()
{
  QString TDAQ_DB_REPOSITORY = getenv("TDAQ_DB_REPOSITORY");
  if (TDAQ_DB_REPOSITORY.isEmpty() == false) {
    QString TDAQ_DB_USER_REPOSITORY = getenv("TDAQ_DB_USER_REPOSITORY");
    FolderPathList = TDAQ_DB_USER_REPOSITORY.split(":", Qt::SkipEmptyParts);
 
    FolderPathList << dbe::MainWindow::findthis()->find_db_repository_dir();
  } else {
    QString DUNEDAQ_DB_PATH = getenv("DUNEDAQ_DB_PATH");
    FolderPathList = DUNEDAQ_DB_PATH.split(":", Qt::SkipEmptyParts);
  }
 
  for (QString& PathName : FolderPathList) {
    if (!PathName.endsWith("/")) {
      PathName.append("/");
    }
  }
}
 
void
dbe::FileModel::initconnections()
{
  connect(
    this, SIGNAL(FileCacheReady(QList<QStringList>&)), &confaccessor::ref(), SLOT(GetFileCache(QList<QStringList>&)));
}
 
QString
dbe::FileModel::GetFullFileName(QString& FileName)
{
  QFileInfo FileInfo(FileName);
 
  if (FileInfo.isRelative()) {
    bool found = false;
    for (const QString& Folder : FolderPathList) {
      QDir TdaqFolder(Folder);
 
      if (TdaqFolder.exists(FileName)) {
        FileName = TdaqFolder.path() + "/" + FileName;
        char* rpath = realpath(FileName.toStdString().c_str(), NULL);
        FileName = QString(rpath);
        free(rpath);
        found = true;
        break;
      }
    }
    if (!found) {
      FileName = FileInfo.absoluteFilePath();
    }
  }
 
  return FileName;
}
 
QList<QStringList>
dbe::FileModel::GetFilesInfo() const
{
  return IncludedFiles;
}
 
void
dbe::FileModel::initmodel()
{
  QStringList sources = dbe::config::api::get::file::inclusions({ confaccessor::dbfullname() });
 
  std::list<std::string> updated = confaccessor::uncommitted_files();
  for (QString const& source : sources) {
    QFileInfo srcinfo(source);
    QString fn = srcinfo.filePath();
    QString dn = GetFullFileName(fn).remove(QRegularExpression("/[a-zA-Z0-9_-]*.data.xml"));
 
    QString modified{};
    auto file = GetFullFileName(fn).toStdString();
    for (auto ufile : updated) {
      if (file == ufile) {
        modified = "Modified";
        break;
      }
    }
 
    IncludedFiles.append(
      QStringList{ srcinfo.fileName(), dn, confaccessor::check_file_rw(source) ? "RW" : "RO", modified });
  }
 
  if (IncludedFiles.size()) {
    emit FileCacheReady(IncludedFiles);
  }
}