import_export_helpers.cpp

#include <torch/csrc/jit/import_export_helpers.h>
#include <c10/util/Exception.h>

#include <algorithm>

namespace torch {
namespace jit {
namespace ImportExportHelpers {

static const std::string exportPrefix = "code/";
static const std::string LEGACY_exportPrefix = "libs/";
static const std::string kExportSuffix = "py";

std::string qualifierToPath(
    const std::string& qualifier,
    size_t proto_version) {
  const auto kExportPrefix =
      proto_version >= 6 ? exportPrefix : LEGACY_exportPrefix;
  std::string path = qualifier;
  std::replace_if(
      path.begin(), path.end(), [](char c) { return c == '.'; }, '/');
  return kExportPrefix + path + "." + kExportSuffix;
}

std::string pathToQualifier(
    const std::string& classPath,
    size_t proto_version) {
  const auto kExportPrefix =
      proto_version >= 6 ? exportPrefix : LEGACY_exportPrefix;
  // strip input suffix
  const auto end = classPath.rfind(kExportSuffix);
  AT_ASSERT(end != std::string::npos);

  // strip input suffix
  size_t libs_idx = classPath.find(kExportPrefix);
  AT_ASSERT(libs_idx == 0);

  AT_ASSERT(classPath.size() > kExportPrefix.size());
  const auto start = kExportPrefix.size();

  std::string class_qualifier = classPath.substr(start, end - start);
  std::replace_if(
      class_qualifier.begin(),
      class_qualifier.end(),
      [](char c) { return c == '/'; },
      '.');

  return class_qualifier;
}
} // namespace ImportExportHelpers
} // namespace jit
} // namespace torch