48 lines
1007 B
Go
48 lines
1007 B
Go
package sql
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/filebrowser/filebrowser/v2/auth"
|
|
"github.com/filebrowser/filebrowser/v2/errors"
|
|
"github.com/filebrowser/filebrowser/v2/settings"
|
|
)
|
|
|
|
type authBackend struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func (s authBackend) Get(t settings.AuthMethod) (auth.Auther, error) {
|
|
var auther auth.Auther
|
|
|
|
switch t {
|
|
case auth.MethodJSONAuth:
|
|
auther = &auth.JSONAuth{}
|
|
case auth.MethodProxyAuth:
|
|
auther = &auth.ProxyAuth{}
|
|
case auth.MethodHookAuth:
|
|
auther = &auth.HookAuth{}
|
|
case auth.MethodNoAuth:
|
|
auther = &auth.NoAuth{}
|
|
default:
|
|
fmt.Println("ERROR: unknown auth method " + t)
|
|
return nil, errors.ErrInvalidAuthMethod
|
|
}
|
|
|
|
val := GetSetting(s.db, "auther")
|
|
if val == "" {
|
|
return auther, nil
|
|
}
|
|
return auther, json.Unmarshal([]byte(val), auther)
|
|
}
|
|
|
|
func (s authBackend) Save(a auth.Auther) error {
|
|
val, err := json.Marshal(a)
|
|
if !checkError(err, "Fail to save auth.Auther") {
|
|
return err
|
|
}
|
|
return SetSetting(s.db, "auther", string(val))
|
|
}
|