2019-07-29 19:21:36 +03:00
|
|
|
module vweb
|
|
|
|
|
|
|
|
import (
|
|
|
|
os
|
|
|
|
strings
|
|
|
|
net
|
|
|
|
http
|
2019-08-12 18:54:28 +03:00
|
|
|
net.urllib
|
2019-07-29 19:21:36 +03:00
|
|
|
)
|
|
|
|
|
2019-08-10 10:12:17 +03:00
|
|
|
const (
|
|
|
|
methods_with_form = ['POST', 'PUT', 'PATCH']
|
|
|
|
)
|
|
|
|
|
2019-07-29 19:21:36 +03:00
|
|
|
struct Context {
|
2019-07-31 07:10:53 +03:00
|
|
|
static_files map[string]string
|
|
|
|
static_mime_types map[string]string
|
2019-07-29 19:21:36 +03:00
|
|
|
pub:
|
|
|
|
req http.Request
|
|
|
|
conn net.Socket
|
2019-08-10 10:12:17 +03:00
|
|
|
form map[string]string
|
2019-07-29 19:21:36 +03:00
|
|
|
// TODO Response
|
2019-07-30 16:46:10 +03:00
|
|
|
headers []string // response headers
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn (ctx Context) text(s string) {
|
|
|
|
h := ctx.headers.join('\n')
|
2019-08-11 15:07:22 +03:00
|
|
|
ctx.conn.write('HTTP/1.1 200 OK\nContent-Type: text/plain\n$h\n$s')
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
|
|
|
|
2019-07-30 06:13:44 +03:00
|
|
|
pub fn (ctx Context) json(s string) {
|
|
|
|
h := ctx.headers.join('\n')
|
2019-08-11 15:07:22 +03:00
|
|
|
ctx.conn.write('HTTP/1.1 200 OK\nContent-Type: application/json\n$h\n$s')
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn (ctx Context) redirect(url string) {
|
|
|
|
h := ctx.headers.join('\n')
|
2019-08-11 15:07:22 +03:00
|
|
|
ctx.conn.write('HTTP/1.1 302 Found\nLocation: $url\n$h')
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
|
|
|
|
2019-08-02 05:04:48 +03:00
|
|
|
pub fn (ctx Context) not_found(s string) {
|
|
|
|
ctx.conn.write('HTTP/1.1 404 Not Found')
|
|
|
|
}
|
|
|
|
|
2019-07-29 19:21:36 +03:00
|
|
|
pub fn (ctx mut Context) set_cookie(key, val string) {
|
|
|
|
ctx.set_header('Set-Cookie', '$key=$val')
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn (ctx Context) get_cookie(key string) string {
|
2019-08-01 19:36:36 +03:00
|
|
|
for h in ctx.req.headers2 {
|
2019-08-02 07:57:35 +03:00
|
|
|
if h.starts_with('Cookie:') || h.starts_with('cookie:') {
|
2019-08-01 19:36:36 +03:00
|
|
|
cookie := h.right(7)
|
2019-08-02 07:57:35 +03:00
|
|
|
return cookie.find_between(' $key=', ';')
|
2019-08-01 19:36:36 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return ''
|
2019-08-11 15:07:22 +03:00
|
|
|
/*
|
2019-07-29 19:21:36 +03:00
|
|
|
cookie := ctx.req.headers['Cookie']
|
2019-08-01 19:36:36 +03:00
|
|
|
println('get cookie $key : "$cookie"')
|
2019-07-29 19:21:36 +03:00
|
|
|
return cookie.find_between('$key=', ';')
|
2019-08-11 15:07:22 +03:00
|
|
|
*/
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
fn (ctx mut Context) set_header(key, val string) {
|
|
|
|
// ctx.resp.headers[key] = val
|
|
|
|
ctx.headers << '$key: $val'
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn (ctx Context) html(html string) {
|
2019-08-03 00:45:36 +03:00
|
|
|
h := ctx.headers.join('\n')
|
2019-08-11 15:07:22 +03:00
|
|
|
ctx.conn.write('HTTP/1.1 200 OK\nContent-Type: text/html\n$h\n\n$html')
|
2019-07-29 19:21:36 +03:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn run<T>(port int) {
|
2019-08-01 02:34:28 +03:00
|
|
|
println('Running vweb app on http://localhost:$port ...')
|
2019-08-10 11:26:42 +03:00
|
|
|
l := net.listen(port) or { panic('failed to listen') }
|
2019-08-03 02:35:36 +03:00
|
|
|
mut app := T{}
|
|
|
|
app.init()
|
2019-07-29 19:21:36 +03:00
|
|
|
for {
|
|
|
|
conn := l.accept() or {
|
|
|
|
panic('accept() failed')
|
|
|
|
}
|
2019-08-13 14:50:19 +03:00
|
|
|
foobar<T>()
|
2019-07-29 19:21:36 +03:00
|
|
|
// TODO move this to handle_conn<T>(conn, app)
|
2019-08-11 15:07:22 +03:00
|
|
|
s := conn.read_line()
|
|
|
|
if s == '' {
|
|
|
|
conn.write('HTTP/1.1 500 Not Found \nContent-Type: text/plain \n\n500')
|
|
|
|
conn.close()
|
|
|
|
continue
|
|
|
|
}
|
2019-08-01 19:36:36 +03:00
|
|
|
// Parse request headers
|
|
|
|
lines := s.split_into_lines()
|
|
|
|
mut headers := []string //map[string]string{}
|
|
|
|
for i, line in lines {
|
2019-08-11 15:07:22 +03:00
|
|
|
if i == 0 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
words := line.split(':')
|
|
|
|
if words.len != 2 {
|
|
|
|
continue
|
|
|
|
}
|
2019-08-01 19:36:36 +03:00
|
|
|
headers << line
|
2019-08-11 15:07:22 +03:00
|
|
|
/*
|
|
|
|
key := words[0]
|
|
|
|
val := words[1]
|
|
|
|
headers[key] = val
|
|
|
|
*/
|
2019-08-01 19:36:36 +03:00
|
|
|
}
|
2019-07-29 19:21:36 +03:00
|
|
|
// Parse the first line
|
|
|
|
// "GET / HTTP/1.1"
|
|
|
|
first_line := s.all_before('\n')
|
|
|
|
vals := first_line.split(' ')
|
|
|
|
mut action := vals[1].right(1).all_before('/')
|
2019-08-01 18:57:01 +03:00
|
|
|
if action.contains('?') {
|
|
|
|
action = action.all_before('?')
|
|
|
|
}
|
2019-07-29 19:21:36 +03:00
|
|
|
if action == '' {
|
|
|
|
action = 'index'
|
|
|
|
}
|
|
|
|
req := http.Request{
|
2019-08-11 15:07:22 +03:00
|
|
|
headers: map[string]string{}
|
2019-08-01 19:36:36 +03:00
|
|
|
headers2: headers
|
2019-08-11 15:07:22 +03:00
|
|
|
ws_func: 0
|
|
|
|
user_ptr: 0
|
|
|
|
method: vals[0]
|
|
|
|
url: vals[1]
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
2019-08-03 00:45:36 +03:00
|
|
|
println('vweb action = "$action"')
|
2019-08-03 02:35:36 +03:00
|
|
|
//mut app := T{
|
|
|
|
app.vweb = Context{
|
2019-08-02 06:54:40 +03:00
|
|
|
req: req
|
|
|
|
conn: conn
|
2019-08-10 10:12:17 +03:00
|
|
|
form: map[string]string{}
|
2019-08-02 06:54:40 +03:00
|
|
|
static_files: map[string]string{}
|
|
|
|
static_mime_types: map[string]string{}
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
2019-08-03 02:35:36 +03:00
|
|
|
//}
|
2019-08-10 10:12:17 +03:00
|
|
|
if req.method in methods_with_form {
|
2019-07-29 19:21:36 +03:00
|
|
|
app.vweb.parse_form(s)
|
|
|
|
}
|
|
|
|
if vals.len < 2 {
|
|
|
|
println('no vals for http')
|
2019-08-02 05:04:48 +03:00
|
|
|
conn.close()
|
|
|
|
continue
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
2019-07-31 07:10:53 +03:00
|
|
|
|
2019-07-30 16:46:10 +03:00
|
|
|
// Serve a static file if it's one
|
2019-07-31 07:10:53 +03:00
|
|
|
// if app.vweb.handle_static() {
|
|
|
|
// conn.close()
|
|
|
|
// continue
|
|
|
|
// }
|
|
|
|
|
2019-07-30 16:46:10 +03:00
|
|
|
// Call the right action
|
2019-08-02 05:04:48 +03:00
|
|
|
app.$action() or {
|
2019-08-11 15:07:22 +03:00
|
|
|
conn.write('HTTP/1.1 404 Not Found \nContent-Type: text/plain \n\n404 not found')
|
|
|
|
}
|
2019-07-29 19:21:36 +03:00
|
|
|
conn.close()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-08-13 14:50:19 +03:00
|
|
|
pub fn foobar<T>() {
|
|
|
|
}
|
|
|
|
|
2019-07-29 19:21:36 +03:00
|
|
|
fn (ctx mut Context) parse_form(s string) {
|
2019-08-10 10:12:17 +03:00
|
|
|
if !(ctx.req.method in methods_with_form) {
|
2019-07-29 19:21:36 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
pos := s.index('\r\n\r\n')
|
|
|
|
if pos > -1 {
|
|
|
|
mut str_form := s.substr(pos, s.len)
|
|
|
|
str_form = str_form.replace('+', ' ')
|
|
|
|
words := str_form.split('&')
|
|
|
|
for word in words {
|
2019-08-02 05:04:48 +03:00
|
|
|
println('parse form keyval="$word"')
|
2019-08-03 02:35:36 +03:00
|
|
|
keyval := word.trim_space().split('=')
|
2019-08-02 05:04:48 +03:00
|
|
|
if keyval.len != 2 { continue }
|
2019-07-29 19:21:36 +03:00
|
|
|
key := keyval[0]
|
2019-08-12 18:54:28 +03:00
|
|
|
val := urllib.query_unescape(keyval[1]) or {
|
|
|
|
continue
|
|
|
|
}
|
2019-08-03 02:35:36 +03:00
|
|
|
println('http form "$key" => "$val"')
|
2019-08-10 10:12:17 +03:00
|
|
|
ctx.form[key] = val
|
2019-07-29 19:21:36 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2019-08-12 21:56:47 +03:00
|
|
|
const (
|
|
|
|
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-07-29 19:21:36 +03:00
|
|
|
|
2019-07-31 07:10:53 +03:00
|
|
|
fn (ctx mut Context) scan_static_directory(directory_path, mount_path string) {
|
|
|
|
files := os.ls(directory_path)
|
|
|
|
if files.len > 0 {
|
|
|
|
for file in files {
|
|
|
|
mut ext := ''
|
|
|
|
mut i := file.len
|
|
|
|
mut flag := true
|
|
|
|
for i > 0 {
|
|
|
|
i--
|
|
|
|
if flag {
|
|
|
|
ext = file.substr(i, i + 1) + ext
|
|
|
|
}
|
|
|
|
if file.substr(i, i + 1) == '.' {
|
|
|
|
flag = false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// todo: os.is_dir is broken now
|
|
|
|
// so we expect that file is dir it has no extension
|
|
|
|
if flag {
|
|
|
|
ctx.scan_static_directory(directory_path + '/' + file, mount_path + '/' + file)
|
|
|
|
} else {
|
|
|
|
ctx.static_files[mount_path + '/' + file] = directory_path + '/' + file
|
|
|
|
ctx.static_mime_types[mount_path + '/' + file] = mime_types[ext]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn (ctx mut Context) handle_static(directory_path string) bool {
|
|
|
|
ctx.scan_static_directory(directory_path, '')
|
|
|
|
|
2019-07-30 16:46:10 +03:00
|
|
|
static_file := ctx.static_files[ctx.req.url]
|
2019-07-31 07:10:53 +03:00
|
|
|
mime_type := ctx.static_mime_types[ctx.req.url]
|
|
|
|
|
2019-07-30 16:46:10 +03:00
|
|
|
if static_file != '' {
|
|
|
|
data := os.read_file(static_file) or { return false }
|
2019-08-11 15:07:22 +03:00
|
|
|
ctx.conn.write('HTTP/1.1 200 OK\nContent-Type: $mime_type\n\n$data')
|
2019-07-30 16:46:10 +03:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2019-07-31 07:10:53 +03:00
|
|
|
pub fn (ctx mut Context) serve_static(url, file_path, mime_type string) {
|
2019-07-30 16:46:10 +03:00
|
|
|
ctx.static_files[url] = file_path
|
2019-07-31 07:10:53 +03:00
|
|
|
ctx.static_mime_types[url] = mime_type
|
2019-08-11 15:07:22 +03:00
|
|
|
}
|