v/compiler/comptime.v

294 lines
7.1 KiB
V
Raw Normal View History

2019-07-29 18:21:36 +02:00
// Copyright (c) 2019 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module main
import (
vweb.tmpl // for `$vweb_html()`
os
2019-09-26 04:28:43 +02:00
strings
)
2019-07-29 18:21:36 +02:00
fn (p mut Parser) comp_time() {
p.check(.dollar)
2019-07-29 18:21:36 +02:00
if p.tok == .key_if {
p.check(.key_if)
p.fspace()
not := p.tok == .not
2019-07-29 18:21:36 +02:00
if not {
p.check(.not)
2019-07-29 18:21:36 +02:00
}
name := p.check_name()
p.fspace()
if name in supported_platforms {
ifdef_name := os_name_to_ifdef(name)
2019-07-29 18:21:36 +02:00
if not {
p.genln('#ifndef $ifdef_name')
}
else {
p.genln('#ifdef $ifdef_name')
}
p.check(.lcbr)
p.statements_no_rcbr()
if ! (p.tok == .dollar && p.peek() == .key_else) {
p.genln('#endif')
}
}
2019-08-16 07:50:36 +02:00
else if name == 'debug' {
p.genln('#ifdef VDEBUG')
p.check(.lcbr)
p.statements_no_rcbr()
p.genln('#endif')
}
2019-07-29 18:21:36 +02:00
else {
println('Supported platforms:')
println(supported_platforms)
2019-07-29 18:21:36 +02:00
p.error('unknown platform `$name`')
}
if_returns := p.returns
p.returns = false
2019-07-08 03:42:36 +02:00
//p.gen('/* returns $p.returns */')
if p.tok == .dollar && p.peek() == .key_else {
p.next()
p.next()
p.check(.lcbr)
p.genln('#else')
p.statements_no_rcbr()
p.genln('#endif')
else_returns := p.returns
p.returns = if_returns && else_returns
2019-07-08 03:42:36 +02:00
//p.gen('/* returns $p.returns */')
}
2019-07-29 18:21:36 +02:00
}
else if p.tok == .key_for {
p.next()
name := p.check_name()
if name != 'field' {
p.error('for field only')
}
p.check(.key_in)
p.check_name()
p.check(.dot)
p.check_name()// fields
p.check(.lcbr)
// for p.tok != .rcbr && p.tok != .eof {
res_name := p.check_name()
println(res_name)
p.check(.dot)
p.check(.dollar)
p.check(.name)
p.check(.assign)
p.cgen.start_tmp()
p.bool_expression()
val := p.cgen.end_tmp()
println(val)
p.check(.rcbr)
// }
}
// $vweb.html()
2019-07-29 18:21:36 +02:00
// Compile vweb html template to V code, parse that V code and embed the resulting V functions
// that returns an html string
2019-07-29 18:21:36 +02:00
else if p.tok == .name && p.lit == 'vweb' {
mut path := p.cur_fn.name + '.html'
2019-07-31 03:41:39 +02:00
if p.pref.is_debug {
println('compiling tmpl $path')
}
2019-07-29 18:21:36 +02:00
if !os.file_exists(path) {
// Can't find the template file in current directory,
// try looking next to the vweb program, in case it's run with
// v path/to/vweb_app.v
path = os.dir(p.scanner.file_path) + '/' + path
if !os.file_exists(path) {
p.error('vweb HTML template "$path" not found')
}
}
p.check(.name) // skip `vweb.html()` TODO
p.check(.dot)
p.check(.name)
p.check(.lpar)
p.check(.rpar)
v_code := tmpl.compile_template(path)
2019-07-31 03:41:39 +02:00
if os.file_exists('.vwebtmpl.v') {
os.rm('.vwebtmpl.v')
}
os.write_file('.vwebtmpl.v', v_code.clone()) // TODO don't need clone, compiler bug
p.genln('')
2019-07-29 18:21:36 +02:00
// Parse the function and embed resulting C code in current function so that
// all variables are available.
pos := p.cgen.lines.len - 1
mut pp := p.v.new_parser_file('.vwebtmpl.v')
if !p.pref.is_debug {
os.rm('.vwebtmpl.v')
}
pp.is_vweb = true
2019-09-07 12:44:41 +02:00
pp.set_current_fn( p.cur_fn ) // give access too all variables in current function
pp.parse(.main)
tmpl_fn_body := p.cgen.lines.slice(pos + 2, p.cgen.lines.len).join('\n').clone()
end_pos := tmpl_fn_body.last_index('Builder_str( sb )') + 19 // TODO
p.cgen.lines = p.cgen.lines.left(pos)
p.genln('/////////////////// tmpl start')
p.genln(tmpl_fn_body.left(end_pos))
p.genln('/////////////////// tmpl end')
// `app.vweb.html(index_view())`
receiver := p.cur_fn.args[0]
dot := if receiver.is_mut { '->' } else { '.' }
p.genln('vweb__Context_html($receiver.name $dot vweb, tmpl_res)')
}
2019-07-29 18:21:36 +02:00
else {
p.error('bad comptime expr')
}
}
// #include, #flag, #v
2019-07-29 18:21:36 +02:00
fn (p mut Parser) chash() {
hash := p.lit.trim_space()
// println('chsh() file=$p.file is_sig=${p.is_sig()} hash="$hash"')
p.next()
is_sig := p.is_sig()
if hash.starts_with('flag ') {
mut flag := hash.right(5)
// expand `@VROOT` `@VMOD` to absolute path
flag = flag.replace('@VROOT', p.vroot)
flag = flag.replace('@VMOD', ModPath)
2019-09-23 22:18:55 +02:00
//p.log('adding flag "$flag"')
p.table.parse_cflag(flag, p.mod)
2019-07-29 18:21:36 +02:00
return
}
if hash.starts_with('include') {
if p.first_pass() && !is_sig {
if p.file_pcguard.len != 0 {
//println('p: $p.file_platform $p.file_pcguard')
p.cgen.includes << '$p.file_pcguard\n#$hash\n#endif'
return
}
2019-07-29 18:21:36 +02:00
p.cgen.includes << '#$hash'
return
}
}
// TODO remove after ui_mac.m is removed
else if hash.contains('embed') {
pos := hash.index('embed') + 5
file := hash.right(pos)
if p.pref.build_mode != BuildMode.default_mode {
p.genln('#include $file')
}
}
else if hash.contains('define') {
// Move defines on top
2019-07-29 18:21:36 +02:00
p.cgen.includes << '#$hash'
}
2019-07-29 18:21:36 +02:00
else if hash == 'v' {
println('v script')
//p.v_script = true
}
2019-09-14 22:48:30 +02:00
// Don't parse a non-JS V file (`#-js` flag)
else if hash == '-js' {
$if js {
for p.tok != .eof {
p.next()
}
} $else {
p.next()
}
}
2019-07-29 18:21:36 +02:00
else {
2019-09-14 22:48:30 +02:00
$if !js {
if !p.can_chash {
println('hash="$hash"')
println(hash.starts_with('include'))
p.error('bad token `#` (embedding C code is no longer supported)')
}
2019-07-29 18:21:36 +02:00
}
p.genln(hash)
}
}
// `user.$method()` (`method` is a string)
2019-07-29 18:21:36 +02:00
fn (p mut Parser) comptime_method_call(typ Type) {
p.cgen.cur_line = ''
2019-07-29 18:21:36 +02:00
p.check(.dollar)
var := p.check_name()
2019-08-02 05:20:18 +02:00
for i, method in typ.methods {
2019-07-29 18:21:36 +02:00
if method.typ != 'void' {
continue
}
receiver := method.args[0]
amp := if receiver.is_mut { '&' } else { '' }
if i > 0 {
2019-08-02 05:20:18 +02:00
p.gen(' else ')
}
p.gen('if ( string_eq($var, _STR("$method.name")) ) ${typ.name}_$method.name($amp $p.expr_var.name);')
}
p.check(.lpar)
p.check(.rpar)
2019-08-02 04:04:48 +02:00
if p.tok == .key_orelse {
p.check(.key_orelse)
p.genln('else {')
2019-08-02 04:04:48 +02:00
p.check(.lcbr)
p.statements()
}
}
2019-07-29 18:21:36 +02:00
2019-08-31 15:38:13 +02:00
fn (p mut Parser) gen_array_str(typ Type) {
2019-09-18 14:37:34 +02:00
p.add_method(typ.name, Fn{
name: 'str'
2019-08-10 23:02:48 +02:00
typ: 'string'
args: [Var{typ: typ.name, is_arg:true}]
is_method: true
is_public: true
receiver_typ: typ.name
})
2019-09-18 14:37:34 +02:00
elm_type := typ.name.right(6)
2019-08-31 15:38:13 +02:00
elm_type2 := p.table.find_type(elm_type)
if p.typ_to_fmt(elm_type, 0) == '' &&
!p.table.type_has_method(elm_type2, 'str') {
p.error('cant print ${elm_type}[], unhandled print of ${elm_type}')
}
p.v.vgen_buf.writeln('
2019-09-26 04:28:43 +02:00
fn (a $typ.name) str() string {
mut sb := strings.new_builder(a.len * 3)
sb.write("[")
for i, elm in a {
sb.write(elm.str())
if i < a.len - 1 {
sb.write(", ")
}
}
2019-09-26 04:28:43 +02:00
sb.write("]")
return sb.str()
}
2019-09-26 04:28:43 +02:00
')
p.cgen.fns << 'string ${typ.name}_str();'
}
// `Foo { bar: 3, baz: 'hi' }` => '{ bar: 3, baz: "hi" }'
fn (p mut Parser) gen_struct_str(typ Type) {
p.add_method(typ.name, Fn{
name: 'str'
typ: 'string'
args: [Var{typ: typ.name, is_arg:true}]
is_method: true
is_public: true
receiver_typ: typ.name
})
mut sb := strings.new_builder(typ.fields.len * 20)
sb.writeln('fn (a $typ.name) str() string {\nreturn')
sb.writeln("'{")
for field in typ.fields {
sb.writeln('\t$field.name: \$a.${field.name}')
}
sb.writeln("\n}'")
sb.writeln('}')
p.v.vgen_buf.writeln(sb.str())
2019-09-26 04:28:43 +02:00
// Need to manually add the definition to `fns` so that it stays
// at the top of the file.
// This function will get parsee by V after the main pass.
p.cgen.fns << 'string ${typ.name}_str();'
}
2019-08-10 23:02:48 +02:00
2019-08-13 13:50:19 +02:00