v/vlib/vweb/vweb.v

390 lines
9.0 KiB
V
Raw Normal View History

2020-02-03 05:00:36 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-10-24 18:44:49 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-07-29 18:21:36 +02:00
module vweb
import (
os
2019-10-24 18:44:49 +02:00
net
2019-12-30 05:42:23 +01:00
net.http
2019-10-24 18:44:49 +02:00
net.urllib
strings
2019-10-24 18:44:49 +02:00
)
2019-07-29 18:21:36 +02:00
pub const (
methods_with_form = ['POST', 'PUT', 'PATCH']
method_all = ['GET','POST','PUT','PATCH','DELETE']
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\n${HEADERS_CLOSE}404 Not Found'
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 14:46:24 +02: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',
'.wasm': 'application/wasm',
'.pdf': 'application/pdf',
'.png': 'image/png',
'.svg': 'image/svg+xml',
'.xml': 'text/xml; charset=utf-8'
}
2019-12-11 17:20:46 +01:00
MAX_HTTP_POST_SIZE = 1024 * 1024
Default_Port = 8080
)
2019-10-24 18:44:49 +02:00
pub struct Context {
static_files map[string]string
static_mime_types map[string]string
pub:
req http.Request
conn net.Socket
form map[string]string
// TODO Response
mut:
headers string // response headers
2019-12-09 15:10:44 +01:00
done bool
2019-09-05 14:46:24 +02:00
}
fn (ctx mut Context) send_response_to_client(mimetype string, res string) bool {
if ctx.done { return false }
ctx.done = true
mut sb := strings.new_builder(1024)
sb.write('HTTP/1.1 200 OK\r\nContent-Type: ') sb.write(mimetype)
sb.write('\r\nContent-Length: ') sb.write(res.len.str())
sb.write(ctx.headers)
sb.write('\r\n')
sb.write(HEADERS_CLOSE)
sb.write(res)
ctx.conn.send_string(sb.str()) or { return false }
sb.free()
return true
}
pub fn (ctx mut Context) html(s string) {
ctx.send_response_to_client('text/html', s)
2019-09-05 14:46:24 +02:00
}
2019-12-09 15:10:44 +01:00
pub fn (ctx mut Context) text(s string) {
ctx.send_response_to_client('text/plain', s)
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
pub fn (ctx mut Context) json(s string) {
ctx.send_response_to_client('application/json', s)
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
pub fn (ctx mut Context) redirect(url string) {
2019-12-09 15:10:44 +01:00
if ctx.done { return }
ctx.done = true
ctx.conn.send_string('HTTP/1.1 302 Found\r\nLocation: ${url}${ctx.headers}\r\n${HEADERS_CLOSE}') or { return }
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
pub fn (ctx mut Context) not_found(s string) {
2019-12-09 15:10:44 +01:00
if ctx.done { return }
ctx.done = true
ctx.conn.send_string(HTTP_404) or { return }
2019-09-05 14:46:24 +02:00
}
2019-08-02 04:04:48 +02:00
pub fn (ctx mut Context) set_cookie(key, val string) {
// TODO support directives, escape cookie value (https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Set-Cookie)
//println('Set-Cookie $key=$val')
ctx.add_header('Set-Cookie', '${key}=${val}; Secure; HttpOnly')
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
pub fn (ctx &Context) get_cookie(key string) ?string { // TODO refactor
2019-12-13 19:11:40 +01: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 14:46:24 +02:00
cookie := if cookie_header.contains(';') {
2019-12-07 23:48:49 +01:00
cookie_header.find_between(' $key=', ';')
2019-09-05 14:46:24 +02:00
} else {
2019-12-07 23:48:49 +01:00
cookie_header.find_between(' $key=', '\r')
2019-09-05 14:46:24 +02:00
}
if cookie != '' {
2019-12-07 23:48:49 +01:00
return cookie.trim_space()
2019-09-05 14:46:24 +02:00
}
return error('Cookie not found')
}
2019-07-29 18:21:36 +02:00
pub fn (ctx mut 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 18:21:36 +02:00
}
pub fn (ctx &Context) get_header(key string) string {
2019-12-06 22:44:22 +01:00
return ctx.req.headers[key]
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
2019-12-20 01:02:16 +01:00
//fn handle_conn(conn net.Socket) {
//println('handle')
//}
pub fn foo<T>() {
}
pub fn run<T>(port int) {
//pub fn run<T>(app mut T, port int) {
2019-12-14 16:59:02 +01:00
println('Running a Vweb app on http://localhost:$port ...')
2019-10-24 18:44:49 +02:00
l := net.listen(port) or { panic('failed to listen') }
2019-12-20 01:02:16 +01:00
mut app := T{}
app.vweb = Context{}
2019-10-24 18:44:49 +02:00
app.init()
2019-12-20 01:02:16 +01:00
//app.reset()
2019-07-29 18:21:36 +02:00
for {
conn := l.accept() or { panic('accept() failed') }
2019-12-20 01:02:16 +01:00
handle_conn(conn, mut app)
2019-09-05 14:46:24 +02:00
//foobar<T>()
2019-07-29 18:21:36 +02:00
// TODO move this to handle_conn<T>(conn, app)
//message := readall(conn)
//println(message)
/*
2019-12-11 17:20:46 +01: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 19:11:40 +01:00
//lines := message.split_into_lines()
//println(lines)
2019-12-13 19:11:40 +01:00
/*
2019-12-11 17:20:46 +01:00
if lines.len < 2 {
conn.send_string(HTTP_500) or {}
conn.close() or {}
continue
}
*/
2019-12-20 01:02:16 +01:00
}
}
2019-12-13 19:11:40 +01:00
2019-12-20 01:02:16 +01:00
fn handle_conn<T>(conn net.Socket, app mut T) {
//first_line := strip(lines[0])
first_line := conn.read_line()
println('firstline="$first_line"')
$if debug { println(first_line) }
// 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')
conn.send_string(HTTP_500) or {}
conn.close() or {}
return
//continue
}
mut headers := []string
mut body := ''
mut in_headers := true
mut len := 0
mut body_len := 0
//for line in lines[1..] {
for j in 0..100 {
2019-12-20 01:02:16 +01: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 01:02:16 +01:00
//} //else {
// End of body
//break
//}
//println('HHH')
in_headers = false
2019-10-24 18:44:49 +02:00
}
2019-12-20 01:02:16 +01: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 10:18:12 +02:00
}
2019-12-20 01:02:16 +01:00
} else {
body += sline + '\r\n'
body_len += body.len
if body_len >= len {
break
}
//println('body:$body')
2019-10-24 18:44:49 +02:00
}
2019-12-20 01:02:16 +01:00
}
2019-12-20 01:02:16 +01:00
mut action := vals[1][1..].all_before('/')
if action.contains('?') {
action = action.all_before('?')
}
if action == '' {
action = 'index'
}
req := http.Request{
headers: http.parse_headers(headers) //s.split_into_lines())
data: strip(body)
2019-12-20 01:02:16 +01:00
ws_func: 0
user_ptr: 0
method: vals[0]
url: vals[1]
}
$if debug {
println('req.headers = ')
println(req.headers)
println('req.data="$req.data"' )
2019-12-20 01:02:16 +01:00
println('vweb action = "$action"')
}
//mut app := T{
app.vweb = Context{
req: req
conn: conn
form: map[string]string
2020-03-07 14:16:03 +01:00
static_files: app.vweb.static_files
static_mime_types: app.vweb.static_mime_types
2019-12-20 01:02:16 +01:00
}
//}
if req.method in methods_with_form {
app.vweb.parse_form(req.data)
2019-12-20 01:02:16 +01:00
}
if vals.len < 2 {
$if debug {
2019-12-20 01:02:16 +01:00
println('no vals for http')
}
conn.close() or {}
2019-12-20 01:02:16 +01:00
return
//continue
}
// Serve a static file if it's one
2020-03-07 14:16:03 +01:00
static_file := app.vweb.static_files[app.vweb.req.url]
mime_type := app.vweb.static_mime_types[app.vweb.req.url]
if static_file != '' && mime_type != '' {
data := os.read_file(static_file) or {
conn.send_string(HTTP_404) or {}
return
}
app.vweb.send_response_to_client(mime_type, data)
return
}
2019-12-20 01:02:16 +01:00
// Call the right action
$if debug {
println('action=$action')
}
app.$action() or {
conn.send_string(HTTP_404) or {}
2019-07-29 18:21:36 +02:00
}
2019-12-20 01:02:16 +01:00
conn.close() or {}
app.reset()
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
2019-10-24 18:44:49 +02:00
fn (ctx mut Context) parse_form(s string) {
if !(ctx.req.method in methods_with_form) {
2019-10-24 18:44:49 +02:00
return
}
2019-11-26 11:54:41 +01: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 := keyval[0]
val := urllib.query_unescape(keyval[1]) or {
continue
}
$if debug {
println('http form "$key" => "$val"')
2019-07-29 18:21:36 +02:00
}
2019-11-26 11:54:41 +01:00
ctx.form[key] = val
2019-07-29 18:21:36 +02:00
}
2019-11-26 11:54:41 +01:00
//}
2019-12-11 17:20:46 +01:00
// todo: parse form-data and application/json
// ...
2019-09-05 14:46:24 +02:00
}
2019-07-29 18:21:36 +02:00
fn (ctx mut Context) scan_static_directory(directory_path, mount_path string) {
2019-10-17 13:30:05 +02:00
files := os.ls(directory_path) or { panic(err) }
2020-03-07 14:16:03 +01:00
if files.len > 0 {
2019-09-05 14:46:24 +02:00
for file in files {
if os.is_dir(file) {
ctx.scan_static_directory(directory_path + '/' + file, mount_path + '/' + file)
2020-03-07 14:16:03 +01:00
} else if file.contains('.') && ! file.starts_with('.') && ! file.ends_with('.') {
2020-03-07 22:26:26 +01:00
ext := os.ext(file)
2020-03-07 14:16:03 +01: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])
}
}
}
}
}
2019-09-05 14:46:24 +02:00
pub fn (ctx mut Context) handle_static(directory_path string) bool {
2020-03-07 14:16:03 +01:00
if ctx.done || ! os.exists(directory_path) {
return false
}
dir_path := directory_path.trim_space().trim_right('/')
2020-03-07 14:16:03 +01:00
mut mount_path := ''
2020-03-07 14:16:03 +01:00
if dir_path != '.' && os.is_dir(dir_path) {
// Mount point hygene, "./assets" => "/assets".
mount_path = '/' + dir_path.trim_left('.').trim('/')
2019-10-24 18:44:49 +02:00
}
2020-03-07 14:16:03 +01:00
ctx.scan_static_directory(dir_path, mount_path)
2020-03-07 14:16:03 +01:00
return true
2019-10-24 18:44:49 +02:00
}
2019-09-05 14:46:24 +02:00
pub fn (ctx mut Context) serve_static(url, file_path, mime_type string) {
2019-10-24 18:44:49 +02:00
ctx.static_files[url] = file_path
ctx.static_mime_types[url] = mime_type
}
2019-12-11 17:20:46 +01:00
/*
2019-12-11 17:20:46 +01:00
fn readall(conn net.Socket) string {
// read all message from socket
//printf("waitall=%d\n", C.MSG_WAITALL)
2019-12-11 17:20:46 +01:00
mut message := ''
buf := [1024]byte
for {
n := C.recv(conn.sockfd, buf, 1024, 0)
2019-12-11 17:20:46 +01:00
m := conn.crecv(buf, 1024)
message += string( byteptr(buf), m )
if message.len > MAX_HTTP_POST_SIZE { break }
if n == m { break }
}
return message
}
*/
2019-12-11 17:20:46 +01:00
fn strip(s string) string {
// strip('\nabc\r\n') => 'abc'
return s.trim('\r\n')
}