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]
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/loveuer/nf/nft/log"
|
|
"github.com/loveuer/ushare/internal/api"
|
|
"github.com/loveuer/ushare/internal/controller"
|
|
"github.com/loveuer/ushare/internal/model"
|
|
"github.com/loveuer/ushare/internal/opt"
|
|
"github.com/loveuer/ushare/internal/pkg/db"
|
|
"github.com/loveuer/ushare/internal/pkg/tool"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
func init() {
|
|
flag.BoolVar(&opt.Cfg.Debug, "debug", false, "debug mode")
|
|
flag.StringVar(&opt.Cfg.Address, "address", "0.0.0.0:9119", "")
|
|
flag.StringVar(&opt.Cfg.DataPath, "data", "/data", "")
|
|
flag.IntVar(&opt.Cfg.CleanInterval, "clean", 24, "清理文件的周期, 单位: 小时, 0 则表示不自动清理")
|
|
flag.Parse()
|
|
|
|
opt.LoadFromEnv()
|
|
|
|
if opt.Cfg.Debug {
|
|
log.SetLogLevel(log.LogLevelDebug)
|
|
tool.TablePrinter(opt.Cfg)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
opt.Init(ctx)
|
|
|
|
if err := os.MkdirAll(opt.Cfg.DataPath, 0755); err != nil {
|
|
log.Fatal("main: create data path failed: %s", err.Error())
|
|
}
|
|
|
|
dbPath := filepath.Join(opt.Cfg.DataPath, ".ushare.db")
|
|
if err := db.Init(ctx, "sqlite::"+dbPath); err != nil {
|
|
log.Fatal("main: init db failed: %s", err.Error())
|
|
}
|
|
log.Debug("main: db initialized at %s", dbPath)
|
|
|
|
if err := db.Default.Migrate(&model.Role{}, &model.User{}); err != nil {
|
|
log.Fatal("main: db migrate failed: %s", err.Error())
|
|
}
|
|
|
|
controller.UserManager.Start(ctx)
|
|
controller.MetaManager.Start(ctx)
|
|
controller.RoomController.Start(ctx)
|
|
api.Start(ctx)
|
|
|
|
<-ctx.Done()
|
|
}
|