function.h

// Copyright (c) 2020-now by the Zeek Project. See LICENSE for details.
 
#pragma once
 
#include <utility>
 
#include <hilti/ast/declaration.h>
#include <hilti/ast/declarations/type.h>
#include <hilti/ast/function.h>
#include <hilti/ast/node-range.h>
#include <hilti/ast/operator-registry.h>
#include <hilti/ast/operator.h>
#include <hilti/ast/types/type.h>
 
namespace hilti::declaration {
 
class Function : public Declaration {
public:
    auto function() const { return child<::hilti::Function>(0); }
 
    auto operator_() const { return _operator; }
 
    declaration::Type* linkedDeclaration(ASTContext* ctx) const {
        if ( _linked_declaration_index ) {
            auto* decl = ctx->lookup(_linked_declaration_index);
            return decl->as<declaration::Type>();
        }
        else
            return nullptr;
    }
 
    auto linkedDeclarationIndex() const { return _linked_declaration_index; }
 
    declaration::Function* linkedPrototype(ASTContext* ctx) const {
        if ( _linked_prototype_index ) {
            auto* decl = ctx->lookup(_linked_prototype_index);
            return decl->as<declaration::Function>();
        }
        else
            return nullptr;
    }
 
    auto linkedPrototypeIndex() const { return _linked_prototype_index; }
 
    void setOperator(const Operator* op) { _operator = op; }
 
    void setLinkedDeclarationIndex(ast::DeclarationIndex index) {
        assert(index);
        _linked_declaration_index = index;
    }
 
    void setLinkedPrototypeIndex(ast::DeclarationIndex index) {
        assert(index);
        _linked_prototype_index = index;
    }
 
    ID functionID(ASTContext* ctx) const;
 
    std::string_view displayName() const final { return "function"; }
 
    node::Properties properties() const final;
 
    static Function* create(ASTContext* ctx,
                            hilti::Function* function,
                            declaration::Linkage linkage = Linkage::Private,
                            Meta meta = {}) {
        return ctx->make<Function>(ctx, {function}, function->id(), linkage, std::move(meta));
    }
 
protected:
    Function(ASTContext* ctx, Nodes children, ID id, declaration::Linkage linkage, Meta meta)
        : Declaration(ctx, NodeTags, std::move(children), std::move(id), linkage, std::move(meta)) {}
 
    HILTI_NODE_1(declaration::Function, Declaration, final);
 
private:
    const Operator* _operator = nullptr;
 
    ast::DeclarationIndex _linked_declaration_index;
    ast::DeclarationIndex _linked_prototype_index;
};
 
} // namespace hilti::declaration