module main import os import time import flag import v.token import v.parser import v.ast import v.pref import v.errors import strings struct Context { mut: is_watch bool is_compile bool is_print bool is_terse bool hide_names map[string]bool } const context = &Context{} fn main() { if os.args.len < 2 { eprintln('not enough parameters') exit(1) } mut ctx := unsafe { context } mut fp := flag.new_flag_parser(os.args[2..]) fp.application('v ast') fp.usage_example('demo.v generate demo.json file.') fp.usage_example('-w demo.v generate demo.json file, and watch for changes.') fp.usage_example('-c demo.v generate demo.json *and* a demo.c file, and watch for changes.') fp.usage_example('-p demo.v print the json output to stdout.') fp.description('Dump a JSON representation of the V AST for a given .v or .vsh file.') fp.description('By default, `v ast` will save the JSON to a .json file, named after the .v file.') fp.description('Pass -p to see it instead.') ctx.is_watch = fp.bool('watch', `w`, false, 'watch a .v file for changes, rewrite the .json file, when a change is detected') ctx.is_print = fp.bool('print', `p`, false, 'print the AST to stdout') ctx.is_compile = fp.bool('compile', `c`, false, 'watch the .v file for changes, rewrite the .json file, *AND* generate a .c file too on any change') ctx.is_terse = fp.bool('terse', `t`, false, 'terse output, only with tree node names (AST structure), no details') hfields := fp.string_multi('hide', 0, 'hide the specified fields. You can give several, by separating them with `,`').join(',') for hf in hfields.split(',') { ctx.hide_names[hf] = true } fp.limit_free_args_to_at_least(1)! rest_of_args := fp.remaining_parameters() for vfile in rest_of_args { file := get_abs_path(vfile) check_file(file) ctx.write_file_or_print(file) if ctx.is_watch || ctx.is_compile { ctx.watch_for_changes(file) } } } fn (ctx Context) write_file_or_print(file string) { if ctx.is_print { println(json(file)) } else { println('${time.now()}: AST written to: ' + json_file(file)) } } // generate ast json file and c source code file fn (ctx Context) watch_for_changes(file string) { println('start watching...') mut timestamp := i64(0) for { new_timestamp := os.file_last_mod_unix(file) if timestamp != new_timestamp { ctx.write_file_or_print(file) if ctx.is_compile { file_name := file[0..(file.len - os.file_ext(file).len)] os.system('v -o ${file_name}.c ${file}') } } timestamp = new_timestamp time.sleep(500 * time.millisecond) } } // get absolute path for file fn get_abs_path(path string) string { if os.is_abs_path(path) { return path } else if path.starts_with('./') { return os.join_path(os.getwd(), path[2..]) } else { return os.join_path(os.getwd(), path) } } // check file is v file and exists fn check_file(file string) { if os.file_ext(file) !in ['.v', '.vv', '.vsh'] { eprintln('the file `${file}` must be a v file or vsh file') exit(1) } if !os.exists(file) { eprintln('the v file `${file}` does not exist') exit(1) } } // generate json file with the same file name fn json_file(file string) string { ast_json := json(file) // support .v and .vsh file file_name := file[0..(file.len - os.file_ext(file).len)] json_file := file_name + '.json' os.write_file(json_file, ast_json) or { panic(err) } return json_file } // generate json string fn json(file string) string { // use as permissive preferences as possible, so that `v ast` // can print the AST of arbitrary V files, even .vsh or ones // that require globals: mut pref_ := &pref.Preferences{} pref_.fill_with_defaults() pref_.enable_globals = true pref_.is_fmt = true // mut t := Tree{ root: new_object() table: ast.new_table() pref: pref_ } // parse file with comment ast_file := parser.parse_file(file, t.table, .parse_comments, t.pref) t.root = t.ast_file(ast_file) // generate the ast string s := json_print(t.root) return s } // the ast tree struct Tree { table &ast.Table = unsafe { nil } pref &pref.Preferences = unsafe { nil } mut: root Node // the root of tree } // tree node pub type Node = C.cJSON // create an object node [inline] fn new_object() &Node { return C.cJSON_CreateObject() } // add item to object node [inline] fn (node &Node) add(key string, child &Node) { if context.hide_names.len > 0 && key in context.hide_names { return } if context.is_terse { return } add_item_to_object(node, key, child) } // add item to object node [inline] fn (node &Node) add_terse(key string, child &Node) { if context.hide_names.len > 0 && key in context.hide_names { return } add_item_to_object(node, key, child) } // create an array node [inline] fn new_array() &Node { return C.cJSON_CreateArray() } // add item to array node [inline] fn (node &Node) add_item(child &Node) { add_item_to_array(node, child) } // string type node fn (t Tree) string_node(val string) &Node { return create_string(val) } // number type node fn (t Tree) number_node(val int) &Node { return create_number(val) } // bool type node fn (t Tree) bool_node(val bool) &Node { if val { return create_true() } else { return create_false() } } // null type node fn (t Tree) null_node() &Node { return create_null() } // type node fn (t Tree) type_node(typ ast.Type) &Node { if typ == 0 { return create_null() } else { type_name := t.table.get_type_name(typ) return create_string(strings.repeat(`&`, typ.nr_muls()) + type_name) } } // token type node fn (t Tree) token_node(tok_kind token.Kind) &Node { return t.string_node('token:${int(tok_kind)}(${tok_kind.str()})') } // enum type node fn (t Tree) enum_node[T](value T) &Node { return t.string_node('enum:${int(value)}(${value})') } // for [][]comment fn (t Tree) two_dimension_comment(node [][]ast.Comment) &Node { mut comments := new_array() for n in node { mut comment_array := new_array() for c in n { comment_array.add_item(t.comment(c)) } comments.add_item(comment_array) } return comments } // ast file root node fn (t Tree) ast_file(node ast.File) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ast.File')) obj.add_terse('path', t.string_node(node.path)) obj.add('path_base', t.string_node(node.path_base)) obj.add_terse('nr_lines', t.number_node(node.nr_lines)) obj.add_terse('nr_bytes', t.number_node(node.nr_bytes)) obj.add_terse('mod', t.mod(node.mod)) obj.add_terse('imports', t.imports(node.imports)) obj.add('global_scope', t.scope(node.global_scope)) obj.add('scope', t.scope(node.scope)) obj.add('errors', t.errors(node.errors)) obj.add('warnings', t.warnings(node.warnings)) obj.add('notices', t.notices(node.notices)) obj.add_terse('auto_imports', t.array_node_string(node.auto_imports)) symbol_obj := new_object() for key, val in node.imported_symbols { symbol_obj.add_terse(key, t.string_node(val)) } obj.add_terse('imported_symbols', symbol_obj) obj.add_terse('generic_fns', t.array_node_generic_fns(node.generic_fns)) obj.add_terse('embedded_files', t.array_node_embed_file(node.embedded_files)) obj.add_terse('global_labels', t.array_node_string(node.global_labels)) obj.add_terse('is_test', t.bool_node(node.is_test)) obj.add_terse('stmts', t.stmts(node.stmts)) return obj } // embed files fn (t Tree) embed_file(node ast.EmbeddedFile) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EmbeddedFile')) obj.add_terse('rpath', t.string_node(node.rpath)) obj.add('apath', t.string_node(node.apath)) obj.add('compression_type', t.string_node(node.compression_type)) obj.add('is_compressed', t.bool_node(node.is_compressed)) obj.add('len', t.number_node(node.len)) obj.add('bytes', t.array_node_u8(node.bytes)) return obj } // ast module node fn (t Tree) mod(node ast.Module) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Module')) obj.add_terse('name', t.string_node(node.name)) obj.add('short_name', t.string_node(node.short_name)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) obj.add('pos', t.pos(node.pos)) obj.add('name_pos', t.pos(node.name_pos)) obj.add_terse('is_skipped', t.bool_node(node.is_skipped)) return obj } fn (t Tree) scope(scope ast.Scope) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Scope')) obj.add_terse('parent', t.string_node(ptr_str(scope.parent))) children_arr := new_array() for s in scope.children { mut children_obj := new_object() children_obj.add_terse('parent', t.string_node(ptr_str(s.parent))) children_obj.add('start_pos', t.number_node(s.start_pos)) children_obj.add('end_pos', t.number_node(s.end_pos)) children_arr.add_item(children_obj) } obj.add_terse('children', children_arr) obj.add('start_pos', t.number_node(scope.start_pos)) obj.add('end_pos', t.number_node(scope.end_pos)) obj.add_terse('objects', t.objects(scope.objects)) obj.add_terse('struct_fields', t.array_node_scope_struct_field(scope.struct_fields)) return obj } fn (t Tree) scope_struct_field(node ast.ScopeStructField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ScopeStructField')) obj.add_terse('struct_type', t.type_node(node.struct_type)) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('orig_type', t.type_node(node.orig_type)) obj.add('pos', t.pos(node.pos)) obj.add_terse('smartcasts', t.array_node_type(node.smartcasts)) return obj } fn (t Tree) objects(so map[string]ast.ScopeObject) &Node { mut obj := new_object() for key, val in so { obj.add_terse(key, t.scope_object(val)) } return obj } fn (t Tree) scope_object(node ast.ScopeObject) &Node { mut obj := new_object() match node { ast.ConstField { t.const_field(node) } ast.GlobalField { t.global_field(node) } ast.Var { t.var(node) } ast.AsmRegister { t.asm_register(node) } } return obj } fn (t Tree) imports(nodes []ast.Import) &Node { mut import_array := new_array() for node in nodes { import_array.add_item(t.import_module(node)) } return import_array } fn (t Tree) errors(errors_ []errors.Error) &Node { mut errs := new_array() for e in errors_ { obj := new_object() obj.add_terse('message', t.string_node(e.message)) obj.add_terse('file_path', t.string_node(e.file_path)) obj.add('pos', t.pos(e.pos)) obj.add('reporter', t.enum_node(e.reporter)) errs.add_item(obj) } return errs } fn (t Tree) warnings(warnings []errors.Warning) &Node { mut warns := new_array() for w in warnings { mut obj := new_object() obj.add_terse('message', t.string_node(w.message)) obj.add_terse('file_path', t.string_node(w.file_path)) obj.add('pos', t.pos(w.pos)) obj.add('reporter', t.enum_node(w.reporter)) warns.add_item(obj) } return warns } fn (t Tree) notices(notices []errors.Notice) &Node { mut notice_array := new_array() for n in notices { mut obj := new_object() obj.add_terse('message', t.string_node(n.message)) obj.add_terse('file_path', t.string_node(n.file_path)) obj.add('pos', t.pos(n.pos)) obj.add('reporter', t.enum_node(n.reporter)) notice_array.add_item(obj) } return notice_array } // stmt array node fn (t Tree) stmts(stmts []ast.Stmt) &Node { mut stmt_array := new_array() for s in stmts { stmt_array.add_item(t.stmt(s)) } return stmt_array } fn (t Tree) stmt(node ast.Stmt) &Node { match node { ast.Module { return t.mod(node) } ast.Import { return t.import_module(node) } ast.ConstDecl { return t.const_decl(node) } ast.FnDecl { return t.fn_decl(node) } ast.StructDecl { return t.struct_decl(node) } ast.EnumDecl { return t.enum_decl(node) } ast.InterfaceDecl { return t.interface_decl(node) } ast.HashStmt { return t.hash_stmt(node) } ast.ComptimeFor { return t.comptime_for(node) } ast.GlobalDecl { return t.global_decl(node) } ast.DeferStmt { return t.defer_stmt(node) } ast.TypeDecl { return t.type_decl(node) } ast.GotoLabel { return t.goto_label(node) } ast.GotoStmt { return t.goto_stmt(node) } ast.AssignStmt { return t.assign_stmt(node) } ast.Return { return t.return_(node) } ast.ForCStmt { return t.for_c_stmt(node) } ast.ForStmt { return t.for_stmt(node) } ast.ForInStmt { return t.for_in_stmt(node) } ast.BranchStmt { return t.branch_stmt(node) } ast.AssertStmt { return t.assert_stmt(node) } ast.ExprStmt { return t.expr_stmt(node) } ast.Block { return t.block(node) } ast.SqlStmt { return t.sql_stmt(node) } ast.AsmStmt { return t.asm_stmt(node) } ast.NodeError { return t.node_error(node) } ast.EmptyStmt { return t.empty_stmt(node) } } return t.null_node() } fn (t Tree) import_module(node ast.Import) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Import')) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('alias', t.string_node(node.alias)) obj.add_terse('syms', t.array_node_import_symbol(node.syms)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('next_comments', t.array_node_comment(node.next_comments)) obj.add('pos', t.pos(node.pos)) obj.add('mod_pos', t.pos(node.mod_pos)) obj.add('alias_pos', t.pos(node.alias_pos)) obj.add('syms_pos', t.pos(node.syms_pos)) return obj } fn (t Tree) import_symbol(node ast.ImportSymbol) &Node { mut obj := new_object() obj.add_terse('name', t.string_node(node.name)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) pos(p token.Pos) &Node { mut obj := new_object() obj.add('line_nr', t.number_node(p.line_nr)) obj.add('last_line', t.number_node(p.last_line)) obj.add('pos', t.number_node(p.pos)) obj.add('len', t.number_node(p.len)) return obj } fn (t Tree) comment(node ast.Comment) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Comment')) obj.add('text', t.string_node(node.text)) obj.add('is_multi', t.bool_node(node.is_multi)) obj.add('is_inline', t.bool_node(node.is_inline)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) const_decl(node ast.ConstDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ConstDecl')) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('is_block', t.bool_node(node.is_block)) obj.add_terse('fields', t.array_node_const_field(node.fields)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) obj.add('end_comments', t.array_node_comment(node.end_comments)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) const_field(node ast.ConstField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ConstField')) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('is_markused', t.bool_node(node.is_markused)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('comptime_expr_value', t.comptime_expr_value(node.comptime_expr_value)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) comptime_expr_value(node ast.ComptTimeConstValue) &Node { match node { ast.EmptyExpr { return t.empty_expr(node) } string { return t.string_node(node) } else { return t.string_node(node.str()) } } } // function declaration fn (t Tree) fn_decl(node ast.FnDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('FnDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('is_deprecated', t.bool_node(node.is_deprecated)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('is_variadic', t.bool_node(node.is_variadic)) obj.add('is_anon', t.bool_node(node.is_anon)) obj.add_terse('is_noreturn', t.bool_node(node.is_noreturn)) obj.add_terse('is_manualfree', t.bool_node(node.is_manualfree)) obj.add('is_main', t.bool_node(node.is_main)) obj.add('is_test', t.bool_node(node.is_test)) obj.add('is_conditional', t.bool_node(node.is_conditional)) obj.add_terse('is_exported', t.bool_node(node.is_exported)) obj.add('is_keep_alive', t.bool_node(node.is_keep_alive)) obj.add_terse('is_unsafe', t.bool_node(node.is_unsafe)) obj.add_terse('receiver', t.struct_field(node.receiver)) obj.add('receiver_pos', t.pos(node.receiver_pos)) obj.add_terse('is_method', t.bool_node(node.is_method)) obj.add('method_type_pos', t.pos(node.method_type_pos)) obj.add('method_idx', t.number_node(node.method_idx)) obj.add_terse('rec_mut', t.bool_node(node.rec_mut)) obj.add('rec_share', t.enum_node(node.rec_share)) obj.add_terse('language', t.enum_node(node.language)) obj.add('file_mode', t.enum_node(node.file_mode)) obj.add('no_body', t.bool_node(node.no_body)) obj.add('is_builtin', t.bool_node(node.is_builtin)) obj.add('is_direct_arr', t.bool_node(node.is_direct_arr)) obj.add('ctdefine_idx', t.number_node(node.ctdefine_idx)) obj.add('pos', t.pos(node.pos)) obj.add('body_pos', t.pos(node.body_pos)) obj.add('return_type_pos', t.pos(node.return_type_pos)) obj.add('file', t.string_node(node.file)) obj.add('has_return', t.bool_node(node.has_return)) obj.add('should_be_skipped', t.bool_node(node.should_be_skipped)) obj.add('ninstances', t.number_node(node.ninstances)) obj.add_terse('has_await', t.bool_node(node.has_await)) obj.add_terse('return_type', t.type_node(node.return_type)) obj.add('source_file', t.number_node(int(node.source_file))) obj.add('scope', t.number_node(int(node.scope))) obj.add('attrs', t.array_node_attr(node.attrs)) obj.add_terse('params', t.array_node_arg(node.params)) obj.add_terse('generic_names', t.array_node_string(node.generic_names)) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('next_comments', t.array_node_comment(node.next_comments)) obj.add('label_names', t.array_node_string(node.label_names)) obj.add('defer_stmts', t.array_node_defer_stmt(node.defer_stmts)) return obj } fn (t Tree) anon_fn(node ast.AnonFn) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AnonFn')) obj.add_terse('decl', t.fn_decl(node.decl)) obj.add('inherited_vars', t.array_node_arg(node.inherited_vars)) obj.add_terse('typ', t.type_node(node.typ)) symbol_obj := new_object() for key, val in node.has_gen { symbol_obj.add_terse(key.str(), t.bool_node(val)) } obj.add_terse('has_gen', symbol_obj) return obj } fn (t Tree) struct_decl(node ast.StructDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StructDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add('pub_pos', t.number_node(node.pub_pos)) obj.add('mut_pos', t.number_node(node.mut_pos)) obj.add('pub_mut_pos', t.number_node(node.pub_mut_pos)) obj.add('global_pos', t.number_node(node.global_pos)) obj.add('module_pos', t.number_node(node.module_pos)) obj.add_terse('language', t.enum_node(node.language)) obj.add_terse('is_union', t.bool_node(node.is_union)) obj.add('pos', t.pos(node.pos)) obj.add_terse('fields', t.array_node_struct_field(node.fields)) obj.add_terse('generic_types', t.array_node_type(node.generic_types)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) obj.add('end_comments', t.array_node_comment(node.end_comments)) obj.add_terse('embeds', t.array_node_embed(node.embeds)) return obj } fn (t Tree) struct_field(node ast.StructField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StructField')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('anon_struct_decl', t.struct_decl(node.anon_struct_decl)) obj.add_terse('unaliased_typ', t.type_node(node.unaliased_typ)) obj.add('type_pos', t.pos(node.type_pos)) obj.add('option_pos', t.pos(node.option_pos)) obj.add_terse('has_default_expr', t.bool_node(node.has_default_expr)) obj.add_terse('default_expr_typ', t.type_node(node.default_expr_typ)) obj.add_terse('default_expr', t.expr(node.default_expr)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add_terse('is_global', t.bool_node(node.is_global)) obj.add_terse('is_volatile', t.bool_node(node.is_volatile)) obj.add_terse('is_deprecated', t.bool_node(node.is_deprecated)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) obj.add_terse('i', t.number_node(node.i)) return obj } fn (t Tree) embed(node ast.Embed) &Node { mut obj := new_object() obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) enum_decl(node ast.EnumDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EnumDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('is_flag', t.bool_node(node.is_flag)) obj.add_terse('is_multi_allowed', t.bool_node(node.is_multi_allowed)) obj.add('pos', t.pos(node.pos)) obj.add_terse('fields', t.array_node_enum_field(node.fields)) obj.add('comments', t.array_node_comment(node.comments)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) obj.add_terse('typ', t.type_node(node.typ)) return obj } fn (t Tree) enum_field(node ast.EnumField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EnumField')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('has_expr', t.bool_node(node.has_expr)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('next_comments', t.array_node_comment(node.next_comments)) return obj } fn (t Tree) interface_decl(node ast.InterfaceDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('InterfaceDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add('mut_pos', t.number_node(node.mut_pos)) obj.add_terse('field_names', t.array_node_string(node.field_names)) obj.add_terse('methods', t.array_node_fn_decl(node.methods)) obj.add_terse('fields', t.array_node_struct_field(node.fields)) obj.add('pre_comments', t.array_node_comment(node.pre_comments)) obj.add('name_pos', t.pos(node.name_pos)) obj.add_terse('language', t.enum_node(node.language)) obj.add('pos', t.pos(node.pos)) obj.add('are_embeds_expanded', t.bool_node(node.are_embeds_expanded)) obj.add_terse('embeds', t.array_node_interface_embedding(node.embeds)) obj.add_terse('attrs', t.array_node_attr(node.attrs)) return obj } fn (t Tree) interface_embedding(node ast.InterfaceEmbedding) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('InterfaceEmbedding')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) attr(node ast.Attr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Attr')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('has_arg', t.bool_node(node.has_arg)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add_terse('ct_expr', t.expr(node.ct_expr)) obj.add_terse('ct_opt', t.bool_node(node.ct_opt)) obj.add_terse('ct_evaled', t.bool_node(node.ct_evaled)) obj.add_terse('ct_skip', t.bool_node(node.ct_skip)) obj.add_terse('arg', t.string_node(node.arg)) return obj } fn (t Tree) hash_stmt(node ast.HashStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('HashStmt')) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('val', t.string_node(node.val)) obj.add_terse('kind', t.string_node(node.kind)) obj.add_terse('main', t.string_node(node.main)) obj.add_terse('msg', t.string_node(node.msg)) obj.add_terse('ct_conds', t.array_node_expr(node.ct_conds)) obj.add_terse('source_file', t.string_node(node.source_file)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) comptime_for(node ast.ComptimeFor) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ComptimeFor')) obj.add_terse('val_var', t.string_node(node.val_var)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add('pos', t.pos(node.pos)) obj.add('typ_pos', t.pos(node.pos)) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) return obj } fn (t Tree) global_decl(node ast.GlobalDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('GlobalDecl')) obj.add_terse('mod', t.string_node(node.mod)) obj.add('pos', t.pos(node.pos)) obj.add_terse('is_block', t.bool_node(node.is_block)) obj.add_terse('fields', t.array_node_global_field(node.fields)) obj.add('end_comments', t.array_node_comment(node.end_comments)) obj.add('attrs', t.array_node_attr(node.attrs)) return obj } fn (t Tree) global_field(node ast.GlobalField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('GlobalField')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('has_expr', t.bool_node(node.has_expr)) obj.add_terse('is_markused', t.bool_node(node.is_markused)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) obj.add('typ_pos', t.pos(node.typ_pos)) return obj } fn (t Tree) defer_stmt(node ast.DeferStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('DeferStmt')) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add_terse('defer_vars', t.array_node_ident(node.defer_vars)) obj.add_terse('ifdef', t.string_node(node.ifdef)) obj.add('idx_in_fn', t.number_node(node.idx_in_fn)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) type_decl(node ast.TypeDecl) &Node { match node { ast.AliasTypeDecl { return t.alias_type_decl(node) } ast.FnTypeDecl { return t.fn_type_decl(node) } ast.SumTypeDecl { return t.sum_type_decl(node) } } } fn (t Tree) alias_type_decl(node ast.AliasTypeDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AliasTypeDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('parent_type', t.type_node(node.parent_type)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) sum_type_decl(node ast.SumTypeDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SumTypeDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add('pos', t.pos(node.pos)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('generic_types', t.array_node_type(node.generic_types)) obj.add('comments', t.array_node_comment(node.comments)) obj.add_terse('variants', t.array_node_type_expr(node.variants)) obj.add('name_pos', t.pos(node.name_pos)) return obj } fn (t Tree) fn_type_decl(node ast.FnTypeDecl) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('FnTypeDecl')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_pub', t.bool_node(node.is_pub)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) arg(node ast.Param) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Param')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) return obj } fn (t Tree) goto_label(node ast.GotoLabel) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('GotoLabel')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_used', t.bool_node(node.is_used)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) goto_stmt(node ast.GotoStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('GotoStmt')) obj.add_terse('name', t.string_node(node.name)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) assign_stmt(node ast.AssignStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AssignStmt')) obj.add_terse('op', t.token_node(node.op)) obj.add_terse('left', t.array_node_expr(node.left)) obj.add_terse('left_types', t.array_node_type(node.left_types)) obj.add_terse('right', t.array_node_expr(node.right)) obj.add_terse('right_types', t.array_node_type(node.left_types)) obj.add_terse('is_static', t.bool_node(node.is_static)) obj.add_terse('is_volatile', t.bool_node(node.is_volatile)) obj.add_terse('is_simple', t.bool_node(node.is_simple)) obj.add_terse('has_cross_var', t.bool_node(node.has_cross_var)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('end_comments', t.array_node_comment(node.end_comments)) return obj } fn (t Tree) var(node ast.Var) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Var')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('orig_type', t.type_node(node.orig_type)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('is_arg', t.bool_node(node.is_arg)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add('is_used', t.bool_node(node.is_used)) obj.add('is_changed', t.bool_node(node.is_changed)) obj.add_terse('ct_type_var', t.enum_node(node.ct_type_var)) obj.add('is_or', t.bool_node(node.is_or)) obj.add('is_tmp', t.bool_node(node.is_tmp)) obj.add('is_autofree_tmp', t.bool_node(node.is_autofree_tmp)) obj.add('is_auto_deref', t.bool_node(node.is_auto_deref)) obj.add('is_inherited', t.bool_node(node.is_inherited)) obj.add('has_inherited', t.bool_node(node.has_inherited)) obj.add('is_auto_heap', t.bool_node(node.is_auto_heap)) obj.add('is_stack_obj', t.bool_node(node.is_stack_obj)) obj.add_terse('share', t.enum_node(node.share)) obj.add('pos', t.pos(node.pos)) obj.add_terse('smartcasts', t.array_node_type(node.smartcasts)) return obj } fn (t Tree) return_(node ast.Return) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Return')) obj.add_terse('exprs', t.array_node_expr(node.exprs)) obj.add_terse('types', t.array_node_type(node.types)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) for_c_stmt(node ast.ForCStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ForCStmt')) obj.add_terse('has_init', t.bool_node(node.has_init)) obj.add_terse('init', t.stmt(node.init)) obj.add_terse('has_cond', t.bool_node(node.has_cond)) obj.add_terse('cond', t.expr(node.cond)) obj.add_terse('has_inc', t.bool_node(node.has_inc)) obj.add_terse('inc', t.stmt(node.inc)) obj.add_terse('is_multi', t.bool_node(node.is_multi)) obj.add_terse('label', t.string_node(node.label)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('scope', t.number_node(int(node.scope))) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) return obj } fn (t Tree) for_stmt(node ast.ForStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ForStmt')) obj.add_terse('cond', t.expr(node.cond)) obj.add_terse('is_inf', t.bool_node(node.is_inf)) obj.add_terse('label', t.string_node(node.label)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('scope', t.number_node(int(node.scope))) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) return obj } fn (t Tree) for_in_stmt(node ast.ForInStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ForInStmt')) obj.add_terse('key_var', t.string_node(node.key_var)) obj.add_terse('val_var', t.string_node(node.val_var)) obj.add_terse('cond', t.expr(node.cond)) obj.add_terse('is_range', t.bool_node(node.is_range)) obj.add_terse('high', t.expr(node.high)) obj.add_terse('key_type', t.type_node(node.key_type)) obj.add_terse('val_type', t.type_node(node.val_type)) obj.add_terse('cond_type', t.type_node(node.cond_type)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add_terse('val_is_mut', t.bool_node(node.val_is_mut)) obj.add_terse('val_is_ref', t.bool_node(node.val_is_ref)) obj.add_terse('label', t.string_node(node.label)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('scope', t.number_node(int(node.scope))) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) return obj } fn (t Tree) branch_stmt(node ast.BranchStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('BranchStmt')) obj.add_terse('kind', t.token_node(node.kind)) obj.add_terse('label', t.string_node(node.label)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) assert_stmt(node ast.AssertStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AssertStmt')) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('is_used', t.bool_node(node.is_used)) if node.extra !is ast.EmptyExpr { obj.add_terse('extra', t.expr(node.extra)) obj.add('extra_pos', t.pos(node.extra_pos)) } obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) block(node ast.Block) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Block')) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add_terse('is_unsafe', t.bool_node(node.is_unsafe)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) comptime_call(node ast.ComptimeCall) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ComptimeCall')) obj.add_terse('method_name', t.string_node(node.method_name)) obj.add_terse('left', t.expr(node.left)) obj.add_terse('is_vweb', t.bool_node(node.is_vweb)) obj.add_terse('vweb_tmpl', t.string_node(node.vweb_tmpl.path)) obj.add_terse('args_var', t.string_node(node.args_var)) obj.add_terse('has_parens', t.bool_node(node.has_parens)) obj.add_terse('is_embed', t.bool_node(node.is_embed)) obj.add_terse('embed_file', t.embed_file(node.embed_file)) obj.add('method_pos', t.pos(node.method_pos)) obj.add_terse('left_type', t.type_node(node.left_type)) obj.add_terse('result_type', t.type_node(node.result_type)) obj.add('scope', t.scope(node.scope)) obj.add_terse('env_value', t.string_node(node.env_value)) obj.add('pos', t.pos(node.pos)) obj.add_terse('args', t.array_node_call_arg(node.args)) return obj } fn (t Tree) comptime_selector(node ast.ComptimeSelector) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ComptimeSelector')) obj.add_terse('has_parens', t.bool_node(node.has_parens)) obj.add_terse('left', t.expr(node.left)) obj.add_terse('field_expr', t.expr(node.field_expr)) obj.add_terse('left_type', t.type_node(node.left_type)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) expr_stmt(node ast.ExprStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ExprStmt')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('is_expr', t.bool_node(node.is_expr)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) return obj } // expr fn (t Tree) expr(expr ast.Expr) &Node { match expr { ast.IntegerLiteral { return t.integer_literal(expr) } ast.FloatLiteral { return t.float_literal(expr) } ast.StringLiteral { return t.string_literal(expr) } ast.CharLiteral { return t.char_literal(expr) } ast.BoolLiteral { return t.bool_literal(expr) } ast.StringInterLiteral { return t.string_inter_literal(expr) } ast.EnumVal { return t.enum_val(expr) } ast.Assoc { return t.assoc(expr) } ast.AtExpr { return t.at_expr(expr) } ast.CastExpr { return t.cast_expr(expr) } ast.AsCast { return t.as_cast(expr) } ast.TypeNode { return t.type_expr(expr) } ast.SizeOf { return t.size_of(expr) } ast.IsRefType { return t.is_ref_type(expr) } ast.PrefixExpr { return t.prefix_expr(expr) } ast.InfixExpr { return t.infix_expr(expr) } ast.IndexExpr { return t.index_expr(expr) } ast.PostfixExpr { return t.postfix_expr(expr) } ast.SelectorExpr { return t.selector_expr(expr) } ast.RangeExpr { return t.range_expr(expr) } ast.IfExpr { return t.if_expr(expr) } ast.Ident { return t.ident(expr) } ast.CallExpr { return t.call_expr(expr) } ast.OrExpr { return t.or_expr(expr) } ast.StructInit { return t.struct_init(expr) } ast.ArrayInit { return t.array_init(expr) } ast.MapInit { return t.map_init(expr) } ast.None { return t.none_expr(expr) } ast.ParExpr { return t.par_expr(expr) } ast.IfGuardExpr { return t.if_guard_expr(expr) } ast.MatchExpr { return t.match_expr(expr) } ast.ConcatExpr { return t.concat_expr(expr) } ast.TypeOf { return t.type_of(expr) } ast.Likely { return t.likely(expr) } ast.SqlExpr { return t.sql_expr(expr) } ast.ComptimeCall { return t.comptime_call(expr) } ast.ComptimeSelector { return t.comptime_selector(expr) } ast.LockExpr { return t.lock_expr(expr) } ast.UnsafeExpr { return t.unsafe_expr(expr) } ast.ChanInit { return t.chan_init(expr) } ast.SelectExpr { return t.select_expr(expr) } ast.Comment { return t.comment(expr) } ast.AnonFn { return t.anon_fn(expr) } ast.ArrayDecompose { return t.array_decompose(expr) } ast.GoExpr { return t.go_expr(expr) } ast.OffsetOf { return t.offset_of(expr) } ast.DumpExpr { return t.dump_expr(expr) } ast.NodeError { return t.node_error(expr) } ast.EmptyExpr { return t.empty_expr(expr) } ast.Nil { return t.nil_expr(expr) } else { // println('unknown expr') return t.null_node() } } } fn (t Tree) integer_literal(node ast.IntegerLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IntegerLiteral')) obj.add_terse('val', t.string_node(node.val)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) float_literal(node ast.FloatLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('FloatLiteral')) obj.add_terse('val', t.string_node(node.val)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) string_literal(node ast.StringLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StringLiteral')) obj.add_terse('val', t.string_node(node.val)) obj.add_terse('is_raw', t.bool_node(node.is_raw)) obj.add_terse('language', t.enum_node(node.language)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) char_literal(node ast.CharLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('CharLiteral')) obj.add_terse('val', t.string_node(node.val)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) bool_literal(node ast.BoolLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('BoolLiteral')) obj.add_terse('val', t.bool_node(node.val)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) string_inter_literal(node ast.StringInterLiteral) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StringInterLiteral')) obj.add_terse('vals', t.array_node_string(node.vals)) obj.add_terse('exprs', t.array_node_expr(node.exprs)) obj.add_terse('expr_types', t.array_node_type(node.expr_types)) obj.add_terse('fwidths', t.array_node_int(node.fwidths)) obj.add_terse('precisions', t.array_node_int(node.precisions)) obj.add_terse('pluss', t.array_node_bool(node.pluss)) obj.add_terse('fills', t.array_node_bool(node.fills)) obj.add_terse('fmt_poss', t.array_node_position(node.fmt_poss)) obj.add_terse('fmts', t.array_node_u8(node.fmts)) obj.add_terse('need_fmts', t.array_node_bool(node.need_fmts)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) enum_val(node ast.EnumVal) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EnumVal')) obj.add_terse('enum_name', t.string_node(node.enum_name)) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('val', t.string_node(node.val)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) assoc(node ast.Assoc) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Assoc')) obj.add_terse('var_name', t.string_node(node.var_name)) obj.add_terse('fields', t.array_node_string(node.fields)) obj.add_terse('exprs', t.array_node_expr(node.exprs)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) obj.add('scope', t.number_node(int(node.scope))) return obj } fn (t Tree) at_expr(node ast.AtExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AtExpr')) obj.add_terse('name', t.string_node(node.name)) obj.add('pos', t.pos(node.pos)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add_terse('val', t.string_node(node.val)) return obj } fn (t Tree) cast_expr(node ast.CastExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('CastExpr')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('ityp', t.number_node(int(node.typ))) obj.add_terse('typname', t.string_node(node.typname)) obj.add_terse('has_arg', t.bool_node(node.has_arg)) obj.add_terse('arg', t.expr(node.arg)) obj.add_terse('expr_type', t.type_node(node.expr_type)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) as_cast(node ast.AsCast) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsCast')) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expr_type', t.type_node(node.expr_type)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) type_expr(node ast.TypeNode) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('TypeNode')) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) size_of(node ast.SizeOf) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SizeOf')) obj.add_terse('guessed_type', t.bool_node(node.guessed_type)) obj.add_terse('is_type', t.bool_node(node.is_type)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) is_ref_type(node ast.IsRefType) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IsRefType')) obj.add_terse('guessed_type', t.bool_node(node.guessed_type)) obj.add_terse('is_type', t.bool_node(node.is_type)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) prefix_expr(node ast.PrefixExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('PrefixExpr')) obj.add_terse('op', t.token_node(node.op)) obj.add_terse('right', t.expr(node.right)) obj.add_terse('right_type', t.type_node(node.right_type)) obj.add_terse('or_block', t.or_expr(node.or_block)) obj.add_terse('is_option', t.bool_node(node.is_option)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) infix_expr(node ast.InfixExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('InfixExpr')) obj.add_terse('op', t.token_node(node.op)) obj.add_terse('left', t.expr(node.left)) obj.add_terse('left_type', t.type_node(node.left_type)) obj.add_terse('right', t.expr(node.right)) obj.add_terse('right_type', t.type_node(node.right_type)) obj.add_terse('promoted_type', t.type_node(node.promoted_type)) obj.add('auto_locked', t.string_node(node.auto_locked)) obj.add_terse('or_block', t.or_expr(node.or_block)) obj.add_terse('is_stmt', t.bool_node(node.is_stmt)) obj.add_terse('ct_left_value_evaled', t.bool_node(node.ct_left_value_evaled)) obj.add_terse('ct_left_value', t.comptime_expr_value(node.ct_left_value)) obj.add_terse('ct_right_value_evaled', t.bool_node(node.ct_right_value_evaled)) obj.add_terse('ct_right_value', t.comptime_expr_value(node.ct_right_value)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) index_expr(node ast.IndexExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IndexExpr')) obj.add_terse('left', t.expr(node.left)) obj.add_terse('left_type', t.type_node(node.left_type)) obj.add_terse('index', t.expr(node.index)) obj.add_terse('is_setter', t.bool_node(node.is_setter)) obj.add_terse('is_direct', t.bool_node(node.is_direct)) obj.add_terse('or_expr', t.or_expr(node.or_expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) postfix_expr(node ast.PostfixExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('PostfixExpr')) obj.add_terse('op', t.token_node(node.op)) obj.add_terse('expr', t.expr(node.expr)) obj.add('auto_locked', t.string_node(node.auto_locked)) obj.add('pos', t.pos(node.pos)) obj.add('is_c2v_prefix', t.bool_node(node.is_c2v_prefix)) return obj } fn (t Tree) selector_expr(node ast.SelectorExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SelectorExpr')) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('expr_type', t.type_node(node.expr_type)) obj.add_terse('field_name', t.string_node(node.field_name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('name_type', t.type_node(node.name_type)) obj.add_terse('or_block', t.or_expr(node.or_block)) obj.add_terse('gkind_field', t.enum_node(node.gkind_field)) obj.add_terse('from_embed_types', t.array_node_type(node.from_embed_types)) obj.add_terse('has_hidden_receiver', t.bool_node(node.has_hidden_receiver)) obj.add_terse('next_token', t.token_node(node.next_token)) obj.add('pos', t.pos(node.pos)) obj.add('scope', t.number_node(int(node.scope))) return obj } fn (t Tree) range_expr(node ast.RangeExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('RangeExpr')) obj.add_terse('low', t.expr(node.low)) obj.add_terse('high', t.expr(node.high)) obj.add_terse('has_high', t.bool_node(node.has_high)) obj.add_terse('has_low', t.bool_node(node.has_low)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) if_expr(node ast.IfExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IfExpr')) obj.add_terse('is_comptime', t.bool_node(node.is_comptime)) obj.add_terse('tok_kind', t.token_node(node.tok_kind)) obj.add_terse('branches', t.array_node_if_branch(node.branches)) obj.add_terse('left', t.expr(node.left)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('has_else', t.bool_node(node.has_else)) obj.add_terse('is_expr', t.bool_node(node.is_expr)) obj.add('pos', t.pos(node.pos)) obj.add('post_comments', t.array_node_comment(node.post_comments)) return obj } fn (t Tree) if_branch(node ast.IfBranch) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IfBranch')) obj.add_terse('cond', t.expr(node.cond)) obj.add('pos', t.pos(node.pos)) obj.add('body_pos', t.pos(node.body_pos)) obj.add_terse('pkg_exist', t.bool_node(node.pkg_exist)) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add('scope', t.number_node(int(node.scope))) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) ident(node ast.Ident) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Ident')) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('language', t.enum_node(node.language)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add_terse('comptime', t.bool_node(node.comptime)) obj.add_terse('tok_kind', t.token_node(node.tok_kind)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add_terse('info', t.ident_info(node.info)) obj.add('pos', t.pos(node.pos)) obj.add('mut_pos', t.pos(node.mut_pos)) obj.add('obj', t.scope_object(node.obj)) obj.add('scope', t.number_node(int(node.scope))) obj.add_terse('or_expr', t.or_expr(node.or_expr)) return obj } fn (t Tree) ident_info(node ast.IdentInfo) &Node { match node { ast.IdentVar { return t.ident_var(node) } ast.IdentFn { return t.ident_fn(node) } } } fn (t Tree) ident_var(node ast.IdentVar) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IdentVar')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add_terse('is_static', t.bool_node(node.is_static)) obj.add_terse('is_volatile', t.bool_node(node.is_volatile)) obj.add_terse('is_option', t.bool_node(node.is_option)) obj.add_terse('share', t.enum_node(node.share)) return obj } fn (t Tree) ident_fn(node ast.IdentFn) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IdentFn')) obj.add_terse('typ', t.type_node(node.typ)) return obj } fn (t Tree) call_expr(node ast.CallExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('CallExpr')) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('const_name', t.string_node(node.const_name)) obj.add_terse('language', t.enum_node(node.language)) obj.add_terse('left_type', t.type_node(node.left_type)) obj.add_terse('receiver_type', t.type_node(node.receiver_type)) obj.add_terse('return_type', t.type_node(node.return_type)) obj.add_terse('fn_var_type', t.type_node(node.fn_var_type)) obj.add_terse('left', t.expr(node.left)) obj.add_terse('is_method', t.bool_node(node.is_method)) obj.add_terse('is_field', t.bool_node(node.is_field)) obj.add_terse('is_fn_var', t.bool_node(node.is_fn_var)) obj.add_terse('is_fn_a_const', t.bool_node(node.is_fn_a_const)) obj.add('is_keep_alive', t.bool_node(node.is_keep_alive)) obj.add_terse('is_noreturn', t.bool_node(node.is_noreturn)) obj.add_terse('is_ctor_new', t.bool_node(node.is_ctor_new)) obj.add('should_be_skipped', t.bool_node(node.should_be_skipped)) obj.add_terse('free_receiver', t.bool_node(node.free_receiver)) obj.add('scope', t.number_node(int(node.scope))) obj.add_terse('args', t.array_node_call_arg(node.args)) obj.add_terse('expected_arg_types', t.array_node_type(node.expected_arg_types)) obj.add_terse('concrete_types', t.array_node_type(node.concrete_types)) obj.add_terse('or_block', t.or_expr(node.or_block)) obj.add('concrete_list_pos', t.pos(node.concrete_list_pos)) obj.add_terse('from_embed_types', t.array_node_type(node.from_embed_types)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) obj.add('name_pos', t.pos(node.name_pos)) return obj } fn (t Tree) call_arg(node ast.CallArg) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('CallArg')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add_terse('share', t.enum_node(node.share)) obj.add_terse('expr', t.expr(node.expr)) obj.add('is_tmp_autofree', t.bool_node(node.is_tmp_autofree)) obj.add('pos', t.pos(node.pos)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) or_expr(node ast.OrExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('OrExpr')) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add_terse('kind', t.enum_node(node.kind)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) struct_init(node ast.StructInit) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StructInit')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('no_keys', t.bool_node(node.no_keys)) obj.add_terse('is_short_syntax', t.bool_node(node.is_short_syntax)) obj.add_terse('is_anon', t.bool_node(node.is_anon)) obj.add_terse('unresolved', t.bool_node(node.unresolved)) obj.add_terse('has_update_expr', t.bool_node(node.has_update_expr)) obj.add_terse('update_expr', t.expr(node.update_expr)) obj.add_terse('update_expr_type', t.type_node(node.update_expr_type)) obj.add('pos', t.pos(node.pos)) obj.add('name_pos', t.pos(node.name_pos)) obj.add('update_expr_comments', t.array_node_comment(node.update_expr_comments)) obj.add_terse('fields', t.array_node_struct_init_field(node.fields)) obj.add_terse('embeds', t.array_node_struct_init_embed(node.embeds)) obj.add('pre_comments', t.array_node_comment(node.pre_comments)) return obj } fn (t Tree) struct_init_field(node ast.StructInitField) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StructInitField')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expected_type', t.type_node(node.expected_type)) obj.add_terse('parent_type', t.type_node(node.parent_type)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('next_comments', t.array_node_comment(node.next_comments)) obj.add('pos', t.pos(node.pos)) obj.add('name_pos', t.pos(node.name_pos)) return obj } fn (t Tree) struct_init_embed(node ast.StructInitEmbed) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('StructInitEmbed')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expected_type', t.type_node(node.expected_type)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('next_comments', t.array_node_comment(node.next_comments)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) array_init(node ast.ArrayInit) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ArrayInit')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('elem_type', t.type_node(node.elem_type)) obj.add_terse('exprs', t.array_node_expr(node.exprs)) obj.add('ecmnts', t.two_dimension_comment(node.ecmnts)) obj.add('pre_cmnts', t.array_node_comment(node.pre_cmnts)) obj.add('elem_type_pos', t.pos(node.elem_type_pos)) obj.add_terse('is_fixed', t.bool_node(node.is_fixed)) obj.add_terse('has_val', t.bool_node(node.has_val)) obj.add_terse('mod', t.string_node(node.mod)) obj.add_terse('len_expr', t.expr(node.len_expr)) obj.add_terse('cap_expr', t.expr(node.cap_expr)) obj.add_terse('default_expr', t.expr(node.default_expr)) obj.add_terse('has_len', t.bool_node(node.has_len)) obj.add_terse('has_cap', t.bool_node(node.has_cap)) obj.add_terse('has_default', t.bool_node(node.has_default)) obj.add_terse('has_index', t.bool_node(node.has_index)) obj.add_terse('expr_types', t.array_node_type(node.expr_types)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) map_init(node ast.MapInit) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('MapInit')) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('key_type', t.type_node(node.key_type)) obj.add_terse('value_type', t.type_node(node.value_type)) obj.add_terse('keys', t.array_node_expr(node.keys)) obj.add_terse('vals', t.array_node_expr(node.vals)) obj.add_terse('val_types', t.array_node_type(node.val_types)) obj.add('comments', t.two_dimension_comment(node.comments)) obj.add('pre_cmnts', t.array_node_comment(node.pre_cmnts)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) none_expr(node ast.None) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('None')) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) par_expr(node ast.ParExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ParExpr')) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) if_guard_expr(node ast.IfGuardExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IfGuardExpr')) obj.add_terse('vars', t.array_node_if_guard_var(node.vars)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('expr_type', t.type_node(node.expr_type)) return obj } fn (t Tree) if_guard_var(node ast.IfGuardVar) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('IfGuardVar')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_mut', t.bool_node(node.is_mut)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) match_expr(node ast.MatchExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('MatchExpr')) obj.add_terse('tok_kind', t.token_node(node.tok_kind)) obj.add_terse('cond', t.expr(node.cond)) obj.add_terse('cond_type', t.type_node(node.cond_type)) obj.add_terse('return_type', t.type_node(node.return_type)) obj.add_terse('expected_type', t.type_node(node.expected_type)) obj.add_terse('is_sum_type', t.bool_node(node.is_sum_type)) obj.add_terse('is_expr', t.bool_node(node.is_expr)) obj.add('pos', t.pos(node.pos)) obj.add_terse('branches', t.array_node_match_branch(node.branches)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) match_branch(node ast.MatchBranch) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('MatchBranch')) obj.add('ecmnts', t.two_dimension_comment(node.ecmnts)) obj.add_terse('stmts', t.array_node_stmt(node.stmts)) obj.add_terse('is_else', t.bool_node(node.is_else)) obj.add('pos', t.pos(node.pos)) obj.add_terse('post_comments', t.array_node_comment(node.post_comments)) obj.add('branch_pos', t.pos(node.branch_pos)) obj.add_terse('exprs', t.array_node_expr(node.exprs)) obj.add('scope', t.number_node(int(node.scope))) return obj } fn (t Tree) concat_expr(node ast.ConcatExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ConcatExpr')) obj.add_terse('vals', t.array_node_expr(node.vals)) obj.add_terse('return_type', t.type_node(node.return_type)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) type_of(node ast.TypeOf) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('TypeOf')) obj.add_terse('is_type', t.bool_node(node.is_type)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('expr', t.expr(node.expr)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) likely(node ast.Likely) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Likely')) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('is_likely', t.bool_node(node.is_likely)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) sql_expr(node ast.SqlExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SqlExpr')) obj.add_terse('type', t.type_node(node.typ)) obj.add_terse('is_count', t.bool_node(node.is_count)) obj.add_terse('db_expr', t.expr(node.db_expr)) obj.add_terse('table_expr', t.type_expr(node.table_expr)) obj.add_terse('has_where', t.bool_node(node.has_where)) obj.add_terse('where_expr', t.expr(node.where_expr)) obj.add_terse('has_order', t.bool_node(node.has_order)) obj.add_terse('order_expr', t.expr(node.order_expr)) obj.add_terse('has_desc', t.bool_node(node.has_desc)) obj.add_terse('is_array', t.bool_node(node.is_array)) obj.add_terse('or_expr', t.or_expr(node.or_expr)) obj.add('pos', t.pos(node.pos)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('has_limit', t.bool_node(node.has_limit)) obj.add_terse('limit_expr', t.expr(node.limit_expr)) obj.add_terse('has_offset', t.bool_node(node.has_offset)) obj.add_terse('offset_expr', t.expr(node.offset_expr)) obj.add_terse('fields', t.array_node_struct_field(node.fields)) sub_struct_map := new_object() for key, val in node.sub_structs { sub_struct_map.add_terse(key.str(), t.sql_expr(val)) } obj.add_terse('sub_structs', sub_struct_map) return obj } fn (t Tree) sql_stmt(node ast.SqlStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SqlStmt')) obj.add_terse('db_expr_type', t.type_node(node.db_expr_type)) obj.add_terse('db_expr', t.expr(node.db_expr)) obj.add_terse('or_expr', t.or_expr(node.or_expr)) obj.add('pos', t.pos(node.pos)) obj.add_terse('lines', t.array_node_sql_stmt_line(node.lines)) return obj } fn (t Tree) sql_stmt_line(node ast.SqlStmtLine) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SqlStmtLine')) obj.add_terse('kind', t.enum_node(node.kind)) obj.add_terse('table_expr', t.type_expr(node.table_expr)) obj.add_terse('object_var_name', t.string_node(node.object_var_name)) obj.add_terse('where_expr', t.expr(node.where_expr)) obj.add_terse('fields', t.array_node_struct_field(node.fields)) obj.add_terse('updated_columns', t.array_node_string(node.updated_columns)) obj.add_terse('update_exprs', t.array_node_expr(node.update_exprs)) obj.add('pos', t.pos(node.pos)) sub_struct_map := new_object() for key, val in node.sub_structs { sub_struct_map.add_terse(key.str(), t.sql_stmt_line(val)) } obj.add_terse('sub_structs', sub_struct_map) return obj } fn (t Tree) lock_expr(expr ast.LockExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('LockExpr')) obj.add_terse('is_expr', t.bool_node(expr.is_expr)) obj.add_terse('typ', t.type_node(expr.typ)) obj.add('pos', t.pos(expr.pos)) obj.add_terse('stmts', t.array_node_stmt(expr.stmts)) obj.add_terse('lockeds', t.array_node_expr(expr.lockeds)) obj.add_terse('r_lock', t.array_node_bool(expr.is_rlock)) return obj } fn (t Tree) unsafe_expr(expr ast.UnsafeExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('UnsafeExpr')) obj.add_terse('expr', t.expr(expr.expr)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) chan_init(expr ast.ChanInit) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ChanInit')) obj.add_terse('has_cap', t.bool_node(expr.has_cap)) obj.add_terse('cap_expr', t.expr(expr.cap_expr)) obj.add_terse('typ', t.type_node(expr.typ)) obj.add_terse('elem_type', t.type_node(expr.elem_type)) obj.add('pos', t.pos(expr.pos)) obj.add('elem_type_pos', t.pos(expr.elem_type_pos)) return obj } fn (t Tree) select_expr(expr ast.SelectExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SelectExpr')) obj.add_terse('branches', t.array_node_select_branch(expr.branches)) obj.add_terse('is_expr', t.bool_node(expr.is_expr)) obj.add_terse('has_exception', t.bool_node(expr.has_exception)) obj.add_terse('expected_type', t.type_node(expr.expected_type)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) select_branch(expr ast.SelectBranch) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('SelectBranch')) obj.add_terse('stmt', t.stmt(expr.stmt)) obj.add_terse('stmts', t.array_node_stmt(expr.stmts)) obj.add('pos', t.pos(expr.pos)) obj.add('comment', t.comment(expr.comment)) obj.add_terse('is_else', t.bool_node(expr.is_else)) obj.add_terse('is_timeout', t.bool_node(expr.is_timeout)) obj.add('post_comments', t.array_node_comment(expr.post_comments)) return obj } fn (t Tree) array_decompose(expr ast.ArrayDecompose) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('ArrayDecompose')) obj.add_terse('expr', t.expr(expr.expr)) obj.add_terse('expr_type', t.type_node(expr.expr_type)) obj.add_terse('arg_type', t.type_node(expr.arg_type)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) go_expr(expr ast.GoExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('GoExpr')) obj.add_terse('call_expr', t.call_expr(expr.call_expr)) obj.add_terse('is_expr', t.bool_node(expr.is_expr)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) offset_of(expr ast.OffsetOf) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('OffsetOf')) obj.add_terse('struct_type', t.type_node(expr.struct_type)) obj.add_terse('field', t.string_node('field')) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) dump_expr(expr ast.DumpExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('DumpExpr')) obj.add_terse('expr', t.expr(expr.expr)) obj.add_terse('expr_type', t.type_node(expr.expr_type)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) node_error(expr ast.NodeError) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('NodeError')) obj.add_terse('idx', t.number_node(expr.idx)) obj.add('pos', t.pos(expr.pos)) return obj } fn (t Tree) empty_expr(expr ast.EmptyExpr) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EmptyExpr')) // obj.add('x', t.number_node(expr.x)) return obj } fn (t Tree) empty_stmt(node ast.EmptyStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('EmptyStmt')) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) nil_expr(node ast.Nil) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('Nil')) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) asm_stmt(node ast.AsmStmt) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmStmt')) obj.add_terse('arch', t.enum_node(node.arch)) obj.add_terse('is_basic', t.bool_node(node.is_basic)) obj.add_terse('is_volatile', t.bool_node(node.is_volatile)) obj.add_terse('is_goto', t.bool_node(node.is_goto)) obj.add('scope', t.scope(node.scope)) // obj.add('scope', t.number_node(int(node.scope))) obj.add('pos', t.pos(node.pos)) obj.add_terse('clobbered', t.array_node_asm_clobbered(node.clobbered)) obj.add_terse('templates', t.array_node_asm_template(node.templates)) obj.add_terse('output', t.array_node_asm_io(node.output)) obj.add_terse('input', t.array_node_asm_io(node.input)) obj.add_terse('global_labels', t.array_node_string(node.global_labels)) obj.add_terse('local_labels', t.array_node_string(node.local_labels)) return obj } fn (t Tree) asm_register(node ast.AsmRegister) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmRegister')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('typ', t.type_node(node.typ)) obj.add_terse('size', t.number_node(node.size)) return obj } fn (t Tree) asm_template(node ast.AsmTemplate) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmTemplate')) obj.add_terse('name', t.string_node(node.name)) obj.add_terse('is_label', t.bool_node(node.is_label)) obj.add_terse('is_directive', t.bool_node(node.is_directive)) obj.add_terse('args', t.array_node_asm_arg(node.args)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) asm_addressing(node ast.AsmAddressing) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmAddressing')) obj.add_terse('scale', t.number_node(node.scale)) obj.add_terse('mode', t.enum_node(node.mode)) obj.add_terse('segment', t.string_node(node.segment)) obj.add_terse('displacement', t.asm_arg(node.displacement)) obj.add_terse('base', t.asm_arg(node.base)) obj.add_terse('index', t.asm_arg(node.index)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) asm_arg(node ast.AsmArg) &Node { match node { ast.AsmAddressing { return t.asm_addressing(node) } ast.AsmAlias { return t.asm_alias(node) } ast.AsmDisp { return t.asm_disp(node) } ast.AsmRegister { return t.asm_register(node) } ast.BoolLiteral { return t.bool_literal(node) } ast.CharLiteral { return t.char_literal(node) } ast.FloatLiteral { return t.float_literal(node) } ast.IntegerLiteral { return t.integer_literal(node) } string { return t.string_node(node) } } } fn (t Tree) asm_alias(node ast.AsmAlias) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmAlias')) obj.add_terse('name', t.string_node(node.name)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) asm_disp(node ast.AsmDisp) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmDisp')) obj.add_terse('val', t.string_node(node.val)) obj.add('pos', t.pos(node.pos)) return obj } fn (t Tree) asm_clobbered(node ast.AsmClobbered) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmClobbered')) obj.add_terse('reg', t.asm_register(node.reg)) obj.add('comments', t.array_node_comment(node.comments)) return obj } fn (t Tree) asm_io(node ast.AsmIO) &Node { mut obj := new_object() obj.add_terse('ast_type', t.string_node('AsmIO')) obj.add_terse('alias', t.string_node(node.alias)) obj.add_terse('constraint', t.string_node(node.constraint)) obj.add_terse('expr', t.expr(node.expr)) obj.add_terse('typ', t.type_node(node.typ)) obj.add('comments', t.array_node_comment(node.comments)) obj.add('pos', t.pos(node.pos)) return obj } // do not support yet by vlang // fn (t Tree) array_node1[T](nodes []T, method_name string) &Node { // mut arr := new_array() // // call method dynamically, V do not support yet // // error: todo: not a string literal // // for node in nodes { // // arr.add_item(t.$method_name(node)) // // } // // temp // $for method in Tree.methods { // if method.name == method_name { // for node in nodes { // res := t.$method(node) // arr.add_item(res) // TODO,waiting for bug fixed // } // } // } // return arr // } // do not support yet by vlang // fn (t Tree) array_node2[T](nodes []T) &Node { // mut arr := new_array() // for node in nodes { // match node { // string { // arr.add_item(t.string_node(node)) // } // ast.Comment { // arr.add_item(t.comment(node)) // } // ast.ConstField { // arr.add_item(t.const_field(node)) // } // else { // panic('unknown array type') // } // } // } // return arr // } // list all the different type of array node,temporarily fn (t Tree) array_node_string(nodes []string) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.string_node(node)) } return arr } fn (t Tree) array_node_position(nodes []token.Pos) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.pos(node)) } return arr } fn (t Tree) array_node_if_branch(nodes []ast.IfBranch) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.if_branch(node)) } return arr } fn (t Tree) array_node_fn_decl(nodes []ast.FnDecl) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.fn_decl(node)) } return arr } fn (t Tree) array_node_generic_fns(nodes []&ast.FnDecl) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.fn_decl(node)) } return arr } fn (t Tree) array_node_embed_file(nodes []ast.EmbeddedFile) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.embed_file(node)) } return arr } fn (t Tree) array_node_attr(nodes []ast.Attr) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.attr(node)) } return arr } fn (t Tree) array_node_scope_struct_field(nodes map[string]ast.ScopeStructField) &Node { mut arr := new_array() for _, node in nodes { arr.add_item(t.scope_struct_field(node)) } return arr } fn (t Tree) array_node_type(nodes []ast.Type) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.type_node(node)) } return arr } fn (t Tree) array_node_type_expr(nodes []ast.TypeNode) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.type_expr(node)) } return arr } fn (t Tree) array_node_import_symbol(nodes []ast.ImportSymbol) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.import_symbol(node)) } return arr } fn (t Tree) array_node_comment(nodes []ast.Comment) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.comment(node)) } return arr } fn (t Tree) array_node_const_field(nodes []ast.ConstField) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.const_field(node)) } return arr } fn (t Tree) array_node_arg(nodes []ast.Param) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.arg(node)) } return arr } fn (t Tree) array_node_stmt(nodes []ast.Stmt) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.stmt(node)) } return arr } fn (t Tree) array_node_defer_stmt(nodes []ast.DeferStmt) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.defer_stmt(node)) } return arr } fn (t Tree) array_node_struct_field(nodes []ast.StructField) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.struct_field(node)) } return arr } fn (t Tree) array_node_embed(nodes []ast.Embed) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.embed(node)) } return arr } fn (t Tree) array_node_enum_field(nodes []ast.EnumField) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.enum_field(node)) } return arr } fn (t Tree) array_node_global_field(nodes []ast.GlobalField) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.global_field(node)) } return arr } fn (t Tree) array_node_expr(nodes []ast.Expr) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.expr(node)) } return arr } fn (t Tree) array_node_call_arg(nodes []ast.CallArg) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.call_arg(node)) } return arr } fn (t Tree) array_node_int(nodes []int) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.number_node(node)) } return arr } fn (t Tree) array_node_u8(nodes []u8) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.number_node(node)) } return arr } fn (t Tree) array_node_bool(nodes []bool) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.bool_node(node)) } return arr } fn (t Tree) array_node_struct_init_field(nodes []ast.StructInitField) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.struct_init_field(node)) } return arr } fn (t Tree) array_node_if_guard_var(nodes []ast.IfGuardVar) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.if_guard_var(node)) } return arr } fn (t Tree) array_node_struct_init_embed(nodes []ast.StructInitEmbed) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.struct_init_embed(node)) } return arr } fn (t Tree) array_node_match_branch(nodes []ast.MatchBranch) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.match_branch(node)) } return arr } fn (t Tree) array_node_ident(nodes []ast.Ident) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.ident(node)) } return arr } fn (t Tree) array_node_select_branch(nodes []ast.SelectBranch) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.select_branch(node)) } return arr } fn (t Tree) array_node_asm_clobbered(nodes []ast.AsmClobbered) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.asm_clobbered(node)) } return arr } fn (t Tree) array_node_asm_template(nodes []ast.AsmTemplate) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.asm_template(node)) } return arr } fn (t Tree) array_node_asm_io(nodes []ast.AsmIO) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.asm_io(node)) } return arr } fn (t Tree) array_node_asm_arg(nodes []ast.AsmArg) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.asm_arg(node)) } return arr } fn (t Tree) array_node_sql_stmt_line(nodes []ast.SqlStmtLine) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.sql_stmt_line(node)) } return arr } fn (t Tree) array_node_interface_embedding(nodes []ast.InterfaceEmbedding) &Node { mut arr := new_array() for node in nodes { arr.add_item(t.interface_embedding(node)) } return arr }