v/vlib/v/parser/parser.v

1213 lines
24 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-12-22 02:34:37 +01:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module parser
2020-04-14 19:32:23 +02:00
import v.scanner
import v.ast
import v.token
import v.table
import v.pref
import v.util
import term
import os
2019-12-22 02:34:37 +01:00
struct Parser {
scanner &scanner.Scanner
2020-04-10 12:01:06 +02:00
file_name string // "/home/user/hello.v"
file_name_dir string // "/home/user"
2019-12-22 02:34:37 +01:00
mut:
tok token.Token
2020-04-18 00:22:03 +02:00
prev_tok token.Token
peek_tok token.Token
table &table.Table
is_c bool
2020-04-16 15:32:11 +02:00
is_js bool
inside_if bool
2020-04-08 19:08:54 +02:00
inside_for bool
inside_fn bool
pref &pref.Preferences
builtin_mod bool // are we in the `builtin` module?
mod string // current module name
attr string
expr_mod string
scope &ast.Scope
global_scope &ast.Scope
imports map[string]string
ast_imports []ast.Import
is_amp bool
returns bool
2020-04-15 16:23:03 +02:00
inside_match bool // to separate `match A { }` from `Struct{}`
inside_match_case bool // to separate `match_expr { }` from `Struct{}`
2020-04-14 17:41:14 +02:00
is_stmt_ident bool // true while the beginning of a statement is an ident/selector
2020-04-16 15:32:11 +02:00
inside_is bool // `is Type`, expecting type
2019-12-22 02:34:37 +01:00
}
// for tests
pub fn parse_stmt(text string, table &table.Table, scope &ast.Scope) ast.Stmt {
2020-02-29 17:51:35 +01:00
s := scanner.new_scanner(text, .skip_comments)
2020-04-15 01:55:51 +02:00
var p := Parser{
2019-12-22 02:34:37 +01:00
scanner: s
table: table
2020-02-07 22:10:48 +01:00
pref: &pref.Preferences{}
scope: scope
global_scope: &ast.Scope{
start_pos: 0
parent: 0
}
2019-12-22 02:34:37 +01:00
}
2020-01-06 16:13:12 +01:00
p.init_parse_fns()
p.read_first_token()
2019-12-28 14:11:05 +01:00
return p.stmt()
2019-12-22 02:34:37 +01:00
}
pub fn parse_file(path string, table &table.Table, comments_mode scanner.CommentsMode, pref &pref.Preferences, global_scope &ast.Scope) ast.File {
2020-02-29 17:51:35 +01:00
// println('parse_file("$path")')
2020-02-29 17:53:04 +01:00
// text := os.read_file(path) or {
// panic(err)
// }
2020-04-15 01:55:51 +02:00
var stmts := []ast.Stmt
var p := Parser{
2020-02-29 17:51:35 +01:00
scanner: scanner.new_scanner_file(path, comments_mode)
table: table
2020-01-01 10:15:05 +01:00
file_name: path
file_name_dir: os.dir(path)
2020-04-05 04:05:09 +02:00
pref: pref
2020-02-16 11:48:29 +01:00
scope: &ast.Scope{
start_pos: 0
parent: 0
}
global_scope: global_scope
}
2020-04-05 04:05:09 +02:00
// comments_mode: comments_mode
2019-12-28 14:11:05 +01:00
p.read_first_token()
for p.tok.kind == .comment {
2020-04-15 01:55:51 +02:00
var stmt := ast.Stmt{} // TODO sum type << bug
com := p.comment()
stmt = com
stmts << stmt
}
2020-04-15 01:55:51 +02:00
var mstmt := ast.Stmt{}
module_decl := p.module_decl()
mstmt = module_decl
stmts << mstmt
// imports
2020-02-29 17:51:35 +01:00
/*
mut imports := []ast.Import
for p.tok.kind == .key_import {
imports << p.import_stmt()
}
*/
// TODO: import only mode
for {
2019-12-28 09:43:22 +01:00
// res := s.scan()
if p.tok.kind == .eof {
2020-02-19 16:12:39 +01:00
// println('EOF, breaking')
break
}
2020-01-06 16:13:12 +01:00
// println('stmt at ' + p.tok.str())
stmts << p.top_stmt()
}
// println('nr stmts = $stmts.len')
2019-12-28 14:11:05 +01:00
// println(stmts[0])
p.scope.end_pos = p.tok.pos
2019-12-30 12:10:46 +01:00
return ast.File{
2020-02-03 07:31:54 +01:00
path: path
mod: module_decl
2020-02-29 17:51:35 +01:00
imports: p.ast_imports
stmts: stmts
scope: p.scope
global_scope: p.global_scope
2019-12-28 14:11:05 +01:00
}
}
/*
struct Queue {
mut:
idx int
mu sync.Mutex
paths []string
table &table.Table
parsed_ast_files []ast.File
}
fn (q mut Queue) run() {
q.mu.lock()
idx := q.idx
if idx >= q.paths.len {
q.mu.unlock()
return
}
q.idx++
q.mu.unlock()
path := q.paths[idx]
file := parse_file(path, q.table, .skip_comments)
q.mu.lock()
q.parsed_ast_files << file
q.mu.unlock()
}
*/
pub fn parse_files(paths []string, table &table.Table, pref &pref.Preferences, global_scope &ast.Scope) []ast.File {
/*
println('\n\n\nparse_files()')
println(paths)
nr_cpus := runtime.nr_cpus()
println('nr_cpus= $nr_cpus')
mut q := &Queue{
paths: paths
table: table
}
for i in 0 .. nr_cpus {
go q.run()
}
time.sleep_ms(100)
return q.parsed_ast_files
*/
// ///////////////
2020-04-15 01:55:51 +02:00
var files := []ast.File
2019-12-30 12:10:46 +01:00
for path in paths {
2020-03-27 08:46:54 +01:00
// println('parse_files $path')
files << parse_file(path, table, .skip_comments, pref, global_scope)
2019-12-30 12:10:46 +01:00
}
return files
}
2020-01-07 01:57:38 +01:00
pub fn (p &Parser) init_parse_fns() {
// p.prefix_parse_fns = make(100, 100, sizeof(PrefixParseFn))
2020-01-06 16:13:12 +01:00
// p.prefix_parse_fns[token.Kind.name] = parse_name
2020-01-07 01:57:38 +01:00
println('')
2020-01-06 16:13:12 +01:00
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) read_first_token() {
2019-12-28 14:11:05 +01:00
// need to call next() twice to get peek token and current token
p.next()
p.next()
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) open_scope() {
p.scope = &ast.Scope{
parent: p.scope
start_pos: p.tok.pos
}
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) close_scope() {
p.scope.end_pos = p.tok.pos
p.scope.parent.children << p.scope
p.scope = p.scope.parent
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) parse_block() []ast.Stmt {
p.open_scope()
// println('parse block')
stmts := p.parse_block_no_scope()
p.close_scope()
// println('nr exprs in block = $exprs.len')
return stmts
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) parse_block_no_scope() []ast.Stmt {
2019-12-31 19:42:16 +01:00
p.check(.lcbr)
2020-04-15 01:55:51 +02:00
var stmts := []ast.Stmt
2020-01-06 16:13:12 +01:00
if p.tok.kind != .rcbr {
for {
2020-01-06 16:13:12 +01:00
stmts << p.stmt()
// p.warn('after stmt(): tok=$p.tok.str()')
if p.tok.kind in [.eof, .rcbr] {
break
}
2019-12-27 13:57:49 +01:00
}
}
2019-12-31 19:42:16 +01:00
p.check(.rcbr)
2019-12-28 14:11:05 +01:00
return stmts
2019-12-27 13:57:49 +01:00
}
2020-04-05 02:08:10 +02:00
/*
fn (p mut Parser) next_with_comment() {
p.tok = p.peek_tok
p.peek_tok = p.scanner.scan()
}
*/
2020-04-16 15:42:40 +02:00
fn (var p Parser) next() {
2020-04-18 00:22:03 +02:00
p.prev_tok = p.tok
p.tok = p.peek_tok
p.peek_tok = p.scanner.scan()
2020-04-05 02:08:10 +02:00
/*
if p.tok.kind==.comment {
p.comments << ast.Comment{text:p.tok.lit, line_nr:p.tok.line_nr}
p.next()
}
*/
2019-12-22 02:34:37 +01:00
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) check(expected token.Kind) {
2020-02-29 17:51:35 +01:00
// for p.tok.kind in [.line_comment, .mline_comment] {
// p.next()
// }
if p.tok.kind != expected {
2020-04-16 12:17:15 +02:00
p.error('unexpected `${p.tok.kind.str()}`, expecting `${expected.str()}`')
2019-12-27 13:57:49 +01:00
}
p.next()
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) check_name() string {
name := p.tok.lit
2019-12-27 13:57:49 +01:00
p.check(.name)
return name
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) top_stmt() ast.Stmt {
match p.tok.kind {
2019-12-31 19:42:16 +01:00
.key_pub {
match p.peek_tok.kind {
.key_const {
return p.const_decl()
}
2019-12-31 19:42:16 +01:00
.key_fn {
return p.fn_decl()
}
2020-04-02 12:53:15 +02:00
.key_struct, .key_union {
2019-12-31 19:42:16 +01:00
return p.struct_decl()
}
2020-04-02 12:53:15 +02:00
.key_interface {
return p.interface_decl()
}
.key_enum {
return p.enum_decl()
}
.key_type {
return p.type_decl()
}
2019-12-31 19:42:16 +01:00
else {
p.error('wrong pub keyword usage')
return ast.Stmt{}
}
}
}
.lsbr {
2020-03-05 16:13:14 +01:00
return p.attribute()
}
.key_interface {
return p.interface_decl()
}
2020-02-29 17:51:35 +01:00
.key_import {
node := p.import_stmt()
p.ast_imports << node
return node[0]
}
.key_global {
return p.global_decl()
}
.key_const {
return p.const_decl()
2019-12-31 19:42:16 +01:00
}
2019-12-28 09:43:22 +01:00
.key_fn {
return p.fn_decl()
}
.key_struct {
return p.struct_decl()
}
.dollar {
return p.comp_if()
2020-01-07 13:10:05 +01:00
}
2020-02-04 09:54:15 +01:00
.hash {
return p.hash()
}
2020-02-12 01:16:38 +01:00
.key_type {
return p.type_decl()
}
.key_enum {
return p.enum_decl()
}
.key_union {
return p.struct_decl()
}
2020-04-05 02:08:10 +02:00
.comment {
return p.comment()
2020-02-18 20:20:15 +01:00
}
2020-01-06 16:13:12 +01:00
else {
2020-04-09 12:46:16 +02:00
if p.pref.is_script && !p.pref.is_test {
p.scanner.add_fn_main_and_rescan()
p.read_first_token()
2020-04-09 12:46:16 +02:00
return p.top_stmt()
} else {
2020-04-09 12:46:16 +02:00
p.error('bad top level statement ' + p.tok.str())
return ast.Stmt{}
}
2019-12-28 09:43:22 +01:00
}
2020-01-06 16:13:12 +01:00
}
}
2020-04-05 02:08:10 +02:00
// TODO [if vfmt]
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) check_comment() ast.Comment {
2020-04-05 02:08:10 +02:00
if p.tok.kind == .comment {
return p.comment()
}
return ast.Comment{}
2020-04-05 02:08:10 +02:00
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) comment() ast.Comment {
pos := p.tok.position()
2020-02-29 17:51:35 +01:00
text := p.tok.lit
p.next()
// p.next_with_comment()
2020-04-05 02:08:10 +02:00
return ast.Comment{
2020-02-29 17:51:35 +01:00
text: text
pos: pos
2020-02-29 17:51:35 +01:00
}
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) stmt() ast.Stmt {
2020-04-14 17:41:14 +02:00
p.is_stmt_ident = p.tok.kind == .name
2020-01-06 16:13:12 +01:00
match p.tok.kind {
2020-03-24 15:44:17 +01:00
.lcbr {
stmts := p.parse_block()
return ast.Block{
stmts: stmts
}
}
.key_assert {
p.next()
assert_pos := p.tok.position()
expr := p.expr(0)
return ast.AssertStmt{
expr: expr
pos: assert_pos
}
}
2020-04-15 01:45:27 +02:00
.key_mut, .key_static, .key_var {
return p.assign_stmt()
2019-12-28 11:02:06 +01:00
}
.key_for {
2020-04-08 19:08:54 +02:00
return p.for_stmt()
}
2020-04-05 02:08:10 +02:00
.comment {
return p.comment()
2020-02-29 17:51:35 +01:00
}
2020-01-06 16:13:12 +01:00
.key_return {
return p.return_stmt()
}
.dollar {
return p.comp_if()
}
.key_continue, .key_break {
tok := p.tok
p.next()
return ast.BranchStmt{
2020-02-18 03:28:39 +01:00
tok: tok
}
}
.key_unsafe {
p.next()
2020-02-26 20:45:03 +01:00
stmts := p.parse_block()
return ast.UnsafeStmt{
stmts: stmts
}
}
2020-03-19 07:59:01 +01:00
.hash {
return p.hash()
}
2020-02-11 10:26:46 +01:00
.key_defer {
p.next()
stmts := p.parse_block()
return ast.DeferStmt{
stmts: stmts
}
}
.key_go {
p.next()
2020-04-03 15:18:17 +02:00
expr := p.expr(0)
// mut call_expr := &ast.CallExpr(0) // TODO
2020-04-03 15:18:17 +02:00
match expr {
ast.CallExpr {
// call_expr = it
2020-04-03 15:18:17 +02:00
}
2020-04-16 12:17:15 +02:00
else {}
2020-04-03 15:18:17 +02:00
}
return ast.GoStmt{
2020-04-03 15:18:17 +02:00
call_expr: expr
}
}
2020-02-17 14:15:42 +01:00
.key_goto {
p.next()
name := p.check_name()
return ast.GotoStmt{
name: name
}
}
2019-12-28 14:11:05 +01:00
else {
2020-01-06 16:13:12 +01:00
// `x := ...`
if p.tok.kind == .name && p.peek_tok.kind in [.decl_assign, .comma] {
return p.assign_stmt()
} else if p.tok.kind == .name && p.peek_tok.kind == .colon {
2020-04-05 04:05:09 +02:00
// `label:`
2020-02-17 14:15:42 +01:00
name := p.check_name()
p.check(.colon)
return ast.GotoLabel{
name: name
}
}
epos := p.tok.position()
expr := p.expr(0)
2019-12-28 14:11:05 +01:00
return ast.ExprStmt{
expr: expr
pos: epos
2019-12-28 14:11:05 +01:00
}
}
}
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) attribute() ast.Attr {
p.check(.lsbr)
if p.tok.kind == .key_if {
p.next()
}
2020-04-15 01:55:51 +02:00
var name := p.check_name()
if p.tok.kind == .colon {
p.next()
if p.tok.kind == .name {
name += p.check_name()
} else if p.tok.kind == .string {
name += p.tok.lit
p.next()
}
}
p.check(.rsbr)
2020-03-05 16:13:14 +01:00
p.attr = name
return ast.Attr{
name: name
}
}
/*
2020-02-02 14:31:54 +01:00
fn (p mut Parser) range_expr(low ast.Expr) ast.Expr {
// ,table.Type) {
if p.tok.kind != .dotdot {
p.next()
}
p.check(.dotdot)
2020-02-07 07:34:18 +01:00
mut high := ast.Expr{}
if p.tok.kind != .rsbr {
high = p.expr(0)
2020-02-07 07:34:18 +01:00
// if typ.typ.kind != .int {
// p.error('non-integer index `$typ.typ.name`')
// }
2020-02-02 14:31:54 +01:00
}
node := ast.RangeExpr{
low: low
high: high
}
return node
}
*/
2019-12-29 06:50:08 +01:00
pub fn (p &Parser) error(s string) {
p.error_with_pos(s, p.tok.position())
}
pub fn (p &Parser) warn(s string) {
p.warn_with_pos(s, p.tok.position())
}
pub fn (p &Parser) error_with_pos(s string, pos token.Position) {
2020-04-15 01:55:51 +02:00
var kind := 'error:'
2020-04-07 00:44:19 +02:00
if p.pref.is_verbose {
print_backtrace()
kind = 'parser error:'
}
ferror := util.formatted_error(kind, s, p.file_name, pos)
eprintln(ferror)
2020-01-02 08:30:15 +01:00
exit(1)
2019-12-30 09:38:12 +01:00
}
pub fn (p &Parser) warn_with_pos(s string, pos token.Position) {
ferror := util.formatted_error('warning:', s, p.file_name, pos)
eprintln(ferror)
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) parse_ident(is_c, is_js bool) ast.Ident {
2020-02-06 17:38:02 +01:00
// p.warn('name ')
pos := p.tok.position()
2020-04-15 01:55:51 +02:00
var name := p.check_name()
if name == '_' {
return ast.Ident{
name: '_'
kind: .blank_ident
pos: pos
}
}
if p.expr_mod.len > 0 {
name = '${p.expr_mod}.$name'
}
2020-04-15 01:55:51 +02:00
var ident := ast.Ident{
kind: .unresolved
2020-02-06 17:38:02 +01:00
name: name
is_c: is_c
2020-04-15 23:16:49 +02:00
is_js: is_js
mod: p.mod
pos: pos
2020-02-06 17:38:02 +01:00
}
return ident
2020-02-06 17:38:02 +01:00
}
2020-04-16 15:42:40 +02:00
pub fn (var p Parser) name_expr() ast.Expr {
2020-04-15 01:55:51 +02:00
var node := ast.Expr{}
2020-04-16 15:32:11 +02:00
if p.inside_is {
2020-04-16 15:35:19 +02:00
p.inside_is = false
// get type position before moving to next
type_pos := p.tok.position()
2020-04-16 15:32:11 +02:00
return ast.Type{
typ: p.parse_type()
pos: type_pos
2020-04-16 15:32:11 +02:00
}
}
is_c := p.tok.lit == 'C'
2020-04-15 23:16:49 +02:00
is_js := p.tok.lit == 'JS'
2020-04-15 01:55:51 +02:00
var mod := ''
2020-02-27 00:12:37 +01:00
// p.warn('resetting')
p.expr_mod = ''
// `map[string]int` initialization
if p.tok.lit == 'map' && p.peek_tok.kind == .lsbr {
map_type := p.parse_map_type()
2020-03-07 16:23:10 +01:00
return ast.MapInit{
2020-03-02 12:34:02 +01:00
typ: map_type
}
2020-02-27 00:12:37 +01:00
}
2020-03-24 17:07:27 +01:00
// Raw string (`s := r'hello \n ')
2020-04-18 00:22:03 +02:00
if p.tok.lit in ['r', 'c', 'js'] && p.peek_tok.kind == .string && p.prev_tok.kind != .str_dollar {
2020-03-24 17:07:27 +01:00
return p.string_expr()
}
2020-03-01 16:14:52 +01:00
known_var := p.scope.known_var(p.tok.lit)
2020-04-16 15:32:11 +02:00
if p.peek_tok.kind == .dot && !known_var && (is_c || is_js || p.known_import(p.tok.lit) ||
p.mod.all_after('.') == p.tok.lit) {
2020-02-29 11:47:47 +01:00
if is_c {
mod = 'C'
2020-04-15 23:16:49 +02:00
} else if is_js {
mod = 'JS'
} else {
// prepend the full import
mod = p.imports[p.tok.lit]
}
2020-01-07 16:06:37 +01:00
p.next()
p.check(.dot)
2020-02-27 00:12:37 +01:00
p.expr_mod = mod
}
// p.warn('name expr $p.tok.lit $p.peek_tok.str()')
2020-02-07 21:29:28 +01:00
// fn call or type cast
2020-01-06 16:13:12 +01:00
if p.peek_tok.kind == .lpar {
2020-04-15 01:55:51 +02:00
var name := p.tok.lit
2020-02-29 11:47:47 +01:00
if mod.len > 0 {
name = '${mod}.$name'
}
2020-03-01 16:14:52 +01:00
name_w_mod := p.prepend_mod(name)
2020-02-07 21:29:28 +01:00
// type cast. TODO: finish
// if name in table.builtin_type_names {
2020-04-08 01:00:58 +02:00
if (name in p.table.type_idxs || name_w_mod in p.table.type_idxs) && !(name in ['C.stat',
2020-04-07 18:51:39 +02:00
'C.sigaction']) {
2020-02-11 10:26:46 +01:00
// TODO handle C.stat()
2020-04-15 01:55:51 +02:00
var to_typ := p.parse_type()
2020-03-10 23:21:26 +01:00
if p.is_amp {
// Handle `&Foo(0)`
to_typ = table.type_to_ptr(to_typ)
}
p.check(.lpar)
2020-04-15 01:55:51 +02:00
var expr := ast.Expr{}
var arg := ast.Expr{}
var has_arg := false
expr = p.expr(0)
// TODO, string(b, len)
if p.tok.kind == .comma && table.type_idx(to_typ) == table.string_type_idx {
p.check(.comma)
arg = p.expr(0) // len
2020-03-07 16:23:10 +01:00
has_arg = true
2020-02-07 21:29:28 +01:00
}
p.check(.rpar)
node = ast.CastExpr{
typ: to_typ
expr: expr
2020-03-07 16:23:10 +01:00
arg: arg
has_arg: has_arg
}
2020-02-27 00:12:37 +01:00
p.expr_mod = ''
2020-02-18 08:58:20 +01:00
return node
} else {
2020-04-05 04:05:09 +02:00
// fn call
2020-02-18 08:58:20 +01:00
// println('calling $p.tok.lit')
2020-04-15 23:16:49 +02:00
x := p.call_expr(is_c, is_js, mod) // TODO `node,typ :=` should work
2020-02-07 21:29:28 +01:00
node = x
}
} else if p.peek_tok.kind == .lcbr && !p.inside_match && !p.inside_match_case && !p.inside_if &&
!p.inside_for {
2020-04-08 19:47:17 +02:00
return p.struct_init(false) // short_syntax: false
} else if p.peek_tok.kind == .dot && (p.tok.lit[0].is_capital() && !known_var) {
2020-02-27 00:12:37 +01:00
// `Color.green`
2020-04-15 01:55:51 +02:00
var enum_name := p.check_name()
2020-02-27 00:12:37 +01:00
if mod != '' {
enum_name = mod + '.' + enum_name
} else {
2020-02-27 00:12:37 +01:00
enum_name = p.prepend_mod(enum_name)
}
// p.warn('Color.green $enum_name ' + p.prepend_mod(enum_name) + 'mod=$mod')
2020-02-25 15:02:34 +01:00
p.check(.dot)
val := p.check_name()
// println('enum val $enum_name . $val')
2020-02-27 00:12:37 +01:00
p.expr_mod = ''
2020-02-25 15:02:34 +01:00
return ast.EnumVal{
enum_name: enum_name
2020-02-25 15:02:34 +01:00
val: val
2020-02-27 00:12:37 +01:00
pos: p.tok.position()
2020-03-15 00:46:08 +01:00
mod: mod
2020-02-25 15:02:34 +01:00
}
} else {
2020-04-15 01:55:51 +02:00
var ident := ast.Ident{}
2020-04-15 23:16:49 +02:00
ident = p.parse_ident(is_c, is_js)
2020-02-06 17:38:02 +01:00
node = ident
2020-01-06 16:13:12 +01:00
}
2020-02-27 00:12:37 +01:00
p.expr_mod = ''
2020-02-18 08:58:20 +01:00
return node
2020-01-06 16:13:12 +01:00
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) index_expr(left ast.Expr) ast.IndexExpr {
// left == `a` in `a[0]`
p.next() // [
2020-04-15 01:55:51 +02:00
var has_low := true
if p.tok.kind == .dotdot {
2020-03-06 22:24:39 +01:00
has_low = false
// [..end]
p.next()
high := p.expr(0)
p.check(.rsbr)
return ast.IndexExpr{
left: left
pos: p.tok.position()
index: ast.RangeExpr{
low: ast.Expr{}
high: high
has_high: true
}
}
2020-02-02 14:31:54 +01:00
}
expr := p.expr(0) // `[expr]` or `[expr..]`
2020-04-15 01:55:51 +02:00
var has_high := false
if p.tok.kind == .dotdot {
// [start..end] or [start..]
p.check(.dotdot)
2020-04-15 01:55:51 +02:00
var high := ast.Expr{}
if p.tok.kind != .rsbr {
2020-03-06 22:24:39 +01:00
has_high = true
high = p.expr(0)
}
p.check(.rsbr)
return ast.IndexExpr{
left: left
pos: p.tok.position()
index: ast.RangeExpr{
low: expr
high: high
has_high: has_high
has_low: has_low
}
}
}
// [expr]
2020-01-07 12:14:10 +01:00
p.check(.rsbr)
return ast.IndexExpr{
2020-01-07 12:14:10 +01:00
left: left
index: expr
2020-02-03 11:29:50 +01:00
pos: p.tok.position()
}
2020-01-07 12:14:10 +01:00
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) filter() {
p.scope.register('it', ast.Var{
name: 'it'
})
2020-02-10 14:42:57 +01:00
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) dot_expr(left ast.Expr) ast.Expr {
2020-01-06 16:13:12 +01:00
p.next()
2020-04-15 01:55:51 +02:00
var name_pos := p.tok.position()
2020-01-06 16:13:12 +01:00
field_name := p.check_name()
2020-03-25 10:09:50 +01:00
is_filter := field_name in ['filter', 'map']
if is_filter {
p.open_scope()
name_pos = p.tok.position()
p.filter()
// wrong tok position when using defer
// defer {
// p.close_scope()
// }
2020-02-10 14:42:57 +01:00
}
// Method call
if p.tok.kind == .lpar {
p.next()
args := p.call_args()
2020-04-10 14:53:06 +02:00
p.check(.rpar)
2020-04-15 01:55:51 +02:00
var or_stmts := []ast.Stmt
var is_or_block_used := false
2020-02-04 17:44:39 +01:00
if p.tok.kind == .key_orelse {
p.next()
p.open_scope()
p.scope.register('errcode', ast.Var{
name: 'errcode'
typ: table.int_type
})
p.scope.register('err', ast.Var{
name: 'err'
typ: table.string_type
})
is_or_block_used = true
or_stmts = p.parse_block_no_scope()
p.close_scope()
2020-02-04 17:44:39 +01:00
}
end_pos := p.tok.position()
pos := token.Position{
line_nr: name_pos.line_nr
pos: name_pos.pos
len: end_pos.pos - name_pos.pos
}
mcall_expr := ast.CallExpr{
left: left
name: field_name
args: args
pos: pos
is_method: true
or_block: ast.OrExpr{
stmts: or_stmts
is_used: is_or_block_used
}
}
2020-04-15 01:55:51 +02:00
var node := ast.Expr{}
node = mcall_expr
2020-03-25 10:09:50 +01:00
if is_filter {
p.close_scope()
}
return node
2020-01-08 10:19:12 +01:00
}
sel_expr := ast.SelectorExpr{
2020-01-06 16:13:12 +01:00
expr: left
field: field_name
pos: name_pos
2019-12-22 02:34:37 +01:00
}
2020-04-15 01:55:51 +02:00
var node := ast.Expr{}
node = sel_expr
2020-03-25 10:09:50 +01:00
if is_filter {
p.close_scope()
}
return node
2019-12-24 18:54:43 +01:00
}
// `.green`
2020-02-20 15:42:56 +01:00
// `pref.BuildMode.default_mode`
2020-04-16 15:42:40 +02:00
fn (var p Parser) enum_val() ast.EnumVal {
p.check(.dot)
2020-02-25 15:02:34 +01:00
val := p.check_name()
return ast.EnumVal{
2020-02-25 15:02:34 +01:00
val: val
2020-02-27 00:12:37 +01:00
pos: p.tok.position()
}
}
2020-01-03 11:36:17 +01:00
2020-04-16 15:42:40 +02:00
fn (var p Parser) string_expr() ast.Expr {
2020-03-24 17:07:27 +01:00
is_raw := p.tok.kind == .name && p.tok.lit == 'r'
is_cstr := p.tok.kind == .name && p.tok.lit == 'c'
if is_raw || is_cstr {
p.next()
}
2020-04-15 01:55:51 +02:00
var node := ast.Expr{}
2020-03-21 07:01:06 +01:00
val := p.tok.lit
2020-04-10 14:40:52 +02:00
pos := p.tok.position()
if p.peek_tok.kind != .str_dollar {
p.next()
node = ast.StringLiteral{
val: val
is_raw: is_raw
is_c: is_cstr
pos: pos
}
return node
}
2020-04-15 01:55:51 +02:00
var exprs := []ast.Expr
var vals := []string
var efmts := []string
// Handle $ interpolation
2020-03-16 03:19:26 +01:00
for p.tok.kind == .string {
2020-03-21 07:01:06 +01:00
vals << p.tok.lit
p.next()
if p.tok.kind != .str_dollar {
continue
}
p.check(.str_dollar)
2020-03-21 07:01:06 +01:00
exprs << p.expr(0)
2020-04-15 01:55:51 +02:00
var efmt := []string
if p.tok.kind == .colon {
efmt << ':'
2020-02-04 09:54:15 +01:00
p.next()
2020-04-11 17:25:39 +02:00
// ${num:-2d}
if p.tok.kind == .minus {
efmt << '-'
p.next()
}
// ${num:2d}
if p.tok.kind == .number {
efmt << p.tok.lit
p.next()
}
2020-02-11 10:26:46 +01:00
if p.tok.lit.len == 1 {
efmt << p.tok.lit
2020-02-11 10:26:46 +01:00
p.next()
}
}
efmts << efmt.join('')
}
2020-03-21 07:01:06 +01:00
node = ast.StringInterLiteral{
vals: vals
exprs: exprs
expr_fmts: efmts
pos: pos
2020-03-21 07:01:06 +01:00
}
return node
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) parse_number_literal() ast.Expr {
lit := p.tok.lit
2020-04-10 00:09:34 +02:00
pos := p.tok.position()
2020-04-15 01:55:51 +02:00
var node := ast.Expr{}
2020-04-17 20:31:32 +02:00
if lit.index_any('.eE') >= 0 && lit[..2] !in ['0x', '0X', '0o', '0O', '0b', '0B'] {
node = ast.FloatLiteral{
val: lit
}
} else {
node = ast.IntegerLiteral{
val: lit
2020-04-10 00:09:34 +02:00
pos: pos
}
}
p.next()
return node
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) module_decl() ast.Module {
2020-04-15 01:55:51 +02:00
var name := 'main'
is_skipped := p.tok.kind != .key_module
if !is_skipped {
p.check(.key_module)
name = p.check_name()
}
full_mod := p.table.qualify_module(name, p.file_name)
p.mod = full_mod
p.builtin_mod = p.mod == 'builtin'
return ast.Module{
name: full_mod
is_skipped: is_skipped
}
2019-12-28 09:43:22 +01:00
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) parse_import() ast.Import {
2020-03-11 18:52:51 +01:00
pos := p.tok.position()
2020-04-15 01:55:51 +02:00
var mod_name := p.check_name()
var mod_alias := mod_name
for p.tok.kind == .dot {
p.check(.dot)
submod_name := p.check_name()
mod_name += '.' + submod_name
mod_alias = submod_name
}
2020-01-06 16:13:12 +01:00
if p.tok.kind == .key_as {
p.check(.key_as)
mod_alias = p.check_name()
2020-01-06 16:13:12 +01:00
}
p.imports[mod_alias] = mod_name
p.table.imports << mod_name
2019-12-31 19:42:16 +01:00
return ast.Import{
mod: mod_name
alias: mod_alias
2020-03-11 18:52:51 +01:00
pos: pos
}
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) import_stmt() []ast.Import {
p.check(.key_import)
2020-04-15 01:55:51 +02:00
var imports := []ast.Import
if p.tok.kind == .lpar {
p.check(.lpar)
for p.tok.kind != .rpar {
imports << p.parse_import()
2020-04-05 02:08:10 +02:00
if p.tok.kind == .comment {
p.comment()
}
}
p.check(.rpar)
} else {
// p.warn('`import module` has been deprecated, use `import ( module )` instead')
imports << p.parse_import()
}
return imports
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) const_decl() ast.ConstDecl {
is_pub := p.tok.kind == .key_pub
if is_pub {
p.next()
}
pos := p.tok.position()
p.check(.key_const)
2020-04-16 06:00:05 +02:00
if p.tok.kind != .lpar {
p.error('consts must be grouped, e.g.\nconst (\n\ta = 1\n)')
}
2020-04-16 15:32:11 +02:00
p.next() // (
2020-04-15 01:55:51 +02:00
var fields := []ast.ConstField
for p.tok.kind != .rpar {
2020-04-05 02:08:10 +02:00
if p.tok.kind == .comment {
p.comment()
}
2020-02-22 13:04:56 +01:00
name := p.prepend_mod(p.check_name())
// name := p.check_name()
2020-02-22 17:07:03 +01:00
// println('!!const: $name')
p.check(.assign)
expr := p.expr(0)
field := ast.ConstField{
name: name
expr: expr
pos: p.tok.position()
}
fields << field
p.global_scope.register(field.name, field)
2019-12-31 19:42:16 +01:00
}
p.check(.rpar)
return ast.ConstDecl{
pos: pos
fields: fields
is_pub: is_pub
}
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) return_stmt() ast.Return {
2020-04-17 16:16:56 +02:00
first_pos := p.tok.position()
p.next()
// return expressions
2020-04-15 01:55:51 +02:00
var exprs := []ast.Expr
if p.tok.kind == .rcbr {
return ast.Return{
2020-04-17 16:16:56 +02:00
pos: first_pos
}
}
for {
expr := p.expr(0)
exprs << expr
if p.tok.kind == .comma {
p.check(.comma)
} else {
break
}
}
2020-04-17 16:16:56 +02:00
last_pos := exprs.last().position()
return ast.Return{
exprs: exprs
2020-04-17 16:16:56 +02:00
pos: token.Position{
line_nr: first_pos.line_nr
pos: first_pos.pos
len: last_pos.pos - first_pos.pos + last_pos.len
}
2019-12-28 09:43:22 +01:00
}
}
// left hand side of `=` or `:=` in `a,b,c := 1,2,3`
2020-04-16 15:42:40 +02:00
fn (var p Parser) global_decl() ast.GlobalDecl {
2020-04-08 01:00:58 +02:00
if !p.pref.translated && !p.pref.is_live && !p.builtin_mod && !p.pref.building_v && p.mod !=
2020-04-07 18:51:39 +02:00
'ui' && p.mod != 'gg2' && p.mod != 'uiold' && !os.getwd().contains('/volt') && !p.pref.enable_globals {
p.error('use `v --enable-globals ...` to enable globals')
}
p.next()
name := p.check_name()
2020-03-31 19:58:44 +02:00
// println(name)
typ := p.parse_type()
2020-04-15 01:55:51 +02:00
var expr := ast.Expr{}
2020-04-07 18:51:39 +02:00
has_expr := p.tok.kind == .assign
if has_expr {
p.next()
2020-04-07 18:51:39 +02:00
expr = p.expr(0)
}
// p.genln(p.table.cgen_name_type_pair(name, typ))
/*
mut g := p.table.cgen_name_type_pair(name, typ)
if p.tok == .assign {
p.next()
g += ' = '
_,expr := p.tmp_expr()
g += expr
}
// p.genln('; // global')
g += '; // global'
if !p.cgen.nogen {
p.cgen.consts << g
}
*/
glob := ast.GlobalDecl{
name: name
2020-03-06 16:31:40 +01:00
typ: typ
2020-04-07 18:51:39 +02:00
has_expr: has_expr
expr: expr
}
p.global_scope.register(name, glob)
return glob
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) enum_decl() ast.EnumDecl {
is_pub := p.tok.kind == .key_pub
if is_pub {
p.next()
}
p.check(.key_enum)
name := p.prepend_mod(p.check_name())
p.check(.lcbr)
2020-04-15 01:55:51 +02:00
var vals := []string
2020-04-09 19:23:49 +02:00
// mut default_exprs := []ast.Expr
2020-04-15 01:55:51 +02:00
var fields := []ast.EnumField
for p.tok.kind != .eof && p.tok.kind != .rcbr {
2020-04-09 19:23:49 +02:00
pos := p.tok.position()
2020-04-10 00:09:34 +02:00
val := p.check_name()
vals << val
2020-04-15 01:55:51 +02:00
var expr := ast.Expr{}
var has_expr := false
2020-02-19 11:31:33 +01:00
// p.warn('enum val $val')
2020-02-12 01:16:38 +01:00
if p.tok.kind == .assign {
p.next()
2020-04-10 14:44:01 +02:00
expr = p.expr(0)
has_expr = true
2020-02-12 01:16:38 +01:00
}
fields << ast.EnumField{
2020-04-17 18:11:04 +02:00
name: val
pos: pos
expr: expr
has_expr: has_expr
}
2020-04-01 12:11:00 +02:00
// Allow commas after enum, helpful for
// enum Color {
2020-04-01 13:38:05 +02:00
// r,g,b
2020-04-01 12:11:00 +02:00
// }
if p.tok.kind == .comma {
p.next()
}
}
p.check(.rcbr)
2020-02-25 13:30:43 +01:00
p.table.register_type_symbol(table.TypeSymbol{
kind: .enum_
name: name
2020-02-25 13:30:43 +01:00
info: table.Enum{
vals: vals
}
2020-02-25 13:30:43 +01:00
})
return ast.EnumDecl{
name: name
is_pub: is_pub
2020-04-09 19:23:49 +02:00
fields: fields
}
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) type_decl() ast.TypeDecl {
is_pub := p.tok.kind == .key_pub
if is_pub {
p.next()
}
p.check(.key_type)
name := p.check_name()
2020-04-15 01:55:51 +02:00
var sum_variants := []table.Type
if p.tok.kind == .assign {
p.next() // TODO require `=`
2020-04-13 15:06:02 +02:00
}
if p.tok.kind == .key_fn {
// function type: `type mycallback fn(string, int)`
fn_name := p.prepend_mod(name)
fn_type := p.parse_fn_type(fn_name)
return ast.FnTypeDecl{
name: fn_name
is_pub: is_pub
typ: fn_type
}
}
first_type := p.parse_type() // need to parse the first type before we can check if it's `type A = X | Y`
2020-04-13 15:06:02 +02:00
if p.tok.kind == .pipe {
p.check(.pipe)
sum_variants << first_type
// type SumType = A | B | c
for {
variant_type := p.parse_type()
sum_variants << variant_type
if p.tok.kind != .pipe {
break
}
p.check(.pipe)
}
p.table.register_type_symbol(table.TypeSymbol{
kind: .sum_type
name: p.prepend_mod(name)
info: table.SumType{
variants: sum_variants
}
})
return ast.SumTypeDecl{
name: name
is_pub: is_pub
sub_types: sum_variants
}
2020-03-02 23:19:04 +01:00
}
// type MyType int
2020-04-13 15:06:02 +02:00
parent_type := first_type
pid := table.type_idx(parent_type)
p.table.register_type_symbol(table.TypeSymbol{
kind: .alias
name: p.prepend_mod(name)
parent_idx: pid
info: table.Alias{
foo: ''
}
})
return ast.AliasTypeDecl{
name: name
is_pub: is_pub
parent_type: parent_type
}
}
2020-04-16 15:42:40 +02:00
fn (var p Parser) assoc() ast.Assoc {
2020-04-01 23:23:20 +02:00
var_name := p.check_name()
pos := p.tok.position()
2020-04-15 01:45:27 +02:00
v := p.scope.find_var(var_name) or {
2020-04-01 23:23:20 +02:00
p.error('unknown variable `$var_name`')
return ast.Assoc{}
}
// println('assoc var $name typ=$var.typ')
2020-04-15 01:55:51 +02:00
var fields := []string
var vals := []ast.Expr
2020-04-01 23:23:20 +02:00
p.check(.pipe)
for {
2020-04-01 23:23:20 +02:00
fields << p.check_name()
p.check(.colon)
expr := p.expr(0)
vals << expr
if p.tok.kind == .comma {
p.check(.comma)
}
if p.tok.kind == .rcbr {
break
}
}
return ast.Assoc{
var_name: var_name
fields: fields
exprs: vals
pos: pos
2020-04-15 01:45:27 +02:00
typ: v.typ
2020-04-01 23:23:20 +02:00
}
}
2020-03-31 20:26:15 +02:00
fn (p &Parser) new_true_expr() ast.Expr {
return ast.BoolLiteral{
val: true
}
}
2019-12-26 11:21:41 +01:00
fn verror(s string) {
util.verror('parser error', s)
2019-12-26 11:21:41 +01:00
}