102 lines
1.8 KiB
Go
102 lines
1.8 KiB
Go
package main
|
|
|
|
type Client struct {
|
|
UUID string
|
|
Auth bool
|
|
}
|
|
|
|
func dbGetAllClients() []Client {
|
|
var ret []Client
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return ret
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
var clientUids []string
|
|
if clientUids, err = db.GetBucketList([]string{"clients"}); err != nil {
|
|
return ret
|
|
}
|
|
for _, v := range clientUids {
|
|
if cl := dbGetClient(v); cl != nil {
|
|
ret = append(ret, *cl)
|
|
}
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func dbGetClient(id string) *Client {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return nil
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
cl := new(Client)
|
|
cl.UUID = id
|
|
cl.Auth = dbClientIsAuth(id)
|
|
return cl
|
|
}
|
|
|
|
func dbAddDeauthClient(cid, ip string) error {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return err
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
err = db.SetBool([]string{"clients", cid}, "auth", false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return db.SetValue([]string{"clients", cid}, "ip", ip)
|
|
}
|
|
|
|
func dbAuthClient(cid, ip string) error {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return err
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
err = db.SetBool([]string{"clients", cid}, "auth", false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return db.SetValue([]string{"clients", cid}, "ip", ip)
|
|
}
|
|
|
|
func dbDeAuthClient(cid string) error {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return err
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
return db.SetBool([]string{"clients", cid}, "auth", false)
|
|
}
|
|
|
|
func dbClientIsAuth(cid string) bool {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return false
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
var isAuth bool
|
|
if isAuth, err = db.GetBool([]string{"clients", cid}, "auth"); err != nil {
|
|
return false
|
|
}
|
|
return isAuth
|
|
}
|
|
|
|
func dbUpdateClientIP(cid, ip string) error {
|
|
var err error
|
|
if err = db.OpenDB(); err != nil {
|
|
return err
|
|
}
|
|
defer db.CloseDB()
|
|
|
|
return db.SetValue([]string{"clients", cid}, "ip", ip)
|
|
}
|