summaryrefslogtreecommitdiff
path: root/src/include/synext.hpp
blob: 05020c8c913aa3733a1798c7fd6b1c7fa68ea0dc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/*
 */
#pragma once
#ifndef _SYNEXT_HPP_
#define _SYNEXT_HPP_

#include "../ast/item.hpp"

namespace AST {
    class Crate;
    class MetaItem;
    class Path;
    
    class Module;
    class Item;
    
    class ExprNode;
    class Expr;
    
    class MacroInvocation;
}
class TokenTree;
class TokenStream;


#include "../common.hpp"   // for mv$ and other things
#include <string>
#include <memory>

enum class AttrStage
{
    EarlyPre,
    EarlyPost,
    LatePre,
    LatePost,
};

class ExpandDecorator
{
public:
    virtual AttrStage   stage() const = 0;
    
    virtual void    handle(const AST::MetaItem& mi, AST::Crate& crate) const {}
    virtual void    handle(const AST::MetaItem& mi, AST::Crate& crate, AST::MacroInvocation& mac) const {}
    virtual void    handle(const AST::MetaItem& mi, AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const {}
    virtual void    handle(const AST::MetaItem& mi, AST::ExprNode& expr) const {};
};

enum class MacroPosition
{
    Item,
    Stmt,
    Expr,
    Type,
    Pattern,
};

class ExpandProcMacro
{
public:
    virtual bool    expand_early() const = 0;
    
    virtual ::std::unique_ptr<TokenStream>  expand(const ::std::string& ident, const TokenTree& tt, AST::Module& mod) = 0;
};

#define STATIC_DECORATOR(ident, _handler_class) \
    struct register_##_handler_class##_c {\
        register_##_handler_class##_c() {\
            Register_Synext_Decorator( ident, ::std::unique_ptr<ExpandDecorator>(new _handler_class()) ); \
        } \
    } s_register_##_handler_class;
#define STATIC_MACRO(ident, _handler_class) \
    struct register_##_handler_class##_c {\
        register_##_handler_class##_c() {\
            Register_Synext_Macro( ident, ::std::unique_ptr<ExpandProcMacro>(new _handler_class()) ); \
        } \
    } s_register_##_handler_class;

extern void Register_Synext_Decorator(::std::string name, ::std::unique_ptr<ExpandDecorator> handler);
extern void Register_Synext_Macro(::std::string name, ::std::unique_ptr<ExpandProcMacro> handler);

#endif