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
|
/*
*/
#include <synext.hpp>
#include "../parse/common.hpp"
#include "../parse/parseerror.hpp"
#include "../parse/tokentree.hpp"
#include "../parse/lex.hpp"
class CFormatArgsExpander:
public ExpandProcMacro
{
bool expand_early() const override { return true; }
::std::unique_ptr<TokenStream> expand(const Span& sp, const ::AST::Crate& crate, const ::std::string& ident, const TokenTree& tt, AST::Module& mod) override
{
Token tok;
auto lex = TTStream(tt);
if( ident != "" )
ERROR(sp, E0000, "format_args! doesn't take an ident");
auto n = Parse_ExprVal(lex);
auto format_string = dynamic_cast<AST::ExprNode_String&>(*n).m_value;
// TODO: Interpolated expression "tokens"
::std::map< ::std::string, TokenTree> named_args;
::std::vector<TokenTree> free_args;
while( GET_TOK(tok, lex) == TOK_COMMA )
{
if( lex.lookahead(0) == TOK_IDENT && lex.lookahead(1) == TOK_EQUAL )
{
GET_CHECK_TOK(tok, lex, TOK_IDENT);
auto name = mv$(tok.str());
GET_CHECK_TOK(tok, lex, TOK_EQUAL);
auto expr_tt = Parse_TT_Expr(lex);
named_args.insert( ::std::make_pair(mv$(name), mv$(expr_tt)) );
}
else
{
auto expr_tt = Parse_TT_Expr(lex);
free_args.push_back( mv$(expr_tt) );
}
}
// TODO: Expand format_args!
::std::vector<TokenTree> toks;
toks.push_back( TokenTree(TOK_PAREN_OPEN) );
toks.push_back( TokenTree(TOK_PAREN_CLOSE) );
return box$( TTStreamO(TokenTree(mv$(toks))) );
}
};
STATIC_MACRO("format_args", CFormatArgsExpander);
|
