v/vlib/vweb/tmpl/tmpl.v

167 lines
5.1 KiB
V
Raw Normal View History

// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
2019-07-30 22:33:20 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module tmpl
2019-07-29 18:21:36 +02:00
import os
import strings
2019-07-29 18:21:36 +02:00
const (
str_start = "sb.write_string('"
2021-01-13 06:05:27 +01:00
str_end = "' ) "
2019-07-29 18:21:36 +02:00
)
// compile_file compiles the content of a file by the given path as a template
pub fn compile_file(path string, fn_name string) string {
basepath := os.dir(path)
html := os.read_file(path) or { panic('html failed') }
return compile_template(basepath, html, fn_name)
}
2020-06-09 23:20:48 +02:00
enum State {
html
css // <style>
js // <script>
// span // span.{
2020-06-09 23:20:48 +02:00
}
pub fn compile_template(basepath string, html_ string, fn_name string) string {
// lines := os.read_lines(path)
mut html := html_.trim_space()
mut header := ''
mut footer := ''
if os.exists('templates/header.html') && html.contains('@header') {
h := os.read_file('templates/header.html') or {
panic('reading file templates/header.html failed')
}
2021-01-13 06:05:27 +01:00
header = h.trim_space().replace("'", '"')
html = header + html
}
if os.exists('templates/footer.html') && html.contains('@footer') {
f := os.read_file('templates/footer.html') or {
panic('reading file templates/footer.html failed')
}
2021-01-13 06:05:27 +01:00
footer = f.trim_space().replace("'", '"')
html += footer
}
2020-06-11 11:13:36 +02:00
mut lines := html.split_into_lines()
lstartlength := lines.len * 30
mut s := strings.new_builder(1000)
// base := path.all_after_last('/').replace('.html', '')
2019-12-19 22:29:37 +01:00
s.writeln("
2020-06-11 11:13:36 +02:00
import strings
// === vweb html template ===
fn vweb_tmpl_${fn_name}() {
mut sb := strings.new_builder($lstartlength)\n
2020-06-11 11:13:36 +02:00
header := \' \' // TODO remove
_ = header
footer := \' \' // TODO remove
_ = footer
2020-06-11 11:13:36 +02:00
")
s.write_string(tmpl.str_start)
2020-06-09 23:20:48 +02:00
mut state := State.html
2020-06-10 01:43:04 +02:00
mut in_span := false
// for _line in lines {
for i := 0; i < lines.len; i++ {
line := lines[i].trim_space()
2019-07-30 22:33:20 +02:00
if line == '<style>' {
2020-06-09 23:20:48 +02:00
state = .css
2020-06-08 12:44:27 +02:00
} else if line == '</style>' {
2020-06-09 23:20:48 +02:00
state = .html
} else if line == '<script>' {
2020-06-09 23:20:48 +02:00
state = .js
} else if line == '</script>' {
2020-06-09 23:20:48 +02:00
state = .html
}
2021-02-04 17:07:04 +01:00
if line.contains('@include ') {
lines.delete(i)
mut file_name := line.split("'")[1]
mut file_ext := os.file_ext(file_name)
if file_ext == '' {
file_ext = '.html'
}
file_name = file_name.replace(file_ext, '')
mut templates_folder := os.join_path(basepath, 'templates')
if file_name.contains('/') {
if file_name.starts_with('/') {
// absolute path
templates_folder = ''
} else {
// relative path, starting with the current folder
templates_folder = os.real_path(basepath)
}
}
file_path := os.real_path(os.join_path(templates_folder, '$file_name$file_ext'))
$if trace_tmpl ? {
eprintln('>>> basepath: "$basepath" , fn_name: "$fn_name" , @include line: "$line" , file_name: "$file_name" , file_ext: "$file_ext" , templates_folder: "$templates_folder" , file_path: "$file_path"')
}
2021-02-04 17:07:04 +01:00
file_content := os.read_file(file_path) or {
panic('Vweb: Reading file $file_name failed.')
2020-06-10 18:53:04 +02:00
}
2021-02-04 17:07:04 +01:00
file_splitted := file_content.split_into_lines().reverse()
for f in file_splitted {
lines.insert(i, f)
}
2021-02-04 17:07:04 +01:00
i--
2020-06-11 20:34:59 +02:00
} else if line.contains('@js ') {
pos := line.index('@js') or { continue }
s.write_string('<script src="')
s.write_string(line[pos + 5..line.len - 1])
s.writeln('"></script>')
2020-06-11 20:34:59 +02:00
} else if line.contains('@css ') {
pos := line.index('@css') or { continue }
s.write_string('<link href="')
s.write_string(line[pos + 6..line.len - 1])
s.writeln('" rel="stylesheet" type="text/css">')
2020-06-10 18:53:04 +02:00
} else if line.contains('@if ') {
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_end)
pos := line.index('@if') or { continue }
2019-12-19 22:29:37 +01:00
s.writeln('if ' + line[pos + 4..] + '{')
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_start)
2020-06-08 12:44:27 +02:00
} else if line.contains('@end') {
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_end)
2019-07-29 18:21:36 +02:00
s.writeln('}')
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_start)
2020-06-08 12:44:27 +02:00
} else if line.contains('@else') {
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_end)
2019-07-29 18:21:36 +02:00
s.writeln(' } else { ')
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_start)
2020-06-08 12:44:27 +02:00
} else if line.contains('@for') {
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_end)
pos := line.index('@for') or { continue }
2019-12-19 22:29:37 +01:00
s.writeln('for ' + line[pos + 4..] + '{')
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_start)
2020-06-09 23:20:48 +02:00
} else if state == .html && line.contains('span.') && line.ends_with('{') {
2020-06-08 12:44:27 +02:00
// `span.header {` => `<span class='header'>`
class := line.find_between('span.', '{').trim_space()
s.writeln('<span class="$class">')
in_span = true
2020-06-09 23:20:48 +02:00
} else if state == .html && line.contains('.') && line.ends_with('{') {
// `.header {` => `<div class='header'>`
2020-06-08 12:44:27 +02:00
class := line.find_between('.', '{').trim_space()
s.writeln('<div class="$class">')
2020-06-09 23:20:48 +02:00
} else if state == .html && line.contains('#') && line.ends_with('{') {
2020-06-08 12:44:27 +02:00
// `#header {` => `<div id='header'>`
class := line.find_between('#', '{').trim_space()
s.writeln('<div id="$class">')
2020-06-09 23:20:48 +02:00
} else if state == .html && line == '}' {
2020-06-10 01:43:04 +02:00
if in_span {
2020-06-08 12:44:27 +02:00
s.writeln('</span>')
2020-06-10 01:43:04 +02:00
in_span = false
2020-06-08 12:44:27 +02:00
} else {
s.writeln('</div>')
}
} else {
// HTML, may include `@var`
// escaped by cgen, unless it's a `vweb.RawHtml` string
s.writeln(line.replace('@', '$').replace("'", '"'))
2019-07-29 18:21:36 +02:00
}
}
2021-01-25 10:26:20 +01:00
s.writeln(tmpl.str_end)
2020-06-10 12:17:49 +02:00
s.writeln('_tmpl_res_$fn_name := sb.str() ')
s.writeln('}')
2020-06-06 21:36:24 +02:00
s.writeln('// === end of vweb html template ===')
2019-07-29 18:21:36 +02:00
return s.str()
}