mirror/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
Code Issues Projects Releases Wiki Activity

vweb: implement live page reload in development, based on polling (useful with watch) (#17683)

Browse Source
This commit is contained in:
Delyan Angelov
2023-03-16 22:00:47 +02:00
committed by GitHub
parent 658b116d07
commit 6e1e406288
8 changed files with 134 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
vlib/vweb/vweb.v
View File

@ -167,7 +167,8 @@ pub mut:
header http.Header // response headers
// ? It doesn't seem to be used anywhere
form_error string
form_error string
livereload_poll_interval_ms int = 250
}
struct FileData {
@ -190,8 +191,8 @@ pub fn (ctx Context) init_server() {
}
// Defining this method is optional.
// This method called before every request (aka middleware).
// Probably you can use it for check user session cookie or add header.
// This method is called before every request (aka middleware).
// You can use it for checking user session cookies or to add headers.
pub fn (ctx Context) before_request() {}
// TODO - test
@ -202,17 +203,22 @@ pub fn (mut ctx Context) send_response_to_client(mimetype string, res string) bo
return false
}
ctx.done = true
// build header
header := http.new_header_from_map({
http.CommonHeader.content_type: mimetype
http.CommonHeader.content_length: res.len.str()
}).join(ctx.header)
//
mut resp := http.Response{
header: header.join(vweb.headers_close)
body: res
}
$if vweb_livereload ? {
if mimetype == 'text/html' {
resp.body = res.replace('</html>', '<script src="/vweb_livereload/${vweb_livereload_server_start}/script.js"></script>\n</html>')
}
}
// build the header after the potential modification of resp.body from above
header := http.new_header_from_map({
http.CommonHeader.content_type: mimetype
http.CommonHeader.content_length: resp.body.len.str()
}).join(ctx.header)
resp.header = header.join(vweb.headers_close)
//
resp.set_version(.v1_1)
resp.set_status(http.status_from_int(ctx.status.int()))
send_string(mut ctx.conn, resp.bytestr()) or { return false }
@ -504,6 +510,19 @@ fn handle_conn[T](mut conn net.TcpConn, mut app T, routes map[string]Route) {
// Calling middleware...
app.before_request()
$if vweb_livereload ? {
if url.path.starts_with('/vweb_livereload/') {
if url.path.ends_with('current') {
app.handle_vweb_livereload_current()
return
}
if url.path.ends_with('script.js') {
app.handle_vweb_livereload_script()
return
}
}
}
// Static handling
if serve_if_static[T](mut app, url) {
// successfully served a static file