vweb: middleware implementation (#17730)

vlib/vweb/tests/middleware_test_server.v

@@ -0,0 +1,274 @@
+module main
+
+import vweb
+import time
+import os
+
+struct App {
+	vweb.Context
+	timeout       int
+	global_config shared Config
+	middlewares   map[string][]vweb.Middleware
+}
+
+struct Config {
+pub mut:
+	middleware_text string
+}
+
+fn exit_after_timeout(timeout_in_ms int) {
+	time.sleep(timeout_in_ms * time.millisecond)
+	println('>> webserver: pid: ${os.getpid()}, 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()
+	spawn exit_after_timeout(timeout)
+
+	shared config := &Config{}
+
+	app := &App{
+		timeout: timeout
+		global_config: config
+		middlewares: {
+			'/single':        [middleware1]
+			'/single_post':   [middleware1]
+			'/multiple':      [middleware1, middleware2]
+			'/multiple_post': [middleware1, middleware2]
+			'/combined':      [middleware1, middleware2]
+			'/combined_post': [middleware1, middleware2]
+			'/admin/':        [middleware1]
+			'/other/':        [middleware1, middleware2]
+			'/redirect':      [middleware_redirect]
+		}
+	}
+	eprintln('>> webserver: pid: ${os.getpid()}, started on http://localhost:${http_port}/ , with maximum runtime of ${app.timeout} milliseconds.')
+	vweb.run_at(app, host: 'localhost', port: http_port, family: .ip)!
+}
+
+// normal routes:
+
+[middleware: app_middleware]
+['/']
+pub fn (mut app App) index() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}index')
+}
+
+['/single']
+pub fn (mut app App) single() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}single')
+}
+
+['/multiple']
+pub fn (mut app App) multiple() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}multiple')
+}
+
+[middleware: app_middleware]
+['/combined']
+pub fn (mut app App) combined() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}combined')
+}
+
+['/admin/nested']
+pub fn (mut app App) nested() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}nested')
+}
+
+// above routes + post
+
+[middleware: app_middleware]
+['/index_post'; post]
+pub fn (mut app App) index_post() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}index_post:${app.req.data}')
+}
+
+['/single_post'; post]
+pub fn (mut app App) single_post() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}single_post:${app.req.data}')
+}
+
+['/multiple_post'; post]
+pub fn (mut app App) multiple_post() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}multiple_post:${app.req.data}')
+}
+
+[middleware: app_middleware]
+['/combined_post'; post]
+pub fn (mut app App) combined_post() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}combined_post:${app.req.data}')
+}
+
+['/admin/nested_post'; post]
+pub fn (mut app App) nested_post() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}nested_post:${app.req.data}')
+}
+
+// dynamic routes
+
+['/admin/:dynamic']
+pub fn (mut app App) admin_dynamic(dynamic string) vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}admin_dynamic:${dynamic}')
+}
+
+[middleware: app_middleware]
+['/other/:dynamic']
+pub fn (mut app App) combined_dynamic(dynamic string) vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}combined_dynamic:${dynamic}')
+}
+
+// redirect routes:
+
+[middleware: app_redirect]
+['/app_redirect']
+pub fn (mut app App) app_redirect_route() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}should_never_reach!')
+}
+
+['/redirect']
+pub fn (mut app App) redirect_route() vweb.Result {
+	mut result := ''
+
+	rlock app.global_config {
+		result = app.global_config.middleware_text
+	}
+
+	return app.text('${result}should_never_reach!')
+}
+
+// middleware functions:
+
+pub fn (mut app App) before_request() {
+	lock app.global_config {
+		app.global_config.middleware_text = '0'
+	}
+}
+
+pub fn (mut app App) app_middleware() bool {
+	lock app.global_config {
+		app.global_config.middleware_text += 'app_middleware'
+	}
+	return true
+}
+
+pub fn (mut app App) app_redirect() bool {
+	app.redirect('/')
+	return false
+}
+
+fn middleware1(mut ctx vweb.Context) bool {
+	ctx.conn.write_string('m1') or { panic(err) }
+	return true
+}
+
+fn middleware2(mut ctx vweb.Context) bool {
+	ctx.conn.write_string('m2') or { panic(err) }
+	return true
+}
+
+fn middleware_redirect(mut ctx vweb.Context) bool {
+	ctx.conn.write_string('m_redirect') or { panic(err) }
+	ctx.redirect('/')
+	return false
+}
+
+// utility functions:
+
+pub fn (mut app App) shutdown() vweb.Result {
+	session_key := app.get_cookie('skey') or { return app.not_found() }
+	if session_key != 'superman' {
+		return app.not_found()
+	}
+	spawn app.gracefull_exit()
+	return app.ok('good bye')
+}
+
+fn (mut app App) gracefull_exit() {
+	eprintln('>> webserver: gracefull_exit')
+	time.sleep(100 * time.millisecond)
+	exit(0)
+}