vweb: middleware implementation (#17730)

2023-08-10 21:13:21 +03:00 · 2023-03-26 00:57:42 +01:00
parent 713c95fcc8
commit 1fe5aca782
11 changed files with 968 additions and 19 deletions
--- a/examples/vweb/middleware/templates/base.html
+++ b/examples/vweb/middleware/templates/base.html
@ -0,0 +1,21 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta http-equiv="X-UA-Compatible" content="IE=edge">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>@title</title>
+</head>
+<body>
+<nav>
+    <a href="/">Home</a>
+    <a href="/admin/secrets">"/admin/secrets"</a>
+    <a href="/admin/dynamic">"/admin/dynamic"</a>
+    <a href="/early">Early exit</a>
+</nav>
+<main>
+    @content
+</main>
+<footer></footer>
+</body>
+</html>
--- a/examples/vweb/middleware/templates/early.html
+++ b/examples/vweb/middleware/templates/early.html
@ -0,0 +1 @@
+<p>Early exit</p>
--- a/examples/vweb/middleware/templates/index.html
+++ b/examples/vweb/middleware/templates/index.html
@ -0,0 +1,3 @@
+
+
+<h1>Hello </h1>
--- a/examples/vweb/middleware/templates/secret.html
+++ b/examples/vweb/middleware/templates/secret.html
@ -0,0 +1 @@
+<p>Super secret stuff</p>
--- a/examples/vweb/middleware/vweb_example.v
+++ b/examples/vweb/middleware/vweb_example.v
@ -0,0 +1,109 @@
+module main
+
+import vweb
+
+// for another example see vlib/vweb/tests/middleware_test_server.v
+const (
+	http_port = 8080
+)
+
+struct App {
+	vweb.Context
+	middlewares map[string][]vweb.Middleware
+mut:
+	is_authenticated bool
+}
+
+fn main() {
+	mut app := new_app()
+	vweb.run(app, http_port)
+}
+
+fn new_app() &App {
+	mut app := &App{
+		middlewares: {
+			// chaining is allowed, middleware will be evaluated in order
+			'/admin/': [other_func1, other_func2]
+			'/early':  [middleware_early]
+		}
+	}
+
+	// do stuff with app
+	// ...
+	return app
+}
+
+['/']
+pub fn (mut app App) index() vweb.Result {
+	println('Index page')
+	title := 'Home Page'
+
+	content := $tmpl('templates/index.html')
+	base := $tmpl('templates/base.html')
+	return app.html(base)
+}
+
+[middleware: check_auth]
+['/admin/secrets']
+pub fn (mut app App) secrets() vweb.Result {
+	println('Secrets page')
+	title := 'Secret Admin Page'
+
+	content := $tmpl('templates/secret.html')
+	base := $tmpl('templates/base.html')
+	return app.html(base)
+}
+
+['/admin/:sub']
+pub fn (mut app App) dynamic(sub string) vweb.Result {
+	println('Dynamic page')
+	title := 'Secret dynamic'
+
+	content := sub
+	base := $tmpl('templates/base.html')
+	return app.html(base)
+}
+
+['/early']
+pub fn (mut app App) early() vweb.Result {
+	println('Early page')
+	title := 'Early Exit'
+
+	content := $tmpl('templates/early.html')
+	base := $tmpl('templates/base.html')
+	return app.html(base)
+}
+
+// is always executed first!
+pub fn (mut app App) before_request() {
+	app.is_authenticated = false
+	println('0')
+}
+
+pub fn (mut app App) check_auth() bool {
+	println('3')
+	if app.is_authenticated == false {
+		app.redirect('/')
+	}
+	return app.is_authenticated
+}
+
+fn other_func1(mut ctx vweb.Context) bool {
+	println('1')
+	return true
+}
+
+fn other_func2(mut ctx vweb.Context) bool {
+	println('2')
+
+	// ...
+	return true
+}
+
+fn middleware_early(mut ctx vweb.Context) bool {
+	println('4')
+	ctx.text(':(')
+
+	// returns false, so the middleware propogation is stopped and the user will see the text ":("
+	return false
+}