Working in memory database for users and cookie authentication
parent
328271edc3
commit
d8f1a9b4d2
@ -0,0 +1,136 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
)
|
||||
|
||||
// PublicUser is the "public" version of the "User" struct, this excludes private information
|
||||
type PublicUser struct {
|
||||
Username string `json:"username"`
|
||||
}
|
||||
|
||||
// User represents a user in the database
|
||||
type User struct {
|
||||
Username string `json:"username"`
|
||||
PasswordBCrypt []byte `json:"passwordBCrypt"`
|
||||
}
|
||||
|
||||
func (u User) PublicUser() PublicUser {
|
||||
return PublicUser{
|
||||
Username: u.Username,
|
||||
}
|
||||
}
|
||||
|
||||
type Database interface {
|
||||
GetUsers() ([]User, error)
|
||||
GetUser(username string) (User, error)
|
||||
CreateUser(user User) error
|
||||
}
|
||||
|
||||
type memDB struct {
|
||||
Users map[string]User `json:"users"`
|
||||
}
|
||||
|
||||
func NewInMemoryDB() (Database, error) {
|
||||
var db memDB
|
||||
|
||||
exampleFile, err := os.Open("example-db.local.json")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(exampleFile).Decode(&db); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &db, nil
|
||||
}
|
||||
|
||||
func (db *memDB) GetUsers() ([]User, error) {
|
||||
users := make([]User, 0, len(db.Users))
|
||||
for _, u := range db.Users {
|
||||
users = append(users, u)
|
||||
}
|
||||
|
||||
return users, nil
|
||||
}
|
||||
|
||||
func (db *memDB) GetUser(username string) (User, error) {
|
||||
user, ok := db.Users[username]
|
||||
if !ok {
|
||||
return User{}, fmt.Errorf(`no user with username %q`, username)
|
||||
}
|
||||
|
||||
return user, nil
|
||||
}
|
||||
|
||||
func (db *memDB) CreateUser(user User) error {
|
||||
if _, ok := db.Users[user.Username]; ok {
|
||||
return fmt.Errorf(`user with username %q already exists`, user.Username)
|
||||
}
|
||||
|
||||
db.Users[user.Username] = user
|
||||
return nil
|
||||
}
|
||||
|
||||
type Auth interface {
|
||||
Register(username, password string) error
|
||||
Login(username, password string) (string, error)
|
||||
UserForSession(token string) (string, error)
|
||||
GetUser(username string) (User, error)
|
||||
}
|
||||
|
||||
type memAuth struct {
|
||||
db Database
|
||||
|
||||
sessions map[string]string
|
||||
}
|
||||
|
||||
func NewInMemoryAuth(db Database) Auth {
|
||||
return &memAuth{db, map[string]string{}}
|
||||
}
|
||||
|
||||
func (auth *memAuth) Register(username, password string) error {
|
||||
passwordBCrypt, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return auth.db.CreateUser(User{
|
||||
Username: username,
|
||||
PasswordBCrypt: passwordBCrypt,
|
||||
})
|
||||
}
|
||||
|
||||
func (auth *memAuth) Login(username, password string) (string, error) {
|
||||
user, err := auth.db.GetUser(username)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
if err := bcrypt.CompareHashAndPassword(user.PasswordBCrypt, []byte(password)); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
token := GenerateRandomString(16)
|
||||
auth.sessions[token] = username
|
||||
|
||||
return token, nil
|
||||
}
|
||||
|
||||
func (auth *memAuth) UserForSession(token string) (string, error) {
|
||||
username, ok := auth.sessions[token]
|
||||
if !ok {
|
||||
return "", fmt.Errorf(`invalid session token`)
|
||||
}
|
||||
|
||||
return username, nil
|
||||
}
|
||||
|
||||
func (auth *memAuth) GetUser(username string) (User, error) {
|
||||
return auth.db.GetUser(username)
|
||||
}
|
||||
@ -0,0 +1,12 @@
|
||||
<!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>Login • Lupus Lite</title>
|
||||
</head>
|
||||
<body>
|
||||
<h1>Crea Partita | Lupus Lite</h1>
|
||||
</body>
|
||||
</html>
|
||||
@ -1,9 +1,106 @@
|
||||
package main
|
||||
|
||||
import "github.com/gofiber/fiber/v2"
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
)
|
||||
|
||||
const UserKey = "github.com/aziis98/lupus-lite/user"
|
||||
|
||||
func requestUser(c *fiber.Ctx) *User {
|
||||
return c.Locals(UserKey).(*User)
|
||||
}
|
||||
|
||||
func RequireLoggedMiddleware(auth Auth) fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
token := c.Cookies("sid")
|
||||
username, err := auth.UserForSession(token)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
user, err := auth.GetUser(username)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
c.Locals(UserKey, &user)
|
||||
|
||||
return c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func mountApiRoutes(api fiber.Router) {
|
||||
db, err := NewInMemoryDB()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
auth := NewInMemoryAuth(db)
|
||||
requireLogged := RequireLoggedMiddleware(auth)
|
||||
|
||||
api.Get("/status", func(c *fiber.Ctx) error {
|
||||
s, err := json.MarshalIndent(db, "", " ")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
log.Println(string(s))
|
||||
|
||||
return c.SendString("ok")
|
||||
})
|
||||
|
||||
api.Post("/login", func(c *fiber.Ctx) error {
|
||||
var loginForm struct {
|
||||
Username string `form:"username"`
|
||||
Password string `form:"password"`
|
||||
}
|
||||
|
||||
if err := c.BodyParser(&loginForm); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
token, err := auth.Login(loginForm.Username, loginForm.Password)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
c.Cookie(&fiber.Cookie{
|
||||
Name: "sid",
|
||||
Value: token,
|
||||
Path: "/",
|
||||
Expires: time.Now().Add(3 * 24 * time.Hour),
|
||||
})
|
||||
|
||||
return c.Redirect("/")
|
||||
})
|
||||
|
||||
api.Post("/logout", func(c *fiber.Ctx) error {
|
||||
panic("TODO: not implemented")
|
||||
})
|
||||
|
||||
api.Post("/register", func(c *fiber.Ctx) error {
|
||||
var loginForm struct {
|
||||
Username string `form:"username"`
|
||||
Password string `form:"password"`
|
||||
}
|
||||
|
||||
if err := c.BodyParser(&loginForm); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := auth.Register(loginForm.Username, loginForm.Password); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return c.Redirect("/login")
|
||||
})
|
||||
|
||||
api.Get("/user", requireLogged, func(c *fiber.Ctx) error {
|
||||
return c.JSON(requestUser(c).PublicUser())
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
@ -0,0 +1,22 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"time"
|
||||
)
|
||||
|
||||
const alphabet = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||
|
||||
func init() {
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
}
|
||||
|
||||
func GenerateRandomString(n int) string {
|
||||
b := make([]byte, n)
|
||||
|
||||
for i := range b {
|
||||
b[i] = alphabet[rand.Intn(len(alphabet))]
|
||||
}
|
||||
|
||||
return string(b)
|
||||
}
|
||||
Loading…
Reference in New Issue