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
|
// Logic for formatting arguments and argument lists
#include "../ast.h"
#include "../stdlib/datatypes.h"
#include "../stdlib/optionals.h"
#include "../stdlib/text.h"
#include "formatter.h"
#include "types.h"
#include "utils.h"
OptionalText_t format_inline_arg(arg_ast_t *arg, Table_t comments) {
if (range_has_comment(arg->start, arg->end, comments)) return NONE_TEXT;
if (arg->name == NULL && arg->value) return must(format_inline_code(arg->value, comments));
Text_t code = Text$from_str(arg->name);
if (arg->type) code = Texts(code, ":", must(format_type(arg->type)));
if (arg->value) code = Texts(code, "=", must(format_inline_code(arg->value, comments)));
return code;
}
Text_t format_arg(arg_ast_t *arg, Table_t comments, Text_t indent) {
OptionalText_t inline_arg = format_inline_arg(arg, comments);
if (inline_arg.length >= 0 && inline_arg.length <= MAX_WIDTH) return inline_arg;
if (arg->name == NULL && arg->value) return format_code(arg->value, comments, indent);
Text_t code = Text$from_str(arg->name);
if (arg->type) code = Texts(code, ":", format_type(arg->type));
if (arg->value) code = Texts(code, "=", format_code(arg->value, comments, indent));
return code;
}
OptionalText_t format_inline_args(arg_ast_t *args, Table_t comments) {
Text_t code = EMPTY_TEXT;
for (arg_ast_t *arg = args; arg; arg = arg->next) {
if (arg->name && arg->next && arg->type == arg->next->type && arg->value == arg->next->value) {
code = Texts(code, Text$from_str(arg->name), ",");
} else {
code = Texts(code, must(format_inline_arg(arg, comments)));
if (arg->next) code = Texts(code, ", ");
}
if (arg->next && range_has_comment(arg->end, arg->next->start, comments)) return NONE_TEXT;
}
return code;
}
Text_t format_args(arg_ast_t *args, Table_t comments, Text_t indent) {
OptionalText_t inline_args = format_inline_args(args, comments);
if (inline_args.length >= 0 && inline_args.length <= MAX_WIDTH) return inline_args;
Text_t code = EMPTY_TEXT;
for (arg_ast_t *arg = args; arg; arg = arg->next) {
if (arg->name && arg->next && arg->type == arg->next->type && arg->value == arg->next->value) {
code = Texts(code, Text$from_str(arg->name), ",");
} else {
code = Texts(code, "\n", indent, single_indent, format_arg(arg, comments, Texts(indent, single_indent)));
if (args->next) code = Texts(code, ",");
}
}
return code;
}
|
