5f187bb5d6
- Introduce SQLite persistence via GORM (stored at <data>/.ushare.db) - Add Role model with two built-in roles: admin (all perms) and user (upload only) - Add three permissions: user_manage, upload, token_manage (reserved) - Rewrite UserManager: DB-backed login with in-memory session tokens - Auto-seed default roles and admin user on first startup - Add AuthPermission middleware for fine-grained permission checks - Add /api/uauth/me endpoint for current session info - Add /api/admin/* CRUD routes for user and role management - Add admin console page (/admin) with user table and role permissions view - Show admin console link in share page for users with user_manage permission 🤖 Generated with [Qoder][https://qoder.com]
57 lines
871 B
Go
57 lines
871 B
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/loveuer/ushare/internal/opt"
|
|
"github.com/loveuer/ushare/internal/pkg/tool"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var Default *Client
|
|
|
|
type DBType string
|
|
|
|
const (
|
|
DBTypeSqlite = "sqlite"
|
|
DBTypeMysql = "mysql"
|
|
DBTypePostgres = "postgres"
|
|
)
|
|
|
|
type Client struct {
|
|
ctx context.Context
|
|
cli *gorm.DB
|
|
dbType DBType
|
|
}
|
|
|
|
func (c *Client) Type() DBType {
|
|
return c.dbType
|
|
}
|
|
|
|
func (c *Client) Session(ctxs ...context.Context) *gorm.DB {
|
|
var ctx context.Context
|
|
if len(ctxs) > 0 && ctxs[0] != nil {
|
|
ctx = ctxs[0]
|
|
} else {
|
|
ctx = tool.Timeout(30)
|
|
}
|
|
|
|
session := c.cli.Session(&gorm.Session{Context: ctx})
|
|
|
|
if opt.Cfg.Debug {
|
|
session = session.Debug()
|
|
}
|
|
|
|
return session
|
|
}
|
|
|
|
func (c *Client) Migrate(models ...interface{}) error {
|
|
return c.cli.AutoMigrate(models...)
|
|
}
|
|
|
|
func (c *Client) Close() {
|
|
d, _ := c.cli.DB()
|
|
d.Close()
|
|
}
|