104 lines
2.1 KiB
V
104 lines
2.1 KiB
V
module main
|
|
|
|
import os
|
|
import vweb
|
|
import time
|
|
|
|
const (
|
|
known_users = ['bilbo', 'kent']
|
|
)
|
|
|
|
struct App {
|
|
port int
|
|
timeout int
|
|
pub mut:
|
|
vweb vweb.Context
|
|
}
|
|
|
|
fn exit_after_timeout(timeout_in_ms int) {
|
|
time.sleep_ms(timeout_in_ms)
|
|
// eprintln('webserver is exiting ...')
|
|
exit(0)
|
|
}
|
|
|
|
fn main() {
|
|
if os.args.len != 3 {
|
|
panic('Usage: `vweb_test_server.exe PORT TIMEOUT_IN_MILLISECONDS`')
|
|
}
|
|
http_port := os.args[1].int()
|
|
assert http_port > 0
|
|
timeout := os.args[2].int()
|
|
assert timeout > 0
|
|
go exit_after_timeout(timeout)
|
|
//
|
|
mut app := App{
|
|
port: http_port
|
|
timeout: timeout
|
|
}
|
|
vweb.run_app<App>(mut app, http_port)
|
|
}
|
|
|
|
pub fn (mut app App) init() {
|
|
}
|
|
|
|
pub fn (mut app App) init_once() {
|
|
eprintln('>> webserver: started on http://127.0.0.1:$app.port/ , with maximum runtime of $app.timeout milliseconds.')
|
|
}
|
|
|
|
pub fn (mut app App) index() {
|
|
app.vweb.text('Welcome to VWeb')
|
|
}
|
|
|
|
pub fn (mut app App) simple() vweb.Result {
|
|
app.vweb.text('A simple result')
|
|
return vweb.Result{}
|
|
}
|
|
|
|
pub fn (mut app App) html_page() vweb.Result {
|
|
app.vweb.html('<h1>ok</h1>')
|
|
return vweb.Result{}
|
|
}
|
|
|
|
// the following serve custom routes
|
|
['/:user/settings']
|
|
pub fn (mut app App) settings(username string) vweb.Result {
|
|
if username !in known_users {
|
|
return app.vweb.not_found()
|
|
}
|
|
app.vweb.html('username: $username')
|
|
return vweb.Result{}
|
|
}
|
|
|
|
['/:user/:repo/settings']
|
|
pub fn (mut app App) user_repo_settings(username, repository string) vweb.Result {
|
|
if username !in known_users {
|
|
return app.vweb.not_found()
|
|
}
|
|
app.vweb.html('username: $username | repository: $repository')
|
|
return vweb.Result{}
|
|
}
|
|
|
|
[post]
|
|
['/json_echo']
|
|
pub fn (mut app App) json() vweb.Result {
|
|
app.vweb.set_content_type(app.vweb.req.headers['Content-Type'])
|
|
return app.vweb.ok(app.vweb.req.data)
|
|
}
|
|
|
|
pub fn (mut app App) shutdown() vweb.Result {
|
|
session_key := app.vweb.get_cookie('skey') or {
|
|
return app.vweb.not_found()
|
|
}
|
|
if session_key != 'superman' {
|
|
return app.vweb.not_found()
|
|
}
|
|
go app.gracefull_exit()
|
|
return app.vweb.ok('good bye')
|
|
}
|
|
|
|
fn (mut app App) gracefull_exit() {
|
|
eprintln('>> webserver: gracefull_exit')
|
|
time.sleep_ms(100)
|
|
exit(0)
|
|
}
|