blob: 7c0b0c2b4fc521554aceee2bf0aceac3a929dad0 [file] [log] [blame]
/*
* Copyright (C) 2018 CESNET, https://photonics.cesnet.cz/
* Copyright (C) 2018 FIT CVUT, https://fit.cvut.cz/
*
* Written by Václav Kubernát <kubervac@fit.cvut.cz>
*
*/
#pragma once
#include <set>
#include <unordered_map>
#include "ast_path.hpp"
#include "schema.hpp"
namespace yang {
enum class ContainerTraits {
Presence,
None,
};
struct container {
yang::ContainerTraits m_presence;
};
struct list {
std::set<std::string> m_keys;
};
struct leaf {
yang::LeafDataTypes m_type;
std::set<std::string> m_enumValues;
ModuleValuePair m_identBase;
std::string m_leafRefSource;
};
struct module {
};
}
using NodeType = boost::variant<yang::container, yang::list, yang::leaf, yang::module>;
/*! \class StaticSchema
* \brief Static schema, used mainly for testing
* */
class StaticSchema : public Schema {
public:
StaticSchema();
yang::NodeTypes nodeType(const std::string& path) const override;
yang::NodeTypes nodeType(const schemaPath_& location, const ModuleNodePair& node) const override;
bool isModule(const std::string& name) const override;
bool leafEnumHasValue(const schemaPath_& location, const ModuleNodePair& node, const std::string& value) const override;
bool leafIdentityIsValid(const schemaPath_& location, const ModuleNodePair& node, const ModuleValuePair& value) const override;
bool listHasKey(const schemaPath_& location, const ModuleNodePair& node, const std::string& key) const override;
const std::set<std::string> listKeys(const schemaPath_& location, const ModuleNodePair& node) const override;
yang::LeafDataTypes leafType(const schemaPath_& location, const ModuleNodePair& node) const override;
yang::LeafDataTypes leafType(const std::string& path) const override;
yang::LeafDataTypes leafrefBaseType(const schemaPath_& location, const ModuleNodePair& node) const override;
const std::set<std::string> enumValues(const schemaPath_& location, const ModuleNodePair& node) const override;
const std::set<std::string> validIdentities(const schemaPath_& location, const ModuleNodePair& node, const Prefixes prefixes) const override;
std::set<std::string> childNodes(const schemaPath_& path, const Recursion) const override;
std::set<std::string> moduleNodes(const module_& module, const Recursion recursion) const override;
std::optional<std::string> description(const std::string& path) const override;
std::optional<std::string> units(const std::string& path) const override;
void addContainer(const std::string& location, const std::string& name, yang::ContainerTraits isPresence = yang::ContainerTraits::None);
void addLeaf(const std::string& location, const std::string& name, const yang::LeafDataTypes& type);
void addLeafEnum(const std::string& location, const std::string& name, std::set<std::string> enumValues);
void addLeafIdentityRef(const std::string& location, const std::string& name, const ModuleValuePair& base);
void addLeafRef(const std::string& location, const std::string& name, const std::string& source);
void addList(const std::string& location, const std::string& name, const std::set<std::string>& keys);
void addModule(const std::string& name);
void addIdentity(const std::optional<ModuleValuePair>& base, const ModuleValuePair& name);
private:
const std::unordered_map<std::string, NodeType>& children(const std::string& name) const;
void getIdentSet(const ModuleValuePair& ident, std::set<ModuleValuePair>& res) const;
bool nodeExists(const std::string& location, const std::string& node) const;
std::unordered_map<std::string, std::unordered_map<std::string, NodeType>> m_nodes;
std::set<std::string> m_modules;
std::map<ModuleValuePair, std::set<ModuleValuePair>> m_identities;
};