oks_check_schema.cxx

Go to the documentation of this file.

 
#include "CLI/CLI.hpp"
#include "logging/Logging.hpp"
#include "oks/class.hpp"
#include "oks/file.hpp"
#include "oks/kernel.hpp"
#include "oks/relationship.hpp"
 
#include <fmt/core.h>
#include <set>
#include <string>
 
using namespace dunedaq::oks;
 
int
main(int argc, char** argv)
{
  enum Exitcode : int
  {
    SUCCESS,
    BADCMD,
    NOFILE,
    LOADFAIL,
    BADRELATIONSHIP,
    UNRESOLVED,
    MISSING_INCLUDE
  };
 
  CLI::App app{
    "Check consistency of OKS schema file\n" +
    fmt::format("Return codes: {} Success, file is OK\n", Exitcode::SUCCESS) +
    fmt::format("              {} Bad command line\n", Exitcode::BADCMD) +
    fmt::format("              {} Failed to find file\n", Exitcode::NOFILE) +
    fmt::format("              {} Failed to load file -- invalid schema\n", Exitcode::LOADFAIL) +
    fmt::format("              {} File contains relationship to non loaded class\n", Exitcode::BADRELATIONSHIP) +
    fmt::format("              {} File contains object with relationship to non loaded class/object\n",
                Exitcode::UNRESOLVED) +
    fmt::format("              {} File refers to class in file not directly included\n", Exitcode::MISSING_INCLUDE)
  };
 
  std::string filename;
  app.add_option("-f,--file", filename, "Schema file")->required();
 
  CLI11_PARSE(app, argc, argv);
 
  OksKernel kernel;
  OksFile* file{ nullptr };
  try {
    file = kernel.load_file(filename);
  } catch (FailedLoadFile& fail) {
    TLOG() << fail.what() << "\n";
    return Exitcode::LOADFAIL;
  } catch (CanNotOpenFile& fail) {
    TLOG() << fail.what() << "\n";
    return Exitcode::NOFILE;
  } catch (std::exception& exc) {
    TLOG() << exc.what() << "\n";
    return Exitcode::LOADFAIL;
  }
 
  if (file == nullptr) {
    TLOG() << "Failed to load " << filename;
    return Exitcode::LOADFAIL;
  }
 
  std::set<std::string> includes;
  for (const auto& ifile : file->get_include_files()) {
    includes.insert(ifile);
    TLOG_DEBUG(2) << "direct inserting " << ifile << "\n";
  }
 
  Exitcode result = Exitcode::SUCCESS;
 
  if (filename.ends_with(".schema.xml")) {
    // Check classes defined in this schema file
    for (auto [name, oks_class] : kernel.classes()) {
 
      if (file->get_full_file_name() != oks_class->get_file()->get_full_file_name()) {
        // Class not defined in this file
        continue;
      }
 
      auto cfile = oks_class->get_file()->get_short_file_name();
      if (cfile != file->get_short_file_name() && !includes.contains(cfile)) {
        TLOG() << "Error " << filename << " misses include of " << cfile << " required by " << name;
        if (result == Exitcode::SUCCESS) {
          result = Exitcode::MISSING_INCLUDE;
        }
      }
      auto relationships = oks_class->direct_relationships();
      if (relationships != nullptr) {
        for (auto rel : *relationships) {
          auto rel_class = rel->get_class_type();
          if (rel_class == nullptr) {
            TLOG() << "Error class '" << name << "' has relationship '" << rel->get_name() << "' to a class '"
                   << rel->get_type() << "' that is not loaded\n";
            if (result == Exitcode::SUCCESS) {
              result = Exitcode::BADRELATIONSHIP;
            }
          }
        }
      }
      // Could check super-classes but we wouldn't have got past the
      // load if they referred to a missing class.
    }
  } else if (filename.ends_with(".data.xml")) {
 
    auto check_relationship = [&](OksData* data) {
      auto relobj = data->data.OBJECT;
      if (relobj == nullptr) {
        return;
      }
      auto ofile_name = relobj->get_file()->get_short_file_name();
      if (file->get_short_file_name() != ofile_name && !includes.contains(ofile_name)) {
        TLOG() << "Error " << filename << " misses include of " << ofile_name << " required by " << relobj->GetId();
        if (result == Exitcode::SUCCESS) {
          result = Exitcode::MISSING_INCLUDE;
        }
      }
    };
 
    auto data_objs = kernel.create_list_of_data_objects(file);
    for (auto obj : *data_objs) {
      // Check that any objects defined in this data file have their
      // corresponding schema file included directly in this file
      auto oks_class = obj->GetClass();
      auto cfile = oks_class->get_file()->get_short_file_name();
      if (!includes.contains(cfile)) {
        TLOG() << "Error " << filename << " misses include of " << cfile << " required by " << obj->GetId()
               << " of class " << oks_class->get_name();
        if (result == Exitcode::SUCCESS) {
          result = Exitcode::MISSING_INCLUDE;
        }
      }
 
      // Check that any objects referenced by objects in this file are
      // either defined in this file or one that is directly included
      auto relationships = oks_class->all_relationships();
      if (relationships != nullptr) {
        for (auto rel : *relationships) {
          auto relval = obj->GetRelationshipValue(rel->get_name());
          if (relval == nullptr) {
            continue;
          }
          if (rel->get_high_cardinality_constraint() == OksRelationship::Many) {
            for (OksData::List::iterator iter = relval->data.LIST->begin(); iter != relval->data.LIST->end(); ++iter) {
              check_relationship(*iter);
            }
          } else {
            check_relationship(relval);
          }
        }
      }
    }
    delete data_objs;
  }
 
  auto status = kernel.get_bind_objects_status();
  if (!status.empty()) {
    if (result == Exitcode::SUCCESS) {
      result = Exitcode::UNRESOLVED;
    }
  }
 
  return result;
}