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

286 lines
6.8 KiB
V
Raw Normal View History

2019-10-24 19:44:49 +03: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.
2019-07-29 19:21:36 +03:00
module vweb
import (
os
2019-10-24 19:44:49 +03:00
net
http
net.urllib
)
2019-07-29 19:21:36 +03:00
const (
methods_with_form = ['POST', 'PUT', 'PATCH']
2019-09-05 15:46:24 +03:00
HEADER_SERVER = 'Server: VWeb\r\n' // TODO add to the headers
HTTP_404 = 'HTTP/1.1 404 Not Found\r\nContent-Type: text/plain\r\n\r\n404 Not Found'
HTTP_500 = 'HTTP/1.1 500 Internal Server Error\r\nContent-Type: text/plain\r\n\r\n500 Internal Server Error'
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-10-24 19:44:49 +03: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 17:10:44 +03:00
done bool
2019-09-05 15:46:24 +03:00
}
pub fn (ctx Context) html(html string) {
2019-12-09 17:10:44 +03:00
if ctx.done { return }
2019-12-07 00:44:22 +03:00
//println('HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n$ctx.headers\r\n\r\n$html')
ctx.conn.write('HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n$ctx.headers\r\n\r\n$html') or { panic(err) }
2019-09-05 15:46:24 +03:00
}
2019-12-09 17:10:44 +03:00
pub fn (ctx mut Context) text(s string) {
if ctx.done { return }
ctx.conn.write('HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n' +
'$ctx.headers\r\n$s') or { panic(err) }
ctx.done = true
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2019-07-30 06:13:44 +03:00
pub fn (ctx Context) json(s string) {
2019-12-09 17:10:44 +03:00
if ctx.done { return }
ctx.conn.write('HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n$ctx.headers\r\n\r\n$s') or { panic(err) }
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
pub fn (ctx Context) redirect(url string) {
2019-12-09 17:10:44 +03:00
if ctx.done { return }
2019-12-08 01:05:57 +03:00
ctx.conn.write('HTTP/1.1 302 Found\r\nLocation: $url\r\n$ctx.headers\r\n\r\n') or { panic(err) }
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2019-08-02 05:04:48 +03:00
pub fn (ctx Context) not_found(s string) {
2019-12-09 17:10:44 +03:00
if ctx.done { return }
ctx.conn.write(HTTP_404) or { panic(err) }
2019-09-05 15:46:24 +03:00
}
2019-08-02 05:04:48 +03:00
2019-09-05 15:46:24 +03: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')
2019-09-05 15:46:24 +03:00
ctx.add_header('Set-Cookie', '$key=$val')
}
2019-07-29 19:21:36 +03:00
pub fn (ctx &Context) get_cookie(key string) ?string { // TODO refactor
2019-12-08 01:48:49 +03:00
cookie_header := ' ' + ctx.get_header('Cookie')
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
2019-09-05 15:46:24 +03:00
fn (ctx mut Context) add_header(key, val string) {
//println('add_header($key, $val)')
ctx.headers = ctx.headers +
if ctx.headers == '' { '$key: $val' } else { '\r\n$key: $val' }
//println(ctx.headers)
2019-07-29 19:21:36 +03:00
}
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-10-24 19:44:49 +03:00
//pub fn run<T>(port int) {
pub fn run<T>(app mut T, port int) {
2019-10-24 19:44:49 +03:00
println('Running vweb app on http://localhost:$port ...')
l := net.listen(port) or { panic('failed to listen') }
//mut app := T{}
app.init()
2019-07-29 19:21:36 +03:00
for {
conn := l.accept() or {
2019-09-05 15:46:24 +03:00
panic('accept() failed')
2019-10-24 19:44:49 +03:00
}
2019-09-05 15:46:24 +03:00
//foobar<T>()
2019-07-29 19:21:36 +03:00
// TODO move this to handle_conn<T>(conn, app)
2019-12-07 00:44:22 +03:00
first_line:= conn.read_line()
if first_line == '' {
conn.write(HTTP_500) or {}
conn.close() or {}
2019-09-05 15:46:24 +03:00
return
}
2019-07-29 19:21:36 +03:00
// Parse the first line
// "GET / HTTP/1.1"
2019-12-07 00:44:22 +03:00
//first_line := s.all_before('\n')
2019-09-05 15:46:24 +03:00
vals := first_line.split(' ')
if vals.len < 2 {
println('no vals for http')
conn.write(HTTP_500) or {}
conn.close() or {}
2019-09-05 15:46:24 +03:00
return
}
2019-12-07 00:44:22 +03:00
mut headers := []string
for _ in 0..30 {
header := conn.read_line()
headers << header
//println('header="$header" len = ' + header.len.str())
if header.len <= 2 {
break
}
}
mut action := vals[1][1..].all_before('/')
2019-08-01 18:57:01 +03:00
if action.contains('?') {
2019-09-05 15:46:24 +03:00
action = action.all_before('?')
}
2019-07-29 19:21:36 +03:00
if action == '' {
2019-09-05 15:46:24 +03:00
action = 'index'
}
2019-07-29 19:21:36 +03:00
req := http.Request{
2019-12-07 00:44:22 +03:00
headers: http.parse_headers(headers) //s.split_into_lines())
ws_func: 0
user_ptr: 0
method: vals[0]
2019-10-24 19:44:49 +03:00
url: vals[1]
}
2019-08-20 11:18:12 +03:00
$if debug {
2019-12-07 00:44:22 +03:00
println('req.headers = ')
println(req.headers)
2019-08-20 11:18:12 +03:00
println('vweb action = "$action"')
}
2019-08-03 02:35:36 +03:00
//mut app := T{
app.vweb = Context{
2019-10-24 19:44:49 +03:00
req: req
conn: conn
2019-08-17 02:55:11 +03:00
form: map[string]string
static_files: map[string]string
static_mime_types: map[string]string
2019-10-24 19:44:49 +03:00
}
//}
if req.method in methods_with_form {
2019-12-08 17:37:23 +03:00
/*
2019-11-26 13:54:41 +03:00
for {
line := conn.read_line()
if line == '' || line == '\r\n' {
break
}
2019-11-26 13:54:41 +03:00
//if line.contains('POST') || line == '' {
//break
//}
}
2019-12-08 17:37:23 +03:00
*/
2019-11-26 13:54:41 +03:00
line := conn.read_line()
app.vweb.parse_form(line)
2019-10-24 19:44:49 +03:00
}
2019-07-29 19:21:36 +03:00
if vals.len < 2 {
2019-08-20 11:18:12 +03:00
$if debug {
println('no vals for http')
}
conn.close() or {}
2019-10-24 19:44:49 +03:00
continue
}
2019-10-24 19:44:49 +03:00
// Serve a static file if it's one
// if app.vweb.handle_static() {
// conn.close()
2019-10-24 19:44:49 +03:00
// continue
2019-09-05 15:46:24 +03:00
// }
2019-09-05 15:46:24 +03:00
// Call the right action
app.$action() or {
conn.write(HTTP_404) or {}
}
conn.close() or {}
2019-12-08 18:07:04 +03:00
reset := 'reset'
app.$reset()
2019-07-29 19:21:36 +03:00
}
2019-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
2019-10-24 19:44:49 +03:00
pub fn foobar<T>() {
}
2019-08-13 14:50:19 +03:00
2019-10-24 19:44:49 +03:00
fn (ctx mut 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 := keyval[0]
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-09-05 15:46:24 +03:00
}
2019-07-29 19:21:36 +03:00
fn (ctx mut 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) }
if files.len > 0 {
2019-09-05 15:46:24 +03:00
for file in files {
mut ext := ''
mut i := file.len
mut flag := true
for i > 0 {
i--
if flag {
ext = file[i..i + 1] + ext
}
if file[i..i + 1] == '.' {
flag = false
}
}
2019-09-05 15:46:24 +03:00
// todo: os.is_dir is broken now so we expect that file is dir it has no extension
// if flag {
if os.is_dir(file) {
ctx.scan_static_directory(directory_path + '/' + file, mount_path + '/' + file)
} else {
2019-09-05 15:46:24 +03:00
ctx.static_files[mount_path + '/' + file] = directory_path + '/' + file
ctx.static_mime_types[mount_path + '/' + file] = mime_types[ext]
}
}
}
}
2019-09-05 15:46:24 +03:00
pub fn (ctx mut Context) handle_static(directory_path string) bool {
ctx.scan_static_directory(directory_path, '')
2019-09-05 15:46:24 +03:00
static_file := ctx.static_files[ctx.req.url]
mime_type := ctx.static_mime_types[ctx.req.url]
2019-10-24 19:44:49 +03:00
if static_file != '' {
data := os.read_file(static_file) or { return false }
ctx.conn.write('HTTP/1.1 200 OK\r\nContent-Type: $mime_type\r\n\r\n$data') or { panic(err) }
2019-10-24 19:44:49 +03:00
return true
}
return false
}
2019-09-05 15:46:24 +03:00
pub fn (ctx mut 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
}