FileProperty.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidAPI/FileProperty.h"
 
#include "MantidAPI/FileFinder.h"
#include "MantidKernel/ConfigService.h"
#include "MantidKernel/DirectoryValidator.h"
#include "MantidKernel/FacilityInfo.h"
#include "MantidKernel/PropertyWithValue.h"
#include "MantidKernel/Strings.h"
 
#include <Poco/File.h>
#include <Poco/Path.h>
 
#include <iterator>
#include <memory>
 
namespace Mantid::API {
 
using Mantid::Kernel::ConfigService;
using Mantid::Kernel::DirectoryValidator;
using Mantid::Kernel::FileValidator;
using Mantid::Kernel::IValidator_sptr;
 
namespace {
IValidator_sptr createValidator(unsigned int action, const std::vector<std::string> &exts) {
  if (action == FileProperty::Directory || action == FileProperty::OptionalDirectory) {
    return std::make_shared<DirectoryValidator>(action == FileProperty::Directory);
  } else {
    return std::make_shared<FileValidator>(exts, (action == FileProperty::Load));
  }
}
 
void addExtension(const std::string &extension, std::vector<std::string> &extensions) {
  if (std::find(extensions.begin(), extensions.end(), extension) != extensions.end())
    return;
  else
    extensions.emplace_back(extension);
}
 
const std::string &getHomePath() {
  static std::string homePath;
  static bool initialised(false);
 
  if (initialised) {
    return homePath;
  }
  initialised = true;
 
  char *home = std::getenv("HOME"); // Usually set on Windows and UNIX
  if (home) {
    homePath = std::string(home);
    return homePath;
  }
 
  char *userProfile = std::getenv("USERPROFILE"); // Not usually set on UNIX
  // Return even if it's an empty string, as we can do no better
  homePath = userProfile ? std::string(userProfile) : "";
  return homePath;
}
 
std::string expandUser(const std::string &filepath) {
  auto start = filepath.begin();
  auto end = filepath.end();
 
  // Filepath empty or contains no user variables
  if (start == end || *start != '~')
    return filepath;
 
  // Position of the first slash after the variable
  auto nextSlash = find_if(start, end, [](const char &c) { return c == '/' || c == '\\'; });
 
  // ~user/blah format - no support for this as yet
  if (std::distance(start, nextSlash) != 1)
    return filepath;
 
  return getHomePath() + std::string(nextSlash, end);
}
 
std::string createDirectory(const std::string &path) {
  Poco::Path stempath(path);
  if (stempath.isFile()) {
    stempath.makeParent();
  }
 
  if (!stempath.toString().empty()) {
    Poco::File stem(stempath);
    if (!stem.exists()) {
      try {
        stem.createDirectories();
      } catch (Poco::Exception &e) {
        std::stringstream msg;
        msg << "Failed to create directory \"" << stempath.toString() << "\": " << e.what();
        return msg.str();
      }
    }
  } else {
    return "Invalid directory.";
  }
  return ""; // everything went fine
}
} // Anonymous namespace
 
//-----------------------------------------------------------------
// Public member functions
//-----------------------------------------------------------------
FileProperty::FileProperty(const std::string &name, const std::string &defaultValue, unsigned int action,
                           const std::vector<std::string> &exts, unsigned int direction)
    : PropertyWithValue<std::string>(name, defaultValue, createValidator(action, exts), direction), m_action(action),
      m_defaultExt((!exts.empty()) ? exts.front() : ""), m_runFileProp(isLoadProperty() && extsMatchRunFiles()),
      m_oldLoadPropValue(""), m_oldLoadFoundFile("") {}
 
FileProperty::FileProperty(const std::string &name, const std::string &default_value, unsigned int action,
                           const std::string &ext, unsigned int direction)
    : FileProperty(name, default_value, action, std::vector<std::string>(1, ext), direction) {}
 
FileProperty::FileProperty(const std::string &name, const std::string &default_value, unsigned int action,
                           std::initializer_list<std::string> exts, unsigned int direction)
    : FileProperty(name, default_value, action, std::vector<std::string>(exts), direction) {}
 
bool FileProperty::isLoadProperty() const { return m_action == Load || m_action == OptionalLoad; }
 
bool FileProperty::isSaveProperty() const { return m_action == Save || m_action == OptionalSave; }
 
bool FileProperty::isDirectoryProperty() const { return m_action == Directory || m_action == OptionalDirectory; }
 
bool FileProperty::isOptional() const {
  return (m_action == OptionalLoad || m_action == OptionalSave || m_action == OptionalDirectory);
}
 
std::string FileProperty::setValue(const std::string &propValue) {
  std::string strippedValue = Kernel::Strings::strip(propValue);
 
  // Empty value is allowed if optional
  if (strippedValue.empty()) {
    PropertyWithValue<std::string>::setValue("");
    return isEmptyValueValid();
  }
 
  // Expand user variables, if there are any
  strippedValue = expandUser(strippedValue);
 
  // If this looks like an absolute path then don't do any searching but make
  // sure the
  // directory exists for a Save property
  if (Poco::Path(strippedValue).isAbsolute()) {
    std::string error;
    if (isSaveProperty()) {
      error = createDirectory(strippedValue);
      if (!error.empty())
        return error;
    }
 
    return PropertyWithValue<std::string>::setValue(strippedValue);
  }
 
  std::string errorMsg;
  // For relative paths, differentiate between load and save types
  if (isLoadProperty()) {
    errorMsg = setLoadProperty(strippedValue);
  } else {
    errorMsg = setSaveProperty(strippedValue);
  }
  return errorMsg;
}
 
std::string FileProperty::isValid() const {
  const std::string &value = (*this)();
  if (value.empty()) {
    return isEmptyValueValid();
  } else {
    return PropertyWithValue<std::string>::isValid();
  }
}
 
std::string FileProperty::isEmptyValueValid() const {
  if (isOptional()) {
    return "";
  } else {
    return "No file specified.";
  }
}
 
bool FileProperty::extsMatchRunFiles() {
  bool match(false);
  try {
    Kernel::FacilityInfo facilityInfo = Kernel::ConfigService::Instance().getFacility();
    const std::vector<std::string> facilityExts = facilityInfo.extensions();
    auto facilityExtsBegin = facilityExts.cbegin();
    auto facilityExtsEnd = facilityExts.cend();
    const std::vector<std::string> allowedExts = this->allowedValues();
 
    for (const auto &ext : allowedExts) {
      if (std::find(facilityExtsBegin, facilityExtsEnd, ext) != facilityExtsEnd) {
        match = true;
        break;
      }
    }
  } catch (Mantid::Kernel::Exception::NotFoundError &) {
  } // facility could not be found, do nothing this will return the default
  // match of false
 
  return match;
}
 
std::string FileProperty::setLoadProperty(const std::string &propValue) {
  // determine the initial version of foundFile
  std::string foundFile;
  if ((propValue == m_oldLoadPropValue) && (!m_oldLoadFoundFile.empty())) {
    foundFile = m_oldLoadFoundFile;
  }
 
  // cache the new version of propValue
  m_oldLoadPropValue = propValue;
 
  // if foundFile is not empty then it is the cached file
  if (foundFile.empty()) {
    if (m_runFileProp) // runfiles go through FileFinder::findRun
    {
      std::vector<std::string> allowedExts(allowedValues());
      std::vector<std::string> exts;
      if (!m_defaultExt.empty()) {
        addExtension(m_defaultExt, exts);
 
        std::string lower = Mantid::Kernel::Strings::toLower(m_defaultExt);
        addExtension(lower, exts);
 
        std::string upper = Mantid::Kernel::Strings::toUpper(m_defaultExt);
        addExtension(upper, exts);
      }
      for (auto &ext : allowedExts) {
        std::string lower(ext);
        std::string upper(ext);
        std::transform(ext.begin(), ext.end(), lower.begin(), tolower);
        std::transform(ext.begin(), ext.end(), upper.begin(), toupper);
        addExtension(ext, exts);
        addExtension(lower, exts);
        addExtension(upper, exts);
      }
      foundFile = FileFinder::Instance().findRun(propValue, exts);
    } else // non-runfiles go through FileFinder::getFullPath
    {
      foundFile = FileFinder::Instance().getFullPath(propValue);
    }
  }
 
  // cache the new version of foundFile
  m_oldLoadFoundFile = foundFile;
 
  if (foundFile.empty()) {
    return PropertyWithValue<std::string>::setValue(propValue);
  } else {
    return PropertyWithValue<std::string>::setValue(foundFile);
  }
}
 
std::string FileProperty::setSaveProperty(const std::string &propValue) {
  if (propValue.empty()) {
    if (m_action == OptionalSave) {
      return PropertyWithValue<std::string>::setValue("");
    } else
      return "Empty filename not allowed.";
  }
  std::string errorMsg;
  // We have a relative save path so just prepend the path that is in the
  // 'defaultsave.directory'
  // Note that this catches the Poco::NotFoundException and returns an empty
  // string in that case
  std::string save_path = ConfigService::Instance().getString("defaultsave.directory");
  Poco::Path save_dir;
  if (save_path.empty()) {
    save_dir = Poco::Path(propValue).parent();
    // If we only have a stem filename, parent() will make save_dir empty and
    // then Poco::File throws
    if (save_dir.toString().empty()) {
      save_dir = Poco::Path::current();
    }
  } else {
    save_dir = Poco::Path(save_path).makeDirectory();
  }
  errorMsg = createDirectory(save_dir.toString());
  if (errorMsg.empty()) {
    std::string fullpath = save_dir.resolve(propValue).toString();
    errorMsg = PropertyWithValue<std::string>::setValue(fullpath);
  }
  return errorMsg;
}
} // namespace Mantid::API