mirror of
https://github.com/AlistGo/alist.git
synced 2025-11-25 03:15:10 +08:00
c64f899a63
* feat(auth): Added device session management - Added the `handleSession` function to manage user device sessions and verify client identity - Updated `auth.go` to call `handleSession` for device handling when a user logs in - Added the `Session` model to database migrations - Added `device.go` and `session.go` files to handle device session logic - Updated `settings.go` to add device-related configuration items, such as the maximum number of devices, device eviction policy, and session TTL * feat(session): Adds session management features - Added `SessionInactive` error type in `device.go` - Added session-related APIs in `router.go` to support listing and evicting sessions - Added `ListSessionsByUser`, `ListSessions`, and `MarkInactive` methods in `session.go` - Returns an appropriate error when the session state is `SessionInactive` * feat(auth): Marks the device session as invalid. - Import the `session` package into the `auth` module to handle device session status. - Add a check in the login logic. If `device_key` is obtained, call `session.MarkInactive` to mark the device session as invalid. - Store the invalid status in the context variable `session_inactive` for subsequent middleware checks. - Add a check in the session refresh logic to abort the process if the current session has been marked invalid. * feat(auth, session): Added device information processing and session management changes - Updated device handling logic in `auth.go` to pass user agent and IP information - Adjusted database queries in `session.go` to optimize session query fields and add `user_agent` and `ip` fields - Modified the `Handle` method to add `ua` and `ip` parameters to store the user agent and IP address - Added the `SessionResp` structure to return a session response containing `user_agent` and `ip` - Updated the `/admin/user/create` and `/webdav` endpoints to pass the user agent and IP address to the device handler
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package db
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/alist-org/alist/v3/internal/conf"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var db *gorm.DB
|
|
|
|
func Init(d *gorm.DB) {
|
|
db = d
|
|
err := AutoMigrate(new(model.Storage), new(model.User), new(model.Meta), new(model.SettingItem), new(model.SearchNode), new(model.TaskItem), new(model.SSHPublicKey), new(model.Role), new(model.Label), new(model.LabelFileBinding), new(model.ObjFile), new(model.Session))
|
|
if err != nil {
|
|
log.Fatalf("failed migrate database: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
func AutoMigrate(dst ...interface{}) error {
|
|
var err error
|
|
if conf.Conf.Database.Type == "mysql" {
|
|
err = db.Set("gorm:table_options", "ENGINE=InnoDB CHARSET=utf8mb4").AutoMigrate(dst...)
|
|
} else {
|
|
err = db.AutoMigrate(dst...)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func GetDb() *gorm.DB {
|
|
return db
|
|
}
|
|
|
|
func Close() {
|
|
log.Info("closing db")
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
log.Errorf("failed to get db: %s", err.Error())
|
|
return
|
|
}
|
|
err = sqlDB.Close()
|
|
if err != nil {
|
|
log.Errorf("failed to close db: %s", err.Error())
|
|
return
|
|
}
|
|
}
|