module vweb import ( os strings net http ) struct Context { static_files map[string]string static_mime_types map[string]string pub: req http.Request conn net.Socket post_form map[string]string // TODO Response headers []string // response headers } pub fn (ctx Context) text(s string) { h := ctx.headers.join('\n') ctx.conn.write('HTTP/1.1 200 OK Content-Type: text/plain $h $s ') } pub fn (ctx Context) json(s string) { h := ctx.headers.join('\n') ctx.conn.write('HTTP/1.1 200 OK Content-Type: application/json $h $s ') } pub fn (ctx Context) redirect(url string) { h := ctx.headers.join('\n') ctx.conn.write('HTTP/1.1 302 Found Location: $url $h ') } pub fn (ctx Context) not_found(s string) { ctx.conn.write('HTTP/1.1 404 Not Found') } 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 { for h in ctx.req.headers2 { if h.starts_with('Cookie:') || h.starts_with('cookie:') { cookie := h.right(7) return cookie.find_between(' $key=', ';') } } return '' /* cookie := ctx.req.headers['Cookie'] println('get cookie $key : "$cookie"') return cookie.find_between('$key=', ';') */ } 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) { h := ctx.headers.join('\n') ctx.conn.write('HTTP/1.1 200 OK Content-Type: text/html $h $html ') } pub fn run(port int) { println('Running vweb app on http://localhost:$port ...') l := net.listen(port) or { panic('failed to listen') return } mut app_init := T{} app_init.init() for { conn := l.accept() or { panic('accept() failed') return } // TODO move this to handle_conn(conn, app) s := conn.read_line() // Parse request headers lines := s.split_into_lines() mut headers := []string //map[string]string{} for i, line in lines { if i == 0 { continue } words := line.split(':') if words.len != 2 { continue } headers << line /* key := words[0] val := words[1] headers[key] = val */ } // 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('/') if action.contains('?') { action = action.all_before('?') } if action == '' { action = 'index' } req := http.Request{ headers: map[string]string{} headers2: headers ws_func: 0 user_ptr: 0 method: vals[0] url: vals[1] } println('vweb action = "$action"') mut app := T{ vweb: Context{ req: req conn: conn post_form: map[string]string{} static_files: map[string]string{} static_mime_types: map[string]string{} } } if req.method == 'POST' { app.vweb.parse_form(s) } if vals.len < 2 { println('no vals for http') conn.close() continue } // Serve a static file if it's one // if app.vweb.handle_static() { // conn.close() // continue // } // Call the right action app.$action() or { conn.write('HTTP/1.1 404 Not Found Content-Type: text/plain 404 not found ') } conn.close() } } fn (ctx mut Context) parse_form(s string) { if ctx.req.method != 'POST' { 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 { println('parse form keyval="$word"') keyval := word.split('=') if keyval.len != 2 { continue } key := keyval[0] val := keyval[1] //println('http form $key => $val') ctx.post_form[key] = http.unescape_url(val) } } } fn (ctx mut Context) scan_static_directory(directory_path, mount_path string) { // mime types mut mime_types := map[string]string{} mime_types['.css'] = 'text/css; charset=utf-8' mime_types['.gif'] = 'image/gif' mime_types['.htm'] = 'text/html; charset=utf-8' mime_types['.html'] = 'text/html; charset=utf-8' mime_types['.jpg'] = 'image/jpeg' mime_types['.js'] = 'application/javascript' mime_types['.wasm'] = 'application/wasm' mime_types['.pdf'] = 'application/pdf' mime_types['.png'] = 'image/png' mime_types['.svg'] = 'image/svg+xml' mime_types['.xml'] = 'text/xml; charset=utf-8' 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, '') static_file := ctx.static_files[ctx.req.url] mime_type := ctx.static_mime_types[ctx.req.url] if static_file != '' { data := os.read_file(static_file) or { return false } ctx.conn.write('HTTP/1.1 200 OK Content-Type: $mime_type $data ') return true } return false } pub fn (ctx mut Context) serve_static(url, file_path, mime_type string) { ctx.static_files[url] = file_path ctx.static_mime_types[url] = mime_type }