herolib/aiprompts/v_veb_webserver/veb_auth.md

module veb.auth

Description

veb.auth is a module that helps with common logic required for authentication.

It allows to easily generate hashed and salted passwords and to compare password hashes.

It also handles authentication tokens, including DB table creation and insertion. All DBs are supported.

Usage

import veb
import db.pg
import veb.auth

pub struct App {
	veb.StaticHandler
pub mut:
	db   pg.DB
	auth auth.Auth[pg.DB] // or auth.Auth[sqlite.DB] etc
}

const port = 8081

pub struct Context {
	veb.Context
	current_user User
}

struct User {
	id            int @[primary; sql: serial]
	name          string
	password_hash string
	salt          string
}

fn main() {
	mut app := &App{
		db: pg.connect(host: 'localhost', user: 'postgres', password: '', dbname: 'postgres')!
	}
	app.auth = auth.new(app.db)
	veb.run[App, Context](mut app, port)
}

@[post]
pub fn (mut app App) register_user(mut ctx Context, name string, password string) veb.Result {
	salt := auth.generate_salt()
	new_user := User{
		name:          name
		password_hash: auth.hash_password_with_salt(password, salt)
		salt:          salt
	}
	sql app.db {
		insert new_user into User
	} or {}

	// Get new user ID (until RETURNING is supported by ORM)
	if x := app.find_user_by_name(name) {
		// Generate and insert the token using user ID
		token := app.auth.add_token(x.id) or { '' }
		// Authenticate the user by adding the token to the cookies
		ctx.set_cookie(name: 'token', value: token)
	}

	return ctx.redirect('/')
}

@[post]
pub fn (mut app App) login_post(mut ctx Context, name string, password string) veb.Result {
	user := app.find_user_by_name(name) or {
		ctx.error('Bad credentials')
		return ctx.redirect('/login')
	}
	// Verify user password using veb.auth
	if !auth.compare_password_with_hash(password, user.salt, user.password_hash) {
		ctx.error('Bad credentials')
		return ctx.redirect('/login')
	}
	// Find the user token in the Token table
	token := app.auth.add_token(user.id) or { '' }
	// Authenticate the user by adding the token to the cookies
	ctx.set_cookie(name: 'token', value: token)
	return ctx.redirect('/')
}

pub fn (mut app App) find_user_by_name(name string) ?User {
	// ... db query
	return User{}
}

Security considerations

hash_password_with_salt and its related functions use sha256 for hashing with a single iteration. This is not secure for production use, and you should use a more secure hashing algorithm and multiple iterations.

See also:- OWASP Password Storage Cheat Sheet

fn compare_password_with_hash(plain_text_password string, salt string, hashed string) bool fn generate_salt() string fn hash_password_with_salt(plain_text_password string, salt string) string fn new[T](db T) Auth[T] fn set_rand_crypto_safe_seed() fn (mut app Auth[T]) add_token(user_id int) !string fn (mut app App) add_token(user_id int, ip string) !string { fn (app &Auth[T]) find_token(value string) ?Token fn (mut app Auth[T]) delete_tokens(user_id int) ! struct Auth[T] { db T // pub: // salt string } struct Request { pub: client_id string client_secret string code string state string } struct Token { pub: id int @[primary; sql: serial] user_id int value string // ip string }