49 lines
910 B
Go
49 lines
910 B
Go
package admin
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"arimelody-web/global"
|
|
)
|
|
|
|
type (
|
|
Session struct {
|
|
Token string
|
|
UserID string
|
|
Expires time.Time
|
|
}
|
|
)
|
|
|
|
const TOKEN_LENGTH = 64
|
|
const TOKEN_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
|
|
|
var ADMIN_BYPASS = func() bool {
|
|
if global.Args["adminBypass"] == "true" {
|
|
fmt.Println("WARN: Admin login is currently BYPASSED. (-adminBypass)")
|
|
return true
|
|
}
|
|
return false
|
|
}()
|
|
|
|
var sessions []*Session
|
|
|
|
func createSession(userID string, expires time.Time) Session {
|
|
return Session{
|
|
Token: string(generateToken()),
|
|
UserID: userID,
|
|
Expires: expires,
|
|
}
|
|
}
|
|
|
|
func generateToken() string {
|
|
var token []byte
|
|
|
|
for i := 0; i < TOKEN_LENGTH; i++ {
|
|
token = append(token, TOKEN_CHARS[rand.Intn(len(TOKEN_CHARS))])
|
|
}
|
|
|
|
return string(token)
|
|
}
|