20065764942

Committed 09 Dec 2025 01:49PM UTC coverage: 90.564% (+0.008%) from 90.556%

Build # 20065764942

Build Type

push

github

Committed by

SuperFola

Commit Message

chore(tests): adding IR generation tests for arg attributes

Run Details

8043 of 8881 relevant lines covered (90.56%)

180911.91 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

92.81

/src/arkreactor/Compiler/AST/Node.cpp

#include <Ark/Compiler/Common.hpp>
#include <Ark/Compiler/AST/Node.hpp>

#include <Ark/Error/Exceptions.hpp>

#include <cassert>
#include <fmt/core.h>

namespace Ark::internal
{
    Node::Node(const NodeType node_type, const std::string& value) :
        m_type(node_type), m_value(value), m_pos()
    {}

    Node::Node(const NodeType node_type) :
        m_type(node_type), m_pos()
    {
        if (m_type == NodeType::List || m_type == NodeType::Macro || m_type == NodeType::Field)
            m_value = std::vector<Node>();
    }

    Node::Node(double value) :
        m_type(NodeType::Number), m_value(value), m_pos()
    {}

    Node::Node(const long value) :
        m_type(NodeType::Number), m_value(static_cast<double>(value)), m_pos()
    {}

    Node::Node(Keyword value) :
        m_type(NodeType::Keyword), m_value(value), m_pos()
    {}

    Node::Node(const Namespace& namespace_) :
        m_type(NodeType::Namespace), m_value(namespace_), m_pos()
    {}

    const std::string& Node::string() const noexcept
    {
        return std::get<std::string>(m_value);
    }

    double Node::number() const noexcept
    {
        return std::get<double>(m_value);
    }

    Keyword Node::keyword() const noexcept
    {
        return std::get<Keyword>(m_value);
    }

    Namespace& Node::arkNamespace() noexcept
    {
        return std::get<Namespace>(m_value);
    }

    const Namespace& Node::constArkNamespace() const noexcept
    {
        return std::get<Namespace>(m_value);
    }

    void Node::push_back(const Node& node) noexcept
    {
        list().push_back(node);
    }

    std::vector<Node>& Node::list() noexcept
    {
        return std::get<std::vector<Node>>(m_value);
    }

    const std::vector<Node>& Node::constList() const noexcept
    {
        return std::get<std::vector<Node>>(m_value);
    }

    NodeType Node::nodeType() const noexcept
    {
        return m_type;
    }

    bool Node::isListLike() const noexcept
    {
        return m_type == NodeType::List || m_type == NodeType::Macro;
    }

    bool Node::isFunction() const noexcept
    {
        return m_type == NodeType::List &&
            !constList().empty() &&
            constList()[0].nodeType() == NodeType::Keyword &&
            constList()[0].keyword() == Keyword::Fun;
    }

    const std::optional<std::string>& Node::getUnqualifiedName() const noexcept
    {
        return m_unqualified_name;
    }

    void Node::updateValueAndType(const Node& source) noexcept
    {
        m_type = source.m_type;
        m_value = source.m_value;
    }

    void Node::setNodeType(const NodeType type) noexcept
    {
        m_type = type;
    }

    void Node::setUnqualifiedName(const std::string& name) noexcept
    {
        m_unqualified_name = name;
    }

    void Node::setString(const std::string& value) noexcept
    {
        m_value = value;
    }

    void Node::setPositionFrom(const Node& source) noexcept
    {
        m_filename = source.m_filename;
        m_pos = source.m_pos;
    }

    Node& Node::attachNearestCommentBefore(const std::string& comment)
    {
        m_comment = comment;
        return *this;
    }

    Node& Node::attachCommentAfter(const std::string& comment)
    {
        if (!m_after_comment.empty())
            m_after_comment += "\n";
        m_after_comment += comment;
        if (!m_after_comment.empty() && m_after_comment.back() == '\n')
            m_after_comment.pop_back();
        return *this;
    }

    void Node::setAltSyntax(const bool toggle)
    {
        m_alt_syntax = toggle;
    }

    void Node::setFunctionKind(const bool anonymous)
    {
        m_is_anonymous_function = anonymous;
    }

    bool Node::isAnonymousFunction() const noexcept
    {
        return m_is_anonymous_function;
    }

    FileSpan Node::position() const noexcept
    {
        return m_pos;
    }

    const std::string& Node::filename() const noexcept
    {
        return m_filename;
    }

    const std::string& Node::comment() const noexcept
    {
        return m_comment;
    }

    const std::string& Node::commentAfter() const noexcept
    {
        return m_after_comment;
    }

    std::string Node::repr() const noexcept
    {
        std::string data;
        switch (m_type)
        {
            case NodeType::Symbol:
                data += string();
                break;

            case NodeType::MutArg:
                data += "(mut " + string() + ")";
                break;

            case NodeType::RefArg:
                data += "(ref " + string() + ")";
                break;

            case NodeType::Capture:
                data += "&" + string();
                break;

            case NodeType::Keyword:
                data += keywords[static_cast<std::size_t>(keyword())];
                break;

            case NodeType::String:
                data += "\"" + string() + "\"";
                break;

            case NodeType::Number:
                data += fmt::format("{}", number());
                break;

            case NodeType::List:
                if (m_alt_syntax)
                {
                    const auto first = constList().front();
                    char open = 0;
                    if (first.nodeType() == NodeType::Keyword && first.keyword() == Keyword::Begin)
                        open = '{';
                    else if (first.nodeType() == NodeType::Symbol && first.string() == "list")
                        open = '[';
                    else
                        assert(false && "Alt syntax nodes can only be begin or list");

                    data += open;
                    for (std::size_t i = 1, end = constList().size(); i < end; ++i)
                    {
                        data += constList()[i].repr();
                        if (i < end - 1)
                            data += " ";
                    }

                    if (open == '{')
                        data += "}";
                    else if (open == '[')
                        data += "]";
                }
                else
                {
                    data += "(";
                    for (std::size_t i = 0, end = constList().size(); i < end; ++i)
                    {
                        data += constList()[i].repr();
                        if (i < end - 1)
                            data += " ";
                    }
                    data += ")";
                }
                break;

            case NodeType::Field:
                for (std::size_t i = 0, end = constList().size(); i < end; ++i)
                {
                    data += constList()[i].repr();
                    if (i < end - 1)
                        data += ".";
                }
                break;

            case NodeType::Macro:
                data += "(macro ";
                for (std::size_t i = 0, end = constList().size(); i < end; ++i)
                {
                    data += constList()[i].repr();
                    if (i < end - 1)
                        data += " ";
                }
                data += ")";
                break;

            case NodeType::Spread:
                data += "..." + string();
                break;

            // namespace node should not have a representation as it is purely internal,
            // and it can't be exploited by macros (unless you try passing an import node
            // to a macro, which should not happen?)
            case NodeType::Namespace:
                data += constArkNamespace().ast->repr();
                break;

            case NodeType::Unused:
                break;
        }
        return data;
    }

    std::ostream& Node::debugPrint(std::ostream& os) const noexcept
    {
        switch (m_type)
        {
            case NodeType::Symbol:
                os << "Symbol:" << string();
                break;

            case NodeType::MutArg:
                os << "MutArg:" << string();
                break;

            case NodeType::RefArg:
                os << "RefArg:" << string();
                break;

            case NodeType::Capture:
                os << "Capture:" << string();
                break;

            case NodeType::Keyword:
                os << "Keyword:";
                switch (keyword())
                {
                    case Keyword::Fun: os << "Fun"; break;
                    case Keyword::Let: os << "Let"; break;
                    case Keyword::Mut: os << "Mut"; break;
                    case Keyword::Set: os << "Set"; break;
                    case Keyword::If: os << "If"; break;
                    case Keyword::While: os << "While"; break;
                    case Keyword::Begin: os << "Begin"; break;
                    case Keyword::Import: os << "Import"; break;
                    case Keyword::Del: os << "Del"; break;
                }
                break;

            case NodeType::String:
                os << "String:" << string();
                break;

            case NodeType::Number:
                os << "Number:" << number();
                break;

            case NodeType::List:
                os << "( ";
                for (const auto& i : constList())
                    i.debugPrint(os) << " ";
                os << ")";
                break;

            case NodeType::Field:
                os << "( Field ";
                for (const auto& i : constList())
                    i.debugPrint(os) << " ";
                os << ")";
                break;

            case NodeType::Macro:
                os << "( Macro ";
                for (const auto& i : constList())
                    i.debugPrint(os) << " ";
                os << ")";
                break;

            case NodeType::Spread:
                os << "Spread:" << string();
                break;

            case NodeType::Namespace:
            {
                const auto details = constArkNamespace();
                os << "( Namespace:" << details.name << " ";
                details.ast->debugPrint(os) << " )";
                break;
            }

            case NodeType::Unused:
                break;
        }
        return os;
    }

    const Node& getTrueNode()
    {
        static const Node TrueNode(NodeType::Symbol, "true");
        return TrueNode;
    }

    const Node& getFalseNode()
    {
        static const Node FalseNode(NodeType::Symbol, "false");
        return FalseNode;
    }

    const Node& getNilNode()
    {
        static const Node NilNode(NodeType::Symbol, "nil");
        return NilNode;
    }

    const Node& getListNode()
    {
        static const Node ListNode(NodeType::Symbol, "list");
        return ListNode;
    }

    bool operator==(const Node& A, const Node& B)
    {
        if (A.m_type != B.m_type)  // should have the same types
            return false;

        if (A.m_type != NodeType::List)
            return A.m_value == B.m_value;
        return false;
    }

    bool operator<(const Node& A, const Node& B)
    {
        if (A.nodeType() != B.nodeType())
            return (static_cast<int>(A.nodeType()) - static_cast<int>(B.nodeType())) < 0;

        switch (A.nodeType())
        {
            case NodeType::Number:
                [[fallthrough]];
            case NodeType::Symbol:
                [[fallthrough]];
            case NodeType::String:
                return A.m_value < B.m_value;

            default:
                return false;
        }
    }
}

1	#include <Ark/Compiler/Common.hpp>
2	#include <Ark/Compiler/AST/Node.hpp>
3
4	#include <Ark/Error/Exceptions.hpp>
5
6	#include <cassert>
7	#include <fmt/core.h>
8
9	namespace Ark::internal
10	{
11	Node::Node(const NodeType node_type, const std::string& value) :	167,150✔
12	m_type(node_type), m_value(value), m_pos()	83,575✔
13	{}	83,575✔
14
15	Node::Node(const NodeType node_type) :	316,809✔
16	m_type(node_type), m_pos()	633,618✔
17	{	316,809✔
18	if (m_type == NodeType::List \|\| m_type == NodeType::Macro \|\| m_type == NodeType::Field)	316,809✔
19	m_value = std::vector<Node>();	315,705✔
20	}	316,809✔
21
22	Node::Node(double value) :	75,976✔
23	m_type(NodeType::Number), m_value(value), m_pos()	151,952✔
24	{}	151,952✔
25
26	Node::Node(const long value) :	106✔
27	m_type(NodeType::Number), m_value(static_cast<double>(value)), m_pos()	212✔
28	{}	212✔
29
30	Node::Node(Keyword value) :	16,862✔
31	m_type(NodeType::Keyword), m_value(value), m_pos()	33,724✔
32	{}	33,724✔
33
34	Node::Node(const Namespace& namespace_) :	278✔
35	m_type(NodeType::Namespace), m_value(namespace_), m_pos()	139✔
36	{}	139✔
37
38	const std::string& Node::string() const noexcept	10,748,743✔
39	{	10,748,743✔
40	return std::get<std::string>(m_value);	10,748,743✔
41	}
42
43	double Node::number() const noexcept	12,948✔
44	{	12,948✔
45	return std::get<double>(m_value);	12,948✔
46	}
47
48	Keyword Node::keyword() const noexcept	144,436✔
49	{	144,436✔
50	return std::get<Keyword>(m_value);	144,436✔
51	}
52
53	Namespace& Node::arkNamespace() noexcept	473✔
54	{	473✔
55	return std::get<Namespace>(m_value);	473✔
56	}
57
58	const Namespace& Node::constArkNamespace() const noexcept	134✔
59	{	134✔
60	return std::get<Namespace>(m_value);	134✔
61	}
62
63	void Node::push_back(const Node& node) noexcept	214,409✔
64	{	214,409✔
65	list().push_back(node);	214,409✔
66	}	214,409✔
67
68	std::vector<Node>& Node::list() noexcept	1,463,898✔
69	{	1,463,898✔
70	return std::get<std::vector<Node>>(m_value);	1,463,898✔
71	}
72
73	const std::vector<Node>& Node::constList() const noexcept	2,449,466✔
74	{	2,449,466✔
75	return std::get<std::vector<Node>>(m_value);	2,449,466✔
76	}
77
78	NodeType Node::nodeType() const noexcept	5,005,301✔
79	{	5,005,301✔
80	return m_type;	5,005,301✔
81	}
82
83	bool Node::isListLike() const noexcept	16,671✔
84	{	16,671✔
85	return m_type == NodeType::List \|\| m_type == NodeType::Macro;	16,671✔
86	}
87
88	bool Node::isFunction() const noexcept	8,155✔
89	{	8,155✔
90	return m_type == NodeType::List &&	14,257✔
91	!constList().empty() &&	6,102✔
92	constList()[0].nodeType() == NodeType::Keyword &&	8,928✔
93	constList()[0].keyword() == Keyword::Fun;	2,826✔
94	}
95
96	const std::optional<std::string>& Node::getUnqualifiedName() const noexcept	909✔
97	{	909✔
98	return m_unqualified_name;	909✔
99	}
100
101	void Node::updateValueAndType(const Node& source) noexcept	27,676✔
102	{	27,676✔
103	m_type = source.m_type;	27,676✔
104	m_value = source.m_value;	27,676✔
105	}	27,676✔
106
107	void Node::setNodeType(const NodeType type) noexcept	265✔
108	{	265✔
109	m_type = type;	265✔
110	}	265✔
111
112	void Node::setUnqualifiedName(const std::string& name) noexcept	227✔
113	{	227✔
114	m_unqualified_name = name;	227✔
115	}	227✔
116
117	void Node::setString(const std::string& value) noexcept	124,824✔
118	{	124,824✔
119	m_value = value;	124,824✔
120	}	124,824✔
121
122	void Node::setPositionFrom(const Node& source) noexcept	175✔
123	{	175✔
124	m_filename = source.m_filename;	175✔
125	m_pos = source.m_pos;	175✔
126	}	175✔
127
128	Node& Node::attachNearestCommentBefore(const std::string& comment)	193,356✔
129	{	193,356✔
130	m_comment = comment;	193,356✔
131	return *this;	193,356✔
132	}
133
134	Node& Node::attachCommentAfter(const std::string& comment)	52,147✔
135	{	52,147✔
136	if (!m_after_comment.empty())	52,147✔
137	m_after_comment += "\n";	37✔
138	m_after_comment += comment;	52,147✔
139	if (!m_after_comment.empty() && m_after_comment.back() == '\n')	52,147✔
140	m_after_comment.pop_back();	107✔
141	return *this;	52,147✔
142	}
143
144	void Node::setAltSyntax(const bool toggle)	3,630✔
145	{	3,630✔
146	m_alt_syntax = toggle;	3,630✔
147	}	3,630✔
148
149	void Node::setFunctionKind(const bool anonymous)	2,786✔
150	{	2,786✔
151	m_is_anonymous_function = anonymous;	2,786✔
152	}	2,786✔
153
154	bool Node::isAnonymousFunction() const noexcept	6,295✔
155	{	6,295✔
156	return m_is_anonymous_function;	6,295✔
157	}
158
159	FileSpan Node::position() const noexcept	169,772✔
160	{	169,772✔
161	return m_pos;	169,772✔
162	}
163
164	const std::string& Node::filename() const noexcept	92,994✔
165	{	92,994✔
166	return m_filename;	92,994✔
167	}
168
169	const std::string& Node::comment() const noexcept	19,916✔
170	{	19,916✔
171	return m_comment;	19,916✔
172	}
173
174	const std::string& Node::commentAfter() const noexcept	2,444✔
175	{	2,444✔
176	return m_after_comment;	2,444✔
177	}
178
179	std::string Node::repr() const noexcept	20,611✔
180	{	20,611✔
181	std::string data;	20,611✔
182	switch (m_type)	20,611✔
183	{	17,327✔
184	case NodeType::Symbol:
185	data += string();	17,327✔
186	break;	17,327✔
187
188	case NodeType::MutArg:
189	data += "(mut " + string() + ")";	×
190	break;	×
191
192	case NodeType::RefArg:
193	data += "(ref " + string() + ")";	×
194	break;	×
195
196	case NodeType::Capture:
197	data += "&" + string();	×
198	break;	92✔
199
200	case NodeType::Keyword:
201	data += keywords[static_cast<std::size_t>(keyword())];	92✔
202	break;	599✔
203
204	case NodeType::String:
205	data += "\"" + string() + "\"";	507✔
206	break;	1,756✔
207
208	case NodeType::Number:
209	data += fmt::format("{}", number());	1,249✔
210	break;	2,531✔
211
212	case NodeType::List:
213	if (m_alt_syntax)	1,282✔
214	{
215	const auto first = constList().front();	283✔
216	char open = 0;	283✔
217	if (first.nodeType() == NodeType::Keyword && first.keyword() == Keyword::Begin)	283✔
218	open = '{';	5✔
219	else if (first.nodeType() == NodeType::Symbol && first.string() == "list")	278✔
220	open = '[';	278✔
221	else
222	assert(false && "Alt syntax nodes can only be begin or list");	×
223
224	data += open;	283✔
225	for (std::size_t i = 1, end = constList().size(); i < end; ++i)	867✔
226	{
227	data += constList()[i].repr();	584✔
228	if (i < end - 1)	584✔
229	data += " ";	372✔
230	}	584✔
231
232	if (open == '{')	283✔
233	data += "}";	5✔
234	else if (open == '[')	278✔
235	data += "]";	278✔
236	}	283✔
237	else
238	{
239	data += "(";	999✔
240	for (std::size_t i = 0, end = constList().size(); i < end; ++i)	3,590✔
241	{
242	data += constList()[i].repr();	2,591✔
243	if (i < end - 1)	2,591✔
244	data += " ";	495,078✔
245	}	2,591✔
246	data += ")";	999✔
247	}
248	break;	1,419✔
249
250	case NodeType::Field:	493,467✔
251	for (std::size_t i = 0, end = constList().size(); i < end; ++i)	493,892✔
252	{
253	data += constList()[i].repr();	288✔
254	if (i < end - 1)	288✔
255	data += ".";	151✔
256	}	288✔
257	break;	153✔
258
259	case NodeType::Macro:
260	data += "(macro ";	16✔
261	for (std::size_t i = 0, end = constList().size(); i < end; ++i)	76✔
262	{
263	data += constList()[i].repr();	60✔
264	if (i < end - 1)	60✔
265	data += " ";	44✔
266	}	60✔
267	data += ")";	16✔
268	break;	17✔
269
270	case NodeType::Spread:
271	data += "..." + string();	1✔
272	break;	1✔
273
274	// namespace node should not have a representation as it is purely internal,
275	// and it can't be exploited by macros (unless you try passing an import node
276	// to a macro, which should not happen?)
277	case NodeType::Namespace:
278	data += constArkNamespace().ast->repr();	×
279	break;	×
280
281	case NodeType::Unused:
282	break;	×
283	}	20,611✔
284	return data;	20,611✔
285	}	20,611✔
286
287	std::ostream& Node::debugPrint(std::ostream& os) const noexcept	537✔
288	{	537✔
289	switch (m_type)	537✔
290	{	155✔
291	case NodeType::Symbol:
292	os << "Symbol:" << string();	155✔
293	break;	161✔
294
295	case NodeType::MutArg:
296	os << "MutArg:" << string();	6✔
297	break;	12✔
298
299	case NodeType::RefArg:
300	os << "RefArg:" << string();	6✔
301	break;	16✔
302
303	case NodeType::Capture:
304	os << "Capture:" << string();	10✔
305	break;	102✔
306
307	case NodeType::Keyword:
308	os << "Keyword:";	92✔
309	switch (keyword())	92✔
310	{	24✔
311	case Keyword::Fun: os << "Fun"; break;	44✔
312	case Keyword::Let: os << "Let"; break;	23✔
313	case Keyword::Mut: os << "Mut"; break;	4✔
314	case Keyword::Set: os << "Set"; break;	16✔
315	case Keyword::If: os << "If"; break;	21✔
316	case Keyword::While: os << "While"; break;	17✔
317	case Keyword::Begin: os << "Begin"; break;	19✔
318	case Keyword::Import: os << "Import"; break;	12✔
319	case Keyword::Del: os << "Del"; break;	4✔
320	}	92✔
321	break;	100✔
322
323	case NodeType::String:
324	os << "String:" << string();	8✔
325	break;	85✔
326
327	case NodeType::Number:
328	os << "Number:" << number();	77✔
329	break;	241✔
330
331	case NodeType::List:
332	os << "( ";	164✔
333	for (const auto& i : constList())	564✔
334	i.debugPrint(os) << " ";	400✔
335	os << ")";	164✔
336	break;	173✔
337
338	case NodeType::Field:
339	os << "( Field ";	9✔
340	for (const auto& i : constList())	30✔
341	i.debugPrint(os) << " ";	21✔
342	os << ")";	9✔
343	break;	17✔
344
345	case NodeType::Macro:
346	os << "( Macro ";	8✔
347	for (const auto& i : constList())	31✔
348	i.debugPrint(os) << " ";	23✔
349	os << ")";	8✔
350	break;	10✔
351
352	case NodeType::Spread:
353	os << "Spread:" << string();	2✔
354	break;	2✔
355
356	case NodeType::Namespace:
357	{
358	const auto details = constArkNamespace();	×
359	os << "( Namespace:" << details.name << " ";	×
360	details.ast->debugPrint(os) << " )";	×
361	break;
362	}	×
363
364	case NodeType::Unused:
365	break;	×
366	}	537✔
367	return os;	537✔
368	}
369
370	const Node& getTrueNode()	77✔
371	{	77✔
372	static const Node TrueNode(NodeType::Symbol, "true");	77✔
373	return TrueNode;	77✔
374	}	×
375
376	const Node& getFalseNode()	60✔
377	{	60✔
378	static const Node FalseNode(NodeType::Symbol, "false");	60✔
379	return FalseNode;	60✔
380	}	×
381
382	const Node& getNilNode()	55✔
383	{	55✔
384	static const Node NilNode(NodeType::Symbol, "nil");	55✔
385	return NilNode;	55✔
386	}	×
387
388	const Node& getListNode()	1,189✔
389	{	1,189✔
390	static const Node ListNode(NodeType::Symbol, "list");	1,189✔
391	return ListNode;	1,189✔
392	}	×
393
394	bool operator==(const Node& A, const Node& B)	382✔
395	{	382✔
396	if (A.m_type != B.m_type) // should have the same types	382✔
397	return false;	31✔
398
399	if (A.m_type != NodeType::List)	351✔
400	return A.m_value == B.m_value;	351✔
401	return false;	×
402	}	382✔
403
404	bool operator<(const Node& A, const Node& B)	58✔
405	{	58✔
406	if (A.nodeType() != B.nodeType())	58✔
407	return (static_cast<int>(A.nodeType()) - static_cast<int>(B.nodeType())) < 0;	18✔
408
409	switch (A.nodeType())	40✔
410	{	40✔
411	case NodeType::Number:
412	[[fallthrough]];
413	case NodeType::Symbol:
414	[[fallthrough]];
415	case NodeType::String:
416	return A.m_value < B.m_value;	40✔
417
418	default:
419	return false;	×
420	}
421	}	58✔
422	}

ArkScript-lang / Ark / 20065764942

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous