1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/vweb/vweb.v

617 lines
15 KiB
V
Raw Normal View History

2020-02-03 07:00:36 +03:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-10-24 19:44:49 +03:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-07-29 19:21:36 +03:00
module vweb
2020-04-26 14:49:31 +03:00
import os
import net
import net.http
import net.urllib
import strings
2020-06-23 22:02:17 +03:00
import time
2019-07-29 19:21:36 +03:00
pub const (
methods_with_form = [http.Method.post, .put, .patch]
2020-05-22 18:36:09 +03:00
header_server = 'Server: VWeb\r\n'
header_connection_close = 'Connection: close\r\n'
headers_close = '${header_server}${header_connection_close}\r\n'
http_404 = 'HTTP/1.1 404 Not Found\r\nContent-Type: text/plain\r\nContent-Length: 13\r\n${headers_close}404 Not Found'
2020-05-22 18:36:09 +03:00
http_500 = 'HTTP/1.1 500 Internal Server Error\r\nContent-Type: text/plain\r\n${headers_close}500 Internal Server Error'
2019-09-05 15:46:24 +03:00
mime_types = {
'.css': 'text/css; charset=utf-8',
'.gif': 'image/gif',
'.htm': 'text/html; charset=utf-8',
'.html': 'text/html; charset=utf-8',
'.jpg': 'image/jpeg',
'.js': 'application/javascript',
'.json': 'application/json',
'.md': 'text/markdown; charset=utf-8',
2019-09-05 15:46:24 +03:00
'.pdf': 'application/pdf',
'.png': 'image/png',
'.svg': 'image/svg+xml',
'.txt': 'text/plain; charset=utf-8',
'.wasm': 'application/wasm',
2019-09-05 15:46:24 +03:00
'.xml': 'text/xml; charset=utf-8'
}
2020-05-22 18:36:09 +03:00
max_http_post_size = 1024 * 1024
default_port = 8080
)
2019-10-24 19:44:49 +03:00
pub struct Context {
2020-05-27 07:53:48 +03:00
mut:
2019-10-24 19:44:49 +03:00
static_files map[string]string
static_mime_types map[string]string
2020-06-28 00:22:37 +03:00
content_type string = 'text/plain'
2020-08-09 19:05:06 +03:00
status string = '200 OK'
2019-10-24 19:44:49 +03:00
pub:
req http.Request
conn net.Socket
// TODO Response
2020-05-27 04:38:21 +03:00
pub mut:
2020-05-27 07:53:48 +03:00
form map[string]string
2020-07-07 14:46:57 +03:00
query map[string]string
2019-10-24 19:44:49 +03:00
headers string // response headers
2019-12-09 17:10:44 +03:00
done bool
2020-06-23 22:02:17 +03:00
page_gen_start i64
2020-07-07 22:35:56 +03:00
form_error string
2019-09-05 15:46:24 +03:00
}
pub struct Cookie {
name string
value string
expires time.Time
secure bool
http_only bool
}
2020-06-20 04:12:35 +03:00
pub struct Result {}
2020-05-17 14:51:18 +03:00
fn (mut ctx Context) send_response_to_client(mimetype string, res string) bool {
if ctx.done { return false }
ctx.done = true
mut sb := strings.new_builder(1024)
2020-06-28 14:56:38 +03:00
defer { sb.free() }
2020-08-09 19:05:06 +03:00
sb.write('HTTP/1.1 ${ctx.status}')
sb.write('\r\nContent-Type: ${mimetype}')
sb.write('\r\nContent-Length: ${res.len}')
sb.write(ctx.headers)
sb.write('\r\n')
2020-05-22 18:36:09 +03:00
sb.write(headers_close)
sb.write(res)
2020-06-28 20:55:53 +03:00
s := sb.str()
defer {
s.free()
}
ctx.conn.send_string(s) or { return false }
return true
}
2020-05-17 14:51:18 +03:00
pub fn (mut ctx Context) html(s string) {
ctx.send_response_to_client('text/html', s)
2019-09-05 15:46:24 +03:00
}
2020-06-27 14:56:15 +03:00
pub fn (mut ctx Context) text(s string) Result {
ctx.send_response_to_client('text/plain', s)
2020-06-28 00:22:37 +03:00
return Result{}
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-06-27 14:56:15 +03:00
pub fn (mut ctx Context) json(s string) Result {
ctx.send_response_to_client('application/json', s)
2020-06-28 00:22:37 +03:00
return Result{}
}
pub fn (mut ctx Context) ok(s string) Result {
ctx.send_response_to_client(ctx.content_type, s)
return Result{}
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-06-30 22:04:00 +03:00
pub fn (mut ctx Context) redirect(url string) Result {
if ctx.done { return Result{} }
ctx.done = true
2020-06-30 22:04:00 +03:00
ctx.conn.send_string('HTTP/1.1 302 Found\r\nLocation: ${url}${ctx.headers}\r\n${headers_close}') or { return Result{} }
return Result{}
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-06-27 14:56:15 +03:00
pub fn (mut ctx Context) not_found() Result {
if ctx.done { return vweb.Result{} }
ctx.done = true
2020-06-27 14:56:15 +03:00
ctx.conn.send_string(http_404) or {}
return vweb.Result{}
2019-09-05 15:46:24 +03:00
}
2019-08-02 05:04:48 +03:00
pub fn (mut ctx Context) set_cookie(cookie Cookie) {
mut cookie_data := []string{}
mut secure := if cookie.secure { "Secure;" } else { "" }
secure += if cookie.http_only { " HttpOnly" } else { " " }
cookie_data << secure
if cookie.expires.unix > 0 {
cookie_data << 'expires=${cookie.expires.utc_string()}'
}
data := cookie_data.join(' ')
ctx.add_header('Set-Cookie', '$cookie.name=$cookie.value; $data')
}
pub fn (mut ctx Context) set_cookie_old(key, val string) {
// TODO support directives, escape cookie value (https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Set-Cookie)
//ctx.add_header('Set-Cookie', '${key}=${val}; Secure; HttpOnly')
ctx.add_header('Set-Cookie', '${key}=${val}; HttpOnly')
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-06-28 00:22:37 +03:00
pub fn (mut ctx Context) set_content_type(typ string) {
ctx.content_type = typ
}
2020-06-29 22:14:36 +03:00
pub fn (mut ctx Context) set_cookie_with_expire_date(key, val string, expire_date time.Time) {
ctx.add_header('Set-Cookie', '$key=$val; Secure; HttpOnly; expires=${expire_date.utc_string()}')
}
pub fn (ctx &Context) get_cookie(key string) ?string { // TODO refactor
2019-12-13 21:11:40 +03:00
mut cookie_header := ctx.get_header('cookie')
if cookie_header == '' {
cookie_header = ctx.get_header('Cookie')
}
cookie_header = ' ' + cookie_header
//println('cookie_header="$cookie_header"')
//println(ctx.req.headers)
2019-09-05 15:46:24 +03:00
cookie := if cookie_header.contains(';') {
2019-12-08 01:48:49 +03:00
cookie_header.find_between(' $key=', ';')
2019-09-05 15:46:24 +03:00
} else {
2019-12-08 01:48:49 +03:00
cookie_header.find_between(' $key=', '\r')
2019-09-05 15:46:24 +03:00
}
if cookie != '' {
2019-12-08 01:48:49 +03:00
return cookie.trim_space()
2019-09-05 15:46:24 +03:00
}
return error('Cookie not found')
}
2019-07-29 19:21:36 +03:00
2020-08-09 19:05:06 +03:00
pub fn (mut ctx Context) set_status(code int, desc string) {
if code < 100 || code > 599 {
ctx.status = '500 Internal Server Error'
} else {
ctx.status = '$code $desc'
}
}
2020-05-17 14:51:18 +03:00
pub fn (mut ctx Context) add_header(key, val string) {
//println('add_header($key, $val)')
ctx.headers = ctx.headers + '\r\n$key: $val'
//println(ctx.headers)
2019-07-29 19:21:36 +03:00
}
pub fn (ctx &Context) get_header(key string) string {
2019-12-07 00:44:22 +03:00
return ctx.req.headers[key]
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2019-12-20 03:02:16 +03:00
//fn handle_conn(conn net.Socket) {
//println('handle')
//}
pub fn run<T>(port int) {
mut app := T{}
run_app<T>(mut app, port)
2020-06-10 12:23:41 +03:00
}
pub fn run_app<T>(mut app T, port int) {
2020-06-10 12:50:06 +03:00
println('Running a Vweb app on http://localhost:$port')
2019-10-24 19:44:49 +03:00
l := net.listen(port) or { panic('failed to listen') }
2019-12-20 03:02:16 +03:00
app.vweb = Context{}
2020-06-10 12:23:41 +03:00
app.init_once()
2019-12-20 03:02:16 +03:00
//app.reset()
2019-07-29 19:21:36 +03:00
for {
conn := l.accept() or { panic('accept() failed') }
//handle_conn<T>(conn, mut app)
2020-06-05 22:04:18 +03:00
handle_conn<T>(conn, mut app)
2020-06-23 22:02:17 +03:00
//app.vweb.page_gen_time = time.ticks() - t
//eprintln('handle conn() took ${time.ticks()-t}ms')
//message := readall(conn)
//println(message)
/*
2020-05-22 18:36:09 +03:00
if message.len > max_http_post_size {
println('message.len = $message.len > max_http_post_size')
conn.send_string(http_500) or {}
conn.close() or {}
continue
}
*/
2019-12-13 21:11:40 +03:00
//lines := message.split_into_lines()
//println(lines)
2019-12-13 21:11:40 +03:00
/*
2019-12-11 19:20:46 +03:00
if lines.len < 2 {
2020-05-22 18:36:09 +03:00
conn.send_string(http_500) or {}
2019-12-11 19:20:46 +03:00
conn.close() or {}
continue
}
*/
2019-12-20 03:02:16 +03:00
}
}
2019-12-13 21:11:40 +03:00
2020-06-05 22:04:18 +03:00
fn handle_conn<T>(conn net.Socket, mut app T) {
defer { conn.close() or {} }
2020-06-05 21:57:02 +03:00
//fn handle_conn<T>(conn net.Socket, app_ T) T {
//mut app := app_
2019-12-20 03:02:16 +03:00
//first_line := strip(lines[0])
2020-06-23 22:02:17 +03:00
page_gen_start := time.ticks()
2019-12-20 03:02:16 +03:00
first_line := conn.read_line()
2020-06-10 12:50:06 +03:00
$if debug {
println('firstline="$first_line"')
2020-06-10 12:50:06 +03:00
}
2019-12-20 03:02:16 +03:00
// Parse the first line
// "GET / HTTP/1.1"
//first_line := s.all_before('\n')
vals := first_line.split(' ')
if vals.len < 2 {
println('no vals for http')
2020-05-22 18:36:09 +03:00
conn.send_string(http_500) or {}
2020-06-05 22:04:18 +03:00
return
2019-12-20 03:02:16 +03:00
//continue
}
2020-04-26 14:49:31 +03:00
mut headers := []string{}
2019-12-20 03:02:16 +03:00
mut body := ''
mut in_headers := true
mut len := 0
//for line in lines[1..] {
2020-05-17 15:15:04 +03:00
for _ in 0..100 {
2019-12-20 03:02:16 +03:00
//println(j)
line := conn.read_line()
sline := strip(line)
if sline == '' {
//if in_headers {
// End of headers, no body => exit
if len == 0 {
break
}
2019-12-20 03:02:16 +03:00
//} //else {
// End of body
//break
//}
in_headers = false
2019-10-24 19:44:49 +03:00
}
2019-12-20 03:02:16 +03:00
if in_headers {
//println(sline)
headers << sline
if sline.starts_with('Content-Length') {
len = sline.all_after(': ').int()
//println('GOT CL=$len')
2019-08-20 11:18:12 +03:00
}
2019-12-20 03:02:16 +03:00
} else {
body += line.trim_left('\r\n')
if body.len >= len {
2019-12-20 03:02:16 +03:00
break
}
//println('body:$body')
2019-10-24 19:44:49 +03:00
}
2019-12-20 03:02:16 +03:00
}
2019-12-20 03:02:16 +03:00
req := http.Request{
headers: http.parse_headers(headers) //s.split_into_lines())
data: strip(body)
2019-12-20 03:02:16 +03:00
ws_func: 0
user_ptr: 0
method: http.method_from_str(vals[0])
2019-12-20 03:02:16 +03:00
url: vals[1]
}
$if debug {
println('req.headers = ')
println(req.headers)
println('req.data="$req.data"' )
//println('vweb action = "$action"')
2019-12-20 03:02:16 +03:00
}
//mut app := T{
app.vweb = Context{
req: req
conn: conn
form: map[string]string
2020-03-07 16:16:03 +03:00
static_files: app.vweb.static_files
static_mime_types: app.vweb.static_mime_types
2020-06-23 22:02:17 +03:00
page_gen_start: page_gen_start
2019-12-20 03:02:16 +03:00
}
//}
if req.method in methods_with_form {
app.vweb.parse_form(req.data)
2019-12-20 03:02:16 +03:00
}
if vals.len < 2 {
$if debug {
2019-12-20 03:02:16 +03:00
println('no vals for http')
}
2020-06-05 22:04:18 +03:00
return
2019-12-20 03:02:16 +03:00
//continue
}
// Serve a static file if it is one
// TODO: handle url parameters properly - for now, ignore them
mut static_file_name := app.vweb.req.url
if static_file_name.contains('?') {
static_file_name = static_file_name.all_before('?')
}
static_file := app.vweb.static_files[static_file_name]
mime_type := app.vweb.static_mime_types[static_file_name]
2020-03-07 16:16:03 +03:00
if static_file != '' && mime_type != '' {
data := os.read_file(static_file) or {
2020-05-22 18:36:09 +03:00
conn.send_string(http_404) or {}
2020-06-05 22:04:18 +03:00
return
2020-03-07 16:16:03 +03:00
}
app.vweb.send_response_to_client(mime_type, data)
2020-06-28 14:56:38 +03:00
data.free()
2020-06-05 22:04:18 +03:00
return
2020-03-07 16:16:03 +03:00
}
app.init()
2020-07-08 16:22:03 +03:00
2019-12-20 03:02:16 +03:00
// Call the right action
2020-08-31 20:39:46 +03:00
$if debug {
println('route matching...')
}
2020-07-07 13:35:45 +03:00
//t := time.ticks()
2020-07-07 14:46:57 +03:00
//mut action := ''
2020-07-23 18:19:37 +03:00
mut route_words_a := [][]string{}
2020-07-08 23:15:37 +03:00
mut url_words := vals[1][1..].split('/').filter(it != '')
2020-07-07 13:35:45 +03:00
if url_words.len == 0 {
app.index()
return
2020-07-07 14:46:57 +03:00
} else {
// Parse URL query
2020-07-07 14:46:57 +03:00
if url_words.last().contains('?') {
tmp_query := url_words.last().all_after('?').split('&').map(it.split('='))
url_words[url_words.len - 1] = url_words.last().all_before('?')
for data in tmp_query {
if data.len == 2 {
app.vweb.query[data[0]] = data[1]
}
}
}
2020-07-07 13:35:45 +03:00
}
2020-07-08 16:22:03 +03:00
2020-07-23 18:19:37 +03:00
mut vars := []string{cap: route_words_a.len}
2020-07-08 16:22:03 +03:00
mut action := ''
$for method in T.methods {
$if method.return_type is Result {
attrs := method.attrs
route_words_a = [][]string{}
if attrs.len == 0 {
// No routing for this method. If it matches, call it and finish matching
// since such methods have a priority.
// For example URL `/register` matches route `/:user`, but `fn register()`
// should be called first.
if (req.method == .get && url_words[0] == method.name && url_words.len == 1) || (req.method == .post && url_words[0] + '_post' == method.name) {
2020-08-31 20:39:46 +03:00
$if debug {
println('easy match method=$method.name')
}
app.$method(vars)
return
2020-07-23 18:19:37 +03:00
}
} else {
// Get methods
// Get is default
2020-07-31 02:26:56 +03:00
if req.method == .post {
if 'post' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'post').map(it[1..].split('/'))
}
2020-07-31 02:26:56 +03:00
} else if req.method == .put {
if 'put' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'put').map(it[1..].split('/'))
}
2020-07-31 02:26:56 +03:00
} else if req.method == .patch {
if 'patch' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'patch').map(it[1..].split('/'))
}
2020-07-31 02:26:56 +03:00
} else if req.method == .delete {
if 'delete' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'delete').map(it[1..].split('/'))
}
2020-07-31 02:26:56 +03:00
} else if req.method == .head {
if 'head' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'head').map(it[1..].split('/'))
}
2020-07-31 02:26:56 +03:00
} else if req.method == .options {
if 'options' in attrs {
route_words_a = attrs.filter(it.to_lower() != 'options').map(it[1..].split('/'))
}
} else {
route_words_a = attrs.filter(it.to_lower() != 'get').map(it[1..].split('/'))
}
if route_words_a.len > 0 {
for route_words in route_words_a {
if url_words.len == route_words.len || (url_words.len >= route_words.len - 1 && route_words.last().ends_with('...')) {
// match `/:user/:repo/tree` to `/vlang/v/tree`
mut matching := false
mut unknown := false
mut variables := []string{cap: route_words.len}
for i in 0..route_words.len {
if url_words.len == i {
variables << ''
matching = true
unknown = true
break
}
if url_words[i] == route_words[i] {
// no parameter
matching = true
continue
} else if route_words[i].starts_with(':') {
// is parameter
if i < route_words.len && !route_words[i].ends_with('...') {
// normal parameter
variables << url_words[i]
} else {
// array parameter only in the end
variables << url_words[i..].join('/')
}
matching = true
unknown = true
continue
2020-07-23 18:19:37 +03:00
} else {
matching = false
break
2020-07-23 18:19:37 +03:00
}
}
if matching && !unknown {
// absolute router words like `/test/site`
app.$method(vars)
return
} else if matching && unknown {
// router words with paramter like `/:test/site`
action = method.name
vars = variables
}
2020-07-08 16:22:03 +03:00
}
}
}
}
}
}
2020-07-08 16:22:03 +03:00
if action == '' {
// site not found
conn.send_string(http_404) or {}
2020-07-07 13:35:45 +03:00
return
}
$for method in T.methods {
$if method.return_type is Result {
// search again for method
if action == method.name && method.attrs.len > 0 {
// call action method
app.$method(vars)
}
2020-07-08 16:22:03 +03:00
}
2019-07-29 19:21:36 +03:00
}
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-05-17 14:51:18 +03:00
fn (mut ctx Context) parse_form(s string) {
if ctx.req.method !in methods_with_form {
2019-10-24 19:44:49 +03:00
return
}
2019-11-26 13:54:41 +03:00
//pos := s.index('\r\n\r\n')
//if pos > -1 {
mut str_form := s//[pos..s.len]
str_form = str_form.replace('+', ' ')
words := str_form.split('&')
for word in words {
$if debug {
println('parse form keyval="$word"')
}
keyval := word.trim_space().split('=')
if keyval.len != 2 { continue }
key := urllib.query_unescape(keyval[0]) or {
continue
}
2019-11-26 13:54:41 +03:00
val := urllib.query_unescape(keyval[1]) or {
continue
}
$if debug {
println('http form "$key" => "$val"')
2019-07-29 19:21:36 +03:00
}
2019-11-26 13:54:41 +03:00
ctx.form[key] = val
2019-07-29 19:21:36 +03:00
}
2019-11-26 13:54:41 +03:00
//}
2019-12-11 19:20:46 +03:00
// todo: parse form-data and application/json
// ...
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2020-05-17 14:51:18 +03:00
fn (mut ctx Context) scan_static_directory(directory_path, mount_path string) {
2019-10-17 14:30:05 +03:00
files := os.ls(directory_path) or { panic(err) }
2020-03-07 16:16:03 +03:00
if files.len > 0 {
2019-09-05 15:46:24 +03:00
for file in files {
if os.is_dir(file) {
ctx.scan_static_directory(directory_path + '/' + file, mount_path + '/' + file)
2020-03-07 16:16:03 +03:00
} else if file.contains('.') && ! file.starts_with('.') && ! file.ends_with('.') {
2020-03-26 16:18:08 +03:00
ext := os.file_ext(file)
2020-03-07 16:16:03 +03:00
// Rudimentary guard against adding files not in mime_types.
// Use serve_static directly to add non-standard mime types.
if ext in mime_types {
ctx.serve_static(mount_path + '/' + file, directory_path + '/' + file, mime_types[ext])
}
}
}
}
}
2020-05-17 14:51:18 +03:00
pub fn (mut ctx Context) handle_static(directory_path string) bool {
2020-03-07 16:16:03 +03:00
if ctx.done || ! os.exists(directory_path) {
return false
}
dir_path := directory_path.trim_space().trim_right('/')
2020-03-07 16:16:03 +03:00
mut mount_path := ''
2020-03-07 16:16:03 +03:00
if dir_path != '.' && os.is_dir(dir_path) {
// Mount point hygene, "./assets" => "/assets".
mount_path = '/' + dir_path.trim_left('.').trim('/')
2019-10-24 19:44:49 +03:00
}
2020-03-07 16:16:03 +03:00
ctx.scan_static_directory(dir_path, mount_path)
2020-03-07 16:16:03 +03:00
return true
2019-10-24 19:44:49 +03:00
}
2020-06-20 04:12:35 +03:00
2020-05-17 14:51:18 +03:00
pub fn (mut ctx Context) serve_static(url, file_path, mime_type string) {
2019-10-24 19:44:49 +03:00
ctx.static_files[url] = file_path
ctx.static_mime_types[url] = mime_type
}
2019-12-11 19:20:46 +03:00
2020-07-16 01:48:10 +03:00
pub fn (ctx &Context) ip() string {
2020-07-16 22:23:35 +03:00
mut ip := ctx.req.headers['X-Forwarded-For']
if ip == '' {
ip = ctx.req.headers['X-Real-IP']
}
if ip.contains(',') {
ip = ip.all_before(',')
}
if ip == '' {
ip = ctx.conn.peer_ip() or { '' }
}
2020-07-16 22:23:35 +03:00
return ip
2020-07-16 01:48:10 +03:00
}
2020-07-07 22:35:56 +03:00
pub fn (mut ctx Context) error(s string) {
ctx.form_error = s
}
2019-12-11 19:20:46 +03:00
/*
2019-12-11 19:20:46 +03:00
fn readall(conn net.Socket) string {
// read all message from socket
//printf("waitall=%d\n", C.MSG_WAITALL)
2019-12-11 19:20:46 +03:00
mut message := ''
buf := [1024]byte
for {
n := C.recv(conn.sockfd, buf, 1024, 0)
2019-12-11 19:20:46 +03:00
m := conn.crecv(buf, 1024)
message += unsafe { byteptr(buf).vstring_with_len(m) }
2020-05-22 18:36:09 +03:00
if message.len > max_http_post_size { break }
2019-12-11 19:20:46 +03:00
if n == m { break }
}
return message
}
*/
2019-12-11 19:20:46 +03:00
fn strip(s string) string {
// strip('\nabc\r\n') => 'abc'
return s.trim('\r\n')
}
2020-06-22 18:13:57 +03:00
pub fn not_found() Result {
return Result{}
}
fn filter(s string) string {
2020-06-24 23:38:25 +03:00
return s.replace_each([
'<', '&lt;',
'"', '&quot;',
'&', '&amp;',
])
}
pub type RawHtml = string